"""
Copyright © 2022 Noah Vogt <noah@noahvogt.com>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

import os

from utils import log, create_min_obs_subdirs, error_msg, expand_dir
from slides import ClassicSongTemplate, ClassicStartSlide, ClassicSongSlide

import config as const

import slidegen


def slide_selection_iterator(ssync):
    iterator_prompt = "Exit now? [y/N]: "
    structure_prompt = (
        "Choose song structure (leave blank for full song)"
        + " eg. [1,R,2,R] / [1-4]: "
    )
    file_list_str = ""
    rclone_local_dir = expand_dir(const.RCLONE_LOCAL_DIR)
    obs_slides_dir = expand_dir(const.OBS_SLIDES_DIR)
    try:
        for file in os.listdir(rclone_local_dir):
            file_list_str += file + "\n"
    except (FileNotFoundError, PermissionError, IOError) as error:
        error_msg(
            "Failed to access items in '{}'. Reason: {}".format(
                rclone_local_dir, error
            )
        )
    file_list_str = file_list_str[:-1]
    tempfile_str = ".chosen-tempfile"

    index = 0
    while True:
        index += 1
        input_song_prompt = "[{}{}] ".format(const.OBS_SUBDIR_NAMING, index)
        prompt_answer = str(input(input_song_prompt + iterator_prompt))
        if prompt_answer.lower() == "y":
            create_min_obs_subdirs()
            break

        file_list_str = file_list_str.replace("\n", "\\n")
        os.system('printf "{}" | fzf > {}'.format(file_list_str, tempfile_str))

        with open(
            tempfile_str, encoding="utf-8", mode="r"
        ) as tempfile_file_opener:
            chosen_song_file = tempfile_file_opener.read()[:-1].strip()

        if len(chosen_song_file) == 0:
            log("no slides chosen, skipping...")
        else:
            structure_prompt_answer = input(
                input_song_prompt + structure_prompt
            )

            log(
                "generating slides '{}' to '{}{}'...".format(
                    chosen_song_file, const.OBS_SUBDIR_NAMING, index
                )
            )
            src_dir = os.path.join(rclone_local_dir, chosen_song_file)
            dest_dir = os.path.join(
                obs_slides_dir,
                const.OBS_SUBDIR_NAMING + str(index),
            )
            os.mkdir(dest_dir)

            slidegen_instance = slidegen.Slidegen(
                ClassicSongTemplate,
                ClassicStartSlide,
                ClassicSongSlide,
                src_dir,
                dest_dir,
                structure_prompt_answer,
            )
            slidegen_instance.execute(ssync.disable_async)

    if os.path.isfile(tempfile_str):
        os.remove(tempfile_str)