using System; using System.CommandLine; using System.CommandLine.Invocation; using System.Diagnostics; using System.IO; using System.Runtime.ExceptionServices; using System.Runtime.InteropServices; using System.Security; using System.Threading; using System.Threading.Tasks; using System.Windows.Threading; using BililiveRecorder.ToolBox; using Sentry; using Serilog; using Serilog.Core; using Serilog.Exceptions; using Serilog.Formatting.Compact; #nullable enable namespace BililiveRecorder.WPF { internal static class Program { private const int CODE__WPF = 0x5F_57_50_46; internal static readonly LoggingLevelSwitch levelSwitchGlobal; internal static readonly LoggingLevelSwitch levelSwitchConsole; internal static readonly Logger logger; internal static readonly Update update; internal static Task? updateTask; static Program() { AttachConsole(-1); levelSwitchGlobal = new LoggingLevelSwitch(Serilog.Events.LogEventLevel.Debug); if (Debugger.IsAttached) levelSwitchGlobal.MinimumLevel = Serilog.Events.LogEventLevel.Verbose; levelSwitchConsole = new LoggingLevelSwitch(Serilog.Events.LogEventLevel.Error); logger = BuildLogger(); AppDomain.CurrentDomain.UnhandledException += CurrentDomain_UnhandledException; TaskScheduler.UnobservedTaskException += TaskScheduler_UnobservedTaskException; Log.Logger = logger; SentrySdk.ConfigureScope(s => { s.SetTag("fullsemver", GitVersionInformation.FullSemVer); }); _ = SentrySdk.ConfigureScopeAsync(async s => { var path = Path.GetFullPath(Path.Combine(Path.GetDirectoryName(typeof(Program).Assembly.Location), "..", "packages", ".betaId")); for (var i = 0; i < 10; i++) { if (i != 0) await Task.Delay(TimeSpan.FromSeconds(5)); try { if (!File.Exists(path)) continue; var content = File.ReadAllText(path); if (Guid.TryParse(content, out var id)) { s.User.Id = id.ToString(); return; } } catch (Exception) { } } }); update = new Update(logger); } [STAThread] public static int Main(string[] args) { try { logger.Debug("Starting, Version: {Version}, CurrentDirectory: {CurrentDirectory}, CommandLine: {CommandLine}", GitVersionInformation.InformationalVersion, Environment.CurrentDirectory, Environment.CommandLine); var code = BuildCommand().Invoke(args); logger.Debug("Exit code: {ExitCode}, RunWpf: {RunWpf}", code, code == CODE__WPF); return code == CODE__WPF ? Commands.RunWpfReal() : code; } finally { logger.Dispose(); } } private static RootCommand BuildCommand() { var run = new Command("run", "Run BililiveRecorder at path") { new Argument("path","Work directory") }; run.Handler = CommandHandler.Create((string path) => Commands.RunWpfHandler(path, false)); var root = new RootCommand("") { run, new Option("--squirrel-firstrun") { IsHidden = true }, new ToolCommand(), }; root.Handler = CommandHandler.Create((bool squirrelFirstrun) => Commands.RunWpfHandler(null, squirrelFirstrun)); return root; } private static class Commands { internal static int RunWpfHandler(string? path, bool squirrelFirstrun) { Pages.RootPage.CommandArgumentRecorderPath = path; Pages.RootPage.CommandArgumentFirstRun = squirrelFirstrun; return CODE__WPF; } internal static int RunWpfReal() { var cancel = new CancellationTokenSource(); var token = cancel.Token; try { var app = new App(); app.InitializeComponent(); app.DispatcherUnhandledException += App_DispatcherUnhandledException; updateTask = Task.Run(async () => { while (!token.IsCancellationRequested) { await update.UpdateAsync().ConfigureAwait(false); await Task.Delay(TimeSpan.FromDays(1), token).ConfigureAwait(false); } }); return app.Run(); } finally { cancel.Cancel(); #pragma warning disable VSTHRD002 // Avoid problematic synchronous waits update.WaitForUpdatesOnShutdownAsync().GetAwaiter().GetResult(); #pragma warning restore VSTHRD002 // Avoid problematic synchronous waits } } } private static Logger BuildLogger() => new LoggerConfiguration() .MinimumLevel.ControlledBy(levelSwitchGlobal) .Enrich.WithProcessId() .Enrich.WithThreadId() .Enrich.WithThreadName() .Enrich.FromLogContext() .Enrich.WithExceptionDetails() .Destructure.ByTransforming(x => new { x.Version, x.ExportTime, x.FileSize, x.FileCreationTime, x.FileModificationTime, }) .WriteTo.Console(levelSwitch: levelSwitchConsole) #if DEBUG .WriteTo.Debug() .WriteTo.Sink(Serilog.Events.LogEventLevel.Debug) #else .WriteTo.Sink(Serilog.Events.LogEventLevel.Information) #endif .WriteTo.File(new CompactJsonFormatter(), "./logs/bilirec.txt", shared: true, rollingInterval: RollingInterval.Day, rollOnFileSizeLimit: true) .WriteTo.Sentry(o => { o.Dsn = "https://ac2e03c2f25249968853611a97fcd6ee@o210546.ingest.sentry.io/5556540"; o.SendDefaultPii = true; o.DisableAppDomainUnhandledExceptionCapture(); o.DisableTaskUnobservedTaskExceptionCapture(); o.AddExceptionFilterForType(); o.MinimumBreadcrumbLevel = Serilog.Events.LogEventLevel.Debug; o.MinimumEventLevel = Serilog.Events.LogEventLevel.Error; #if DEBUG o.Environment = "debug-build"; #else o.Environment = "release-build"; #endif }) .CreateLogger(); [DllImport("kernel32")] private static extern bool AttachConsole(int pid); [HandleProcessCorruptedStateExceptions, SecurityCritical] private static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs e) { if (e.ExceptionObject is Exception ex) logger.Fatal(ex, "Unhandled exception from AppDomain.UnhandledException"); } [HandleProcessCorruptedStateExceptions, SecurityCritical] private static void TaskScheduler_UnobservedTaskException(object sender, UnobservedTaskExceptionEventArgs e) => logger.Error(e.Exception, "Unobserved exception from TaskScheduler.UnobservedTaskException"); [HandleProcessCorruptedStateExceptions, SecurityCritical] private static void App_DispatcherUnhandledException(object sender, DispatcherUnhandledExceptionEventArgs e) => logger.Fatal(e.Exception, "Unhandled exception from Application.DispatcherUnhandledException"); } }