vps-manager/app/adapters/namesilo.py

"""Namesilo 适配器（REST API v1，API Key 查询参数）

API 文档：https://www.namesilo.com/api_reference.php
所需配置：{"api_key": "..."}

覆盖：注册域名清单（含到期日）、DNS 记录增删改查。
Namesilo 是纯域名注册商，无 VPS，故不实现 list_vps / 实例动作。

响应约定：所有命令都返回 {"request": ..., "reply": {...}}，
reply.code == 300 才是成功，其余（如 301 未授权、400 参数错）需读 reply.detail。
"""

import httpx

from app.adapters.base import (
    BaseAdapter,
    NormalizedDNSRecord,
    NormalizedDomain,
    parse_iso_date,
)
from app.adapters.registry import register

SUCCESS_CODE = 300


@register("namesilo-api")
class NamesiloAdapter(BaseAdapter):
    required_config = ["api_key"]
    BASE = "https://www.namesilo.com/api"

    def _call(self, command: str, params: dict = None) -> dict:
        query = {"version": "1", "type": "json", "key": self.config.get("api_key", "")}
        if params:
            query.update({k: v for k, v in params.items() if v is not None})
        with httpx.Client(timeout=30) as client:
            resp = client.get(f"{self.BASE}/{command}", params=query)
            resp.raise_for_status()
            data = resp.json()
        reply = data.get("reply", {}) or {}
        code = reply.get("code")
        # Namesilo 的 code 可能是 int 也可能是 str，统一按 int 比较
        try:
            code = int(code)
        except (TypeError, ValueError):
            code = None
        if code != SUCCESS_CODE:
            raise RuntimeError(f"Namesilo {command} 失败（code={reply.get('code')}）：{reply.get('detail')}")
        return reply

    def test_connection(self) -> dict:
        try:
            reply = self._call("checkApiKey")
            return {
                "ok": True,
                "message": f"连接成功，令牌属于 {reply.get('username') or '未知账户'}",
            }
        except httpx.HTTPStatusError as e:
            return {"ok": False, "message": f"HTTP {e.response.status_code}：API Key 无效"}
        except Exception as e:  # noqa: BLE001
            return {"ok": False, "message": str(e)}

    # ---------------- 只读清单 ----------------

    def list_domains(self) -> list:
        reply = self._call("listDomains")
        result = []
        for d in reply.get("domains", []) or []:
            result.append(
                NormalizedDomain(
                    external_id=d.get("domain"),
                    domain_name=d.get("domain"),
                    registrar="namesilo",
                    # expires 形如 "2027-01-01"；Namesilo 也可能给 "Jan 01, 2027"，解析失败归 None
                    expiry_date=parse_iso_date(d.get("expires")),
                    status="active",
                    raw=d,
                )
            )
        return result

    # ---------------- DNS 记录管理 ----------------
    # zone_id 即域名字符串；record_id 用 Namesilo 的 record_id

    def list_dns_records(self, zone_id: str) -> list:
        reply = self._call("listDnsRecords", {"domain": zone_id})
        result = []
        for r in reply.get("dns_records", reply.get("resource_record", [])) or []:
            host = r.get("host") or ""
            result.append(
                NormalizedDNSRecord(
                    record_id=str(r.get("record_id") or r.get("id") or ""),
                    zone_id=zone_id,
                    name=self._fqdn(host, zone_id),
                    type=r.get("type"),
                    content=r.get("value"),
                    ttl=int(r["ttl"]) if r.get("ttl") else None,
                    priority=int(r["distance"]) if r.get("distance") else None,
                    raw=r,
                )
            )
        return result

    @staticmethod
    def _fqdn(host: str, zone_id: str) -> str:
        """Namesilo 的 host 是子域（空或 @ 表示裸域），拼成完整域名供台账展示"""
        if not host or host == "@":
            return zone_id
        return f"{host}.{zone_id}"

    @staticmethod
    def _host(name: str, zone_id: str) -> str:
        """反向：把完整域名拆回子域（裸域用空串）"""
        if not name or name == zone_id:
            return ""
        suffix = f".{zone_id}"
        return name[: -len(suffix)] if name.endswith(suffix) else name

    def create_dns_record(self, zone_id: str, record: dict) -> dict:
        # Namesilo 用 dnsRecordUpdate 同时承担新增（不带 record_id）与修改（带 record_id）
        params = {
            "domain": zone_id,
            "rr_type": record.get("type", "A"),
            "rr_host": self._host(record.get("name", ""), zone_id),
            "rr_value": record.get("content", ""),
            "rr_ttl": record.get("ttl") or 3600,
        }
        if record.get("priority") is not None:
            params["rr_distance"] = record["priority"]
        try:
            self._call("dnsRecordUpdate", params)
            return {"ok": True, "message": "记录已创建"}
        except httpx.HTTPStatusError as e:
            return {"ok": False, "message": f"HTTP {e.response.status_code}：创建失败"}
        except Exception as e:  # noqa: BLE001
            return {"ok": False, "message": f"创建失败：{e}"}

    def update_dns_record(self, zone_id: str, record_id: str, record: dict) -> dict:
        params = {
            "domain": zone_id,
            "rr_id": record_id,
            "rr_type": record.get("type", "A"),
            "rr_host": self._host(record.get("name", ""), zone_id),
            "rr_value": record.get("content", ""),
            "rr_ttl": record.get("ttl") or 3600,
        }
        if record.get("priority") is not None:
            params["rr_distance"] = record["priority"]
        try:
            self._call("dnsRecordUpdate", params)
            return {"ok": True, "message": "记录已更新"}
        except httpx.HTTPStatusError as e:
            return {"ok": False, "message": f"HTTP {e.response.status_code}：更新失败"}
        except Exception as e:  # noqa: BLE001
            return {"ok": False, "message": f"更新失败：{e}"}

    def delete_dns_record(self, zone_id: str, record_id: str) -> dict:
        try:
            self._call("dnsRecordDelete", {"domain": zone_id, "rr_id": record_id})
            return {"ok": True, "message": "记录已删除"}
        except httpx.HTTPStatusError as e:
            return {"ok": False, "message": f"HTTP {e.response.status_code}：删除失败"}
        except Exception as e:  # noqa: BLE001
            return {"ok": False, "message": f"删除失败：{e}"}