using Newtonsoft.Json; using Tranga.Connectors; namespace Tranga; /// /// Manages all TrangaTasks. /// Provides a Threaded environment to execute Tasks, and still manage the Task-Collection /// public class TaskManager { private readonly Dictionary> _chapterCollection; private readonly HashSet _allTasks; private bool _continueRunning = true; private readonly Connector[] _connectors; private string downloadLocation { get; } public Komga? komga { get; private set; } /// Local path to save data (Manga) to /// The Url of the Komga-instance that you want to update public TaskManager(string folderPath, string? komgaBaseUrl = null, string? komgaUsername = null, string? komgaPassword = null) { this.downloadLocation = folderPath; if (komgaBaseUrl != null && komgaUsername != null && komgaPassword != null) this.komga = new Komga(komgaBaseUrl, komgaUsername, komgaPassword); this._connectors = new Connector[]{ new MangaDex(folderPath) }; _chapterCollection = new(); _allTasks = new HashSet(); Thread taskChecker = new(TaskCheckerThread); taskChecker.Start(); } public TaskManager(SettingsData settings) { this._connectors = new Connector[]{ new MangaDex(settings.downloadLocation) }; _chapterCollection = new(); this.downloadLocation = settings.downloadLocation; this.komga = settings.komga; _allTasks = settings.allTasks; Thread taskChecker = new(TaskCheckerThread); taskChecker.Start(); } private void TaskCheckerThread() { while (_continueRunning) { foreach (TrangaTask task in _allTasks) { if(task.ShouldExecute()) TaskExecutor.Execute(this, task, this._chapterCollection); //TODO Might crash here, when adding new Task while another Task is running. Check later } Thread.Sleep(1000); } } /// /// Forces the execution of a given task /// /// Task to execute public void ExecuteTaskNow(TrangaTask task) { if (!this._allTasks.Contains(task)) return; Task t = new Task(() => { TaskExecutor.Execute(this, task, this._chapterCollection); }); t.Start(); } /// /// Creates and adds a new Task to the task-Collection /// /// TrangaTask.Task to later execute /// Name of the connector to use /// Publication to execute Task on, can be null in case of unrelated Task /// Time-Interval between Executions /// language, should Task require parameter. Can be empty /// Is thrown when connectorName is not a available Connector public TrangaTask AddTask(TrangaTask.Task task, string? connectorName, Publication? publication, TimeSpan reoccurrence, string language = "") { if (task != TrangaTask.Task.UpdateKomgaLibrary && connectorName is null) throw new ArgumentException($"connectorName can not be null for task {task}"); TrangaTask newTask; if (task == TrangaTask.Task.UpdateKomgaLibrary) { newTask = new TrangaTask(task, null, null, reoccurrence, language); //Check if same task already exists // ReSharper disable once SimplifyLinqExpressionUseAll readabilty if (!_allTasks.Any(trangaTask => trangaTask.task == task)) { _allTasks.Add(newTask); } } else { //Get appropriate Connector from available Connectors for TrangaTask Connector? connector = _connectors.FirstOrDefault(c => c.name == connectorName); if (connector is null) throw new ArgumentException($"Connector {connectorName} is not a known connector."); newTask = new TrangaTask(task, connector.name, publication, reoccurrence, language); //Check if same task already exists if (!_allTasks.Any(trangaTask => trangaTask.task == task && trangaTask.connectorName == connector.name && trangaTask.publication?.downloadUrl == publication?.downloadUrl)) { if(task != TrangaTask.Task.UpdatePublications) _chapterCollection.Add((Publication)publication!, new List()); _allTasks.Add(newTask); } } ExportData(Directory.GetCurrentDirectory()); return newTask; } /// /// Removes Task from task-collection /// /// TrangaTask.Task type /// Name of Connector that was used /// Publication that was used public void RemoveTask(TrangaTask.Task task, string connectorName, Publication? publication) { _allTasks.RemoveWhere(trangaTask => trangaTask.task == task && trangaTask.connectorName == connectorName && trangaTask.publication?.downloadUrl == publication?.downloadUrl); ExportData(Directory.GetCurrentDirectory()); } /// All available Connectors public Dictionary GetAvailableConnectors() { return this._connectors.ToDictionary(connector => connector.name, connector => connector); } /// All TrangaTasks in task-collection public TrangaTask[] GetAllTasks() { TrangaTask[] ret = new TrangaTask[_allTasks.Count]; _allTasks.CopyTo(ret); return ret; } /// All added Publications public Publication[] GetAllPublications() { return this._chapterCollection.Keys.ToArray(); } public void NewKomga(Komga? pKomga) { this.komga = pKomga; this.ExportData(Directory.GetCurrentDirectory()); } /// /// Shuts down the taskManager. /// /// If force is true, tasks are aborted. public void Shutdown(bool force = false) { _continueRunning = false; ExportData(Directory.GetCurrentDirectory()); if(force) Environment.Exit(_allTasks.Count(task => task.isBeingExecuted)); //Wait for tasks to finish while(_allTasks.Any(task => task.isBeingExecuted)) Thread.Sleep(10); Environment.Exit(0); } public static SettingsData ImportData(string importFolderPath) { string importPath = Path.Join(importFolderPath, "data.json"); if (!File.Exists(importPath)) return new SettingsData("", null, new HashSet()); string toRead = File.ReadAllText(importPath); SettingsData data = JsonConvert.DeserializeObject(toRead)!; return data; } private void ExportData(string exportFolderPath) { SettingsData data = new SettingsData(this.downloadLocation, this.komga, this._allTasks); string exportPath = Path.Join(exportFolderPath, "data.json"); string serializedData = JsonConvert.SerializeObject(data); File.WriteAllText(exportPath, serializedData); } public class SettingsData { public string downloadLocation { get; set; } public Komga? komga { get; set; } public HashSet allTasks { get; } public SettingsData(string downloadLocation, Komga? komga, HashSet allTasks) { this.downloadLocation = downloadLocation; this.komga = komga; this.allTasks = allTasks; } } }