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