#!/usr/bin/env python3
from gtts import gTTS
from pathlib import Path
from mutagen.mp3 import MP3
from utils.console import print_step, print_substep
from rich.progress import track
from rich.console import Console
console = Console()
import re

def save_text_to_mp3(reddit_obj):
    """Saves Text to MP3 files.

    Args:
        reddit_obj : The reddit object you received from the reddit API in the askreddit.py file.
    """
    print_step("Saving Text to MP3 files...")
    length = 0

    # Create a folder for the mp3 files.
    Path("assets/mp3").mkdir(parents=True, exist_ok=True)

    tts = gTTS(text=reddit_obj["thread_title"], lang="en", slow=False)
    tts.save("assets/mp3/title.mp3")
    length += MP3("assets/mp3/title.mp3").info.length

    try:
        Path("assets/mp3/posttext.mp3").unlink()
    except OSError:
        pass

    if reddit_obj["thread_post"] != "":
        tts = gTTS(text=reddit_obj["thread_post"], lang="en", slow=False)
        tts.save("assets/mp3/posttext.mp3")
        length += MP3("assets/mp3/posttext.mp3").info.length

    for idx, comment in track(enumerate(reddit_obj["comments"])):

        #allow user to see what comment is being saved
        print_substep(f"Saving MP3 {idx + 1} ")

        # ! Stop creating mp3 files if the length is greater than 50 seconds. This can be longer, but this is just a good starting point
        if length > 50:
            break
        comment=comment["comment_body"]
        text=re.sub('((http|https)\:\/\/)?[a-zA-Z0-9\.\/\?\:@\-_=#]+\.([a-zA-Z]){2,6}([a-zA-Z0-9\.\&\/\?\:@\-_=#])*', '', comment)
        tts = gTTS(text, lang="en", slow=False)
        tts.save(f"assets/mp3/{idx}.mp3")
        length += MP3(f"assets/mp3/{idx}.mp3").info.length

    #let user know that the MP3 files are saved
    console.log(f"[bold green]Saved {idx + 1} MP3 Files.")
    # ! Return the index so we know how many screenshots of comments we need to make.
    return length, idx