from minio import Minio
from config import settings

# MinIO 客户端
minio_client = Minio(
    settings.MINIO_ENDPOINT,
    access_key=settings.MINIO_ACCESS_KEY,
    secret_key=settings.MINIO_SECRET_KEY,
    secure=settings.MINIO_USE_SSL,
)


async def init_buckets():
    """初始化 MinIO 存储桶（应用启动时调用）"""
    buckets = [
        settings.MINIO_BUCKET_TEMPLATES,  # 原始模板文件
        settings.MINIO_BUCKET_UPLOADS,    # 用户上传的参考文件
        settings.MINIO_BUCKET_OUTPUTS,    # 生成的文档
    ]
    for bucket in buckets:
        if not minio_client.bucket_exists(bucket):
            minio_client.make_bucket(bucket)
            print(f"[MinIO] 创建存储桶: {bucket}")


def get_file_url(bucket: str, object_name: str) -> str:
    """获取文件的公开访问 URL"""
    if settings.MINIO_USE_SSL:
        protocol = "https"
    else:
        protocol = "http"
    return f"{protocol}://{settings.MINIO_ENDPOINT}/{bucket}/{object_name}"


def get_presigned_url(bucket: str, object_name: str, expires: int = 3600) -> str:
    """获取预签名下载 URL（带过期时间）"""
    return minio_client.presigned_get_object(bucket, object_name, expires=expires)