feat(server): add optional analytics tracker tag

lukewhrit · lukewhrit · commit 53963d6f7cdf · 2024-07-13T00:41:04.000-04:00
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -29,7 +29,8 @@ type Cfg struct {
 	ConnectionURI    string `env:"CONNECTION_URI" json:"-"`
 
 	// Web
-	Headless bool `env:"HEADLESS" envDefault:"false" json:"headless"` // Enable website
+	Headless  bool   `env:"HEADLESS" envDefault:"false" json:"headless"` // Enable website
+	Analytics string `env:"ANALYTICS" envDefault:"" json:"analytics"`    // <script> tag for analytics (leave blank to disable)
 
 	// Document
 	IDLength      int      `env:"ID_LENGTH" envDefault:"8" json:"id_length"`
diff --git a/internal/server/fetch.go b/internal/server/fetch.go
@@ -26,6 +26,7 @@ import (
 	"strings"
 
 	"github.com/go-chi/chi/v5"
+	"github.com/lukewhrit/spacebin/internal/config"
 	"github.com/lukewhrit/spacebin/internal/database"
 	"github.com/lukewhrit/spacebin/internal/util"
 	"golang.org/x/exp/slices"
@@ -81,6 +82,7 @@ func (s *Server) StaticDocument(w http.ResponseWriter, r *http.Request) {
 		"Lines":     util.CountLines(document.Content),
 		"Content":   document.Content,
 		"Extension": extension,
+		"Analytics": template.HTML(config.Config.Analytics),
 	}
 
 	if err := t.Execute(w, data); err != nil {
diff --git a/internal/server/server.go b/internal/server/server.go
@@ -21,6 +21,7 @@ import (
 	"io/fs"
 	"net/http"
 	"strings"
+	"text/template"
 
 	"github.com/go-chi/chi/v5"
 	"github.com/go-chi/chi/v5/middleware"
@@ -143,14 +144,21 @@ func (s *Server) MountStatic() {
 	})
 
 	s.Router.Get("/", func(w http.ResponseWriter, r *http.Request) {
-		file, err := resources.ReadFile("web/index.html")
+		t, err := template.ParseFS(resources, "web/index.html")
 
 		if err != nil {
 			util.WriteError(w, http.StatusInternalServerError, err)
 			return
 		}
 
-		w.Write(file)
+		err = t.Execute(w, map[string]interface{}{
+			"Analytics": config.Config.Analytics,
+		})
+
+		if err != nil {
+			util.WriteError(w, http.StatusInternalServerError, err)
+			return
+		}
 	})
 }
 
diff --git a/internal/server/web/document.html b/internal/server/web/document.html
@@ -21,6 +21,8 @@
     <link rel="stylesheet" type="text/css" href="/static/normalize.css">
     <link rel="stylesheet" type="text/css" href="/static/monokai.min.css">
     <link rel="stylesheet" type="text/css" href="/static/global.css">
+
+    {{.Analytics}}
 </head>
 
 <body>
diff --git a/internal/server/web/index.html b/internal/server/web/index.html
@@ -17,6 +17,8 @@
         content="Spacebin is a highly-reliable pastebin server, built with Go, that's capable of serving notes, code, or any other documents." />
     <meta property="og:color" content="#e34b4a" />
 
+    {{.Analytics}}
+
     <link rel="icon" type="image/x-icon" href="/static/favicon.ico">
     <link rel="stylesheet" type="text/css" href="/static/normalize.css">
     <link rel="stylesheet" type="text/css" href="/static/global.css">
