import os
import boto3

from botocore.exceptions import ClientError
from fastapi.responses import StreamingResponse
from src.core.auth import require_enterprise_or_front
from fastapi import APIRouter, HTTPException, Depends

router = APIRouter()

BUKCET = os.getenv("BUCKET")
BUCKET_ENDPOINT = os.getenv("BUCKET_ENDPOINT")
BUCKET_KEY_ID = os.getenv("BUCKET_KEY_ID")
BUCKET_APP_KEY = os.getenv("BUCKET_APP_KEY")
BUCKET_FOLDER = os.getenv("BUCKET_FOLDER")

ALLOWED_TYPES = {
    "image/jpeg",
    "image/png",
    "image/gif",
    "image/webp",
    "image/jpg",
}

MAX_SIZE = 5 * 1024 * 1024

s3_client = boto3.client(
    "s3",
    endpoint_url=f"https://{BUCKET_ENDPOINT}",
    aws_access_key_id=BUCKET_KEY_ID,
    aws_secret_access_key=BUCKET_APP_KEY,
    region_name="us-east-005",
)


@router.get(
        "/images/{folder}/{filename}"
)
async def download_image(
    folder: str,
    filename: str,
    _auth: dict = Depends(require_enterprise_or_front)
):
    """
    Download/visualizar imagem da bucket

    Exemplo: GET /api/images/screenshots-pomind/1234567890.jpg
    """
    try:
        file_key = f"{folder}/{filename}"

        response = s3_client.get_object(
            Bucket=BUKCET,
            Key=file_key
        )

        return StreamingResponse(
            response["Body"].iter_chunks(),
            media_type=response["ContentType"],
            headers={
                "Content-Disposition": f'inline; filename="{filename}"'
            }
        )

    except ClientError as e:
        error_code = e.response['Error']['Code']
        if error_code == "NoSuchKey":
            raise HTTPException(
                status_code=404,
                detail="Imagem não encontrada"
            )
        raise HTTPException(
            status_code=500,
            detail="Erro ao obter imagem"
        )
    except Exception as e:
        raise HTTPException(
            status_code=500,
            detail=str(e)
        )