add -theme dark_light option

綾人 · 綾人 · commit abb6ef839b98 · 2025-04-28T22:23:23.000+09:00
diff --git a/cmd/cli/main.go b/cmd/cli/main.go
@@ -14,12 +14,13 @@ import (
 func main() {
 	// Define command line options
 	output := flag.String("o", "generated_code.png", "Output file name")
+	theme := flag.String("theme", "dark", "Theme: dark or light")
 	flag.Parse()
 
 	// Get remaining arguments (non-optional)
 	args := flag.Args()
 	if len(args) < 1 {
-		fmt.Println("Usage: snapcode [-o output.png] '<code string>'")
+		fmt.Println("Usage: snapcode [-o output.png] [-theme dark|light] '<code string>'")
 		os.Exit(1)
 	}
 
@@ -31,7 +32,7 @@ func main() {
 	defer browser.MustClose()
 
 	// Dynamically generate HTML from code Browser startup
-	htmlContent := generateHTML(code)
+	htmlContent := generateHTML(code, *theme)
 
 	// Convert HTML to Data URL and open
 	dataURL := "data:text/html;base64," + base64.StdEncoding.EncodeToString([]byte(htmlContent))
@@ -48,19 +49,25 @@ func main() {
 }
 
 // Generate an HTML template from the given code string
-func generateHTML(code string) string {
+func generateHTML(code string, theme string) string {
+	darkCSS := `
+	  body { background: #1e1e1e;color: #61dafb; }
+	  pre { background: #282c34; color: #61dafb; }	
+	  `
+	lightCSS := `
+	  body { background: #ffffff; color: #333; }
+	  pre { background: #f6f8fa; color: #111; }
+	  `
+	css := darkCSS
+	if theme == "light" {
+		css = lightCSS
+	}
+
 	return fmt.Sprintf(`
-<!DOCTYPE html>
-<html lang="en">
-<head><meta charset="UTF-8"><title>SnapCode</title>
-  <style>
-    body { background: #1e1e1e; display:flex;justify-content:center;align-items:center;height:100vh;margin:0; }
-    pre  { background:#282c34;color:#61dafb;padding:16px;border-radius:8px;font-size:16px;overflow:auto; }
-  </style>
-</head>
-<body>
-  <pre><code>%s</code></pre>
-</body>
-</html>
-`, code)
+<!DOCTYPE html><html><head><meta charset="utf-8">
+<title>SnapCode</title><style>%s
+pre{padding:16px;border-radius:8px;font-size:16px;max-width:90%%;overflow:auto;}
+body,html{display:flex;justify-content:center;align-items:center;height:100vh;margin:0;}
+</style></head><body><pre><code>%s</code></pre></body></html>
+`, css, code)
 }
