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

import os
import sys  # sysモジュールを追加
import argparse
from pathlib import Path
from typing import List
import mimetypes
from dotenv import load_dotenv
import dropbox
from dropbox.files import FileMetadata
from dropbox.exceptions import ApiError, AuthError
import uuid  # 追加

def get_content_type(file_path: str) -> str:
    """ファイルのContent-Typeを取得"""
    content_type, _ = mimetypes.guess_type(file_path)
    return content_type or 'application/octet-stream'

def upload_to_dropbox(dbx: dropbox.Dropbox, file_path: str) -> str:
    """ファイルをDropboxにアップロードし、共有リンクを返す"""
    try:
        # ファイル存在チェックを追加
        if not os.path.isfile(file_path):
            print(f"Error: File not found - {file_path}")
            return ""

        # オリジナルのファイル名から拡張子を取得
        original_name = os.path.basename(file_path)
        _, ext = os.path.splitext(original_name)
        
        # UUID4を使用してランダムなファイル名を生成
        random_filename = f"{uuid.uuid4()}{ext.lower()}"  # 拡張子を小文字に統一
        dropbox_path = f"/typora_img/{random_filename}"

        # ファイルをアップロード
        try:
            with open(file_path, 'rb') as f:
                file_metadata: FileMetadata = dbx.files_upload(
                    f.read(),
                    dropbox_path,
                    mute=True  # content_type引数を削除
                )
        except IOError as e:
            print(f"Error reading file {file_path}: {str(e)}")
            return ""

        # 共有リンクを作成
        try:
            shared_link = dbx.sharing_create_shared_link_with_settings(dropbox_path)
            return shared_link.url.replace('dl=0', 'raw=1')
        except dropbox.exceptions.ApiError as e:
            if e.error.is_shared_link_already_exists():
                # リンクが既に存在する場合は既存のリンクを取得
                shared_link = dbx.sharing_get_shared_links(dropbox_path)
                return shared_link.links[0].url.replace('dl=0', 'raw=1')
            raise

    except Exception as e:
        print(f"Unexpected error uploading {file_path}: {str(e)}")
        return ""

def main():
    try:
        # 環境変数の読み込み
        load_dotenv()
        
        # Dropbox認証情報の取得と検証
        app_key = os.getenv('DROPBOX_APP_KEY')
        app_secret = os.getenv('DROPBOX_APP_SECRET')
        refresh_token = os.getenv('DROPBOX_REFRESH_TOKEN')
        
        # 認証情報の存在チェック
        missing_vars = []
        if not app_key:
            missing_vars.append('DROPBOX_APP_KEY')
        if not app_secret:
            missing_vars.append('DROPBOX_APP_SECRET')
        if not refresh_token:
            missing_vars.append('DROPBOX_REFRESH_TOKEN')
        
        if missing_vars:
            raise ValueError(f"以下の環境変数が設定されていません: {', '.join(missing_vars)}")

        # Dropboxクライアントの初期化とリフレッシュトークンの検証
        try:
            dbx = dropbox.Dropbox(
                oauth2_refresh_token=refresh_token,
                app_key=app_key,
                app_secret=app_secret
            )
            # 接続テスト
            try:
                dbx.users_get_current_account()
            except AuthError:
                raise ValueError("リフレッシュトークンが無効です。新しいトークンを取得してください。")
            except Exception as e:
                raise ValueError(f"Dropbox APIの呼び出しに失敗しました: {str(e)}")
                
        except Exception as e:
            raise ValueError(f"Dropboxクライアントの初期化に失敗しました: {str(e)}")

        # コマンドライン引数の設定
        parser = argparse.ArgumentParser(description='Upload images to Dropbox')
        parser.add_argument('--images-path', nargs='+', required=True,
                           help='Path to image files to upload')
        args = parser.parse_args()

        # 各ファイルをアップロード
        for image_path in args.images_path:
            if os.path.exists(image_path):
                shared_link = upload_to_dropbox(dbx, image_path)
                if shared_link:
                    print(shared_link)
            else:
                print(f"File not found: {image_path}")

    except Exception as e:
        print(f"Error: {str(e)}", file=sys.stderr)
        sys.exit(1)

if __name__ == "__main__":
    main()