Qiniu MCP Server

import logging
import os
from typing import List
from attr import dataclass
from dotenv import load_dotenv

from ..consts import consts

_CONFIG_ENV_KEY_ACCESS_KEY = "QINIU_ACCESS_KEY"
_CONFIG_ENV_KEY_SECRET_KEY = "QINIU_SECRET_KEY"
_CONFIG_ENV_LIVE_API_KEY = "QINIU_LIVE_API_KEY"
_CONFIG_ENV_LIVE_ENDPOINT = "QINIU_LIVE_ENDPOINT"
_CONFIG_ENV_KEY_ENDPOINT_URL = "QINIU_ENDPOINT_URL"
_CONFIG_ENV_KEY_REGION_NAME = "QINIU_REGION_NAME"
_CONFIG_ENV_KEY_BUCKETS = "QINIU_BUCKETS"

logger = logging.getLogger(consts.LOGGER_NAME)

# Load environment variables at package initialization
load_dotenv()


@dataclass
class Config:
    access_key: str
    secret_key: str
    live_api_key: str
    live_endpoint: str
    endpoint_url: str
    region_name: str
    buckets: List[str]


def load_config() -> Config:
    config = Config(
        access_key=os.getenv(_CONFIG_ENV_KEY_ACCESS_KEY),
        secret_key=os.getenv(_CONFIG_ENV_KEY_SECRET_KEY),
        live_api_key=os.getenv(_CONFIG_ENV_LIVE_API_KEY),
        live_endpoint=os.getenv(_CONFIG_ENV_LIVE_ENDPOINT),
        endpoint_url=os.getenv(_CONFIG_ENV_KEY_ENDPOINT_URL),
        region_name=os.getenv(_CONFIG_ENV_KEY_REGION_NAME),
        buckets=_get_configured_buckets_from_env(),
    )

    if not config.access_key or len(config.access_key) == 0:
        config.access_key = "YOUR_QINIU_ACCESS_KEY"
    if not config.secret_key or len(config.secret_key) == 0:
        config.secret_key = "YOUR_QINIU_SECRET_KEY"
    if not config.live_api_key or len(config.live_api_key) == 0:
        config.live_api_key = "YOUR_QINIU_LIVE_API_KEY"
    if not config.live_endpoint or len(config.live_endpoint) == 0:
        config.live_endpoint = "mls.cn-east-1.qiniumiku.com"
    if not config.endpoint_url or len(config.endpoint_url) == 0:
        config.endpoint_url = "YOUR_QINIU_ENDPOINT_URL"
    if not config.region_name or len(config.region_name) == 0:
        config.region_name = "YOUR_QINIU_REGION_NAME"

    logger.info(f"Configured   access_key: {config.access_key}")
    logger.info(f"Configured endpoint_url: {config.endpoint_url}")
    logger.info(f"Configured  region_name: {config.region_name}")
    logger.info(f"Configured      buckets: {config.buckets}")
    return config


def _get_configured_buckets_from_env() -> List[str]:
    bucket_list = os.getenv(_CONFIG_ENV_KEY_BUCKETS)
    if bucket_list:
        buckets = [b.strip() for b in bucket_list.split(",")]
        return buckets
    else:
        return []