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