import datetime
import os
import random
import shlex
import time
import traceback
import uuid
import asyncio
from pathlib import Path

from pyrogram import enums
from pyrogram.types import InlineKeyboardButton, InputMediaPhoto


async def run_subprocess(cmd):
    process = await asyncio.create_subprocess_shell(cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
    return await process.communicate()


def get_random_start_at(seconds, dur=0):
    return random.randint(0, seconds - dur)


async def get_duration(input_file_link):
    ffmpeg_dur_cmd = f"ffprobe -v error -show_entries format=duration -of csv=p=0:s=x -select_streams v:0 {shlex.quote(input_file_link)}"
    # print(ffmpeg_dur_cmd)
    out, err = await run_subprocess(ffmpeg_dur_cmd)
    if out := out.decode().strip():
        return duration if (duration := round(float(out))) else "No duration!"
    else:
        return err.decode()


def is_url(text):
    return text.startswith("http")


async def get_dimentions(input_file_link):
    ffprobe_cmd = f"ffprobe -v error -show_entries stream=width,height -of csv=p=0:s=x -select_streams v:0 {shlex.quote(input_file_link)}"
    output = await run_subprocess(ffprobe_cmd)
    try:
        width, height = [int(i.strip()) for i in output[0].decode().split("x")]
    except Exception as e:
        print(e)
        width, height = 1280, 534
    return width, height


async def screenshot_flink(c, m):
    m.from_user.id
    # if c.CURRENT_PROCESSES.get(chat_id, 0) == 1:
    #     return await m.answer('You have reached the maximum parallel processes! Try again after one of them finishes.', show_alert=True)
    # if not c.CURRENT_PROCESSES.get(chat_id):
    #     c.CURRENT_PROCESSES[chat_id] = 0
    # c.CURRENT_PROCESSES[chat_id] += 1

    _, num_screenshots = m.data.split("+")
    num_screenshots = int(num_screenshots)
    media_msg = m.message.reply_to_message
    # print(media_msg)
    if media_msg.empty:
        await m.message.edit_msg("Why did you delete the file 😠, Now i cannot help you 😒.")
        # c.CURRENT_PROCESSES[chat_id] -= 1
        return

    uid = str(uuid.uuid4())
    output_folder = Path("GenSS/").joinpath(uid)
    if not output_folder.exists():
        os.makedirs(output_folder)

    try:
        start_time = time.time()

        await m.message.edit_msg("Give me some time bruh!! 😴")

        await m.message.edit_msg("😀 Taking Snaps!")
        file_link = m.message.reply_to_message.command[1]
        duration = await get_duration(file_link)
        if isinstance(duration, str):
            return await m.message.edit_msg("Oops, What's that? Couldn't Open the file😟.")

        reduced_sec = duration - int(duration * 2 / 100)
        print(f"Total seconds: {duration}, Reduced seconds: {reduced_sec}")
        screenshots = []
        ffmpeg_errors = ""

        screenshot_secs = [get_random_start_at(reduced_sec) for _ in range(1, 1 + num_screenshots)]
        width, height = await get_dimentions(file_link)

        for i, sec in enumerate(screenshot_secs):
            thumbnail_template = output_folder.joinpath(f"{i+1}.png")
            ffmpeg_cmd = f"mediaextract -hide_banner -ss {sec} -i {shlex.quote(file_link)} -vframes 1 '{thumbnail_template}'"
            output = await run_subprocess(ffmpeg_cmd)
            await m.message.edit_msg(f"😀 `{i+1}` of `{num_screenshots}` generated!")
            if thumbnail_template.exists():
                screenshots.append(InputMediaPhoto(str(thumbnail_template), caption=f"ScreenShot at {datetime.timedelta(seconds=sec)}"))
                continue
            ffmpeg_errors += output[1].decode() + "\n\n"

        if not screenshots:
            return await m.message.edit_msg("😟 Sorry! Screenshot generation failed possibly due to some infrastructure failure 😥.")

        await m.message.edit_msg("🤓 Its done , Now starting to upload!")
        await media_msg.reply_chat_action(enums.ChatAction.UPLOAD_PHOTO)
        await media_msg.reply_media_group(screenshots, True)

        await m.message.edit_msg(f"Completed in {datetime.timedelta(seconds=int(time.time()-start_time))}\n\nJoin @YasirPediaChannel\n\n©️ https://yasirpedia.eu.org")

    except:
        traceback.print_exc()
        await m.message.edit_msg("😟 Sorry! Screenshot generation failed, ERR: {aa} 😥.")


def gen_ik_buttons():
    btns = []
    i_keyboard = []
    for i in range(2, 11):
        i_keyboard.append(InlineKeyboardButton(f"{i}", f"scht+{i}"))
        if (i > 2) and (i % 2) == 1:
            btns.append(i_keyboard)
            i_keyboard = []
        if i == 10:
            btns.append(i_keyboard)
    return btns