f29e9e2b8a
# Conflicts: # Discord Media Loader.Application/Classes/Core.cs # Discord Media Loader.Application/Classes/JobScheduler.cs # Discord Media Loader.Application/Classes/Settings.cs
359 lines
14 KiB
C#
359 lines
14 KiB
C#
using Discord;
|
|
using Discord.Net;
|
|
using Discord.WebSocket;
|
|
using DML.AppCore.Classes;
|
|
using DML.Application.Classes.RPC;
|
|
using DML.Application.Dialogs;
|
|
using DML.Client;
|
|
using LiteDB;
|
|
using SharpRaven;
|
|
using SharpRaven.Data;
|
|
using SweetLib.Utils;
|
|
using SweetLib.Utils.Logger;
|
|
using SweetLib.Utils.Logger.Memory;
|
|
using System;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Runtime;
|
|
using System.Threading.Tasks;
|
|
using System.Windows.Forms;
|
|
using Logger = SweetLib.Utils.Logger.Logger;
|
|
|
|
namespace DML.Application.Classes
|
|
{
|
|
public static class Core
|
|
{
|
|
//internal static DiscordSocketClient Client { get; set; }
|
|
internal static LiteDatabase Database { get; set; }
|
|
internal static Settings Settings { get; set; }
|
|
internal static JobScheduler Scheduler { get; } = new JobScheduler();
|
|
internal static RavenClient Raven { get; } = new RavenClient("https://0de964231669473e9098b9f6cc1d6278:79d9f2eb24034de199b2a37cc058e0f2@sentry.io/257114");
|
|
internal static bool ShuttingDown { get; private set; } = false;
|
|
internal static string DataDirectory
|
|
=> Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), @"Serraniel\Discord Media Loader");
|
|
|
|
internal static DiscordRpc.RichPresence RpcPresence;
|
|
internal static DiscordRpc.EventHandlers RpcHandlers = new DiscordRpc.EventHandlers();
|
|
|
|
public static async Task Run(string[] paramStrings)
|
|
{
|
|
try
|
|
{
|
|
var splash = new FrmInternalSplash();
|
|
splash.Show();
|
|
System.Windows.Forms.Application.DoEvents();
|
|
|
|
Logger.Info("Starting up Discord Media Loader application...");
|
|
var useTrace = false;
|
|
#if DEBUG
|
|
//temporary add debug log level if debugging...
|
|
Logger.GlobalLogLevel |= LogLevel.Debug;
|
|
Logger.Debug("Running in debug configuartion. Added log level debug.");
|
|
#endif
|
|
|
|
Logger.Debug($"Parameters: {string.Join(", ", paramStrings)}");
|
|
if (paramStrings.Contains("--trace") || paramStrings.Contains("-t"))
|
|
{
|
|
useTrace = true;
|
|
Logger.GlobalLogLevel |= LogLevel.Trace;
|
|
Logger.Trace("Trace parameter found. Added log level trace.");
|
|
}
|
|
|
|
Logger.Debug($"Application data folder: {DataDirectory}");
|
|
|
|
Logger.Trace("Checking application data folder...");
|
|
if (!Directory.Exists(DataDirectory))
|
|
{
|
|
Logger.Debug("Creating application data folder...");
|
|
Directory.CreateDirectory(DataDirectory);
|
|
Logger.Trace("Creating application data folder.");
|
|
}
|
|
|
|
Logger.Trace("Initializing profile optimizations...");
|
|
ProfileOptimization.SetProfileRoot(System.Windows.Forms.Application.UserAppDataPath);
|
|
ProfileOptimization.StartProfile("profile.opt");
|
|
Logger.Trace("Finished initializing profile optimizations.");
|
|
|
|
Logger.Trace("Trying to identify log memory...");
|
|
var logMemory = Logger.DefaultLogMemory as ArchivableConsoleLogMemory;
|
|
if (logMemory != null)
|
|
{
|
|
var logFolder = Path.Combine(DataDirectory, "logs");
|
|
if (!Directory.Exists(logFolder))
|
|
{
|
|
Logger.Debug("Creating log folder...");
|
|
Directory.CreateDirectory(logFolder);
|
|
Logger.Trace("Created log folder.");
|
|
}
|
|
|
|
var logFile = Path.Combine(logFolder,
|
|
SweetUtils.LegalizeFilename($"{DateTime.Now.ToString(CultureInfo.CurrentCulture.DateTimeFormat.SortableDateTimePattern)}.log.zip"));
|
|
|
|
Logger.Trace($"Setting log file: {logFile}");
|
|
logMemory.AutoArchiveOnDispose = true;
|
|
logMemory.ArchiveFile = logFile;
|
|
}
|
|
|
|
var databasePath = Path.Combine(DataDirectory, "config.db");
|
|
#if DEBUG
|
|
databasePath = Path.Combine(DataDirectory, "config.debug.db");
|
|
#endif
|
|
Logger.Debug("Loading database...");
|
|
Database = new LiteDatabase(databasePath);
|
|
Database.Log.Logging += (message) => Logger.Trace($"LiteDB: {message}");
|
|
#endif
|
|
|
|
Logger.Debug("Loading settings collection out of database...");
|
|
var settingsDB = Database.GetCollection<Settings>("settings");
|
|
if (settingsDB.Count() > 1)
|
|
{
|
|
Logger.Warn("Found more than one setting. Loading first one...");
|
|
}
|
|
Settings = settingsDB.FindAll().FirstOrDefault();
|
|
if (Settings == null)
|
|
{
|
|
Logger.Warn("Settings not found. Creating new one. This is normal on first start up...");
|
|
Settings = new Settings();
|
|
Settings.Store();
|
|
}
|
|
|
|
if (Settings.ShowStartUpHints)
|
|
{
|
|
if (MessageBox.Show(splash, "This tool is considered as a selfbot which may violate the Discord TOS. By using this tool you take the risk to get your account banned. Although this never happened yet (as far as I know) you have to confirm to this.\n\r\n\rDo you wish to continue?", "HOLD UP!!", MessageBoxButtons.YesNo) != DialogResult.Yes)
|
|
{
|
|
splash.Close();
|
|
return;
|
|
}
|
|
|
|
Settings.ShowStartUpHints = false;
|
|
Settings.Store();
|
|
}
|
|
|
|
Logger.Debug("Loading jobs collection out of database...");
|
|
Scheduler.JobList = Job.RestoreJobs().ToList();
|
|
|
|
Logger.Info("Loaded settings.");
|
|
Logger.Debug(
|
|
$"Settings: Email: {Settings.Email}, password: {(string.IsNullOrEmpty(Settings.Password) ? "not set" : "is set")}, use username: {Settings.UseUserData}, loginToken: {Settings.LoginToken}");
|
|
|
|
Logger.Trace("Updating log level...");
|
|
Logger.GlobalLogLevel = Settings.ApplicactionLogLevel;
|
|
#if DEBUG
|
|
//temporary add debug log level if debugging...
|
|
Logger.GlobalLogLevel |= LogLevel.Debug;
|
|
Logger.Debug("Running in debug configuartion. Added log level debug.");
|
|
#endif
|
|
if (useTrace)
|
|
{
|
|
Logger.GlobalLogLevel |= LogLevel.Trace;
|
|
Logger.Trace("Creating application data folder.");
|
|
}
|
|
|
|
Logger.Debug("Creating discord client...");
|
|
|
|
var config = new DiscordSocketConfig()
|
|
{
|
|
DefaultRetryMode = RetryMode.AlwaysRetry,
|
|
};
|
|
|
|
//Client = new DiscordSocketClient(config);
|
|
DMLClient.Client.Log += (arg) =>
|
|
{
|
|
var logMessage = $"DiscordClient: {arg.Message}";
|
|
switch (arg.Severity)
|
|
{
|
|
case LogSeverity.Verbose:
|
|
Logger.Trace(logMessage);
|
|
break;
|
|
case LogSeverity.Debug:
|
|
Logger.Trace(logMessage);
|
|
break;
|
|
case LogSeverity.Info:
|
|
Logger.Info(logMessage);
|
|
break;
|
|
case LogSeverity.Warning:
|
|
Logger.Warn(logMessage);
|
|
break;
|
|
case LogSeverity.Error:
|
|
Logger.Error(logMessage);
|
|
break;
|
|
}
|
|
|
|
return Task.CompletedTask;
|
|
};
|
|
|
|
|
|
Logger.Info("Trying to log into discord...");
|
|
|
|
DMLClient.Client.Connected += Client_Connected;
|
|
|
|
var loggedIn = false;
|
|
|
|
while (!loggedIn)
|
|
{
|
|
try
|
|
{
|
|
if (!string.IsNullOrEmpty(Settings.LoginToken))
|
|
{
|
|
Logger.Debug("Trying to login with last known token...");
|
|
loggedIn = await DMLClient.Login(Settings.LoginToken);
|
|
}
|
|
|
|
}
|
|
catch (HttpException)
|
|
{
|
|
Logger.Warn("HTTPException occured during login. Probably from login token.");
|
|
}
|
|
|
|
if (!loggedIn)
|
|
{
|
|
Logger.Debug("Showing dialog for username and password...");
|
|
var loginDlg = new LoginDialog();
|
|
if (loginDlg.ShowDialog() != DialogResult.OK)
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
Logger.Debug("Start checking for invalid jobs...");
|
|
|
|
//Client
|
|
|
|
while (DMLClient.Client.Guilds.Count == 0)
|
|
{
|
|
// wait until guilds are loaded
|
|
}
|
|
|
|
for (var i = Scheduler.JobList.Count - 1; i >= 0; i--)
|
|
{
|
|
var job = Scheduler.JobList[i];
|
|
var isError = false;
|
|
var guild = FindServerById(job.GuildId);
|
|
if (guild == null)
|
|
{
|
|
isError = true;
|
|
}
|
|
else
|
|
{
|
|
var channel = FindChannelById(guild, job.ChannelId);
|
|
if (channel == null)
|
|
{
|
|
isError = true;
|
|
}
|
|
}
|
|
|
|
if (isError)
|
|
{
|
|
MessageBox.Show($"Invalid job for guild {job.GuildId}, channel {job.ChannelId} found. Guild or channel may not exist any more. This job will be deleted...", "Invalid job",
|
|
MessageBoxButtons.OK, MessageBoxIcon.Warning);
|
|
|
|
Scheduler.JobList.Remove(job);
|
|
Scheduler.RunningJobs.Remove(job.Id);
|
|
job.Stop();
|
|
job.Delete();
|
|
}
|
|
|
|
if (Settings.RescanRequired)
|
|
{
|
|
Logger.Info("Restting timestamps");
|
|
job.KnownTimestamp = 0;
|
|
job.Store();
|
|
}
|
|
}
|
|
|
|
Settings.RescanRequired = false;
|
|
Settings.Store();
|
|
|
|
if (Settings.UseRPC)
|
|
{
|
|
Logger.Info("Initializing RPC client");
|
|
Logger.Trace("Registering RPC handlers");
|
|
RpcHandlers.readyCallback += RpcReadyCallback;
|
|
RpcHandlers.disconnectedCallback += RpcDisconnectedCallback;
|
|
RpcHandlers.errorCallback += RpcErrorCallback;
|
|
RpcPresence.startTimestamp = DiscordRpcHelper.DateTimeToTimestamp(DateTime.UtcNow);
|
|
|
|
Logger.Debug("Calling RPC initialize");
|
|
DiscordRpc.Initialize("430025401851707393", ref RpcHandlers, true, null);
|
|
|
|
// Do not await ;)
|
|
Task.Run(async () =>
|
|
{
|
|
while (!ShuttingDown)
|
|
{
|
|
Logger.Trace("Running RPC callbacks");
|
|
await Task.Delay(5000);
|
|
}
|
|
});
|
|
}
|
|
|
|
splash.Close();
|
|
|
|
Logger.Info("Starting scheduler...");
|
|
Scheduler.ScanAll();
|
|
|
|
System.Windows.Forms.Application.Run(new MainForm());
|
|
|
|
// shutting down
|
|
ShuttingDown = true;
|
|
|
|
Logger.Info("Stopping scheduler...");
|
|
Scheduler.Stop();
|
|
|
|
Logger.Info("Shutting down RPC client");
|
|
DiscordRpc.Shutdown();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.Error($"{ex.Message} [{ex.GetType().Name}] occured at: {ex.StackTrace}");
|
|
if (MessageBox.Show($"An error occured while running Discord Media Loader:\n{ex.GetType().Name}: {ex.Message}\n\nDo you aggree to sending the error report to the creator of the tool?", "Discord Media Loader", MessageBoxButtons.YesNo) == DialogResult.Yes)
|
|
{
|
|
Raven.Capture(new SentryEvent(ex));
|
|
}
|
|
}
|
|
}
|
|
|
|
private static Task Client_Connected()
|
|
{
|
|
Logger.Debug("Connected");
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
//TODO: this is thrid time we implement this.....this has to be fixed!!!
|
|
private static SocketGuild FindServerById(ulong id)
|
|
{
|
|
Logger.Trace($"Trying to find server by Id: {id}");
|
|
return (from s in DMLClient.Client.Guilds where s.Id == id select s).FirstOrDefault();
|
|
}
|
|
|
|
private static SocketTextChannel FindChannelById(SocketGuild server, ulong id)
|
|
{
|
|
Logger.Trace($"Trying to find channel in {server} by id: {id}");
|
|
return (from c in server.TextChannels where c.Id == id select c).FirstOrDefault();
|
|
}
|
|
|
|
internal static void RpcUpdatePresence()
|
|
{
|
|
Logger.Debug("Updating RPC presence");
|
|
DiscordRpc.UpdatePresence(ref RpcPresence);
|
|
}
|
|
|
|
private static void RpcReadyCallback()
|
|
{
|
|
Logger.Debug("RpcReady");
|
|
}
|
|
|
|
private static void RpcDisconnectedCallback(int errorCode, string message)
|
|
{
|
|
Logger.Warn($"RPC disconnect received: {errorCode} - {message}");
|
|
}
|
|
|
|
private static void RpcErrorCallback(int errorCode, string message)
|
|
{
|
|
Logger.Error($"RPC error received: {errorCode} - {message}");
|
|
}
|
|
}
|
|
}
|