using System; using System.Reactive.Concurrency; using System.Reactive.Disposables; using System.Runtime.InteropServices; using System.Security.Principal; using System.Threading.Tasks; using System.Windows; using System.Windows.Threading; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Hosting; using Microsoft.Extensions.Logging; using NLog.Extensions.Logging; using NLog.Targets; using Orc.FileAssociation; using ReactiveUI; using Wabbajack.CLI.Builder; using Wabbajack.DTOs; using Wabbajack.DTOs.Interventions; using Wabbajack.Interventions; using Wabbajack.LoginManagers; using Wabbajack.Models; using Wabbajack.Paths; using Wabbajack.Paths.IO; using Wabbajack.Services.OSIntegrated; using Wabbajack.UserIntervention; using Wabbajack.Util; using Ext = Wabbajack.Common.Ext; namespace Wabbajack { /// /// Interaction logic for App.xaml /// public partial class App { private IHost _host; private void OnStartup(object sender, StartupEventArgs e) { if (IsAdmin()) { var messageBox = MessageBox.Show("Don't run Wabbajack as Admin!", "Error", MessageBoxButton.OK, MessageBoxImage.Error, MessageBoxResult.OK, MessageBoxOptions.DefaultDesktopOnly); if (messageBox == MessageBoxResult.OK) { Environment.Exit(1); } else { Environment.Exit(1); } } RxApp.MainThreadScheduler = new DispatcherScheduler(Dispatcher.CurrentDispatcher); _host = Host.CreateDefaultBuilder(Array.Empty()) .ConfigureLogging(AddLogging) .ConfigureServices((host, services) => { ConfigureServices(services); }) .Build(); var args = e.Args; RxApp.MainThreadScheduler.Schedule(0, (_, _) => { if (args.Length == 1) { var arg = args[0].ToAbsolutePath(); if (arg.FileExists() && arg.Extension == Ext.Wabbajack) { var mainWindow = _host.Services.GetRequiredService(); mainWindow!.Show(); return Disposable.Empty; } } else if (args.Length > 0) { var builder = _host.Services.GetRequiredService(); builder.Run(e.Args).ContinueWith(async x => { Environment.Exit(await x); }); return Disposable.Empty; } else { var mainWindow = _host.Services.GetRequiredService(); mainWindow!.Show(); return Disposable.Empty; } return Disposable.Empty; }); } private static bool IsAdmin() { if (!RuntimeInformation.IsOSPlatform(OSPlatform.Windows)) return false; try { var identity = WindowsIdentity.GetCurrent(); var owner = identity.Owner; if (owner is not null) return owner.IsWellKnown(WellKnownSidType.BuiltinAdministratorsSid); var principle = new WindowsPrincipal(identity); return principle.IsInRole(WindowsBuiltInRole.Administrator); } catch (Exception) { return false; } } private void AddLogging(ILoggingBuilder loggingBuilder) { var config = new NLog.Config.LoggingConfiguration(); var logFolder = KnownFolders.LauncherAwarePath.Combine("logs"); if (!logFolder.DirectoryExists()) logFolder.CreateDirectory(); var fileTarget = new FileTarget("file") { FileName = logFolder.Combine("Wabbajack.current.log").ToString(), ArchiveFileName = logFolder.Combine("Wabbajack.{##}.log").ToString(), ArchiveOldFileOnStartup = true, MaxArchiveFiles = 10, Layout = "${processtime} [${level:uppercase=true}] (${logger}) ${message:withexception=true}", Header = "############ Wabbajack log file - ${longdate} ############" }; var consoleTarget = new ConsoleTarget("console"); var uiTarget = new LogStream { Name = "ui", Layout = "${message:withexception=false}", }; loggingBuilder.Services.AddSingleton(uiTarget); config.AddRuleForAllLevels(fileTarget); config.AddRuleForAllLevels(consoleTarget); config.AddRuleForAllLevels(uiTarget); loggingBuilder.ClearProviders(); loggingBuilder.SetMinimumLevel(LogLevel.Information); loggingBuilder.AddNLog(config); } private static IServiceCollection ConfigureServices(IServiceCollection services) { services.AddOSIntegrated(); // Orc.FileAssociation services.AddSingleton(new ApplicationRegistrationService()); services.AddSingleton(); services.AddSingleton(); services.AddTransient(); services.AddTransient(); services.AddTransient(); services.AddSingleton(); services.AddSingleton(); services.AddSingleton(); services.AddSingleton(); services.AddTransient(); services.AddTransient(); services.AddTransient(); services.AddTransient(); services.AddTransient(); services.AddTransient(); services.AddTransient(); services.AddTransient(); // Login Handlers services.AddTransient(); services.AddTransient(); services.AddTransient(); // Login Managers services.AddAllSingleton(); services.AddAllSingleton(); services.AddAllSingleton(); services.AddSingleton(); services.AddSingleton(); // Verbs services.AddSingleton(); services.AddCLIVerbs(); return services; } private void OnExit(object sender, ExitEventArgs e) { using (_host) { _host.StopAsync(); } base.OnExit(e); } } }