align API response format with maqt.top real specs: weapon-categories, weapons, schemes, schemes_aob, favorites check, encryption key derivation

origet/client.py

@@ -0,0 +1,211 @@
+"""
+码枪堂原版 (maqt.top) API 客户端
+
+用法:
+    from client import MaqtClient
+
+    c = MaqtClient()
+    c.login("username", "password")
+    schemes = c.get_schemes(page=1, sort="hot")
+    print(schemes)
+"""
+import json
+import requests
+from typing import Optional, Any
+from api_spec import BASE_URL, ENDPOINTS
+
+# ---- Unicode-safe Base64 ----
+def b64_encode_unicode(s: str) -> str:
+    import base64
+    encoded = base64.b64encode(s.encode("utf-8")).decode("ascii")
+    return encoded
+
+# ---- Client ----
+class MaqtClient:
+    def __init__(self, base_url: str = BASE_URL, token: Optional[str] = None):
+        self.base_url = base_url
+        self.token = token
+        self.session = requests.Session()
+        self.session.headers["User-Agent"] = "MaqiangTang/7.0.4"
+
+    # ---- low-level helpers ----
+
+    def _auth_headers(self) -> dict:
+        if self.token:
+            return {"Authorization": f"Bearer {self.token}"}
+        return {}
+
+    def _get(self, path: str, params: Optional[dict] = None, auth: bool = False) -> dict:
+        h = self._auth_headers() if auth else {}
+        r = self.session.get(f"{self.base_url}{path}", params=params, headers=h, timeout=15)
+        r.raise_for_status()
+        return r.json()
+
+    def _post(self, path: str, body: dict, extra_headers: Optional[dict] = None, auth: bool = False) -> dict:
+        h = {"Content-Type": "application/json"}
+        if auth:
+            h.update(self._auth_headers())
+        if extra_headers:
+            h.update(extra_headers)
+        r = self.session.post(f"{self.base_url}{path}", json=body, headers=h, timeout=15)
+        r.raise_for_status()
+        return r.json()
+
+    def _delete(self, path: str, auth: bool = True) -> dict:
+        h = self._auth_headers() if auth else {}
+        r = self.session.delete(f"{self.base_url}{path}", headers=h, timeout=15)
+        r.raise_for_status()
+        return r.json()
+
+    # ---- auth ----
+
+    def login(self, username: str, password: str) -> dict:
+        resp = self._post("/api/login", {"username": username, "password": password})
+        if resp.get("success") and resp.get("token"):
+            self.token = resp["token"]
+        return resp
+
+    def register(self, username: str, password: str, email: str) -> dict:
+        return self._post("/api/register", {"username": username, "password": password, "email": email})
+
+    def session_status(self) -> dict:
+        return self._get("/api/session-status", auth=True)
+
+    # ---- VIP ----
+
+    def vip_status(self) -> dict:
+        return self._get("/api/vip-status", auth=True)
+
+    def activate_vip(self, card_key: str) -> dict:
+        return self._post("/api/activate-vip", {"cardKey": card_key}, auth=True)
+
+    # ---- weapon data ----
+
+    def get_weapon_categories(self) -> dict:
+        return self._get("/api/weapon-categories")
+
+    def get_weapons(self, category: Optional[str] = None) -> dict:
+        """category 传中文名如'突击步枪'，不传返回全部"""
+        params = {"category": category} if category else None
+        return self._get("/api/weapons", params=params)
+
+    # ---- schemes ----
+
+    def get_schemes(
+        self,
+        page: int = 1,
+        sort: str = "hot",
+        limit: int = 12,
+        mode: str = "schemes",
+        weapon_category: Optional[str] = None,
+        weapon_name: Optional[str] = None,
+        min_price: Optional[int] = None,
+        max_price: Optional[int] = None,
+        search: Optional[str] = None,
+    ) -> dict:
+        params = {"sort": sort, "page": str(page), "limit": str(limit)}
+        if weapon_category:
+            params["weaponCategory"] = weapon_category
+        if weapon_name:
+            params["weaponName"] = weapon_name
+        if min_price is not None:
+            params["minPrice"] = str(min_price)
+        if max_price is not None:
+            params["maxPrice"] = str(max_price)
+        if search:
+            params["search"] = search
+        return self._get(f"/api/{mode}", params=params)
+
+    def get_favorites(self, page: int = 1, sort: str = "hot") -> dict:
+        return self._get("/api/favorites", params={"sort": sort, "page": str(page), "limit": "12"}, auth=True)
+
+    def record_use(self, scheme_id: str, mode: str = "schemes") -> dict:
+        return self._post(f"/api/{mode}/{scheme_id}/use", {})
+
+    def report_scheme(self, scheme_id: str, reason: str, description: str = "", mode: str = "schemes") -> dict:
+        return self._post(
+            f"/api/{mode}/{scheme_id}/report",
+            {"reason": reason, "description": description},
+            auth=True,
+        )
+
+    def share_scheme(
+        self,
+        description: str,
+        category: str,
+        weapon_name: str,
+        scheme: str,
+        price: Optional[int] = None,
+        user_id: int = 0,
+        username: str = "",
+        mode: str = "schemes",
+    ) -> dict:
+        user_info = json.dumps({"id": user_id, "username": username}, ensure_ascii=False)
+        body = {
+            "description": description,
+            "category": category,
+            "weaponName": weapon_name,
+            "scheme": scheme,
+            "tags": [],
+        }
+        if price is not None:
+            body["price"] = price
+        return self._post(
+            f"/api/{mode}",
+            body,
+            extra_headers={"x-user-info": b64_encode_unicode(user_info)},
+        )
+
+    # ---- favorites ----
+
+    def add_favorite(self, scheme_id: str, source: str = "烽火地带") -> dict:
+        return self._post("/api/favorites", {"schemeId": scheme_id, "source": source}, auth=True)
+
+    def remove_favorite(self, scheme_id: str, source: str = "烽火地带") -> dict:
+        import urllib.parse
+        qs = urllib.parse.urlencode({"source": source})
+        return self._delete(f"/api/favorites/{scheme_id}?{qs}")
+
+    def check_favorite(self, scheme_id: str, source: str = "烽火地带") -> dict:
+        import urllib.parse
+        qs = urllib.parse.urlencode({"schemeId": scheme_id, "source": source})
+        return self._get(f"/api/favorites/check?{qs}", auth=True)
+
+    def favorites_count(self) -> dict:
+        return self._get("/api/favorites/count", auth=True)
+
+    # ---- adverts ----
+
+    def get_adverts(self) -> dict:
+        return self._get("/api/adverts/list")
+
+    def click_advert(self, advert_id: str) -> dict:
+        return self._post(f"/api/adverts/{advert_id}/click", {})
+
+    # ---- user ----
+
+    def user_stats(self, user_id: int) -> dict:
+        return self._get(f"/api/user/stats/{user_id}")
+
+    def user_schemes(self, user_id: int) -> dict:
+        return self._get(f"/api/user/schemes/{user_id}")
+
+    def user_favorited_count(self, user_id: int) -> dict:
+        return self._get(f"/api/user/favorited-count/{user_id}", auth=True)
+
+    # ---- misc ----
+
+    def get_filter_share_categories(self) -> dict:
+        return self._get("/api/filter-share/categories")
+
+    def get_filter_shares(self) -> dict:
+        return self._get("/api/filter-shares")
+
+    def copy_filter_share(self, share_id: str, slot: str = "primary") -> dict:
+        return self._post(f"/api/filter-shares/{share_id}/copy", {"slot": slot})
+
+    def activity_ping(self) -> dict:
+        return self._get("/api/activity/ping", auth=True)
+
+    def get_game_map_password(self) -> dict:
+        return self._get("/api/game/map-password")