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 gpui2::{
 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// gpui2::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("Zed Pro Moonlight".to_string());
 52
 53    let asset_source = Arc::new(Assets);
 54    gpui2::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(cx);
 64
 65        let selector =
 66            story_selector.unwrap_or(StorySelector::Component(ComponentStory::Workspace));
 67
 68        let theme_registry = cx.global::<ThemeRegistry>();
 69
 70        let mut theme_settings = ThemeSettings::get_global(cx).clone();
 71        theme_settings.active_theme = theme_registry.get(&theme_name).unwrap();
 72        ThemeSettings::override_global(theme_settings, cx);
 73
 74        ui::settings::init(cx);
 75
 76        let window = cx.open_window(
 77            WindowOptions {
 78                bounds: WindowBounds::Fixed(Bounds {
 79                    origin: Default::default(),
 80                    size: size(px(1700.), px(980.)).into(),
 81                }),
 82                ..Default::default()
 83            },
 84            move |cx| cx.build_view(|cx| StoryWrapper::new(selector.story(cx))),
 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<Self>;
104
105    fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
106        div()
107            .flex()
108            .flex_col()
109            .size_full()
110            .child(self.story.clone())
111    }
112}
113
114fn load_embedded_fonts(cx: &AppContext) -> gpui2::Result<()> {
115    let font_paths = cx.asset_source().list("fonts")?;
116    let mut embedded_fonts = Vec::new();
117    for font_path in font_paths {
118        if font_path.ends_with(".ttf") {
119            let font_bytes = cx.asset_source().load(&font_path)?.to_vec();
120            embedded_fonts.push(Arc::from(font_bytes));
121        }
122    }
123
124    cx.text_system().add_fonts(&embedded_fonts)
125}