Coupang Product Search

#!/usr/bin/env python3
"""Bootstrap and run retention-corp/coupang_partners Coupang MCP tools.

The k-skill repo intentionally does not vendor the third-party implementation.
This wrapper keeps the skill pointed at the approved upstream repository, clones it
into a user cache when needed, and then delegates to its local MCP-compatible CLI.
"""

from __future__ import annotations

import argparse
import os
import pathlib
import subprocess
import sys
from typing import Sequence

UPSTREAM_REPO_URL = "https://github.com/retention-corp/coupang_partners.git"
DEFAULT_MCP_ENDPOINT = "local://coupang-mcp"
DEFAULT_REPO_DIR = pathlib.Path(os.getenv("COUPANG_PARTNERS_REPO_DIR", "~/.cache/k-skill/coupang_partners")).expanduser()
UPSTREAM_CLI = pathlib.Path("bin") / "coupang_mcp.py"


class BootstrapError(RuntimeError):
    """Raised when the upstream checkout cannot be prepared."""


def parse_args(argv: Sequence[str] | None = None) -> argparse.Namespace:
    parser = argparse.ArgumentParser(
        description="Run the retention-corp/coupang_partners local Coupang MCP-compatible CLI.",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog=(
            "Examples:\n"
            "  coupang_partners_mcp.py tools\n"
            "  coupang_partners_mcp.py init\n"
            "  coupang_partners_mcp.py search 생수\n"
            "  coupang_partners_mcp.py budget 키보드 --max-price 100000\n"
            "\n"
            "Honored upstream environment variables (forwarded as-is):\n"
            "  COUPANG_ACCESS_KEY, COUPANG_SECRET_KEY\n"
            "      Operator Coupang Partners API credentials. When set, upstream\n"
            "      uses the local HMAC-signed Coupang Partners path.\n"
            "  OPENCLAW_SHOPPING_CLIENT_ID\n"
            "      Allowlisted client id for the hosted fallback. upstream sends\n"
            "      openclaw-skill by default, which is the value currently on the\n"
            "      Retention Corp allowlist; k-skill does not override this.\n"
            "  OPENCLAW_SHOPPING_FORCE_HOSTED=1\n"
            "      Force the hosted fallback even when Coupang keys are present.\n"
            "  OPENCLAW_SHOPPING_BASE_URL\n"
            "      Override the hosted backend base URL. Default upstream target\n"
            "      is https://a.retn.kr and /v1/public/assist is the public entry.\n"
            "\n"
            "When both COUPANG_ACCESS_KEY and COUPANG_SECRET_KEY are missing,\n"
            "upstream falls back to the hosted Retention Corp backend so this\n"
            "skill keeps working without Coupang Partners credentials."
        ),
    )
    parser.add_argument(
        "--repo-dir",
        default=str(DEFAULT_REPO_DIR),
        help="Checkout directory for retention-corp/coupang_partners (default: %(default)s).",
    )
    parser.add_argument(
        "--no-clone",
        action="store_true",
        help="Do not clone the upstream repository if it is missing; fail with setup guidance instead.",
    )
    parser.add_argument(
        "--update",
        action="store_true",
        help="Run git pull --ff-only in an existing upstream checkout before delegating.",
    )
    parser.add_argument(
        "upstream_args",
        nargs=argparse.REMAINDER,
        help="Arguments passed to bin/coupang_mcp.py, for example: tools, search 생수, rocket 에어팟.",
    )
    args = parser.parse_args(argv)
    if args.upstream_args and args.upstream_args[0] == "--":
        args.upstream_args = args.upstream_args[1:]
    if not args.upstream_args:
        parser.error("missing upstream command; try: tools, init, search <keyword>, rocket <keyword>, budget <keyword>")
    return args


def upstream_cli_path(repo_dir: pathlib.Path) -> pathlib.Path:
    return repo_dir / UPSTREAM_CLI


def ensure_repo(repo_dir: pathlib.Path, *, clone: bool = True, update: bool = False) -> pathlib.Path:
    cli_path = upstream_cli_path(repo_dir)
    if cli_path.exists():
        if update:
            run_checked([