47 lines
989 B
Go
47 lines
989 B
Go
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")
|
|
}
|