import base64
import os
import os.path
import tempfile

import discord
import requests
import yt_dlp
from discord.ext import commands
from dotenv import load_dotenv

# Load environment variables
load_dotenv()
PIXELDRAIN_API_KEY = os.getenv('PIXELDRAIN_API_KEY')


def _get_auth_headers():
    """Generate authentication headers for Pixeldrain API"""
    if not PIXELDRAIN_API_KEY:
        raise Exception("Pixeldrain API key not found in environment variables")

    auth_string = f":{PIXELDRAIN_API_KEY}"
    auth_bytes = auth_string.encode('ascii')
    base64_auth = base64.b64encode(auth_bytes).decode('ascii')
    return {"Authorization": f"Basic {base64_auth}"}


class Downloader(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.pixeldrain_url = "https://pixeldrain.com/api/file"

    @commands.slash_command(
        name="download",
        description="download video from internet",
    )
    async def download(
            self,
            ctx: discord.ApplicationContext,
            url: discord.Option(str, "url", required=True)
    ):
        await ctx.defer()
        try:
            # Create a temporary directory
            with tempfile.TemporaryDirectory() as temp_dir:
                # Configure yt-dlp to use the temp directory
                outtmpl = os.path.join(temp_dir, '%(title)s.%(ext)s')
                ydl_opts = {
                    'outtmpl': outtmpl,
                    'merge_output_format': 'mp4',
                }

                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                    info = ydl.extract_info(url, download=True)
                    actual_filename = ydl.prepare_filename(info)

                await ctx.followup.send("Video downloaded successfully, Uploading to pixeldrain...")
                # upload to pixeldrain
                try:
                    with open(actual_filename, 'rb') as f:
                        files = {'file': f}
                        response = requests.post(
                            self.pixeldrain_url,
                            files=files,
                            headers={
                                "Authorization": f"Basic {base64.b64encode((':' + PIXELDRAIN_API_KEY).encode()).decode()}"}
                        )

                        try:
                            response_data = response.json()
                            print("Pixeldrain response:", response_data)  # Debug log

                            if response.status_code == 201:  # 201 means successful upload
                                file_id = response_data.get('id')
                                if file_id:
                                    file_link = f"https://pixeldrain.com/u/{file_id}"
                                    await ctx.followup.send(f"oto twoje videło\n {file_link}")
                                else:
                                    await ctx.followup.send(
                                        f"⚠️ Upload successful but no file ID returned. Response: {response_data}")
                            else:
                                error_msg = response_data.get('error', 'Unknown error')
                                if response.status_code == 401:
                                    error_msg = "Invalid API key. Please check your PIXELDRAIN_API_KEY in .env file."
                                await ctx.followup.send(f"❌ Upload failed (Status {response.status_code}): {error_msg}")
                        except ValueError as e:
                            # If the response is not JSON
                            error_msg = f"Failed to parse server response: {str(e)}"
                            if response.text:
                                error_msg += f"\nResponse: {response.text[:200]}"
                            await ctx.followup.send(f"❌ {error_msg}")
                except Exception as e:
                    await ctx.followup.send(f"Error: {str(e)}")
                # No need for explicit cleanup as a tempfile.TemporaryDirectory handles it automatically

        except Exception as e:
            await ctx.followup.send(f"Error: {str(e)}")


def setup(bot):
    bot.add_cog(Downloader(bot))