"""
This module contains functions to manage audio files.
"""

import os
import shutil
from pathlib import PurePath

import gradio as gr

from backend.common import INTERMEDIATE_AUDIO_DIR, OUTPUT_AUDIO_DIR, display_progress
from backend.exceptions import InputMissingError, InvalidPathError, PathNotFoundError

from common import GRADIO_TEMP_DIR


def get_output_audio() -> list[tuple[str, str]]:
    """
    Get the name and path of all output audio files.

    Returns
    -------
    list[tuple[str, str]]
        A list of tuples containing the name and path of each output audio file.
    """
    if os.path.isdir(OUTPUT_AUDIO_DIR):
        named_output_files = [
            (file_name, os.path.join(OUTPUT_AUDIO_DIR, file_name))
            for file_name in os.listdir(OUTPUT_AUDIO_DIR)
        ]
        return sorted(named_output_files, key=lambda x: x[0])
    return []


def delete_intermediate_audio(
    song_dirs: list[str],
    progress_bar: gr.Progress | None = None,
    percentage: float = 0.0,
) -> str:
    """
    Delete intermediate audio files in provided song directories.

    Parameters
    ----------
    song_dirs : list[str]
        Paths of song directories to delete intermediate audio files for.
    progress_bar : gr.Progress, optional
        Gradio progress bar to update.
    percentage : float, default=0.0
        Percentage to display in the progress bar.

    Returns
    -------
    str
        Success message.

    Raises
    ------
    InputMissingError
        If no song directories are provided.
    PathNotFoundError
        If a song directory does not exist.
    InvalidPathError
        If a song directory is not located in the root of the intermediate audio directory.
    """
    if not song_dirs:
        raise InputMissingError(
            "Song directories missing! Please provide a non-empty list of song"
            " directories."
        )
    display_progress(
        "[~] Deleting intermediate audio files for selected songs...",
        percentage,
        progress_bar,
    )
    for song_dir in song_dirs:
        if not os.path.isdir(song_dir):
            raise PathNotFoundError(f"Song directory '{song_dir}' does not exist.")

        if not PurePath(song_dir).parent == PurePath(INTERMEDIATE_AUDIO_DIR):
            raise InvalidPathError(
                f"Song directory '{song_dir}' is not located in the root of the"
                " intermediate audio directory."
            )
        shutil.rmtree(song_dir)
    return "[+] Successfully deleted intermediate audio files for selected songs!"


def delete_all_intermediate_audio(
    progress_bar: gr.Progress | None = None, percentage: float = 0.0
) -> str:
    """
    Delete all intermediate audio files.

    Parameters
    ----------
    progress_bar : gr.Progress, optional
        Gradio progress bar to update.
    percentage : float, default=0.0

    Returns
    -------
    str
        Success message.
    """
    display_progress(
        "[~] Deleting all intermediate audio files...", percentage, progress_bar
    )
    if os.path.isdir(INTERMEDIATE_AUDIO_DIR):
        shutil.rmtree(INTERMEDIATE_AUDIO_DIR)

    return "[+] All intermediate audio files successfully deleted!"


def delete_output_audio(
    output_audio_files: list[str],
    progress_bar: gr.Progress | None = None,
    percentage: float = 0.0,
) -> str:
    """
    Delete selected output audio files.

    Parameters
    ----------
    output_audio_files : list[str]
        Paths of output audio files to delete.
    progress_bar : gr.Progress, optional
        Gradio progress bar to update.
    percentage : float, default=0.0
        Percentage to display in the progress bar.

    Returns
    -------
    str
        Success message.

    Raises
    ------
    InputMissingError
        If no output audio files are provided.
    PathNotFoundError
        If an output audio file does not exist.
    InvalidPathError
        If an output audio file is not located in the root of the output audio directory.
    """
    if not output_audio_files:
        raise InputMissingError(
            "Output audio files missing! Please provide a non-empty list of output"
            " audio files."
        )
    display_progress(
        "[~] Deleting selected output audio files...", percentage, progress_bar
    )
    for output_audio_file in output_audio_files:
        if not os.path.isfile(output_audio_file):
            raise PathNotFoundError(
                f"Output audio file '{output_audio_file}' does not exist."
            )
        if not PurePath(output_audio_file).parent == PurePath(OUTPUT_AUDIO_DIR):
            raise InvalidPathError(
                f"Output audio file '{output_audio_file}' is not located in the root of"
                " the output audio directory."
            )
        os.remove(output_audio_file)
    return "[+] Successfully deleted selected output audio files!"


def delete_all_output_audio(
    progress_bar: gr.Progress | None = None, percentage: float = 0.0
) -> str:
    """
    Delete all output audio files.

    Parameters
    ----------
    progress_bar : gr.Progress, optional
        Gradio progress bar to update.
    percentage : float, default=0.0
        Percentage to display in the progress bar.

    Returns
    -------
    str
        Success message.
    """
    display_progress("[~] Deleting all output audio files...", percentage, progress_bar)
    if os.path.isdir(OUTPUT_AUDIO_DIR):
        shutil.rmtree(OUTPUT_AUDIO_DIR)

    return "[+] All output audio files successfully deleted!"


def delete_all_audio(
    progress_bar: gr.Progress | None = None, percentage: float = 0.0
) -> str:
    """
    Delete all audio files.

    Parameters
    ----------
    progress_bar : gr.Progress, optional
        Gradio progress bar to update.
    percentage : float, default=0.0
        Percentage to display in the progress bar.

    Returns
    -------
    str
        Success message.
    """
    display_progress("[~] Deleting all audio files...", percentage, progress_bar)
    if os.path.isdir(INTERMEDIATE_AUDIO_DIR):
        shutil.rmtree(INTERMEDIATE_AUDIO_DIR)
    if os.path.isdir(OUTPUT_AUDIO_DIR):
        shutil.rmtree(OUTPUT_AUDIO_DIR)

    return "[+] All audio files successfully deleted!"


def delete_gradio_temp_dir() -> None:
    """
    Delete the directory where Gradio stores temporary files.
    """
    if os.path.isdir(GRADIO_TEMP_DIR):
        shutil.rmtree(GRADIO_TEMP_DIR)