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("Zed Pro Moonlight".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(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(1500.), px(780.)).into(),
 81                }),
 82                ..Default::default()
 83            },
 84            move |cx| {
 85                let ui_font_size = ThemeSettings::get_global(cx).ui_font_size;
 86                cx.set_rem_size(ui_font_size);
 87
 88                cx.build_view(|cx| StoryWrapper::new(selector.story(cx)))
 89            },
 90        );
 91
 92        cx.activate(true);
 93    });
 94}
 95
 96#[derive(Clone)]
 97pub struct StoryWrapper {
 98    story: AnyView,
 99}
100
101impl StoryWrapper {
102    pub(crate) fn new(story: AnyView) -> Self {
103        Self { story }
104    }
105}
106
107impl Render for StoryWrapper {
108    type Element = Div<Self>;
109
110    fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
111        div()
112            .flex()
113            .flex_col()
114            .size_full()
115            .child(self.story.clone())
116    }
117}
118
119fn load_embedded_fonts(cx: &AppContext) -> gpui::Result<()> {
120    let font_paths = cx.asset_source().list("fonts")?;
121    let mut embedded_fonts = Vec::new();
122    for font_path in font_paths {
123        if font_path.ends_with(".ttf") {
124            let font_bytes = cx.asset_source().load(&font_path)?.to_vec();
125            embedded_fonts.push(Arc::from(font_bytes));
126        }
127    }
128
129    cx.text_system().add_fonts(&embedded_fonts)
130}