using System;
using System.Collections.ObjectModel;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Threading;
using BililiveRecorder.ToolBox;
using BililiveRecorder.ToolBox.Tool.DanmakuMerger;
using BililiveRecorder.ToolBox.Tool.DanmakuStartTime;
using BililiveRecorder.WPF.Controls;
using BililiveRecorder.WPF.Models;
using Microsoft.WindowsAPICodePack.Dialogs;
using Serilog;
using WPFLocalizeExtension.Extensions;
#nullable enable
namespace BililiveRecorder.WPF.Pages
{
///
/// Interaction logic for ToolboxDanmakuMergerPage.xaml
///
public partial class ToolboxDanmakuMergerPage
{
private static readonly ILogger logger = Log.ForContext();
private static readonly string DesktopPath = Environment.GetFolderPath(Environment.SpecialFolder.Desktop);
private readonly ObservableCollection Files = new();
public ToolboxDanmakuMergerPage()
{
this.InitializeComponent();
this.listView.ItemsSource = this.Files;
}
private void RemoveFile_Click(object sender, RoutedEventArgs e)
{
var b = (Button)sender;
var f = (DanmakuFileWithOffset)b.DataContext;
this.Files.Remove(f);
this.CalculateOffsets();
}
private async void DragDrop(object sender, DragEventArgs e)
{
try
{
if (e.Data.GetDataPresent(DataFormats.FileDrop))
{
var files = (string[])e.Data.GetData(DataFormats.FileDrop);
await this.AddFilesAsync(files.Where(x => x.EndsWith(".xml", StringComparison.InvariantCultureIgnoreCase)).ToArray());
}
}
catch (Exception)
{ }
}
private async void AddFile_Click(object sender, RoutedEventArgs e)
{
var d = new CommonOpenFileDialog
{
Title = LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_Merge_OpenFileDialogTitle"),
AllowNonFileSystemItems = false,
EnsureFileExists = true,
EnsurePathExists = true,
EnsureValidNames = true,
DefaultDirectory = DesktopPath,
DefaultExtension = "xml",
Multiselect = true,
};
d.Filters.Add(new CommonFileDialogFilter(LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_Merge_XmlDanmakuFiles"), "*.xml"));
if (d.ShowDialog() != CommonFileDialogResult.Ok)
return;
await this.AddFilesAsync(d.FileNames.ToArray());
}
private async Task AddFilesAsync(string[] paths)
{
var req = new DanmakuStartTimeRequest { Inputs = paths };
var handler = new DanmakuStartTimeHandler();
var resp = await handler.Handle(req, default, default).ConfigureAwait(true);
if (resp.Status != ResponseStatus.OK || resp.Data is null)
return;
var toBeAdded = resp.Data.StartTimes.Select(x => new DanmakuFileWithOffset(x.Path) { StartTime = x.StartTime });
foreach (var file in toBeAdded)
this.Files.Add(file);
_ = this.Dispatcher.BeginInvoke(DispatcherPriority.Normal, (Action)this.CalculateOffsets);
}
private void CalculateOffsets()
{
if (this.Files.Count == 0)
return;
var minTime = this.Files.Min(x => x.StartTime);
foreach (var item in this.Files)
{
item.Offset = (int)(item.StartTime - minTime).TotalSeconds;
}
this.listView.DataContext = null;
this.listView.DataContext = this.Files;
}
private async void Merge_Click(object sender, RoutedEventArgs e)
{
AutoFixProgressDialog? progressDialog = null;
try
{
var inputPaths = this.Files.Distinct().ToArray();
if (inputPaths.Length < 2)
{
MessageBox.Show(LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_Merge_Error_AtLeastTwo"),
LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_Merge_Title"),
MessageBoxButton.OK,
MessageBoxImage.Warning);
return;
}
logger.Debug("合并弹幕文件 {Paths}", inputPaths);
progressDialog = new AutoFixProgressDialog()
{
CancelButtonVisibility = Visibility.Collapsed,
CancellationTokenSource = new CancellationTokenSource()
};
var token = progressDialog.CancellationTokenSource.Token;
var showTask = progressDialog.ShowAsync();
string? outputPath;
{
var fileDialog = new CommonSaveFileDialog()
{
Title = LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_AutoFix_SelectOutputDialog_Title"),
AlwaysAppendDefaultExtension = true,
AddToMostRecentlyUsedList = false,
DefaultExtension = "xml",
EnsurePathExists = true,
EnsureValidNames = true,
NavigateToShortcut = true,
OverwritePrompt = false,
InitialDirectory = Path.GetDirectoryName(inputPaths[0].Path),
};
fileDialog.Filters.Add(new CommonFileDialogFilter(LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_Merge_XmlDanmakuFiles"), "*.xml"));
if (fileDialog.ShowDialog() == CommonFileDialogResult.Ok)
outputPath = fileDialog.FileName;
else
return;
}
var req = new DanmakuMergerRequest
{
Inputs = inputPaths.Select(x => x.Path).ToArray(),
Offsets = inputPaths.Select(x => x.Offset).ToArray(),
Output = outputPath,
};
var handler = new DanmakuMergerHandler();
var resp = await handler.Handle(req, token, async p =>
{
await this.Dispatcher.InvokeAsync(() =>
{
progressDialog.Progress = (int)(p * 98d);
});
}).ConfigureAwait(true);
logger.Debug("弹幕合并结果 {@Response}", resp);
if (resp.Status != ResponseStatus.Cancelled && resp.Status != ResponseStatus.OK)
{
logger.Warning(resp.Exception, "弹幕合并时发生错误 (@Status)", resp.Status);
await Task.Run(() => ShowErrorMessageBox(resp)).ConfigureAwait(true);
}
else
{
this.Files.Clear();
}
progressDialog.Hide();
await showTask.ConfigureAwait(true);
}
catch (Exception ex)
{
logger.Error(ex, "弹幕合并时发生未处理的错误");
}
finally
{
try
{
_ = this.Dispatcher.BeginInvoke((Action)(() => progressDialog?.Hide()));
progressDialog?.CancellationTokenSource?.Cancel();
}
catch (Exception) { }
}
}
private static void ShowErrorMessageBox(CommandResponse resp) where T : IResponseData
{
var title = LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_AutoFix_Error_Title");
var type = LocExtension.GetLocalizedValue("BililiveRecorder.WPF:Strings:Toolbox_AutoFix_Error_Type_" + resp.Status.ToString());
MessageBox.Show($"{type}\n{resp.ErrorMessage}", title, MessageBoxButton.OK, MessageBoxImage.Warning);
}
}
}