#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os
import argparse
from dotenv import load_dotenv
import boto3
import mimetypes
import uuid

# .envファイルから環境変数を読み込む
load_dotenv()

# .envファイルからCloudflare R2のエンドポイントを読み込む
ENDPOINT = os.getenv('R2_ENDPOINT')

def upload_to_r2(image_path):
    """
    画像をCloudflare R2にアップロードする。

    Args:
        image_path (str): アップロードする画像のパス。

    Returns:
        str: アップロードされた画像のURL。
        int: アップロードに失敗した場合のエラーコード。
    """

    try:
        # R2のクライアントを作成
        s3 = boto3.client(
            's3',
            endpoint_url=ENDPOINT,
            aws_access_key_id=os.getenv('R2_ACCESS_KEY_ID'),
            aws_secret_access_key=os.getenv('R2_SECRET_ACCESS_KEY'),
            config=boto3.session.Config(signature_version='s3v4')
        )

        # ランダムなファイル名を生成（拡張子は元のまま）
        random_filename = str(uuid.uuid4()) + os.path.splitext(image_path)[1]

        # 画像のContentTypeをファイル名から推測
        content_type, _ = mimetypes.guess_type(image_path)

        # 画像をアップロード
        with open(image_path, 'rb') as f:
            bucket_name = os.getenv('R2_BUCKET_NAME')
            s3.put_object(Bucket=bucket_name, Key=random_filename, Body=f, ContentType=content_type)

        # アップロードされた画像のURLを返す
        return f"{os.getenv('R2_CUSTOM_URL')}{random_filename}"

    except Exception as e:
        print(f"Error uploading image: {e}")
        return None

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Upload an image to Cloudflare R2.')
    parser.add_argument('--image-path', required=True, help='Path to the image file.')
    args = parser.parse_args()

    image_path = args.image_path

    # 画像が存在するか確認
    if not os.path.exists(image_path):
        print(f"Error: Image not found at {image_path}")
        exit(1)

    # R2に画像をアップロード
    image_url = upload_to_r2(image_path)

    if image_url:
        print(f"{image_url}")
    else:
        print("Image upload failed.")