import base64

import requests
from pyrogram import Client, filters
from pyrogram.types import Message
from utils.misc import modules_help, prefix


@Client.on_message(filters.command(['imgur'], prefix) & filters.me)
async def imgur(_, message: Message):
    # Check if a reply exists
    msg = await message.edit_text('🎉 Please wait. trying to upload...')
    if message.reply_to_message and message.reply_to_message.photo:
        # Download the photo
        photo_path = await message.reply_to_message.download()
        # Read the photo file and encode as base64
        with open(photo_path, 'rb') as file:
            data = file.read()
            base64_data = base64.b64encode(data)
        # Set API endpoint and headers for image upload
        url = 'https://api.imgur.com/3/image'
        headers = {'Authorization': 'Client-ID a10ad04550b0648'}
        # Upload image to Imgur and get URL
        response = requests.post(url, headers=headers, data={'image': base64_data})
        result = response.json()
        await msg.edit_text(result['data']['link'])
    elif message.reply_to_message and message.reply_to_message.animation:
        # Download the animation (GIF)
        animation_path = await message.reply_to_message.download()
        # Read the animation file and encode as base64
        with open(animation_path, 'rb') as file:
            data = file.read()
            base64_data = base64.b64encode(data)
        # Set API endpoint and headers for animation upload
        url = 'https://api.imgur.com/3/image'
        headers = {'Authorization': 'Client-ID a10ad04550b0648'}
        # Upload animation to Imgur and get URL
        response = requests.post(url, headers=headers, data={'image': base64_data})
        result = response.json()
        await msg.edit_text(result['data']['link'])
    else:
        await msg.edit_text('Please reply to a photo or animation (GIF) to upload to Imgur.')


modules_help['imgur'] = {
    'imgur [img]*': 'upload a photo or animation (GIF) to imgur',
}