Source code for gamesheet_sdk.common.shared.image_upload

# Copyright (c) 2026 bdperkin
# SPDX-License-Identifier: MIT

"""Shared image upload functionality for GameSheet resources."""

from __future__ import annotations

import mimetypes
from http import HTTPStatus
from pathlib import Path
from typing import TYPE_CHECKING, Any

from gamesheet_sdk.common import errors
from gamesheet_sdk.common.constants import (
    BFF_API_BASE_URL,
    BFF_ASSETS_UPLOAD_URL,
    CLOUDFLARE_IMAGE_DELIVERY_BASE,
)
from gamesheet_sdk.common.exceptions import AuthenticationError, GameSheetError

if TYPE_CHECKING:
    from gamesheet_sdk.common.session import Session




def upload_image(session: Session, image_path: str, image_type: str = "image") -> str:
    """Upload an image file and return its Cloudflare CDN URL.

    Args:
        session (Session): An authenticated :class:`Session`.
        image_path (str): Path to a local image file.
        image_type (str): Type of image for error messages (e.g., "photo", "logo").

    Returns:
        str: The Cloudflare CDN URL for the uploaded image.

    Raises:
        GameSheetError: If the file is not found, is not a valid image, or the upload fails.
        AuthenticationError: If the server returns 401.

    """
    image_file_path = Path(image_path)
    if not image_file_path.exists():
        err_msg = f"{image_type.capitalize()} file not found: {image_path}"
        raise GameSheetError(err_msg)

    mime_type, _ = mimetypes.guess_type(image_path)
    if not mime_type or not mime_type.startswith("image/"):
        err_msg = f"Invalid image file: {image_path}"
        raise GameSheetError(err_msg)

    upload_url_endpoint = f"{BFF_API_BASE_URL}{BFF_ASSETS_UPLOAD_URL}"
    upload_url_response = session.post(upload_url_endpoint)
    if upload_url_response.status_code == HTTPStatus.UNAUTHORIZED:
        raise AuthenticationError(errors.ERROR_MSG_401_EXPIRED)

    if upload_url_response.status_code >= HTTPStatus.BAD_REQUEST:
        err_msg = errors.ERROR_MSG_HTTP_POST.format(
            endpoint=upload_url_endpoint,
            status_code=upload_url_response.status_code,
            text=repr(upload_url_response.text[:200]),
        )
        raise GameSheetError(err_msg)

    upload_data: dict[str, Any] = upload_url_response.json()
    if upload_data.get("status") != "success":
        err_msg = f"Failed to get upload URL: {upload_data}"
        raise GameSheetError(err_msg)

    upload_url: str = upload_data["data"]["uploadURL"]
    image_id: str = upload_data["data"]["id"]
    with image_file_path.open("rb") as f:
        upload_response = session.post(
            upload_url,
            files={"file": (image_file_path.name, f, mime_type)},
        )

    if upload_response.status_code >= HTTPStatus.BAD_REQUEST:
        err_msg = errors.ERROR_MSG_HTTP_POST.format(
            endpoint=upload_url,
            status_code=upload_response.status_code,
            text=repr(upload_response.text[:200]),
        )
        raise GameSheetError(err_msg)

    upload_result: dict[str, Any] = upload_response.json()
    if not upload_result.get("success"):
        err_msg = f"Failed to upload {image_type}: {upload_result}"
        raise GameSheetError(err_msg)

    return f"{CLOUDFLARE_IMAGE_DELIVERY_BASE}/{image_id}"