Update util.ts

Author: mon-jai

src/util.ts

@@ -1,9 +1,17 @@
-import { glob, readFile, readdir, unlink, writeFile } from "fs/promises"
-import { join } from "path"
+import { glob, readFile, readdir, unlink, writeFile } from "fs/promises";
+import { join } from "path";
+
+
+
+import type { Browser } from "playwright";
+
+
+
+import { ARCHIVE_DIR, HOOK_SCRIPT, SCRIPT, ZIP_SCRIPT } from "./constants.js";
+
+
 
-import type { Browser } from "playwright"
 
-import { ARCHIVE_DIR, HOOK_SCRIPT, SCRIPT, ZIP_SCRIPT } from "./constants.js"
 
 export async function getGeminiIdsFromMarkdowns(): Promise<Set<string>> {
   const ids = new Set<string>()
@@ -97,13 +105,36 @@ export async function archiveConversation(browser: Browser, id: string) {
       removeFrames: true,
       insertSingleFileComment: true
     })
-    const fileContent = pageData.content.replaceAll(/@font-face\s*{[^}]*}/g, (fontFaceRule: string) => {
-      const fontFamilyMatch = fontFaceRule.match(/font-family:\s*(?<quote>['"]?)(?<fontFamily>[^'"]+)\k<quote>/)
-      const fontFamily = fontFamilyMatch?.groups?.["fontFamily"]?.trim() ?? ""
 
-      if (includesKatex && fontFamily.startsWith("KaTeX")) return fontFaceRule
-      else return ""
-    })
+    const variablesUsedInDocument = new Set(
+      // Variable values could contain other values, so /var\(([^\)]+)/g won't work
+      // e.g. --a: var(--b, var(--c));
+      Array.from(pageData.content.matchAll(/var\s*\(\s*(?<variableName>--[a-zA-z0-9\-]+)/g)).map(
+        regExpExecArray => regExpExecArray.groups!["variableName"]!
+      )
+    )
+
+    const fileContent = pageData.content
+      // Remove Google Sans family and Google Symbol font face
+      .replaceAll(/@font-face\s*{[^}]*}/g, (fontFaceRule: string) => {
+        const fontFamilyMatch = fontFaceRule.match(/font-family:\s*(?<quote>['"]?)(?<fontFamily>[^'"]+)\k<quote>/)
+        const fontFamily = fontFamilyMatch?.groups?.["fontFamily"]?.trim() ?? ""
+
+        if (includesKatex && fontFamily.startsWith("KaTeX")) return fontFaceRule
+        else return ""
+      })
+      .replaceAll(
+        // --a: 0px; } .class { ...
+        // <div style="--a: 0px"> ...
+        // <div style='--a: 0px'> ...
+        /(?<variableName>--[a-zA-Z0-9\-]+)\s*:\s*(?<value>[^;\n\}]+)\s*[;\n]?(?<curlyBrace>\})?(?<quote>["'])?/gm,
+        (_match, variableName: string, value: string, curlyBrace: string | undefined, quote: string | undefined) => {
+          const suffix = `${curlyBrace ?? ""}${quote ?? ""}`
+               console.log(_match, suffix)
+     if (variablesUsedInDocument.has(variableName)) return `${variableName}:${value};${suffix}`
+          return suffix
+        }
+      )
 
     // Remove illegal filename chars
     const sanitizedTitle = title.replace(/[\\/:*?"<>|\n]/g, "").substring(0, 100)
@@ -126,4 +157,4 @@ export function buildCommitMessage(added: string[], deleted: string[]): string {
   if (deleted.length > 0) msg += `\nDeleted conversations: ${deleted.join(", ")}`
 
   return msg
-}
+}