Merge pull request #1070 from wzxjohn/feature/umami

feat: support UMAMI analytics

Merge pull request #1070 from wzxjohn/feature/umami
feat: support UMAMI analytics
e81005d1 · Seefs · GitHub · 83fc8cf8 · 0a858460 · e81005d1
Commit e81005d1 authored Sep 27, 2025 by Seefs Committed by GitHub Sep 27, 2025
Show whitespace changes
Inline Side-by-side

Showing with 19 additions and 0 deletions

main.go
+18 -0

web/index.html
+1 -0

No files found.
--- a/main.go
+++ b/main.go
 package main
 import (
+	"bytes"
 	"embed"
 	"fmt"
 	"log"
@@ -16,6 +17,7 @@ import (
 	"one-api/setting/ratio_setting"
 	"os"
 	"strconv"
+	"strings"
 	"github.com/bytedance/gopkg/util/gopool"
 	"github.com/gin-contrib/sessions"
@@ -145,6 +147,22 @@ func main() {
 	})
 	server.Use(sessions.Sessions("session", store))
+	analyticsInjectBuilder := &strings.Builder{}
+	if os.Getenv("UMAMI_WEBSITE_ID") != "" {
+		umamiSiteID := os.Getenv("UMAMI_WEBSITE_ID")
+		umamiScriptURL := os.Getenv("UMAMI_SCRIPT_URL")
+		if umamiScriptURL == "" {
+			umamiScriptURL = "https://analytics.umami.is/script.js"
+		}
+		analyticsInjectBuilder.WriteString("<script defer src=\"")
+		analyticsInjectBuilder.WriteString(umamiScriptURL)
+		analyticsInjectBuilder.WriteString("\" data-website-id=\"")
+		analyticsInjectBuilder.WriteString(umamiSiteID)
+		analyticsInjectBuilder.WriteString("\"></script>")
+	}
+	analyticsInject := analyticsInjectBuilder.String()
+	indexPage = bytes.ReplaceAll(indexPage, []byte("<analytics></analytics>\n"), []byte(analyticsInject))
 	router.SetRouter(server, buildFS, indexPage)
 	var port = os.Getenv("PORT")
 	if port == "" {

--- a/web/index.html
+++ b/web/index.html
@@ -10,6 +10,7 @@
      content="OpenAI 接口聚合管理，支持多种渠道包括 Azure，可用于二次分发管理 key，仅单可执行文件，已打包好 Docker 镜像，一键部署，开箱即用"
    />
    <title>New API</title>
+<analytics></analytics>
  </head>
  <body>