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