Spaces:

Moge-Row
/

Row-proxy

Paused

App Files Files Community

Moge-Row commited on Apr 20

Commit

b5962b7

1 Parent(s): 8d0d977

fix: total rebuild of key handler for stability

Browse files

Files changed (1) hide show

internal/handler/keys.go +24 -50

internal/handler/keys.go CHANGED Viewed

@@ -6,65 +6,39 @@ import (
     "net/http"
     "os"
     "sync"
-    "time"
 )
-type KeyData struct {
-    Name     string `json:"name"`
-    Key      string `json:"key"`
-    Requests int    `json:"requests"`
-    Tokens   int    `json:"tokens"`
-}
-type KeyStore struct {
-    mu   sync.RWMutex
-    Keys map[string]*KeyData `json:"keys"`
-}
-var store = &KeyStore{Keys: make(map[string]*KeyData)}
-const storePath = "/data/keys.json"
 func init() {
     os.MkdirAll("/data", 0755)
-    if _, err := os.Stat(storePath); err == nil {
-        data, _ := os.ReadFile(storePath)
-        json.Unmarshal(data, store)
-    }
-    if store.Keys == nil { store.Keys = make(map[string]*KeyData) }
-}
-func save() {
-    data, _ := json.Marshal(store)
-    os.WriteFile(storePath, data, 0644)
 }
-func GenerateKey(name string) string {
-    const chars = "ABCDEF1234567890"
-    b := make([]byte, 8)
-    for i := range b { b[i] = chars[rand.Intn(len(chars))] }
-    key := "RWPX-" + string(b)
-    store.mu.Lock()
-    defer store.mu.Unlock()
-    store.Keys[key] = &KeyData{Name: name, Key: key}
-    save()
-    return key
 }
 func HandleGenKey(w http.ResponseWriter, r *http.Request) {
-    name := r.FormValue("name")
-    if name == "" { name = "Unknown" }
-    key := GenerateKey(name)
-    fmt.Fprintf(w, "Key creada para %s", name)
 }
 func HandleStats(w http.ResponseWriter, r *http.Request) {
-    store.mu.RLock()
-    defer store.mu.RUnlock()
-    type Pub struct { Name string `json:"name"`; Tokens int `json:"tokens"` }
-    res := make(map[string]Pub)
-    for _, v := range store.Keys { res[v.Name] = Pub{Name: v.Name, Tokens: v.Tokens} }
-    json.NewEncoder(w).Encode(map[string]interface{}{"keys": res})
 }
 func HandleSecretReveal(w http.ResponseWriter, r *http.Request) {
-    if r.URL.Query().Get("pass") != "RWPX_S3cr3t_M0d3_#2026_NzQ!" {
-        w.WriteHeader(403); return
-    }
-    store.mu.RLock()
-    defer store.mu.RUnlock()
     w.Header().Set("Content-Type", "text/html")
-    fmt.Fprint(w, "<body style='background:#000;color:#0f0;font-family:monospace;'><h2>MASTER ACCESS</h2>")
-    for _, k := range store.Keys { fmt.Fprintf(w, "<p>%s: <code>%s</code></p>", k.Name, k.Key) }
-    fmt.Fprint(w, "</body>")
 }

     "net/http"
     "os"
     "sync"
 )
+type KeyData struct { Name string; Key string; Tokens int }
+type Store struct { mu sync.Mutex; Keys map[string]*KeyData }
+var s = &Store{Keys: make(map[string]*KeyData)}
+const p = "/data/keys.json"
 func init() {
     os.MkdirAll("/data", 0755)
+    f, _ := os.ReadFile(p)
+    json.Unmarshal(f, &s.Keys)
+    if s.Keys == nil { s.Keys = make(map[string]*KeyData) }
 }
+func save() { b, _ := json.Marshal(s.Keys); os.WriteFile(p, b, 0644) }
+func GenerateKey(n string) string {
+    k := fmt.Sprintf("RWPX-%d", rand.Intn(999999))
+    s.mu.Lock(); defer s.mu.Unlock()
+    s.Keys[k] = &KeyData{Name: n, Key: k}; save(); return k
 }
 func HandleGenKey(w http.ResponseWriter, r *http.Request) {
+    n := r.FormValue("name"); if n == "" { n = "User" }
+    GenerateKey(n); fmt.Fprint(w, "OK")
 }
 func HandleStats(w http.ResponseWriter, r *http.Request) {
+    s.mu.Lock(); defer s.mu.Unlock()
+    w.Header().Set("Content-Type", "application/json")
+    m := make(map[string]interface{})
+    for _, v := range s.Keys { m[v.Name] = map[string]int{"tokens": v.Tokens} }
+    json.NewEncoder(w).Encode(map[string]interface{}{"keys": m})
 }
 func HandleSecretReveal(w http.ResponseWriter, r *http.Request) {
+    if r.URL.Query().Get("pass") != "RWPX_S3cr3t_M0d3_#2026_NzQ!" { w.WriteHeader(403); return }
+    s.mu.Lock(); defer s.mu.Unlock()
     w.Header().Set("Content-Type", "text/html")
+    fmt.Fprint(w, "<html><body style='background:#000;color:#0f0;font-family:monospace;'>")
+    for _, k := range s.Keys { fmt.Fprintf(w, "<p>%s: %s</p>", k.Name, k.Key) }
+    fmt.Fprint(w, "</body></html>")
 }