wabbajack/Wabbajack.CLI/Program.cs

91 lines
3.2 KiB
C#
Raw Normal View History

2021-09-27 12:42:46 +00:00
using System;
using System.CommandLine;
using System.CommandLine.IO;
using System.Net.Http;
using System.Text.Json;
using System.Threading.Tasks;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
2022-05-29 04:19:25 +00:00
using Microsoft.Extensions.Logging;
using NLog.Extensions.Logging;
using NLog.Targets;
2021-09-27 12:42:46 +00:00
using Octokit;
using Wabbajack.DTOs.Interventions;
2021-09-27 12:42:46 +00:00
using Wabbajack.Networking.Http;
using Wabbajack.Networking.Http.Interfaces;
using Wabbajack.Paths.IO;
using Wabbajack.Server.Lib;
using Wabbajack.Services.OSIntegrated;
using Wabbajack.VFS;
using Client = Wabbajack.Networking.GitHub.Client;
using Wabbajack.CLI.Builder;
2020-01-31 22:38:56 +00:00
2021-10-23 16:51:17 +00:00
namespace Wabbajack.CLI;
internal class Program
2020-01-31 22:38:56 +00:00
{
2021-10-23 16:51:17 +00:00
private static async Task<int> Main(string[] args)
2020-01-31 22:38:56 +00:00
{
2021-10-23 16:51:17 +00:00
var host = Host.CreateDefaultBuilder(Array.Empty<string>())
2022-05-29 04:19:25 +00:00
.ConfigureLogging(AddLogging)
2021-10-23 16:51:17 +00:00
.ConfigureServices((host, services) =>
{
services.AddSingleton(new JsonSerializerOptions());
services.AddSingleton<HttpClient, HttpClient>();
services.AddSingleton<IHttpDownloader, SingleThreadedDownloader>();
services.AddSingleton<IConsole, SystemConsole>();
services.AddSingleton<CommandLineBuilder, CommandLineBuilder>();
services.AddSingleton<TemporaryFileManager>();
services.AddSingleton<FileExtractor.FileExtractor>();
services.AddSingleton(new ParallelOptions {MaxDegreeOfParallelism = Environment.ProcessorCount});
services.AddSingleton<Client>();
services.AddSingleton<Networking.WabbajackClientApi.Client>();
services.AddSingleton(s => new GitHubClient(new ProductHeaderValue("wabbajack")));
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
services.AddOSIntegrated();
services.AddServerLib();
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
services.AddTransient<Context>();
services.AddSingleton<CommandLineBuilder>();
services.AddCLIVerbs();
services.AddSingleton<IUserInterventionHandler, UserInterventionHandler>();
2021-10-23 16:51:17 +00:00
}).Build();
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
var service = host.Services.GetService<CommandLineBuilder>();
return await service!.Run(args);
2020-01-31 22:38:56 +00:00
}
2022-05-29 04:19:25 +00:00
private static void AddLogging(ILoggingBuilder loggingBuilder)
{
var config = new NLog.Config.LoggingConfiguration();
var fileTarget = new FileTarget("file")
{
FileName = "logs/wabbajack-cli.current.log",
ArchiveFileName = "logs/wabbajack-cli.{##}.log",
ArchiveOldFileOnStartup = true,
MaxArchiveFiles = 10,
Layout = "${processtime} [${level:uppercase=true}] (${logger}) ${message:withexception=true}",
Header = "############ Wabbajack log file - ${longdate} ############"
};
var consoleTarget = new ConsoleTarget("console")
{
Layout = "${processtime} [${level:uppercase=true}] ${message:withexception=true}",
};
config.AddRuleForAllLevels(fileTarget);
config.AddRuleForAllLevels(consoleTarget);
loggingBuilder.ClearProviders();
loggingBuilder.SetMinimumLevel(LogLevel.Trace);
loggingBuilder.AddNLog(config);
}
2021-09-27 12:42:46 +00:00
}