Tranga/Tranga-CLI/Tranga_Cli.cs

using System.Globalization;
using Logging;
using Tranga;
using Tranga.LibraryManagers;
using Tranga.TrangaTasks;

namespace Tranga_CLI;

/*
 * This is written with pure hatred for readability.
 * At some point do this properly.
 * Read at own risk.
 */

public static class Tranga_Cli
{
    public static void Main(string[] args)
    {
        string applicationFolderPath =  Path.Join(Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData), "Tranga");
        string logsFolderPath = Path.Join(applicationFolderPath, "logs");
        string logFilePath = Path.Join(logsFolderPath, $"log-{DateTime.Now:dd-M-yyyy-HH-mm-ss}.txt");
        string settingsFilePath = Path.Join(applicationFolderPath, "settings.json");

        Directory.CreateDirectory(applicationFolderPath);
        Directory.CreateDirectory(logsFolderPath);
        
        Console.WriteLine($"Logfile-Path: {logFilePath}");
        Console.WriteLine($"Settings-File-Path: {settingsFilePath}");

        Logger logger = new(new[] { Logger.LoggerType.FileLogger }, null, Console.Out.Encoding, logFilePath);
        
        logger.WriteLine("Tranga_CLI", "Loading Taskmanager.");
        TrangaSettings settings = File.Exists(settingsFilePath) ? TrangaSettings.LoadSettings(settingsFilePath, logger) : new TrangaSettings(Directory.GetCurrentDirectory(), applicationFolderPath, new HashSet<LibraryManager>());

            
        logger.WriteLine("Tranga_CLI", "User Input");
        Console.WriteLine($"Output folder path [{settings.downloadLocation}]:");
        string? tmpPath = Console.ReadLine();
        while(tmpPath is null)
            tmpPath = Console.ReadLine();
        if (tmpPath.Length > 0)
            settings.downloadLocation = tmpPath;

        Console.WriteLine($"Komga BaseURL [{settings.libraryManagers.FirstOrDefault(lm => lm.GetType() == typeof(Komga))?.baseUrl}]:");
        string? tmpUrlKomga = Console.ReadLine();
        while (tmpUrlKomga is null)
            tmpUrlKomga = Console.ReadLine();
        if (tmpUrlKomga.Length > 0)
        {
            Console.WriteLine("Username:");
            string? tmpKomgaUser = Console.ReadLine();
            while (tmpKomgaUser is null || tmpKomgaUser.Length < 1)
                tmpKomgaUser = Console.ReadLine();
            
            Console.WriteLine("Password:");
            string tmpKomgaPass = string.Empty;
            ConsoleKey key;
            do
            {
                var keyInfo = Console.ReadKey(intercept: true);
                key = keyInfo.Key;

                if (key == ConsoleKey.Backspace && tmpKomgaPass.Length > 0)
                {
                    Console.Write("\b \b");
                    tmpKomgaPass = tmpKomgaPass[0..^1];
                }
                else if (!char.IsControl(keyInfo.KeyChar))
                {
                    Console.Write("*");
                    tmpKomgaPass += keyInfo.KeyChar;
                }
            } while (key != ConsoleKey.Enter);

            settings.libraryManagers.RemoveWhere(lm => lm.GetType() == typeof(Komga));
            settings.libraryManagers.Add(new Komga(tmpUrlKomga, tmpKomgaUser, tmpKomgaPass, logger));
        }
        
        Console.WriteLine($"Kavita BaseURL [{settings.libraryManagers.FirstOrDefault(lm => lm.GetType() == typeof(Kavita))?.baseUrl}]:");
        string? tmpUrlKavita = Console.ReadLine();
        while (tmpUrlKavita is null)
            tmpUrlKavita = Console.ReadLine();
        if (tmpUrlKavita.Length > 0)
        {
            Console.WriteLine("Username:");
            string? tmpKavitaUser = Console.ReadLine();
            while (tmpKavitaUser is null || tmpKavitaUser.Length < 1)
                tmpKavitaUser = Console.ReadLine();
            
            Console.WriteLine("Password:");
            string tmpKavitaPass = string.Empty;
            ConsoleKey key;
            do
            {
                var keyInfo = Console.ReadKey(intercept: true);
                key = keyInfo.Key;

                if (key == ConsoleKey.Backspace && tmpKavitaPass.Length > 0)
                {
                    Console.Write("\b \b");
                    tmpKavitaPass = tmpKavitaPass[0..^1];
                }
                else if (!char.IsControl(keyInfo.KeyChar))
                {
                    Console.Write("*");
                    tmpKavitaPass += keyInfo.KeyChar;
                }
            } while (key != ConsoleKey.Enter);

            settings.libraryManagers.RemoveWhere(lm => lm.GetType() == typeof(Kavita));
            settings.libraryManagers.Add(new Kavita(tmpUrlKavita, tmpKavitaUser, tmpKavitaPass, logger));
        }
        
        logger.WriteLine("Tranga_CLI", "Loaded.");
        TaskMode(settings, logger);
    }

    private static void TaskMode(TrangaSettings settings, Logger logger)
    {
        TaskManager taskManager = new (settings, logger);
        ConsoleKey selection = ConsoleKey.EraseEndOfFile;
        PrintMenu(taskManager, taskManager.settings.downloadLocation);
        while (selection != ConsoleKey.Q)
        {
            int taskCount = taskManager.GetAllTasks().Length;
            int taskRunningCount = taskManager.GetAllTasks().Count(task => task.state == TrangaTask.ExecutionState.Running);
            int taskEnqueuedCount =
                taskManager.GetAllTasks().Count(task => task.state == TrangaTask.ExecutionState.Enqueued);
            Console.SetCursorPosition(0,1);
            Console.WriteLine($"Tasks (Running/Queue/Total)): {taskRunningCount}/{taskEnqueuedCount}/{taskCount}");

            if (Console.KeyAvailable)
            {
                selection = Console.ReadKey().Key;
                switch (selection)
                {
                    case ConsoleKey.L:
                        PrintTasks(taskManager.GetAllTasks(), logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.C:
                        CreateTask(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.D:
                        DeleteTask(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.E:
                        ExecuteTaskNow(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.S:
                        SearchTasks(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.R:
                        PrintTasks(
                            taskManager.GetAllTasks().Where(eTask => eTask.state == TrangaTask.ExecutionState.Running)
                                .ToArray(), logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.K:
                        PrintTasks(
                            taskManager.GetAllTasks().Where(qTask => qTask.state is TrangaTask.ExecutionState.Enqueued)
                                .ToArray(), logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.F:
                        TailLog(logger);
                        Console.ReadKey();
                        break;
                    case ConsoleKey.G:
                        RemoveTaskFromQueue(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.B:
                        AddTaskToQueue(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break;
                    case ConsoleKey.M:
                        AddMangaTaskToQueue(taskManager, logger);
                        Console.WriteLine("Press any key.");
                        Console.ReadKey();
                        break; 
                }
                PrintMenu(taskManager, taskManager.settings.downloadLocation);
            }
            Thread.Sleep(200);
        }

        logger.WriteLine("Tranga_CLI", "Exiting.");
        Console.Clear();
        Console.WriteLine("Exiting.");
        if (taskManager.GetAllTasks().Any(task => task.state == TrangaTask.ExecutionState.Running))
        {
            Console.WriteLine("Force quit (Even with running tasks?) y/N");
            selection = Console.ReadKey().Key;
            while(selection != ConsoleKey.Y && selection != ConsoleKey.N)
                selection = Console.ReadKey().Key;
            taskManager.Shutdown(selection == ConsoleKey.Y);
        }else
            // ReSharper disable once RedundantArgumentDefaultValue Better readability
            taskManager.Shutdown(false);
    }

    private static void PrintMenu(TaskManager taskManager, string folderPath)
    {
        int taskCount = taskManager.GetAllTasks().Length;
        int taskRunningCount = taskManager.GetAllTasks().Count(task => task.state == TrangaTask.ExecutionState.Running);
        int taskEnqueuedCount =
            taskManager.GetAllTasks().Count(task => task.state == TrangaTask.ExecutionState.Enqueued);
        Console.Clear();
        Console.WriteLine($"Download Folder: {folderPath}");
        Console.WriteLine($"Tasks (Running/Queue/Total)): {taskRunningCount}/{taskEnqueuedCount}/{taskCount}");
        Console.WriteLine();
        Console.WriteLine($"{"C: Create Task",-30}{"L: List tasks",-30}{"B: Enqueue Task", -30}");
        Console.WriteLine($"{"D: Delete Task",-30}{"S: Search Tasks", -30}{"K: List Task Queue", -30}");
        Console.WriteLine($"{"E: Execute Task now",-30}{"R: List Running Tasks", -30}{"G: Remove Task from Queue", -30}");
        Console.WriteLine($"{"M: New Download Manga Task",-30}{"", -30}{"", -30}");
        Console.WriteLine($"{"",-30}{"F: Show Log",-30}{"Q: Exit",-30}");
    }

    private static void PrintTasks(TrangaTask[] tasks, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Printing Tasks");
        int taskCount = tasks.Length;
        int taskRunningCount = tasks.Count(task => task.state == TrangaTask.ExecutionState.Running);
        int taskEnqueuedCount = tasks.Count(task => task.state == TrangaTask.ExecutionState.Enqueued);
        Console.Clear();
        int tIndex = 0;
        Console.WriteLine($"Tasks (Running/Queue/Total): {taskRunningCount}/{taskEnqueuedCount}/{taskCount}");
        string header =
            $"{"",-5}{"Task",-20} | {"Last Executed",-20} | {"Reoccurrence",-12} | {"State",-10} | {"Progress",-9} | {"Connector",-15} | Publication/Manga ";
        Console.WriteLine(header);
        Console.WriteLine(new string('-', header.Length));
        foreach (TrangaTask trangaTask in tasks)
        {
            string[] taskSplit = trangaTask.ToString().Split(", ");
            Console.WriteLine($"{tIndex++:000}: {taskSplit[0],-20} | {taskSplit[1],-20} | {taskSplit[2],-12} | {taskSplit[3],-10} | {(taskSplit.Length > 4 ? taskSplit[4] : ""),-9} | {(taskSplit.Length > 5 ? taskSplit[5] : ""),-15} | {(taskSplit.Length > 6 ? taskSplit[6] : "")} {(taskSplit.Length > 7 ? taskSplit[7] : "")} {(taskSplit.Length > 8 ? taskSplit[8] : "")}");
        }
            
    }

    private static TrangaTask[] SelectTasks(TrangaTask[] tasks, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Select task");
        if (tasks.Length < 1)
        {
            Console.Clear();
            Console.WriteLine("There are no available Tasks.");
            logger.WriteLine("Tranga_CLI", "No available Tasks.");
            return Array.Empty<TrangaTask>();
        }
        PrintTasks(tasks, logger);
        
        logger.WriteLine("Tranga_CLI", "Selecting Task to Remove (from queue)");
        Console.WriteLine("Enter q to abort");
        Console.WriteLine($"Select Task(s) (0-{tasks.Length - 1}):");

        string? selectedTask = Console.ReadLine();
        while(selectedTask is null || selectedTask.Length < 1)
            selectedTask = Console.ReadLine();
        
        if (selectedTask.Length == 1 && selectedTask.ToLower() == "q")
        {
            Console.Clear();
            Console.WriteLine("aborted.");
            logger.WriteLine("Tranga_CLI", "aborted");
            return Array.Empty<TrangaTask>();
        }

        if (selectedTask.Contains('-'))
        {
            int start = Convert.ToInt32(selectedTask.Split('-')[0]);
            int end = Convert.ToInt32(selectedTask.Split('-')[1]);
            return tasks[start..end];
        }
        else
        {
            int selectedTaskIndex = Convert.ToInt32(selectedTask);
            return new[] { tasks[selectedTaskIndex] };
        }
    }
    
    private static void AddMangaTaskToQueue(TaskManager taskManager, Logger logger)
    {
        Console.Clear();
        logger.WriteLine("Tranga_CLI", "Menu: Add Manga Download to queue");
        
        Connector? connector = SelectConnector(taskManager.GetAvailableConnectors().Values.ToArray(), logger);
        if (connector is null)
            return;
                    
        Publication? publication = SelectPublication(taskManager, connector, logger);
        if (publication is null)
            return;
        
        TimeSpan reoccurrence = SelectReoccurrence(logger);
        logger.WriteLine("Tranga_CLI", "Sending Task to TaskManager");
        TrangaTask? newTask = taskManager.AddTask(TrangaTask.Task.DownloadNewChapters, connector.name, publication.Value.publicationId, reoccurrence, "en");
        Console.WriteLine(newTask);
    }

    private static void AddTaskToQueue(TaskManager taskManager, Logger logger)
    {
        Console.Clear();
        logger.WriteLine("Tranga_CLI", "Menu: Add Task to queue");

        TrangaTask[] tasks = taskManager.GetAllTasks().Where(rTask =>
            rTask.state is not TrangaTask.ExecutionState.Enqueued and not TrangaTask.ExecutionState.Running).ToArray();
        
        TrangaTask[] selectedTasks = SelectTasks(tasks, logger);
        logger.WriteLine("Tranga_CLI", $"Sending {selectedTasks.Length} Tasks to TaskManager");
        foreach(TrangaTask task in selectedTasks)
            taskManager.AddTaskToQueue(task);
    }

    private static void RemoveTaskFromQueue(TaskManager taskManager, Logger logger)
    {
        Console.Clear();
        logger.WriteLine("Tranga_CLI", "Menu: Remove Task from queue");
        
        TrangaTask[] tasks = taskManager.GetAllTasks().Where(rTask => rTask.state is TrangaTask.ExecutionState.Enqueued).ToArray();

        TrangaTask[] selectedTasks = SelectTasks(tasks, logger);
        logger.WriteLine("Tranga_CLI", $"Sending {selectedTasks.Length} Tasks to TaskManager");
        foreach(TrangaTask task in selectedTasks)
            taskManager.RemoveTaskFromQueue(task);
    }

    private static void TailLog(Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Show Log-lines");
        Console.Clear();
        
        string[] lines = logger.Tail(20);
        foreach (string message in lines)
            Console.Write(message);

        while (!Console.KeyAvailable)
        {
            string[] newLines = logger.GetNewLines();
            foreach(string message in newLines)
                Console.Write(message);
            Thread.Sleep(40);
        }
    }

    private static void CreateTask(TaskManager taskManager, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Creating Task");
        TrangaTask.Task? tmpTask = SelectTaskType(logger);
        if (tmpTask is null)
            return;
        TrangaTask.Task task = (TrangaTask.Task)tmpTask;
        
        Connector? connector = null;
        if (task != TrangaTask.Task.UpdateLibraries)
        {
            connector = SelectConnector(taskManager.GetAvailableConnectors().Values.ToArray(), logger);
            if (connector is null)
                return;
        }
                    
        Publication? publication = null;
        if (task != TrangaTask.Task.UpdateLibraries)
        {
            publication = SelectPublication(taskManager, connector!, logger);
            if (publication is null)
                return;
        }

        if (task is TrangaTask.Task.DownloadNewChapters)
        {
            TimeSpan reoccurrence = SelectReoccurrence(logger);
            logger.WriteLine("Tranga_CLI", "Sending Task to TaskManager");

            TrangaTask newTask = new DownloadNewChaptersTask(TrangaTask.Task.DownloadNewChapters, connector!.name, (Publication)publication!, reoccurrence, "en");
            taskManager.AddTask(newTask);
            Console.WriteLine(newTask);
        }else if (task is TrangaTask.Task.DownloadChapter)
        {
            foreach (Chapter chapter in SelectChapters(connector!, (Publication)publication!, logger))
            {
                TrangaTask newTask = new DownloadChapterTask(TrangaTask.Task.DownloadChapter, connector!.name,
                    (Publication)publication!, chapter, "en");
                taskManager.AddTask(newTask);
                Console.WriteLine(newTask);
            }
        }
    }

    private static void ExecuteTaskNow(TaskManager taskManager, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Executing Task");
        TrangaTask[] tasks = taskManager.GetAllTasks().Where(nTask => nTask.state is not TrangaTask.ExecutionState.Running).ToArray();
        
        TrangaTask[] selectedTasks = SelectTasks(tasks, logger);
        logger.WriteLine("Tranga_CLI", $"Sending {selectedTasks.Length} Tasks to TaskManager");
        foreach(TrangaTask task in selectedTasks)
            taskManager.ExecuteTaskNow(task);
    }

    private static void DeleteTask(TaskManager taskManager, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Delete Task");
        TrangaTask[] tasks = taskManager.GetAllTasks();
        
        TrangaTask[] selectedTasks = SelectTasks(tasks, logger);
        logger.WriteLine("Tranga_CLI", $"Sending {selectedTasks.Length} Tasks to TaskManager");
        foreach(TrangaTask task in selectedTasks)
            taskManager.DeleteTask(task);
    }

    private static TrangaTask.Task? SelectTaskType(Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Select TaskType");
        Console.Clear();
        string[] taskNames = Enum.GetNames<TrangaTask.Task>();
        
        int tIndex = 0;
        Console.WriteLine("Available Tasks:");
        foreach (string taskName in taskNames)
            Console.WriteLine($"{tIndex++}: {taskName}");
        
        Console.WriteLine("Enter q to abort");
        Console.WriteLine($"Select Task (0-{taskNames.Length - 1}):");

        string? selectedTask = Console.ReadLine();
        while(selectedTask is null || selectedTask.Length < 1)
            selectedTask = Console.ReadLine();

        if (selectedTask.Length == 1 && selectedTask.ToLower() == "q")
        {
            Console.Clear();
            Console.WriteLine("aborted.");
            logger.WriteLine("Tranga_CLI", "aborted.");
            return null;
        }
        
        try
        {
            int selectedTaskIndex = Convert.ToInt32(selectedTask);
            string selectedTaskName = taskNames[selectedTaskIndex];
            return Enum.Parse<TrangaTask.Task>(selectedTaskName);
        }
        catch (Exception e)
        {
            Console.WriteLine($"Exception: {e.Message}");
            logger.WriteLine("Tranga_CLI", e.Message);
        }

        return null;
    }

    private static TimeSpan SelectReoccurrence(Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Select Reoccurrence");
        Console.WriteLine("Select reoccurrence Timer (Format hh:mm:ss):");
        return TimeSpan.Parse(Console.ReadLine()!, new CultureInfo("en-US"));
    }

    private static Chapter[] SelectChapters(Connector connector, Publication publication, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Select Chapters");
        Chapter[] availableChapters = connector.GetChapters(publication, "en");
        int cIndex = 0;
        Console.WriteLine("Chapters:");
        foreach(Chapter chapter in availableChapters)
            Console.WriteLine($"{cIndex++}: Vol.{chapter.volumeNumber} Ch.{chapter.chapterNumber} - {chapter.name}");
        
        Console.WriteLine("Enter q to abort");
        Console.WriteLine($"Select Chapter(s):");

        string? selectedChapters = Console.ReadLine();
        while(selectedChapters is null || selectedChapters.Length < 1)
            selectedChapters = Console.ReadLine();

        if (selectedChapters.Length == 1 && selectedChapters.ToLower() == "q")
        {
            Console.Clear();
            Console.WriteLine("aborted.");
            logger.WriteLine("Tranga_CLI", "aborted.");
            return Array.Empty<Chapter>();
        }
        
        if (selectedChapters.Contains('-'))
        {
            int start = Convert.ToInt32(selectedChapters.Split('-')[0]);
            int end = Convert.ToInt32(selectedChapters.Split('-')[1]);
            return availableChapters[start..end];
        }
        else
            return new Chapter[] { availableChapters[Convert.ToInt32(selectedChapters)] };
    }

    private static Connector? SelectConnector(Connector[] connectors, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Select Connector");
        Console.Clear();
        
        int cIndex = 0;
        Console.WriteLine("Connectors:");
        foreach (Connector connector in connectors)
            Console.WriteLine($"{cIndex++}: {connector.name}");
        
        Console.WriteLine("Enter q to abort");
        Console.WriteLine($"Select Connector (0-{connectors.Length - 1}):");

        string? selectedConnector = Console.ReadLine();
        while(selectedConnector is null || selectedConnector.Length < 1)
            selectedConnector = Console.ReadLine();

        if (selectedConnector.Length == 1 && selectedConnector.ToLower() == "q")
        {
            Console.Clear();
            Console.WriteLine("aborted.");
            logger.WriteLine("Tranga_CLI", "aborted.");
            return null;
        }
        
        try
        {
            int selectedConnectorIndex = Convert.ToInt32(selectedConnector);
            return connectors[selectedConnectorIndex];
        }
        catch (Exception e)
        {
            Console.WriteLine($"Exception: {e.Message}");
            logger.WriteLine("Tranga_CLI", e.Message);
        }

        return null;
    }

    private static Publication? SelectPublication(TaskManager taskManager, Connector connector, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Select Publication");
        
        Console.Clear();
        Console.WriteLine($"Connector: {connector.name}");
        Console.WriteLine("Publication search query (leave empty for all):");
        string? query = Console.ReadLine();

        Publication[] publications = taskManager.GetPublicationsFromConnector(connector, query ?? "");

        if (publications.Length < 1)
        {
            logger.WriteLine("Tranga_CLI", "No publications returned");
            Console.WriteLine($"No publications for query '{query}' returned;");
            return null;
        }
        
        int pIndex = 0;
        Console.WriteLine("Publications:");
        foreach(Publication publication in publications)
            Console.WriteLine($"{pIndex++}: {publication.sortName}");
        
        Console.WriteLine("Enter q to abort");
        Console.WriteLine($"Select publication to Download (0-{publications.Length - 1}):");

        string? selectedPublication = Console.ReadLine();
        while(selectedPublication is null || selectedPublication.Length < 1)
            selectedPublication = Console.ReadLine();

        if (selectedPublication.Length == 1 && selectedPublication.ToLower() == "q")
        {
            Console.Clear();
            Console.WriteLine("aborted.");
            logger.WriteLine("Tranga_CLI", "aborted.");
            return null;
        }
        
        try
        {
            int selectedPublicationIndex = Convert.ToInt32(selectedPublication);
            return publications[selectedPublicationIndex];
        }
        catch (Exception e)
        {
            Console.WriteLine($"Exception: {e.Message}");
            logger.WriteLine("Tranga_CLI", e.Message);
        }

        return null;
    }

    private static void SearchTasks(TaskManager taskManager, Logger logger)
    {
        logger.WriteLine("Tranga_CLI", "Menu: Search task");
        Console.Clear();
        Console.WriteLine("Enter search query:");
        string? query = Console.ReadLine();
        while (query is null || query.Length < 4)
            query = Console.ReadLine();
        PrintTasks(taskManager.GetAllTasks().Where(qTask =>
            qTask.ToString().ToLower().Contains(query, StringComparison.OrdinalIgnoreCase)).ToArray(), logger);
    }
}