Consolidate styles by token type length

dtolnay · Oct 23, 2024 · b5cf3b1 · b5cf3b1
1 parent 3eff940
commit b5cf3b1
Show file tree

Hide file tree

Showing 2 changed files with 155 additions and 405 deletions.
diff --git a/codegen/src/css.rs b/codegen/src/css.rs
@@ -2,6 +2,7 @@ use crate::workspace_path;
 use anyhow::Result;
 use indoc::{formatdoc, indoc};
 use std::cmp::Ordering;
+use std::collections::BTreeMap;
 use std::fs;
 use syn_codegen::Definitions;
 
@@ -30,17 +31,19 @@ pub fn generate(defs: &Definitions) -> Result<()> {
         \twidth: 0;
         }
     "};
+    let mut shrink = BTreeMap::new();
+    let mut grow = BTreeMap::new();
     for (ty, repr) in &defs.tokens {
         let macro_len = "Token![]".len() + repr.len();
         let ty_len = ty.len();
         styles.push('\n');
         styles += &match Ord::cmp(&macro_len, &ty_len) {
             Ordering::Less => {
+                shrink
+                    .entry((macro_len, ty_len))
+                    .or_insert_with(Vec::new)
+                    .push(ty);
                 formatdoc! {"
-                    a.struct[title=\"struct syn::token::{ty}\"] {{
-                    \tfont-size: calc(100% * {macro_len} / {ty_len});
-                    }}
-
                     a.struct[title=\"struct syn::token::{ty}\"]::before {{
                     \tcontent: \"Token![{repr}]\";
                     \tfont-size: calc(100% * {ty_len} / {macro_len});
@@ -49,19 +52,39 @@ pub fn generate(defs: &Definitions) -> Result<()> {
             }
             Ordering::Equal => unreachable!(),
             Ordering::Greater => {
-                let padding = ".".repeat(macro_len.saturating_sub(ty.len()));
+                let padding = macro_len.saturating_sub(ty.len());
+                grow.entry(padding).or_insert_with(Vec::new).push(ty);
                 formatdoc! {"
                     a.struct[title=\"struct syn::token::{ty}\"]::before {{
                     \tcontent: \"Token![{repr}]\";
                     }}
-
-                    a.struct[title=\"struct syn::token::{ty}\"]::after {{
-                    \tcontent: \"{padding}\";
-                    }}
                 "}
             }
         };
     }
+    for ((macro_len, ty_len), types) in shrink {
+        for ty in types {
+            styles += &format!("\na.struct[title=\"struct syn::token::{ty}\"],");
+        }
+        styles.truncate(styles.len() - 1);
+        styles += &formatdoc! {"
+             {{
+            \tfont-size: calc(100% * {macro_len} / {ty_len});
+            }}
+        "};
+    }
+    for (padding, types) in grow {
+        for ty in types {
+            styles += &format!("\na.struct[title=\"struct syn::token::{ty}\"]::after,");
+        }
+        styles.truncate(styles.len() - 1);
+        let padding = ".".repeat(padding);
+        styles += &formatdoc! {"
+             {{
+            \tcontent: \"{padding}\";
+            }}
+        "};
+    }
 
     let css_path = workspace_path::get("src/gen/token.css");
     fs::write(css_path, styles)?;