mirror of
https://github.com/GreemDev/Ryujinx
synced 2024-12-04 23:42:10 +01:00
340 lines
12 KiB
C#
340 lines
12 KiB
C#
|
using Avalonia.Collections;
|
||
|
using Avalonia.Controls;
|
||
|
using Avalonia.Controls.ApplicationLifetimes;
|
||
|
using Avalonia.Threading;
|
||
|
using DynamicData;
|
||
|
using LibHac.Common;
|
||
|
using LibHac.Fs;
|
||
|
using LibHac.Fs.Fsa;
|
||
|
using LibHac.FsSystem;
|
||
|
using LibHac.Tools.Fs;
|
||
|
using LibHac.Tools.FsSystem;
|
||
|
using LibHac.Tools.FsSystem.NcaUtils;
|
||
|
using Ryujinx.Ava.Common.Locale;
|
||
|
using Ryujinx.Ava.UI.Helpers;
|
||
|
using Ryujinx.Ava.UI.Models;
|
||
|
using Ryujinx.Common.Configuration;
|
||
|
using Ryujinx.Common.Logging;
|
||
|
using Ryujinx.Common.Utilities;
|
||
|
using Ryujinx.HLE.FileSystem;
|
||
|
using System;
|
||
|
using System.Collections.Generic;
|
||
|
using System.IO;
|
||
|
using System.Linq;
|
||
|
using System.Text;
|
||
|
using System.Threading.Tasks;
|
||
|
using Path = System.IO.Path;
|
||
|
|
||
|
namespace Ryujinx.Ava.UI.ViewModels
|
||
|
{
|
||
|
public class DownloadableContentManagerViewModel : BaseModel
|
||
|
{
|
||
|
private readonly List<DownloadableContentContainer> _downloadableContentContainerList;
|
||
|
private readonly string _downloadableContentJsonPath;
|
||
|
|
||
|
private VirtualFileSystem _virtualFileSystem;
|
||
|
private AvaloniaList<DownloadableContentModel> _downloadableContents = new();
|
||
|
private AvaloniaList<DownloadableContentModel> _views = new();
|
||
|
private AvaloniaList<DownloadableContentModel> _selectedDownloadableContents = new();
|
||
|
|
||
|
private string _search;
|
||
|
private ulong _titleId;
|
||
|
private string _titleName;
|
||
|
|
||
|
public AvaloniaList<DownloadableContentModel> DownloadableContents
|
||
|
{
|
||
|
get => _downloadableContents;
|
||
|
set
|
||
|
{
|
||
|
_downloadableContents = value;
|
||
|
OnPropertyChanged();
|
||
|
OnPropertyChanged(nameof(UpdateCount));
|
||
|
Sort();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public AvaloniaList<DownloadableContentModel> Views
|
||
|
{
|
||
|
get => _views;
|
||
|
set
|
||
|
{
|
||
|
_views = value;
|
||
|
OnPropertyChanged();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public AvaloniaList<DownloadableContentModel> SelectedDownloadableContents
|
||
|
{
|
||
|
get => _selectedDownloadableContents;
|
||
|
set
|
||
|
{
|
||
|
_selectedDownloadableContents = value;
|
||
|
OnPropertyChanged();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public string Search
|
||
|
{
|
||
|
get => _search;
|
||
|
set
|
||
|
{
|
||
|
_search = value;
|
||
|
OnPropertyChanged();
|
||
|
Sort();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public string UpdateCount
|
||
|
{
|
||
|
get => string.Format(LocaleManager.Instance[LocaleKeys.DlcWindowHeading], DownloadableContents.Count);
|
||
|
}
|
||
|
|
||
|
public DownloadableContentManagerViewModel(VirtualFileSystem virtualFileSystem, ulong titleId, string titleName)
|
||
|
{
|
||
|
_virtualFileSystem = virtualFileSystem;
|
||
|
|
||
|
_titleId = titleId;
|
||
|
_titleName = titleName;
|
||
|
|
||
|
_downloadableContentJsonPath = Path.Combine(AppDataManager.GamesDirPath, titleId.ToString("x16"), "dlc.json");
|
||
|
|
||
|
try
|
||
|
{
|
||
|
_downloadableContentContainerList = JsonHelper.DeserializeFromFile<List<DownloadableContentContainer>>(_downloadableContentJsonPath);
|
||
|
}
|
||
|
catch
|
||
|
{
|
||
|
Logger.Error?.Print(LogClass.Configuration, "Downloadable Content JSON failed to deserialize.");
|
||
|
_downloadableContentContainerList = new List<DownloadableContentContainer>();
|
||
|
}
|
||
|
|
||
|
LoadDownloadableContents();
|
||
|
}
|
||
|
|
||
|
private void LoadDownloadableContents()
|
||
|
{
|
||
|
foreach (DownloadableContentContainer downloadableContentContainer in _downloadableContentContainerList)
|
||
|
{
|
||
|
if (File.Exists(downloadableContentContainer.ContainerPath))
|
||
|
{
|
||
|
using FileStream containerFile = File.OpenRead(downloadableContentContainer.ContainerPath);
|
||
|
|
||
|
PartitionFileSystem partitionFileSystem = new(containerFile.AsStorage());
|
||
|
|
||
|
_virtualFileSystem.ImportTickets(partitionFileSystem);
|
||
|
|
||
|
foreach (DownloadableContentNca downloadableContentNca in downloadableContentContainer.DownloadableContentNcaList)
|
||
|
{
|
||
|
using UniqueRef<IFile> ncaFile = new();
|
||
|
|
||
|
partitionFileSystem.OpenFile(ref ncaFile.Ref, downloadableContentNca.FullPath.ToU8Span(), OpenMode.Read).ThrowIfFailure();
|
||
|
|
||
|
Nca nca = TryOpenNca(ncaFile.Get.AsStorage(), downloadableContentContainer.ContainerPath);
|
||
|
if (nca != null)
|
||
|
{
|
||
|
var content = new DownloadableContentModel(nca.Header.TitleId.ToString("X16"),
|
||
|
downloadableContentContainer.ContainerPath,
|
||
|
downloadableContentNca.FullPath,
|
||
|
downloadableContentNca.Enabled);
|
||
|
|
||
|
DownloadableContents.Add(content);
|
||
|
|
||
|
if (content.Enabled)
|
||
|
{
|
||
|
SelectedDownloadableContents.Add(content);
|
||
|
}
|
||
|
|
||
|
OnPropertyChanged(nameof(UpdateCount));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// NOTE: Save the list again to remove leftovers.
|
||
|
Save();
|
||
|
Sort();
|
||
|
}
|
||
|
|
||
|
public void Sort()
|
||
|
{
|
||
|
DownloadableContents.AsObservableChangeSet()
|
||
|
.Filter(Filter)
|
||
|
.Bind(out var view).AsObservableList();
|
||
|
|
||
|
_views.Clear();
|
||
|
_views.AddRange(view);
|
||
|
OnPropertyChanged(nameof(Views));
|
||
|
}
|
||
|
|
||
|
private bool Filter(object arg)
|
||
|
{
|
||
|
if (arg is DownloadableContentModel content)
|
||
|
{
|
||
|
return string.IsNullOrWhiteSpace(_search) || content.FileName.ToLower().Contains(_search.ToLower()) || content.TitleId.ToLower().Contains(_search.ToLower());
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
private Nca TryOpenNca(IStorage ncaStorage, string containerPath)
|
||
|
{
|
||
|
try
|
||
|
{
|
||
|
return new Nca(_virtualFileSystem.KeySet, ncaStorage);
|
||
|
}
|
||
|
catch (Exception ex)
|
||
|
{
|
||
|
Dispatcher.UIThread.InvokeAsync(async () =>
|
||
|
{
|
||
|
await ContentDialogHelper.CreateErrorDialog(string.Format(LocaleManager.Instance[LocaleKeys.DialogLoadNcaErrorMessage], ex.Message, containerPath));
|
||
|
});
|
||
|
}
|
||
|
|
||
|
return null;
|
||
|
}
|
||
|
|
||
|
public async void Add()
|
||
|
{
|
||
|
OpenFileDialog dialog = new OpenFileDialog()
|
||
|
{
|
||
|
Title = LocaleManager.Instance[LocaleKeys.SelectDlcDialogTitle],
|
||
|
AllowMultiple = true
|
||
|
};
|
||
|
|
||
|
dialog.Filters.Add(new FileDialogFilter
|
||
|
{
|
||
|
Name = "NSP",
|
||
|
Extensions = { "nsp" }
|
||
|
});
|
||
|
|
||
|
if (Avalonia.Application.Current.ApplicationLifetime is IClassicDesktopStyleApplicationLifetime desktop)
|
||
|
{
|
||
|
string[] files = await dialog.ShowAsync(desktop.MainWindow);
|
||
|
|
||
|
if (files != null)
|
||
|
{
|
||
|
foreach (string file in files)
|
||
|
{
|
||
|
await AddDownloadableContent(file);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private async Task AddDownloadableContent(string path)
|
||
|
{
|
||
|
if (!File.Exists(path) || DownloadableContents.FirstOrDefault(x => x.ContainerPath == path) != null)
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
using FileStream containerFile = File.OpenRead(path);
|
||
|
|
||
|
PartitionFileSystem partitionFileSystem = new(containerFile.AsStorage());
|
||
|
bool containsDownloadableContent = false;
|
||
|
|
||
|
_virtualFileSystem.ImportTickets(partitionFileSystem);
|
||
|
|
||
|
foreach (DirectoryEntryEx fileEntry in partitionFileSystem.EnumerateEntries("/", "*.nca"))
|
||
|
{
|
||
|
using var ncaFile = new UniqueRef<IFile>();
|
||
|
|
||
|
partitionFileSystem.OpenFile(ref ncaFile.Ref, fileEntry.FullPath.ToU8Span(), OpenMode.Read).ThrowIfFailure();
|
||
|
|
||
|
Nca nca = TryOpenNca(ncaFile.Get.AsStorage(), path);
|
||
|
if (nca == null)
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
if (nca.Header.ContentType == NcaContentType.PublicData)
|
||
|
{
|
||
|
if ((nca.Header.TitleId & 0xFFFFFFFFFFFFE000) != _titleId)
|
||
|
{
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
var content = new DownloadableContentModel(nca.Header.TitleId.ToString("X16"), path, fileEntry.FullPath, true);
|
||
|
DownloadableContents.Add(content);
|
||
|
SelectedDownloadableContents.Add(content);
|
||
|
|
||
|
OnPropertyChanged(nameof(UpdateCount));
|
||
|
Sort();
|
||
|
|
||
|
containsDownloadableContent = true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!containsDownloadableContent)
|
||
|
{
|
||
|
await ContentDialogHelper.CreateErrorDialog(LocaleManager.Instance[LocaleKeys.DialogDlcNoDlcErrorMessage]);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public void Remove(DownloadableContentModel model)
|
||
|
{
|
||
|
DownloadableContents.Remove(model);
|
||
|
OnPropertyChanged(nameof(UpdateCount));
|
||
|
Sort();
|
||
|
}
|
||
|
|
||
|
public void RemoveAll()
|
||
|
{
|
||
|
DownloadableContents.Clear();
|
||
|
OnPropertyChanged(nameof(UpdateCount));
|
||
|
Sort();
|
||
|
}
|
||
|
|
||
|
public void EnableAll()
|
||
|
{
|
||
|
SelectedDownloadableContents = new(DownloadableContents);
|
||
|
}
|
||
|
|
||
|
public void DisableAll()
|
||
|
{
|
||
|
SelectedDownloadableContents.Clear();
|
||
|
}
|
||
|
|
||
|
public void Save()
|
||
|
{
|
||
|
_downloadableContentContainerList.Clear();
|
||
|
|
||
|
DownloadableContentContainer container = default;
|
||
|
|
||
|
foreach (DownloadableContentModel downloadableContent in DownloadableContents)
|
||
|
{
|
||
|
if (container.ContainerPath != downloadableContent.ContainerPath)
|
||
|
{
|
||
|
if (!string.IsNullOrWhiteSpace(container.ContainerPath))
|
||
|
{
|
||
|
_downloadableContentContainerList.Add(container);
|
||
|
}
|
||
|
|
||
|
container = new DownloadableContentContainer
|
||
|
{
|
||
|
ContainerPath = downloadableContent.ContainerPath,
|
||
|
DownloadableContentNcaList = new List<DownloadableContentNca>()
|
||
|
};
|
||
|
}
|
||
|
|
||
|
container.DownloadableContentNcaList.Add(new DownloadableContentNca
|
||
|
{
|
||
|
Enabled = downloadableContent.Enabled,
|
||
|
TitleId = Convert.ToUInt64(downloadableContent.TitleId, 16),
|
||
|
FullPath = downloadableContent.FullPath
|
||
|
});
|
||
|
}
|
||
|
|
||
|
if (!string.IsNullOrWhiteSpace(container.ContainerPath))
|
||
|
{
|
||
|
_downloadableContentContainerList.Add(container);
|
||
|
}
|
||
|
|
||
|
using (FileStream downloadableContentJsonStream = File.Create(_downloadableContentJsonPath, 4096, FileOptions.WriteThrough))
|
||
|
{
|
||
|
downloadableContentJsonStream.Write(Encoding.UTF8.GetBytes(JsonHelper.Serialize(_downloadableContentContainerList, true)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
}
|