vps-manager/app/adapters/vultr.py

"""Vultr 适配器（REST API v2，Bearer Token）

API 文档：https://www.vultr.com/api/
所需配置：{"api_key": "..."}
"""

import httpx

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


@register("vultr-api")
class VultrAdapter(BaseAdapter):
    required_config = ["api_key"]
    BASE = "https://api.vultr.com/v2"

    def _headers(self) -> dict:
        return {"Authorization": f"Bearer {self.config.get('api_key', '')}"}

    def _get(self, path: str) -> dict:
        with httpx.Client(timeout=30) as client:
            resp = client.get(self.BASE + path, headers=self._headers())
            resp.raise_for_status()
            return resp.json()

    def test_connection(self) -> dict:
        try:
            data = self._get("/account")
            balance = data.get("account", {}).get("balance")
            return {"ok": True, "message": f"连接成功，账户余额 {balance} USD"}
        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 get_account(self) -> AccountInfo:
        acc = self._get("/account").get("account", {})
        return AccountInfo(
            balance=acc.get("balance"),
            currency="USD",
            pending_charges=acc.get("pending_charges"),
            raw=acc,
        )

    def list_vps(self) -> list:
        result = []
        cursor = ""
        while True:
            path = "/instances?per_page=100"
            if cursor:
                path += f"&cursor={cursor}"
            data = self._get(path)
            for inst in data.get("instances", []):
                ram_mb = inst.get("ram") or 0
                result.append(
                    NormalizedVPS(
                        external_id=inst.get("id"),
                        name=inst.get("label") or inst.get("id"),
                        ip_address=inst.get("main_ip"),
                        region=inst.get("region"),
                        os=inst.get("os"),
                        cpu_cores=inst.get("vcpu_count"),
                        memory_gb=round(ram_mb / 1024, 1) if ram_mb else None,
                        disk_gb=inst.get("disk"),
                        status="active" if inst.get("status") == "active" else (inst.get("status") or "unknown"),
                        monthly_cost=inst.get("monthly_cost"),
                        currency="USD",
                        raw=inst,
                    )
                )
            cursor = data.get("meta", {}).get("links", {}).get("next") or ""
            if not cursor:
                break
        return result

    def list_domains(self) -> list:
        result = []
        cursor = ""
        while True:
            path = "/domains?per_page=100"
            if cursor:
                path += f"&cursor={cursor}"
            data = self._get(path)
            for d in data.get("domains", []):
                result.append(
                    NormalizedDomain(
                        external_id=d.get("domain"),
                        domain_name=d.get("domain"),
                        registrar="vultr",
                        # 非 Vultr 注册的域名 expiry_date 为空，解析失败统一归为 None
                        expiry_date=parse_iso_date(d.get("expiry_date")),
                        raw=d,
                    )
                )
            cursor = data.get("meta", {}).get("links", {}).get("next") or ""
            if not cursor:
                break
        return result