完善自定义 JS/CSS 功能

cmd/patch-html/main.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"strings"
+)
+
+func main() {
+	// 读取 web/dist/index.html
+	htmlPath := "web/dist/index.html"
+	content, err := os.ReadFile(htmlPath)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "读取文件失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	html := string(content)
+
+	// 检查是否已经包含模板代码（避免重复插入）
+	if strings.Contains(html, "{{.CustomJS}}") {
+		fmt.Println("模板代码已存在，跳过插入")
+		return
+	}
+
+	// 在 </head> 前插入模板代码
+	customTemplate := `    <script type="application/javascript">
+        {{.CustomJS}}
+    </script>
+    <style type="text/css">
+        {{.CustomCSS}}
+    </style>
+`
+
+	// 替换 </head>
+	newHTML := strings.Replace(html, "</head>", customTemplate+"</head>", 1)
+
+	// 写回文件
+	err = os.WriteFile(htmlPath, []byte(newHTML), 0644)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "写入文件失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	fmt.Println("✓ 成功添加自定义模板到 index.html")
+}