yifaban-cli 0.1.0__py3-none-any.whl

yifaban/__init__.py

@@ -0,0 +1,29 @@
+# -*- coding=utf8 -*-
+
+"""
+yifaban - 本地直连应用市场发布 CLI
+
+复用 apk-publisher-python 仓库中的渠道发布器，直接从本地电脑
+把 APK（渠道包）上传到华为/荣耀/小米/OPPO/vivo/腾讯应用宝。
+"""
+
+from .logger import log_t, get_g_log
+from .constants import (
+    HUAWEI_CHANNEL_NAME,
+    XIAOMI_CHANNEL_NAME,
+    OPPO_CHANNEL_NAME,
+    TENCENT_CHANNEL_NAME,
+    VIVO_CHANNEL_NAME,
+    HONOR_CHANNEL_NAME,
+)
+
+__all__ = [
+    "log_t",
+    "get_g_log",
+    "HUAWEI_CHANNEL_NAME",
+    "XIAOMI_CHANNEL_NAME",
+    "OPPO_CHANNEL_NAME",
+    "TENCENT_CHANNEL_NAME",
+    "VIVO_CHANNEL_NAME",
+    "HONOR_CHANNEL_NAME",
+]

yifaban/api.py

@@ -0,0 +1,219 @@
+# -*- coding=utf8 -*-
+
+"""
+易发版后端 API 客户端
+
+签名协议与 apk-publisher-python 后端 core/utils.py 完全一致：
+  1. 参数按 key 排序（排除 key 字段），过滤 None / 空字符串 / 字典值
+  2. 值含中文（[\u4e00-\u9fff]）时做 URL 编码（safe="!~*'()"，先归一化换行）
+  3. 拼接 key=value&... 后追加 &token={token}123D77aHE@m
+  4. 取 MD5 作为 key 参数
+
+登录使用空 token 签名；业务请求 header 携带 Authorization/uid，body 携带 key。
+响应格式统一为 {code, message, data}，data 键为驼峰命名。
+"""
+
+import hashlib
+import re
+import time
+import urllib.parse
+
+import requests
+
+from yifaban.config import apply_env_overrides
+
+SIGN_SALT = "123D77aHE@m"
+_CHINESE_RE = re.compile(r"[\u4e00-\u9fff]+")
+
+
+class ApiError(Exception):
+    pass
+
+
+def calculate_api_key(data: dict, token: str) -> str:
+    """计算 API 签名（与后端 core/utils.py 的 calculate_api_key 一致）"""
+    valid_items = []
+    for key in sorted(data.keys()):
+        if key == "key":
+            continue
+        value = data[key]
+        if value is None or isinstance(value, dict) or len(str(value)) == 0:
+            continue
+        value_str = str(value)
+        if _CHINESE_RE.search(value_str):
+            value_str = urllib.parse.quote(
+                value_str.replace("\r\n", "\n").replace("\r", "\n"),
+                safe="!~*'()",
+                encoding="utf-8",
+            )
+        valid_items.append(f"{key}={value_str}")
+    s = "&".join(valid_items)
+    return hashlib.md5(f"{s}&token={token}{SIGN_SALT}".encode("utf-8")).hexdigest()
+
+
+def _post(server, path, data, token="", uid="", timeout=30):
+    payload = dict(data or {})
+    payload["key"] = calculate_api_key(payload, token)
+    headers = {"Content-Type": "application/json"}
+    if token:
+        headers["Authorization"] = token
+        headers["uid"] = uid
+    try:
+        resp = requests.post(f"{server}{path}", json=payload, headers=headers, timeout=timeout)
+    except requests.RequestException as e:
+        raise ApiError(f"网络请求失败: {e}") from e
+    try:
+        body = resp.json()
+    except ValueError:
+        raise ApiError(f"服务器返回非 JSON（HTTP {resp.status_code}）: {resp.text[:200]}")
+    if resp.status_code == 401 and body.get("code") == 4001:
+        raise ApiError("登录已过期，请重新运行 yifaban login")
+    if body.get("code") != 200:
+        raise ApiError(f"{body.get('message') or '请求失败'}（code={body.get('code')}）")
+    return body.get("data", {})
+
+
+def _post_public(server, path, data, timeout=30):
+    """无 token 签名 POST（登录、Q助理 exchange/bind/cli poll 等公开接口）"""
+    payload = dict(data or {})
+    payload["key"] = calculate_api_key(payload, "")
+    try:
+        resp = requests.post(f"{server}{path}", json=payload, timeout=timeout)
+    except requests.RequestException as e:
+        raise ApiError(f"网络请求失败: {e}") from e
+    try:
+        body = resp.json()
+    except ValueError:
+        raise ApiError(f"服务器返回非 JSON（HTTP {resp.status_code}）: {resp.text[:200]}")
+    if body.get("code") != 200:
+        raise ApiError(body.get("message") or "请求失败")
+    return body.get("data", {})
+
+
+def save_login_session(server: str, result: dict, fallback_username: str = ""):
+    """保存登录会话到 ~/.yifaban/auth.json（登录/扫码兑换/绑定登录共用）"""
+    from yifaban.config import save_auth
+
+    save_auth({
+        "server": server,
+        "token": result.get("token", ""),
+        "uid": result.get("uid", ""),
+        "username": result.get("username", fallback_username),
+    })
+
+
+def login(username: str, password: str, server: str):
+    """账号密码登录易发版，保存会话到 ~/.yifaban/auth.json"""
+    data = {
+        "username": username,
+        "password": password,
+        "t": str(int(time.time())),
+    }
+    result = _post_public(server, "/api/auth/login", data)
+    save_login_session(server, result, username)
+    return result
+
+
+def qzhuli_auth_url(server: str):
+    """获取 Q助理扫码授权地址与轮询会话 ID（响应 data 为驼峰键）"""
+    try:
+        resp = requests.get(f"{server}/api/qzhuli/cli/auth-url", timeout=30)
+    except requests.RequestException as e:
+        raise ApiError(f"网络请求失败: {e}") from e
+    try:
+        body = resp.json()
+    except ValueError:
+        raise ApiError(f"服务器返回非 JSON（HTTP {resp.status_code}）: {resp.text[:200]}")
+    if body.get("code") != 200:
+        raise ApiError(body.get("message") or "获取授权地址失败")
+    data = body.get("data", {})
+    url, request_id = data.get("url", ""), data.get("requestId", "")
+    if not url or not request_id:
+        raise ApiError("服务器未返回授权地址")
+    return {"url": url, "request_id": request_id}
+
+
+def qzhuli_poll(server: str, request_id: str):
+    """轮询 Q助理扫码结果：{status: pending|ok|expired, ticket?}"""
+    return _post_public(server, "/api/qzhuli/cli/poll", {"request_id": request_id})
+
+
+def qzhuli_exchange(server: str, ticket: str):
+    """
+    用 ticket 兑换登录（响应 data 为驼峰键）：
+      已绑定 -> {token, uid, username, ...}
+      未绑定 -> {needBind: true, bindToken, nickname, qUid}
+    """
+    return _post_public(server, "/api/qzhuli/exchange", {"ticket": ticket})
+
+
+def qzhuli_bind(server: str, bind_token: str, username: str, password: str):
+    """未绑定 Q助理：用易发版账号密码完成绑定并登录"""
+    return _post_public(server, "/api/qzhuli/bind", {
+        "bind_token": bind_token,
+        "username": username,
+        "password": password,
+    })
+
+
+def require_auth():
+    from yifaban.config import load_auth
+
+    auth = load_auth()
+    if not auth.get("token") or not auth.get("uid"):
+        raise ApiError("未登录，请先运行 yifaban login")
+    return auth
+
+
+def get_apps():
+    """获取账号下的应用列表"""
+    auth = require_auth()
+    data = _post(
+        auth["server"],
+        "/api/app/get_apps_data",
+        {"uid": auth["uid"], "with_channel_config": 0, "with_latest_task": 0},
+        auth["token"],
+        auth["uid"],
+    )
+    return data.get("apps", [])
+
+
+def get_channel_configs(app_id: str):
+    """获取某应用在所有市场的渠道配置"""
+    auth = require_auth()
+    return _post(
+        auth["server"],
+        "/api/config/get_app_channel_configs",
+        {"uid": auth["uid"], "app_id": app_id},
+        auth["token"],
+        auth["uid"],
+    )
+
+
+_CAMEL_TO_SNAKE = {
+    "clientId": "client_id",
+    "clientSecret": "client_secret",
+    "appId": "app_id",
+    "accessKey": "access_key",
+    "accessSecret": "access_secret",
+    "publicKey": "public_key",
+    "userId": "user_id",
+}
+
+
+def channel_publisher_config(server_cfg: dict, channel: str) -> dict:
+    """
+    把服务器返回的渠道配置（驼峰键）转换为发布器需要的配置（下划线键）。
+    小米 publicKey 服务器侧已做 base64 解码，这里直接透传。
+    """
+    config_data = (server_cfg or {}).get("configData") or {}
+    result = {}
+    for k, v in config_data.items():
+        if v is None:
+            continue
+        snake = _CAMEL_TO_SNAKE.get(k, k)
+        result[snake] = v
+    # 渠道行自身的 app_id 兜底（华为/荣耀/腾讯发布器需要）
+    if "app_id" not in result and server_cfg.get("appId"):
+        result["app_id"] = server_cfg.get("appId")
+    return apply_env_overrides(channel, result)