goofish-cli 0.1.0__py3-none-any.whl

goofish_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

goofish_cli/cli.py

@@ -0,0 +1,119 @@
+"""Typer 入口。扫描 registry → 按 namespace 聚合为子命令树 → 统一注入 --format。"""
+from __future__ import annotations
+
+import sys
+from collections import defaultdict
+
+import typer
+from loguru import logger
+
+from goofish_cli.core import GoofishError, iter_commands
+from goofish_cli.core.output import Format, render
+from goofish_cli.core.registry import Command, discover
+
+app = typer.Typer(
+    name="goofish",
+    help="闲鱼 CLI — 支持 MCP，未来支持 Skills。为 AI Agent 提供闲鱼自动化基础能力。",
+    no_args_is_help=True,
+    add_completion=False,
+)
+
+
+def _format_option() -> typer.Option:
+    return typer.Option(
+        Format.TABLE.value,
+        "--format",
+        "-f",
+        help="输出格式：json/yaml/table/md/csv",
+    )
+
+
+def _wrap(cmd: Command):
+    """把 registry 里的函数包装成 Typer 回调，注入 --format 并统一渲染/异常处理。"""
+
+    # 直接复用原函数的签名，加一个 --format
+    import functools
+    import inspect
+
+    sig = inspect.signature(cmd.func)
+
+    @functools.wraps(cmd.func)
+    def wrapper(**kwargs):
+        fmt = Format(kwargs.pop("format"))
+        try:
+            result = cmd.func(**kwargs)
+        except GoofishError as e:
+            typer.secho(f"[{type(e).__name__}] {e}", fg=typer.colors.RED, err=True)
+            raise typer.Exit(code=e.exit_code) from e
+        except Exception as e:  # noqa: BLE001
+            typer.secho(f"[Error] {e}", fg=typer.colors.RED, err=True)
+            logger.exception(e)
+            raise typer.Exit(code=1) from e
+        render(result, fmt=fmt, columns=cmd.columns or None)
+
+    # 重建签名以让 Typer 正确推导
+    new_params = list(sig.parameters.values()) + [
+        inspect.Parameter(
+            "format",
+            kind=inspect.Parameter.KEYWORD_ONLY,
+            default=Format.TABLE.value,
+            annotation=str,
+        )
+    ]
+    wrapper.__signature__ = sig.replace(parameters=new_params)  # type: ignore[attr-defined]
+    return wrapper
+
+
+def build_app() -> typer.Typer:
+    discover()
+    # namespace → Typer sub
+    subs: dict[str, typer.Typer] = defaultdict(lambda: typer.Typer(no_args_is_help=True))
+
+    by_ns: dict[str, list[Command]] = defaultdict(list)
+    for cmd in iter_commands():
+        by_ns[cmd.namespace].append(cmd)
+
+    for ns, cmds in by_ns.items():
+        sub = typer.Typer(
+            name=ns,
+            help=f"{ns} 子命令（{len(cmds)} 个）",
+            no_args_is_help=True,
+            add_completion=False,
+        )
+        for cmd in cmds:
+            sub.command(name=cmd.name, help=cmd.description)(_wrap(cmd))
+        subs[ns] = sub
+        app.add_typer(sub, name=ns)
+
+    return app
+
+
+@app.command(name="version")
+def version_cmd():
+    """打印版本号。"""
+    from goofish_cli import __version__
+    print(f"goofish-cli {__version__}")
+
+
+@app.command(name="list-commands")
+def list_commands(format: str = Format.TABLE.value):  # noqa: A002
+    """列出所有已注册命令（调试用）。"""
+    discover()
+    rows = [
+        {
+            "command": f"goofish {c.namespace} {c.name}",
+            "description": c.description,
+            "strategy": c.strategy.value,
+            "write": c.write,
+        }
+        for c in iter_commands()
+    ]
+    render(rows, fmt=Format(format), columns=["command", "description", "strategy", "write"])
+
+
+# 模块导入时即构建（Typer scripts 需要 top-level app）
+build_app()
+
+
+if __name__ == "__main__":
+    sys.exit(app())

goofish_cli/commands/auth/login.py

@@ -0,0 +1,66 @@
+"""auth login — 导入 cookie 到 ~/.goofish-cli/cookies.json。
+
+支持两种格式：
+- Chrome 扩展导出的 JSON 数组 [{"name":"...","value":"..."}]
+- cookie 字符串 "k=v; k=v"（--raw）
+"""
+
+import json
+from pathlib import Path
+
+from goofish_cli.core import Strategy, command
+from goofish_cli.core.session import DEFAULT_COOKIE_PATH
+
+
+@command(
+    namespace="auth",
+    name="login",
+    description="从 JSON 文件或 cookie 字符串导入登录态",
+    strategy=Strategy.PUBLIC,
+    columns=["path", "unb", "tracknick", "cookies_count"],
+)
+def login(source: str, *, raw: bool = False) -> dict[str, object]:
+    if raw:
+        cookies = _parse_raw(source)
+    else:
+        p = Path(source).expanduser()
+        cookies = _parse_json(p.read_text())
+
+    if "unb" not in cookies or "_m_h5_tk" not in cookies:
+        raise ValueError("cookie 缺失关键字段 unb / _m_h5_tk，请重新从浏览器导出")
+
+    target = DEFAULT_COOKIE_PATH
+    target.parent.mkdir(parents=True, exist_ok=True)
+    target.write_text(json.dumps(
+        [{"name": k, "value": v} for k, v in cookies.items()],
+        ensure_ascii=False,
+        indent=2,
+    ))
+    target.chmod(0o600)
+
+    return {
+        "path": str(target),
+        "unb": cookies.get("unb", ""),
+        "tracknick": cookies.get("tracknick", ""),
+        "cookies_count": len(cookies),
+    }
+
+
+def _parse_raw(raw: str) -> dict[str, str]:
+    out: dict[str, str] = {}
+    for part in raw.split(";"):
+        part = part.strip()
+        if not part or "=" not in part:
+            continue
+        k, _, v = part.partition("=")
+        out[k.strip()] = v.strip()
+    return out
+
+
+def _parse_json(text: str) -> dict[str, str]:
+    data = json.loads(text)
+    if isinstance(data, list):
+        return {c["name"]: c["value"] for c in data if "name" in c and "value" in c}
+    if isinstance(data, dict):
+        return {str(k): str(v) for k, v in data.items()}
+    raise ValueError("cookie JSON 格式不识别（需 list 或 dict）")

goofish_cli/commands/auth/reset_guard.py

@@ -0,0 +1,16 @@
+"""auth reset-guard — 手动解除风控熔断状态。"""
+
+from goofish_cli.core import Strategy, command
+from goofish_cli.core.guard import reset
+
+
+@command(
+    namespace="auth",
+    name="reset-guard",
+    description="手动解除风控熔断状态",
+    strategy=Strategy.PUBLIC,
+    columns=["ok"],
+)
+def reset_guard() -> dict[str, bool]:
+    reset()
+    return {"ok": True}

goofish_cli/commands/auth/status.py

@@ -0,0 +1,40 @@
+"""auth status — 检查登录态是否有效。调用 mtop.taobao.idlemessage.pc.loginuser.get"""
+
+from typing import Any
+
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.mtop import call
+
+
+@command(
+    namespace="auth",
+    name="status",
+    description="检查登录态是否有效，返回 unb / tracknick / 昵称",
+    strategy=Strategy.COOKIE,
+    columns=["unb", "tracknick", "nick", "valid"],
+)
+def status() -> dict[str, Any]:
+    session = Session.load()
+    try:
+        raw = call(
+            session,
+            api="mtop.taobao.idlemessage.pc.loginuser.get",
+            data={},
+            version="1.0",
+            spm_cnt="a21ybx.im.0.0",
+        )
+        user = raw.get("data", {}) or {}
+        return {
+            "unb": session.unb,
+            "tracknick": session.tracknick,
+            "nick": user.get("nick", ""),
+            "valid": True,
+        }
+    except Exception as e:  # noqa: BLE001
+        return {
+            "unb": session.unb,
+            "tracknick": session.tracknick,
+            "nick": "",
+            "valid": False,
+            "error": str(e),
+        }

goofish_cli/commands/category/recommend.py

@@ -0,0 +1,64 @@
+"""category recommend — AI 识别商品类目（发布前置）。
+
+接口：mtop.taobao.idle.kgraph.property.recommend v2.0
+"""
+
+import json
+from typing import Any
+
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.mtop import call
+
+
+@command(
+    namespace="category",
+    name="recommend",
+    description="AI 识别商品类目，输入标题+图片返回 catId/catName",
+    strategy=Strategy.COOKIE,
+    columns=["cat_id", "cat_name", "channel_cat_id", "tb_cat_id", "confidence"],
+)
+def recommend(
+    title: str,
+    images_json: str = "[]",
+) -> dict[str, Any]:
+    """images_json 是 JSON 字符串：[{"url":"...","width":1024,"height":1024}, ...]"""
+    session = Session.load()
+    images = json.loads(images_json) if images_json else []
+    image_infos: list[dict[str, Any]] = []
+    for img in images:
+        image_infos.append({
+            "extraInfo": {"isH": "false", "isT": "false", "raw": "false"},
+            "isQrCode": False,
+            "url": img["url"],
+            "heightSize": img["height"],
+            "widthSize": img["width"],
+            "major": True,
+            "type": 0,
+            "status": "done",
+        })
+
+    raw = call(
+        session,
+        api="mtop.taobao.idle.kgraph.property.recommend",
+        data={
+            "title": title,
+            "lockCpv": False,
+            "multiSKU": False,
+            "publishScene": "mainPublish",
+            "scene": "newPublishChoice",
+            "description": title,
+            "imageInfos": image_infos,
+            "uniqueCode": "1775905618164677",
+        },
+        version="2.0",
+        spm_cnt="a21ybx.publish.0.0",
+    )
+    predict = (raw.get("data", {}) or {}).get("categoryPredictResult", {}) or {}
+    return {
+        "cat_id": str(predict.get("catId", "")),
+        "cat_name": predict.get("catName", ""),
+        "channel_cat_id": str(predict.get("channelCatId", "")),
+        "tb_cat_id": str(predict.get("tbCatId", "")),
+        "confidence": predict.get("confidence", 0),
+        "raw": raw,
+    }

goofish_cli/commands/item/delete.py

@@ -0,0 +1,32 @@
+"""item delete — 下架商品。接口 com.taobao.idle.item.delete v1.1（写操作）"""
+
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.guard import watch
+from goofish_cli.core.limiter import acquire
+from goofish_cli.core.mtop import call
+
+
+@command(
+    namespace="item",
+    name="delete",
+    description="下架/删除商品（写操作，受限流和风控护栏保护）",
+    strategy=Strategy.COOKIE,
+    columns=["item_id", "ok", "message"],
+    write=True,
+)
+def delete(item_id: str) -> dict[str, object]:
+    session = Session.load()
+    with acquire("item.write"), watch():
+        raw = call(
+            session,
+            api="com.taobao.idle.item.delete",
+            data={"itemId": str(item_id)},
+            version="1.1",
+            spm_cnt="a21ybx.item.0.0",
+        )
+    ret = raw.get("ret") or []
+    return {
+        "item_id": str(item_id),
+        "ok": any("SUCCESS" in r for r in ret),
+        "message": " | ".join(ret),
+    }

goofish_cli/commands/item/get.py

@@ -0,0 +1,34 @@
+"""item get — 查询商品详情。接口 mtop.taobao.idle.pc.detail v1.0（只读）"""
+
+from typing import Any
+
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.mtop import call
+
+
+@command(
+    namespace="item",
+    name="get",
+    description="查询闲鱼商品详情（只读）",
+    strategy=Strategy.COOKIE,
+    columns=["item_id", "title", "price", "seller_nick", "status"],
+)
+def get(item_id: str) -> dict[str, Any]:
+    session = Session.load()
+    raw = call(
+        session,
+        api="mtop.taobao.idle.pc.detail",
+        data={"itemId": str(item_id)},
+        version="1.0",
+        spm_cnt="a21ybx.item.0.0",
+    )
+    data = raw.get("data", {}) or {}
+    track = data.get("trackParams", {}) or {}
+    return {
+        "item_id": track.get("id", item_id),
+        "title": track.get("title", ""),
+        "price": track.get("soldPrice") or track.get("price", ""),
+        "seller_nick": track.get("seller_nick", ""),
+        "status": track.get("itemStatus", ""),
+        "raw": raw,
+    }

goofish_cli/commands/item/publish.py

@@ -0,0 +1,175 @@
+"""item publish — 发布新商品。
+
+流程：upload_images → category.recommend → location.default → publish
+接口：mtop.idle.pc.idleitem.publish v1.0（写操作）
+"""
+
+import json
+from typing import Any, Literal
+
+from goofish_cli.commands.category.recommend import recommend
+from goofish_cli.commands.location.default import default as get_default_location
+from goofish_cli.commands.media.upload import upload
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.guard import watch
+from goofish_cli.core.limiter import acquire
+from goofish_cli.core.mtop import call
+
+
+@command(
+    namespace="item",
+    name="publish",
+    description="发布商品（自动识别类目 + 默认地址），价格单位元",
+    strategy=Strategy.COOKIE,
+    columns=["item_id", "title", "price", "cat_name", "ok"],
+    write=True,
+)
+def publish(
+    title: str,
+    desc: str,
+    images: list[str],
+    price: float,
+    original_price: float | None = None,
+    delivery: Literal["包邮", "按距离计费", "一口价", "无需邮寄"] = "无需邮寄",
+    post_price: float = 0,
+    can_self_pickup: bool = True,
+) -> dict[str, Any]:
+    session = Session.load()
+
+    # 1. 上传图片（每张独立限流）
+    image_infos: list[dict[str, Any]] = []
+    for img_path in images:
+        with acquire("item.write"):
+            r = upload(img_path)
+        image_infos.append({"url": r["url"], "width": r["width"], "height": r["height"]})
+
+    # 2. AI 类目
+    cat = recommend(title, json.dumps(image_infos))
+
+    # 3. 默认地址
+    loc = get_default_location()
+
+    # 4. 发布（最后一步，走熔断）
+    data = _build_publish_data(
+        title=title,
+        desc=desc,
+        image_infos=image_infos,
+        price=price,
+        original_price=original_price,
+        delivery=delivery,
+        post_price=post_price,
+        can_self_pickup=can_self_pickup,
+        cat_info=cat,
+        location=loc,
+    )
+
+    with acquire("item.write"), watch():
+        raw = call(
+            session,
+            api="mtop.idle.pc.idleitem.publish",
+            data=data,
+            version="1.0",
+            spm_cnt="a21ybx.publish.0.0",
+        )
+
+    data_out = raw.get("data", {}) or {}
+    return {
+        "item_id": data_out.get("itemId", ""),
+        "title": title,
+        "price": price,
+        "cat_name": cat["cat_name"],
+        "ok": any("SUCCESS" in r for r in raw.get("ret", [])),
+    }
+
+
+def _build_publish_data(
+    *,
+    title: str,
+    desc: str,
+    image_infos: list[dict[str, Any]],
+    price: float,
+    original_price: float | None,
+    delivery: str,
+    post_price: float,
+    can_self_pickup: bool,
+    cat_info: dict[str, Any],
+    location: dict[str, Any],
+) -> dict[str, Any]:
+    image_do_list = [
+        {
+            "extraInfo": {"isH": "false", "isT": "false", "raw": "false"},
+            "isQrCode": False,
+            "url": img["url"],
+            "heightSize": img["height"],
+            "widthSize": img["width"],
+            "major": True,
+            "type": 0,
+            "status": "done",
+        }
+        for img in image_infos
+    ]
+
+    post_fee: dict[str, Any] = {
+        "canFreeShipping": False,
+        "supportFreight": False,
+        "onlyTakeSelf": False,
+    }
+    if delivery == "包邮":
+        post_fee["canFreeShipping"] = True
+        post_fee["supportFreight"] = True
+    elif delivery == "按距离计费":
+        post_fee["supportFreight"] = True
+        post_fee["templateId"] = "-100"
+    elif delivery == "一口价":
+        post_fee["supportFreight"] = True
+        post_fee["postPriceInCent"] = str(int(post_price * 100))
+        post_fee["templateId"] = "0"
+    elif delivery == "无需邮寄":
+        post_fee["templateId"] = "0"
+
+    price_dto: dict[str, str] = {}
+    default_price = price <= 0
+    if not default_price:
+        price_dto["priceInCent"] = str(int(price * 100))
+    if original_price and original_price > 0:
+        price_dto["origPriceInCent"] = str(int(original_price * 100))
+
+    item_addr: dict[str, Any] = {}
+    if location.get("division_id"):
+        all_addrs = location.get("all", []) or []
+        first = all_addrs[0] if all_addrs else {}
+        item_addr = {
+            "area": first.get("area", ""),
+            "city": first.get("city", ""),
+            "divisionId": first.get("divisionId", ""),
+            "gps": f"{first.get('longitude', '')},{first.get('latitude', '')}",
+            "poiId": first.get("poiId", ""),
+            "poiName": first.get("poi", ""),
+            "prov": first.get("prov", ""),
+        }
+
+    return {
+        "freebies": False,
+        "itemTypeStr": "b",
+        "quantity": "1",
+        "simpleItem": "true",
+        "imageInfoDOList": image_do_list,
+        "itemTextDTO": {"desc": desc, "title": title, "titleDescSeparate": True},
+        "itemLabelExtList": [],
+        "itemPriceDTO": price_dto,
+        "userRightsProtocols": [{"enable": False, "serviceCode": "SKILL_PLAY_NO_MIND"}],
+        "itemPostFeeDTO": post_fee,
+        "itemAddrDTO": item_addr,
+        "defaultPrice": default_price,
+        "itemCatDTO": {
+            "catId": cat_info["cat_id"],
+            "catName": cat_info["cat_name"],
+            "channelCatId": cat_info["channel_cat_id"],
+            "tbCatId": cat_info["tb_cat_id"],
+        },
+        "onlyTakeSelf": can_self_pickup,
+        "uniqueCode": "1775897582791680",
+        "sourceId": "pcMainPublish",
+        "bizcode": "pcMainPublish",
+        "publishScene": "pcMainPublish",
+    }

goofish_cli/commands/location/default.py

@@ -0,0 +1,40 @@
+"""location default — 获取默认发布地址。
+
+接口：mtop.taobao.idle.local.poi.get v1.0
+"""
+
+from typing import Any
+
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.mtop import call
+
+
+@command(
+    namespace="location",
+    name="default",
+    description="获取账号默认发布地址",
+    strategy=Strategy.COOKIE,
+    columns=["prov", "city", "area", "poi", "division_id"],
+)
+def default(longitude: float = 121.4737, latitude: float = 31.2304) -> dict[str, Any]:
+    session = Session.load()
+    raw = call(
+        session,
+        api="mtop.taobao.idle.local.poi.get",
+        data={"longitude": longitude, "latitude": latitude},
+        version="1.0",
+        spm_cnt="a21ybx.publish.0.0",
+    )
+    data = raw.get("data", {}) or {}
+    addrs = data.get("commonAddresses", []) or []
+    selected = data.get("selectedPoi") or (addrs[0] if addrs else None)
+    if not selected:
+        return {"prov": "", "city": "", "area": "", "poi": "", "division_id": "", "all": []}
+    return {
+        "prov": selected.get("prov", ""),
+        "city": selected.get("city", ""),
+        "area": selected.get("area", ""),
+        "poi": selected.get("poi", ""),
+        "division_id": str(selected.get("divisionId", "")),
+        "all": addrs or [selected],
+    }

goofish_cli/commands/media/upload.py

@@ -0,0 +1,53 @@
+"""media upload — 上传图片到闲鱼 CDN。"""
+
+import os
+
+from goofish_cli.core import Session, Strategy, command
+from goofish_cli.core.session import USER_AGENT
+
+UPLOAD_URL = "https://stream-upload.goofish.com/api/upload.api"
+
+
+@command(
+    namespace="media",
+    name="upload",
+    description="上传图片到闲鱼 CDN，返回图片 URL + 尺寸",
+    strategy=Strategy.COOKIE,
+    columns=["url", "width", "height", "size"],
+    write=True,
+)
+def upload(path: str) -> dict[str, object]:
+    session = Session.load()
+    abs_path = os.path.expanduser(path)
+    if not os.path.exists(abs_path):
+        raise FileNotFoundError(f"图片不存在：{abs_path}")
+
+    headers = {
+        "accept": "*/*",
+        "origin": "https://www.goofish.com",
+        "referer": "https://www.goofish.com/",
+        "user-agent": USER_AGENT,
+    }
+    params = {"floderId": "0", "appkey": "xy_chat", "_input_charset": "utf-8"}
+
+    with open(abs_path, "rb") as f:
+        resp = session.http.post(
+            UPLOAD_URL,
+            headers=headers,
+            params=params,
+            files={"file": (os.path.basename(abs_path), f, "image/png")},
+            timeout=60,
+        )
+    raw = resp.json()
+    obj = raw.get("object") or {}
+    pix = str(obj.get("pix", "0x0"))
+    try:
+        width, height = map(int, pix.split("x"))
+    except ValueError:
+        width = height = 0
+    return {
+        "url": obj.get("url", ""),
+        "width": width,
+        "height": height,
+        "size": obj.get("size", 0),
+    }