using System.Net; using System.Text.Json.Serialization; using Logging; using Newtonsoft.Json; using Newtonsoft.Json.Linq; using Tranga.TrangaTasks; using JsonConverter = Newtonsoft.Json.JsonConverter; namespace Tranga; /// /// Stores information on Task, when implementing new Tasks also update the serializer /// [JsonDerivedType(typeof(MonitorPublicationTask), 2)] [JsonDerivedType(typeof(UpdateLibrariesTask), 3)] [JsonDerivedType(typeof(DownloadChapterTask), 4)] public abstract class TrangaTask { // ReSharper disable once CommentTypo ...Tell me why! // ReSharper disable once MemberCanBePrivate.Global I want it thaaat way public TimeSpan reoccurrence { get; } public DateTime lastExecuted { get; set; } [Newtonsoft.Json.JsonIgnore] public ExecutionState state { get; set; } public Task task { get; } public string taskId { get; init; } [Newtonsoft.Json.JsonIgnore] public TrangaTask? parentTask { get; set; } public string? parentTaskId { get; set; } [Newtonsoft.Json.JsonIgnore] internal HashSet childTasks { get; } public double progress => GetProgress(); [Newtonsoft.Json.JsonIgnore]public DateTime executionStarted { get; private set; } [Newtonsoft.Json.JsonIgnore]public DateTime lastChange { get; internal set; } [Newtonsoft.Json.JsonIgnore]public DateTime executionApproximatelyFinished => lastChange.Add(GetRemainingTime()); public TimeSpan executionApproximatelyRemaining => executionApproximatelyFinished.Subtract(DateTime.Now); [Newtonsoft.Json.JsonIgnore]public DateTime nextExecution => lastExecuted.Add(reoccurrence); public enum ExecutionState { Waiting, Enqueued, Running, Failed, Success } protected TrangaTask(Task task, TimeSpan reoccurrence, TrangaTask? parentTask = null) { this.reoccurrence = reoccurrence; this.lastExecuted = DateTime.Now.Subtract(reoccurrence); this.task = task; this.executionStarted = DateTime.UnixEpoch; this.lastChange = DateTime.MaxValue; this.taskId = Convert.ToBase64String(BitConverter.GetBytes(new Random().Next())); this.childTasks = new(); this.parentTask = parentTask; this.parentTaskId = parentTask?.taskId; } /// /// BL for concrete Tasks /// /// /// /// protected abstract HttpStatusCode ExecuteTask(TaskManager taskManager, Logger? logger, CancellationToken? cancellationToken = null); public abstract TrangaTask Clone(); protected abstract double GetProgress(); /// /// Execute the task /// /// Should be the parent taskManager /// /// public void Execute(TaskManager taskManager, Logger? logger, CancellationToken? cancellationToken = null) { logger?.WriteLine(this.GetType().ToString(), $"Executing Task {this}"); this.state = ExecutionState.Running; this.executionStarted = DateTime.Now; this.lastChange = DateTime.Now; if(parentTask is not null && parentTask.childTasks.All(ct => ct.state is ExecutionState.Waiting or ExecutionState.Failed)) parentTask.executionStarted = DateTime.Now; HttpStatusCode statusCode = ExecuteTask(taskManager, logger, cancellationToken); while(childTasks.Any(ct => ct.state is ExecutionState.Enqueued or ExecutionState.Running)) Thread.Sleep(1000); if((int)statusCode >= 200 && (int)statusCode < 300 && parentTask is null) foreach(NotificationManager nm in taskManager.settings.notificationManagers) switch (this.task) { case Task.MonitorPublication: MonitorPublicationTask mpt = (MonitorPublicationTask)this; int successfulCount = this.childTasks.Count(ct => ct.state is ExecutionState.Success); if(successfulCount > 0) nm.SendNotification("Downloaded new chapters", $"{mpt.publication.sortName}: {successfulCount} new chapters."); break; case Task.DownloadChapter: DownloadChapterTask dct = (DownloadChapterTask)this; nm.SendNotification("Chapter downloaded", $"{dct.publication.sortName} {dct.chapter.chapterNumber} {dct.chapter.name}"); break; } if ((int)statusCode >= 200 && (int)statusCode < 300) { this.lastExecuted = DateTime.Now; this.state = ExecutionState.Success; }else if (statusCode is HttpStatusCode.NotFound) { this.state = ExecutionState.Waiting; this.lastExecuted = DateTime.MaxValue; } else { this.state = ExecutionState.Failed; this.lastExecuted = DateTime.MaxValue; } foreach (TrangaTask childTask in this.childTasks.Where(ct => ct is DownloadChapterTask).ToArray()) taskManager.DeleteTask(childTask); logger?.WriteLine(this.GetType().ToString(), $"Finished Executing Task {this}"); } public void AddChildTask(TrangaTask childTask) { this.childTasks.Add(childTask); } public void RemoveChildTask(TrangaTask childTask) { this.childTasks.Remove(childTask); } private TimeSpan GetRemainingTime() { if(progress == 0 || state is ExecutionState.Enqueued or ExecutionState.Waiting or ExecutionState.Failed || lastChange == DateTime.MaxValue) return DateTime.MaxValue.Subtract(lastChange).Subtract(TimeSpan.FromHours(1)); TimeSpan elapsed = lastChange.Subtract(executionStarted); return elapsed.Divide(progress).Multiply(1 - progress); } public enum Task : byte { MonitorPublication = 2, UpdateLibraries = 3, DownloadChapter = 4, } public override string ToString() { return $"{task}, {lastExecuted}, {reoccurrence}, {state}, {progress:P2}, {executionApproximatelyFinished}, {executionApproximatelyRemaining}"; } public class TrangaTaskJsonConverter : JsonConverter { public override bool CanConvert(Type objectType) { return objectType == typeof(TrangaTask); } public override object ReadJson(JsonReader reader, Type objectType, object? existingValue, JsonSerializer serializer) { JObject jo = JObject.Load(reader); if (jo["task"]!.Value() == (Int64)Task.MonitorPublication) return jo.ToObject(serializer)!; if (jo["task"]!.Value() == (Int64)Task.UpdateLibraries) return jo.ToObject(serializer)!; if (jo["task"]!.Value() == (Int64)Task.DownloadChapter) return jo.ToObject(serializer)!; throw new Exception(); } public override bool CanWrite => false; /// /// Don't call this /// public override void WriteJson(JsonWriter writer, object? value, JsonSerializer serializer) { throw new Exception("Dont call this"); } } }