1mod assets;
  2mod stories;
  3mod story_selector;
  4
  5use std::sync::Arc;
  6
  7use clap::Parser;
  8use dialoguer::FuzzySelect;
  9use gpui::{
 10    div, px, size, AnyView, AppContext, Bounds, Render, ViewContext, VisualContext, WindowBounds,
 11    WindowOptions,
 12};
 13use log::LevelFilter;
 14use settings::{default_settings, Settings, SettingsStore};
 15use simplelog::SimpleLogger;
 16use strum::IntoEnumIterator;
 17use theme::{ThemeRegistry, ThemeSettings};
 18use ui::prelude::*;
 19
 20use crate::assets::Assets;
 21use crate::story_selector::{ComponentStory, StorySelector};
 22pub use indoc::indoc;
 23
 24#[derive(Parser)]
 25#[command(author, version, about, long_about = None)]
 26struct Args {
 27    #[arg(value_enum)]
 28    story: Option<StorySelector>,
 29
 30    /// The name of the theme to use in the storybook.
 31    ///
 32    /// If not provided, the default theme will be used.
 33    #[arg(long)]
 34    theme: Option<String>,
 35}
 36
 37fn main() {
 38    // unsafe { backtrace_on_stack_overflow::enable() };
 39
 40    SimpleLogger::init(LevelFilter::Info, Default::default()).expect("could not initialize logger");
 41
 42    let args = Args::parse();
 43
 44    let story_selector = args.story.clone().unwrap_or_else(|| {
 45        let stories = ComponentStory::iter().collect::<Vec<_>>();
 46
 47        ctrlc::set_handler(move || {}).unwrap();
 48
 49        let result = FuzzySelect::new()
 50            .with_prompt("Choose a story to run:")
 51            .items(&stories)
 52            .interact();
 53
 54        let Ok(selection) = result else {
 55            dialoguer::console::Term::stderr().show_cursor().unwrap();
 56            std::process::exit(0);
 57        };
 58
 59        StorySelector::Component(stories[selection])
 60    });
 61    let theme_name = args.theme.unwrap_or("One Dark".to_string());
 62
 63    let asset_source = Arc::new(Assets);
 64    gpui::App::production(asset_source).run(move |cx| {
 65        load_embedded_fonts(cx).unwrap();
 66
 67        let mut store = SettingsStore::default();
 68        store
 69            .set_default_settings(default_settings().as_ref(), cx)
 70            .unwrap();
 71        cx.set_global(store);
 72
 73        theme::init(theme::LoadThemes::All, cx);
 74
 75        let selector = story_selector;
 76
 77        let theme_registry = cx.global::<ThemeRegistry>();
 78        let mut theme_settings = ThemeSettings::get_global(cx).clone();
 79        theme_settings.active_theme = theme_registry.get(&theme_name).unwrap();
 80        ThemeSettings::override_global(theme_settings, cx);
 81
 82        language::init(cx);
 83        editor::init(cx);
 84
 85        let _window = cx.open_window(
 86            WindowOptions {
 87                bounds: WindowBounds::Fixed(Bounds {
 88                    origin: Default::default(),
 89                    size: size(px(1500.), px(780.)).into(),
 90                }),
 91                ..Default::default()
 92            },
 93            move |cx| {
 94                let ui_font_size = ThemeSettings::get_global(cx).ui_font_size;
 95                cx.set_rem_size(ui_font_size);
 96
 97                cx.new_view(|cx| StoryWrapper::new(selector.story(cx)))
 98            },
 99        );
100
101        cx.activate(true);
102    });
103}
104
105#[derive(Clone)]
106pub struct StoryWrapper {
107    story: AnyView,
108}
109
110impl StoryWrapper {
111    pub(crate) fn new(story: AnyView) -> Self {
112        Self { story }
113    }
114}
115
116impl Render for StoryWrapper {
117    fn render(&mut self, _cx: &mut ViewContext<Self>) -> impl IntoElement {
118        div()
119            .flex()
120            .flex_col()
121            .size_full()
122            .font("Zed Mono")
123            .child(self.story.clone())
124    }
125}
126
127fn load_embedded_fonts(cx: &AppContext) -> gpui::Result<()> {
128    let font_paths = cx.asset_source().list("fonts")?;
129    let mut embedded_fonts = Vec::new();
130    for font_path in font_paths {
131        if font_path.ends_with(".ttf") {
132            let font_bytes = cx.asset_source().load(&font_path)?.to_vec();
133            embedded_fonts.push(Arc::from(font_bytes));
134        }
135    }
136
137    cx.text_system().add_fonts(&embedded_fonts)
138}