Update internal/handler/keys.go

internal/handler/keys.go

@@ -26,15 +26,7 @@ const p = "/data/keys.json"
 func init() {
 	os.MkdirAll("/data", 0755)
 	f, _ := os.ReadFile(p)
-	if len(f) > 0 {
-		json.Unmarshal(f, &s)
-	} else if backup := os.Getenv("BACKUP_KEYS"); backup != "" {
-		var keys map[string]*KeyData
-		if json.Unmarshal([]byte(backup), &keys) == nil {
-			s.Keys = keys
-			save()
-		}
-	}
+	json.Unmarshal(f, &s)
 	if s.Keys == nil { s.Keys = make(map[string]*KeyData) }
 }
 
@@ -81,7 +73,7 @@ func HandleStats(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json")
 	public := make(map[string]interface{})
 	for _, v := range s.Keys {
-		public[v.Name] = map[string]interface{}{"name": v.Name, "requests": v.Requests}
+		public[v.Name] = map[string]interface{}{"name": v.Name, "requests": v.Requests, "tokens": v.Tokens}
 	}
 	json.NewEncoder(w).Encode(map[string]interface{}{"keys": public})
 }
@@ -95,4 +87,4 @@ func HandleSecretReveal(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte("<p>" + k.Name + ": " + k.Key + "</p>"))
 	}
 	w.Write([]byte("</body>"))
-}
+}