emergency repair: fix keys privacy and build error

internal/handler/keys.go

@@ -1,136 +1,91 @@
 package handler
-
 import (
-	"encoding/json"
-	"math/rand"
-	"net/http"
-	"os"
-	"sync"
-	"time"
+    "encoding/json"
+    "fmt"
+    "math/rand"
+    "net/http"
+    "os"
+    "sync"
+    "time"
 )
-
 type KeyData struct {
-	Name     string    `json:"name"`
-	Key      string    `json:"key"`
-	Requests int       `json:"requests"`
-	Tokens   int       `json:"tokens"`
-	Created  time.Time `json:"created"`
-	LastUsed time.Time `json:"last_used"`
+    Name      string    `json:"name"`
+    Key       string    `json:"key"`
+    Requests  int       `json:"requests"`
+    Tokens    int       `json:"tokens"`
+    Created   time.Time `json:"created"`
+    LastUsed  time.Time `json:"last_used"`
 }
 type KeyStore struct {
-	mu   sync.Mutex
-	Keys map[string]*KeyData `json:"keys"`
+    mu   sync.Mutex
+    Keys map[string]*KeyData `json:"keys"`
 }
-
 var store = &KeyStore{Keys: make(map[string]*KeyData)}
 var storePath = "/data/keys.json"
-
 func init() {
-	os.MkdirAll("/data", 0755)
-	data, err := os.ReadFile(storePath)
-	if err == nil {
-		json.Unmarshal(data, store)
-	}
-	if store.Keys == nil {
-		store.Keys = make(map[string]*KeyData)
-	}
+    os.MkdirAll("/data", 0755)
+    data, err := os.ReadFile(storePath)
+    if err == nil { json.Unmarshal(data, store) }
+    if store.Keys == nil { store.Keys = make(map[string]*KeyData) }
 }
 func saveStore() {
-	data, _ := json.MarshalIndent(store, "", "  ")
-	os.WriteFile(storePath, data, 0644)
+    data, _ := json.MarshalIndent(store, "", "  ")
+    os.WriteFile(storePath, data, 0644)
 }
 func TrackUsage(key string, tokens int) {
-	if key == "free" {
-		return
-	}
-	store.mu.Lock()
-	defer store.mu.Unlock()
-	if kd, ok := store.Keys[key]; ok {
-		kd.Requests++
-		kd.Tokens += tokens
-		kd.LastUsed = time.Now()
-		saveStore()
-	}
+    if key == "free" { return }
+    store.mu.Lock()
+    defer store.mu.Unlock()
+    if kd, ok := store.Keys[key]; ok {
+        kd.Requests++
+        kd.Tokens += tokens
+        kd.LastUsed = time.Now()
+        saveStore()
+    }
 }
 func CheckAndTrack(key string, tokens int) (bool, string) {
-	if key == "free" {
-		return true, ""
-	}
-	store.mu.Lock()
-	defer store.mu.Unlock()
-	if _, ok := store.Keys[key]; ok {
-		return true, ""
-	}
-	return false, "Key invalida"
+    if key == "free" { return true, "" }
+    store.mu.Lock()
+    defer store.mu.Unlock()
+    if _, ok := store.Keys[key]; ok { return true, "" }
+    return false, "Key invalida"
 }
 func GenerateKey(name string) string {
-	const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
-	b := make([]byte, 8)
-	for i := range b {
-		b[i] = chars[rand.Intn(len(chars))]
-	}
-	key := "RWPX-" + string(b)
-	store.mu.Lock()
-	store.Keys[key] = &KeyData{Name: name, Key: key, Created: time.Now()}
-	saveStore()
-	store.mu.Unlock()
-	return key
+    const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+    b := make([]byte, 8)
+    for i := range b { b[i] = chars[rand.Intn(len(chars))] }
+    key := "RWPX-" + string(b)
+    store.mu.Lock()
+    store.Keys[key] = &KeyData{Name: name, Key: key, Created: time.Now()}
+    saveStore()
+    store.mu.Unlock()
+    return key
 }
 func HandleGenKey(w http.ResponseWriter, r *http.Request) {
-	name := r.FormValue("name")
-	if name == "" {
-		name = "usuario"
-	}
-	key := GenerateKey(name)
-	json.NewEncoder(w).Encode(map[string]string{"key": key, "name": name})
+    name := r.FormValue("name")
+    if name == "" { name = "usuario" }
+    key := GenerateKey(name)
+    json.NewEncoder(w).Encode(map[string]string{"key": key, "name": name})
 }
 func HandleStats(w http.ResponseWriter, r *http.Request) {
-store.mu.Lock()
-defer store.mu.Unlock()
-w.Header().Set("Content-Type", "application/json")
-
-type KeyPublic struct {
-Name     string `json:"name"`
-Key      string `json:"key"`
-Requests int    `json:"requests"`
-Tokens   int    `json:"tokens"`
+    store.mu.Lock()
+    defer store.mu.Unlock()
+    type Pub struct { Name string `json:"name"`; Key string `json:"key"`; Requests int `json:"requests"`; Tokens int `json:"tokens"` }
+    res := make(map[string]Pub)
+    for k, v := range store.Keys { masked := k[:8] + "..."; res[masked] = Pub{Name: v.Name, Key: masked, Requests: v.Requests, Tokens: v.Tokens} }
+    json.NewEncoder(w).Encode(map[string]interface{}{"keys": res})
+}
+func HandleReset(w http.ResponseWriter, r *http.Request) {
+    store.mu.Lock()
+    for _, v := range store.Keys { v.Requests = 0; v.Tokens = 0 }
+    saveStore()
+    store.mu.Unlock()
+    w.Write([]byte("Reset OK"))
 }
-
-publicKeys := make(map[string]KeyPublic)
-for k, v := range store.Keys {
-masked := k[:8] + "..."
-publicKeys[masked] = KeyPublic{
-Name:     v.Name,
-Key:      masked,
-Requests: v.Requests,
-Tokens:   v.Tokens,
-}
-}
-json.NewEncoder(w).Encode(map[string]interface{}{"keys": publicKeys})
-}
-publicKeys := make(map[string]KeyPublic)
-for k, v := range store.Keys {
-// Mostramos solo el prefijo y los primeros 3 caracteres de la key
-masked := k[:8] + "..."
-publicKeys[masked] = KeyPublic{
-Name:     v.Name,
-Key:      masked,
-Requests: v.Requests,
-Tokens:   v.Tokens,
-}
-}
-json.NewEncoder(w).Encode(map[string]interface{}{"keys": publicKeys})
-}
-
 func HandleSecretReveal(w http.ResponseWriter, r *http.Request) {
-store.mu.Lock()
-defer store.mu.Unlock()
-w.Header().Set("Content-Type", "text/html")
-fmt.Fprintf(w, "<html><body style='font-family:monospace;background:#000;color:#0f0;padding:20px;'>")
-fmt.Fprintf(w, "<h2>SISTEMA MU/TH/UR 6000 - ACCESO DE COMANDANTE</h2><hr>")
-fmt.Fprintf(w, "<table border='1' cellpadding='10' style='border-color:#0f0;'><tr><th>NOMBRE</th><th>KEY COMPLETA</th></tr>")
-for _, k := range store.Keys {
-fmt.Fprintf(w, "<tr><td>%s</td><td><code style='background:#111;padding:5px;'>%s</code></td></tr>", k.Name, k.Key)
-}
-fmt.Fprintf(w, "</table></body></html>")
+    store.mu.Lock()
+    defer store.mu.Unlock()
+    fmt.Fprintf(w, "<html><body style='background:#000;color:#0f0;font-family:monospace;'><h2>MU/TH/UR 6000 - MASTER ACCESS</h2>")
+    for _, k := range store.Keys { fmt.Fprintf(w, "<p>%s: %s</p>", k.Name, k.Key) }
+    fmt.Fprintf(w, "</body></html>")
 }