"""方舟（Ark）API Key 管理服务"""

import logging
from .volcengine_client import VolcengineClient, VolcengineAPIError, get_ark_client

logger = logging.getLogger(__name__)


class ArkService:
    """方舟 API Key 管理"""

    def __init__(self, ak: str, sk: str):
        self.client = get_ark_client(ak, sk)

    def list_api_keys(self, project_name: str, page_size: int = 100, page_number: int = 1) -> dict:
        """列出项目下的 API Key"""
        return self.client.call_json("ListApiKeys", {
            "ProjectName": project_name,
            "PageSize": page_size,
            "PageNumber": page_number,
        })

    def create_api_key(self, project_name: str, name: str, resource_type: str = "all") -> dict:
        """在项目下创建 API Key"""
        return self.client.call_json("CreateApiKey", {
            "ProjectName": project_name,
            "Name": name,
            "ResourceInstances": [{"ResourceId": "*", "ResourceType": resource_type}],
        })

    def delete_api_key(self, api_key_id: int) -> dict:
        """删除 API Key"""
        return self.client.call_json("DeleteApiKey", {
            "Id": api_key_id,
        })

    def update_api_key_status(self, api_key_id: int, status: str) -> dict:
        """启用/停用 API Key (status: Active / Inactive)"""
        return self.client.call_json("UpdateApiKey", {
            "Id": api_key_id,
            "Status": status,
        })