From 613973d2b156fca0df97e18e526db5b45b3295e5 Mon Sep 17 00:00:00 2001
From: Marshall Bowers <elliott.codes@gmail.com>
Date: Mon, 9 Oct 2023 15:52:57 -0400
Subject: [PATCH] Add support for switching between the two hardcoded themes

---
 crates/storybook2/src/storybook2.rs | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/crates/storybook2/src/storybook2.rs b/crates/storybook2/src/storybook2.rs
index 9be0a7b3fe8b8eb6293d3edbb2720232066daf01..2ec3f06737b2450d8fa2d1fa19371ea7e1964e7e 100644
--- a/crates/storybook2/src/storybook2.rs
+++ b/crates/storybook2/src/storybook2.rs
@@ -21,7 +21,6 @@ use ui::themed;
 
 use crate::assets::Assets;
 use crate::story_selector::StorySelector;
-use crate::themes::rose_pine_dawn;
 use crate::workspace::workspace;
 
 // gpui2::actions! {
@@ -51,6 +50,15 @@ fn main() {
 
     let story_selector = args.story.clone();
 
+    let theme = match args.theme.as_ref().map(|theme| theme.as_str()) {
+        Some("Rosé Pine") => themes::rose_pine(),
+        Some("Rosé Pine Dawn") => themes::rose_pine_dawn(),
+        Some(theme_name) => {
+            panic!("Only 'Rosé Pine' and 'Rosé Pine Dawn' are currently supported.")
+        }
+        None => themes::rose_pine_dawn(),
+    };
+
     let asset_source = Arc::new(Assets);
     gpui3::App::production(asset_source).run(move |cx| {
         match story_selector {
@@ -65,7 +73,7 @@ fn main() {
                     },
                     move |cx| {
                         view(
-                            cx.entity(|cx| StoryWrapper::new(selector)),
+                            cx.entity(|cx| StoryWrapper::new(selector, theme)),
                             StoryWrapper::render,
                         )
                     },
@@ -92,15 +100,16 @@ fn main() {
 #[derive(Clone)]
 pub struct StoryWrapper {
     selector: StorySelector,
+    theme: Theme,
 }
 
 impl StoryWrapper {
-    pub(crate) fn new(selector: StorySelector) -> Self {
-        Self { selector }
+    pub(crate) fn new(selector: StorySelector, theme: Theme) -> Self {
+        Self { selector, theme }
     }
 
     fn render(&mut self, cx: &mut ViewContext<Self>) -> impl Element<State = Self> {
-        themed(rose_pine_dawn(), cx, |cx| {
+        themed(self.theme.clone(), cx, |cx| {
             div()
                 .flex()
                 .flex_col()