autoreg-skills-max: initial public release (restored, redacted, hardened)

skills/github-oauth-automation/scripts/token_pool.py

@@ -0,0 +1,73 @@
+"""
+token_pool.py — GoRouter token pool with rotation
+Usage:
+  python token_pool.py                    # show pool status
+  python token_pool.py next               # get next token (round-robin)
+  python token_pool.py random             # get random token
+  python token_pool.py refresh            # reload from tokens.json
+"""
+import json, random, sys
+from pathlib import Path
+
+TOKENS_FILE = Path(__file__).parent / "tokens.json"
+POOL_FILE = Path(__file__).parent / "pool_state.json"
+
+class TokenPool:
+    def __init__(self):
+        self.tokens = []
+        self.index = 0
+        self._load()
+    
+    def _load(self):
+        if TOKENS_FILE.exists():
+            raw = json.loads(TOKENS_FILE.read_text(encoding="utf-8"))
+            self.tokens = [t for t in raw if t.get("platform") == "gorouter"]
+        if POOL_FILE.exists():
+            state = json.loads(POOL_FILE.read_text(encoding="utf-8"))
+            self.index = state.get("index", 0)
+    
+    def save_state(self):
+        POOL_FILE.write_text(json.dumps({"index": self.index, "total": len(self.tokens)}))
+    
+    def refresh(self):
+        self._load()
+        return len(self.tokens)
+    
+    def next(self) -> dict | None:
+        if not self.tokens:
+            return None
+        token = [REDACTED] % len(self.tokens)]
+        self.index = (self.index + 1) % len(self.tokens)
+        self.save_state()
+        return token
+    
+    def random_token(self) -> dict | None:
+        if not self.tokens:
+            return None
+        return random.choice(self.tokens)
+    
+    def status(self):
+        return {
+            "total": len(self.tokens),
+            "index": self.index,
+            "value": f"${len(self.tokens) * 70}",
+            "tokens": [{"email": t["github_email"], "token_preview": t["token"][:8] + "..."} for t in self.tokens]
+        }
+
+if __name__ == "__main__":
+    pool = TokenPool()
+    cmd = sys.argv[1] if len(sys.argv) > 1 else "status"
+    
+    if cmd == "next":
+        t = pool.next()
+        if t: print(t["token"])
+        else: print("NO_TOKENS")
+    elif cmd == "random":
+        t = pool.random_token()
+        if t: print(t["token"])
+        else: print("NO_TOKENS")
+    elif cmd == "refresh":
+        n = pool.refresh()
+        print(f"Pool refreshed: {n} tokens")
+    else:
+        print(json.dumps(pool.status(), indent=2, ensure_ascii=False))