From d1636bf16b9b37dfc8e0d4e237d8732e53d28b08 Mon Sep 17 00:00:00 2001
From: Marshall Bowers <elliott.codes@gmail.com>
Date: Thu, 2 Nov 2023 19:14:51 -0400
Subject: [PATCH] Finish up theme printer

---
 crates/theme2/src/theme2.rs                |  2 +-
 crates/theme_importer/src/main.rs          | 55 ++++++++++++++++++++--
 crates/theme_importer/src/theme_printer.rs | 23 +++++++--
 3 files changed, 73 insertions(+), 7 deletions(-)

diff --git a/crates/theme2/src/theme2.rs b/crates/theme2/src/theme2.rs
index 4232fde324..9cd5fe75c0 100644
--- a/crates/theme2/src/theme2.rs
+++ b/crates/theme2/src/theme2.rs
@@ -17,7 +17,7 @@ pub use syntax::*;
 use gpui2::{AppContext, Hsla, SharedString};
 use settings2::Settings;
 
-#[derive(Debug, Clone, PartialEq)]
+#[derive(Debug, PartialEq, Clone, Copy)]
 pub enum Appearance {
     Light,
     Dark,
diff --git a/crates/theme_importer/src/main.rs b/crates/theme_importer/src/main.rs
index 5e44ae4ea2..ca2b2e3195 100644
--- a/crates/theme_importer/src/main.rs
+++ b/crates/theme_importer/src/main.rs
@@ -2,10 +2,12 @@ mod theme_printer;
 mod vscode;
 
 use std::fs::{self, File};
+use std::io::Write;
 use std::path::PathBuf;
 use std::str::FromStr;
 
 use anyhow::{anyhow, Context, Result};
+use convert_case::{Case, Casing};
 use gpui::serde_json;
 use log::LevelFilter;
 use serde::Deserialize;
@@ -65,8 +67,6 @@ pub struct ThemeMetadata {
 fn main() -> Result<()> {
     SimpleLogger::init(LevelFilter::Info, Default::default()).expect("could not initialize logger");
 
-    let themes_path = PathBuf::from_str("crates/theme2/src/themes")?;
-
     let vscode_themes_path = PathBuf::from_str("assets/themes/src/vscode/")?;
 
     let mut theme_families = Vec::new();
@@ -116,9 +116,58 @@ fn main() -> Result<()> {
         theme_families.push(theme_family);
     }
 
+    let themes_output_path = PathBuf::from_str("crates/theme2/src/themes")?;
+
+    let mut theme_modules = Vec::new();
+
     for theme_family in theme_families {
-        println!("{:#?}", ThemeFamilyPrinter::new(theme_family));
+        let theme_family_slug = theme_family.name.to_string().to_case(Case::Snake);
+
+        let mut output_file =
+            File::create(themes_output_path.join(format!("{theme_family_slug}.rs")))?;
+
+        let theme_module = format!(
+            r#"
+            use gpui2::rgba;
+
+            use crate::{{
+                default_color_scales, Appearance, GitStatusColors, PlayerColor, PlayerColors, StatusColors,
+                SyntaxTheme, SystemColors, ThemeColors, ThemeFamily, ThemeStyles, ThemeVariant,
+            }};
+
+            pub fn {theme_family_slug}() -> ThemeFamily {{
+                {theme_family_definition}
+            }}
+            "#,
+            theme_family_definition = format!("{:#?}", ThemeFamilyPrinter::new(theme_family))
+        );
+
+        output_file.write_all(theme_module.as_bytes())?;
+
+        theme_modules.push(theme_family_slug);
     }
 
+    let mut mod_rs_file = File::create(themes_output_path.join(format!("mod.rs")))?;
+
+    let mod_rs_contents = format!(
+        r#"
+        {mod_statements}
+
+        {use_statements}
+        "#,
+        mod_statements = theme_modules
+            .iter()
+            .map(|module| format!("mod {module};"))
+            .collect::<Vec<_>>()
+            .join("\n"),
+        use_statements = theme_modules
+            .iter()
+            .map(|module| format!("pub use {module}::*;"))
+            .collect::<Vec<_>>()
+            .join("\n")
+    );
+
+    mod_rs_file.write_all(mod_rs_contents.as_bytes())?;
+
     Ok(())
 }
diff --git a/crates/theme_importer/src/theme_printer.rs b/crates/theme_importer/src/theme_printer.rs
index 82446030e2..7619734eb1 100644
--- a/crates/theme_importer/src/theme_printer.rs
+++ b/crates/theme_importer/src/theme_printer.rs
@@ -2,10 +2,18 @@ use std::fmt::{self, Debug};
 
 use gpui::{Hsla, Rgba};
 use theme::{
-    GitStatusColors, PlayerColor, PlayerColors, StatusColors, SyntaxTheme, SystemColors,
-    ThemeColors, ThemeFamily, ThemeStyles, ThemeVariant,
+    Appearance, GitStatusColors, PlayerColor, PlayerColors, StatusColors, SyntaxTheme,
+    SystemColors, ThemeColors, ThemeFamily, ThemeStyles, ThemeVariant,
 };
 
+struct RawSyntaxPrinter<'a>(&'a str);
+
+impl<'a> Debug for RawSyntaxPrinter<'a> {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{}", self.0)
+    }
+}
+
 struct HslaPrinter(Hsla);
 
 impl Debug for HslaPrinter {
@@ -55,6 +63,7 @@ impl Debug for ThemeFamilyPrinter {
                         .collect(),
                 ),
             )
+            .field("scales", &RawSyntaxPrinter("default_color_scales()"))
             .finish()
     }
 }
@@ -66,12 +75,20 @@ impl<'a> Debug for ThemeVariantPrinter<'a> {
         f.debug_struct("ThemeVariant")
             .field("id", &IntoPrinter(&self.0.id))
             .field("name", &IntoPrinter(&self.0.name))
-            .field("appearance", &self.0.appearance)
+            .field("appearance", &AppearancePrinter(self.0.appearance))
             .field("styles", &ThemeStylesPrinter(&self.0.styles))
             .finish()
     }
 }
 
+pub struct AppearancePrinter(Appearance);
+
+impl Debug for AppearancePrinter {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "Appearance::{:?}", self.0)
+    }
+}
+
 pub struct ThemeStylesPrinter<'a>(&'a ThemeStyles);
 
 impl<'a> Debug for ThemeStylesPrinter<'a> {