mirror of
https://github.com/wabbajack-tools/wabbajack.git
synced 2024-08-30 18:42:17 +00:00
Add install CLI command
This commit is contained in:
parent
a589c2e0d4
commit
8cceacf78d
@ -72,6 +72,7 @@ internal class Program
|
||||
services.AddSingleton<IVerb, ListModlists>();
|
||||
services.AddSingleton<IVerb, Extract>();
|
||||
services.AddSingleton<IVerb, DumpZipInfo>();
|
||||
services.AddSingleton<IVerb, Install>();
|
||||
|
||||
services.AddSingleton<IUserInterventionHandler, UserInterventionHandler>();
|
||||
}).Build();
|
||||
|
107
Wabbajack.CLI/Verbs/Install.cs
Normal file
107
Wabbajack.CLI/Verbs/Install.cs
Normal file
@ -0,0 +1,107 @@
|
||||
using System;
|
||||
using System.CommandLine;
|
||||
using System.CommandLine.Invocation;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Wabbajack.Downloaders;
|
||||
using Wabbajack.Downloaders.GameFile;
|
||||
using Wabbajack.DTOs;
|
||||
using Wabbajack.DTOs.JsonConverters;
|
||||
using Wabbajack.Installer;
|
||||
using Wabbajack.Networking.WabbajackClientApi;
|
||||
using Wabbajack.Paths;
|
||||
using Wabbajack.Paths.IO;
|
||||
using Wabbajack.VFS;
|
||||
|
||||
namespace Wabbajack.CLI.Verbs;
|
||||
|
||||
public class Install : IVerb
|
||||
{
|
||||
private readonly ILogger<Install> _logger;
|
||||
private readonly Client _wjClient;
|
||||
private readonly DownloadDispatcher _dispatcher;
|
||||
private readonly IServiceProvider _serviceProvider;
|
||||
private readonly DTOSerializer _dtos;
|
||||
private readonly FileHashCache _cache;
|
||||
private readonly GameLocator _gameLocator;
|
||||
|
||||
public Install(ILogger<Install> logger, Client wjClient, DownloadDispatcher dispatcher, DTOSerializer dtos,
|
||||
FileHashCache cache, GameLocator gameLocator, IServiceProvider serviceProvider)
|
||||
{
|
||||
_logger = logger;
|
||||
_wjClient = wjClient;
|
||||
_dispatcher = dispatcher;
|
||||
_dtos = dtos;
|
||||
_serviceProvider = serviceProvider;
|
||||
_cache = cache;
|
||||
_gameLocator = gameLocator;
|
||||
}
|
||||
public Command MakeCommand()
|
||||
{
|
||||
var command = new Command("install");
|
||||
command.Add(new Option<AbsolutePath>(new[] {"-w", "-wabbajack"}, "Wabbajack file"));
|
||||
command.Add(new Option<AbsolutePath>(new[] {"-m", "-machineUrl"}, "Machine url to download"));
|
||||
command.Add(new Option<AbsolutePath>(new[] {"-o", "-output"}, "Output path"));
|
||||
command.Add(new Option<AbsolutePath>(new[] {"-d", "-downloads"}, "Downloads path"));
|
||||
command.Description = "Installs a wabbajack file";
|
||||
command.Handler = CommandHandler.Create(Run);
|
||||
return command;
|
||||
}
|
||||
public async Task<int> Run(AbsolutePath wabbajack, AbsolutePath output, AbsolutePath downloads, string machineUrl, CancellationToken token)
|
||||
{
|
||||
if (!string.IsNullOrEmpty(machineUrl))
|
||||
{
|
||||
if (!await DownloadMachineUrl(machineUrl, wabbajack, token))
|
||||
return 1;
|
||||
}
|
||||
|
||||
var modlist = await StandardInstaller.LoadFromFile(_dtos, wabbajack);
|
||||
|
||||
var installer = StandardInstaller.Create(_serviceProvider, new InstallerConfiguration
|
||||
{
|
||||
Downloads = downloads,
|
||||
Install = output,
|
||||
ModList = modlist,
|
||||
Game = modlist.GameType,
|
||||
ModlistArchive = wabbajack,
|
||||
GameFolder = _gameLocator.GameLocation(modlist.GameType)
|
||||
});
|
||||
|
||||
var result = await installer.Begin(token);
|
||||
|
||||
return result ? 0 : 2;
|
||||
}
|
||||
|
||||
private async Task<bool> DownloadMachineUrl(string machineUrl, AbsolutePath wabbajack, CancellationToken token)
|
||||
{
|
||||
_logger.LogInformation("Downloading {MachineUrl}", machineUrl);
|
||||
|
||||
var lists = await _wjClient.LoadLists();
|
||||
var list = lists.FirstOrDefault(l => l.NamespacedName == machineUrl);
|
||||
if (list == null)
|
||||
{
|
||||
_logger.LogInformation("Couldn't find list {MachineUrl}", machineUrl);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (wabbajack.FileExists() && await _cache.FileHashCachedAsync(wabbajack, token) == list.DownloadMetadata!.Hash)
|
||||
{
|
||||
_logger.LogInformation("File already exists, using cached file");
|
||||
return true;
|
||||
}
|
||||
|
||||
var state = _dispatcher.Parse(new Uri(list.Links.Download));
|
||||
|
||||
await _dispatcher.Download(new Archive
|
||||
{
|
||||
Name = wabbajack.FileName.ToString(),
|
||||
Hash = list.DownloadMetadata!.Hash,
|
||||
Size = list.DownloadMetadata.Size,
|
||||
State = state!
|
||||
}, wabbajack, token);
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user