feat: get teler-waf version

teler-sh · Sep 6, 2023 · 8d62d97 · 8d62d97
1 parent f9af063
commit 8d62d97
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 4 deletions.
diff --git a/Makefile b/Makefile
@@ -1,2 +1,4 @@
+TELER_WAF_VERSION := $(shell go list -m -f '{{ .Version }}' github.com/kitabisa/teler-waf)
+
 build:
-	go build -ldflags "-s -w" -o playground .
+	go build -ldflags "-s -w -X 'main.telerWAFVersion=$(TELER_WAF_VERSION)'" -o playground .
diff --git a/handler.go b/handler.go
@@ -7,13 +7,28 @@ import (
 	"strings"
 
 	"encoding/json"
+	"html/template"
 	"net/http"
 	"net/http/httptest"
 
 	"github.com/kitabisa/teler-waf"
 	"github.com/kitabisa/teler-waf/option"
 )
 
+func indexHandler(w http.ResponseWriter, r *http.Request) {
+	tmpl, err := template.ParseFiles("www/index.html")
+	if err != nil {
+		handleError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	err = tmpl.Execute(w, meta)
+	if err != nil {
+		handleError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+}
+
 func playHandler(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json")
 

diff --git a/main.go b/main.go
@@ -13,14 +13,19 @@ func init() {
     if portEnv != "" {
         port = portEnv
     }
-
     hostPort = net.JoinHostPort("0.0.0.0", port)
+
+    meta = map[string]interface{}{
+        "Version": telerWAFVersion,
+    }
 }
 
 func main() {
-    http.Handle("/", http.FileServer(http.Dir("www")))
+    http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(http.Dir("www/assets/"))))
+    http.HandleFunc("/", indexHandler)
     http.HandleFunc("/api/play", playHandler)
 
+    log.Printf("teler-waf version %s\n", telerWAFVersion)
     log.Printf("Server started on http://%s\n", hostPort)
     if err := http.ListenAndServe(hostPort, nil); err != nil {
         log.Fatal(err)

diff --git a/vars.go b/vars.go
@@ -3,4 +3,8 @@ package main
 var (
 	port     string = "3000"
 	hostPort string
+
+	telerWAFVersion string = "Unknown"
+
+	meta map[string]any
 )
diff --git a/www/index.html b/www/index.html
@@ -36,7 +36,7 @@
 </head>
 <body>
 <h1>teler WAF Playground</h1>
-
+<pre>> teler-waf version <a href="https://github.com/kitabisa/teler-waf/releases/tag/{{ .Version }}">{{ .Version }}</a></pre>
 <div class="playground">
   <button id="play" type="button" onclick="return bake(this);">Play</button>
   <button id="clear" type="button" onclick="return clearLogs();">Clear logs</button>