You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
864 lines
28 KiB
864 lines
28 KiB
using System; |
|
using System.Collections.Generic; |
|
using System.Collections.Immutable; |
|
using System.Collections.ObjectModel; |
|
using System.ComponentModel; |
|
using System.Diagnostics; |
|
using System.Globalization; |
|
using System.IO; |
|
using System.Linq; |
|
using System.Reflection; |
|
using System.Text; |
|
using System.Text.Json; |
|
using System.Threading.Tasks; |
|
using Avalonia; |
|
using Avalonia.Controls.Notifications; |
|
using Avalonia.Controls.Primitives; |
|
using Avalonia.Media.Imaging; |
|
using Avalonia.Platform.Storage; |
|
using Avalonia.Styling; |
|
using Avalonia.Threading; |
|
using CommunityToolkit.Mvvm.ComponentModel; |
|
using CommunityToolkit.Mvvm.Input; |
|
using FluentAvalonia.UI.Controls; |
|
using NLog; |
|
using SkiaSharp; |
|
using StabilityMatrix.Avalonia.Controls; |
|
using StabilityMatrix.Avalonia.Extensions; |
|
using StabilityMatrix.Avalonia.Helpers; |
|
using StabilityMatrix.Avalonia.Languages; |
|
using StabilityMatrix.Avalonia.Models; |
|
using StabilityMatrix.Avalonia.Models.TagCompletion; |
|
using StabilityMatrix.Avalonia.Services; |
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
|
using StabilityMatrix.Avalonia.ViewModels.Dialogs; |
|
using StabilityMatrix.Avalonia.ViewModels.Inference; |
|
using StabilityMatrix.Avalonia.Views; |
|
using StabilityMatrix.Avalonia.Views.Dialogs; |
|
using StabilityMatrix.Core.Attributes; |
|
using StabilityMatrix.Core.Extensions; |
|
using StabilityMatrix.Core.Helper; |
|
using StabilityMatrix.Core.Models; |
|
using StabilityMatrix.Core.Models.FileInterfaces; |
|
using StabilityMatrix.Core.Python; |
|
using StabilityMatrix.Core.Services; |
|
using Symbol = FluentIcons.Common.Symbol; |
|
using SymbolIconSource = FluentIcons.FluentAvalonia.SymbolIconSource; |
|
|
|
namespace StabilityMatrix.Avalonia.ViewModels; |
|
|
|
[View(typeof(SettingsPage))] |
|
public partial class SettingsViewModel : PageViewModelBase |
|
{ |
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); |
|
|
|
private readonly INotificationService notificationService; |
|
private readonly ISettingsManager settingsManager; |
|
private readonly IPrerequisiteHelper prerequisiteHelper; |
|
private readonly IPyRunner pyRunner; |
|
private readonly ServiceManager<ViewModelBase> dialogFactory; |
|
private readonly ICompletionProvider completionProvider; |
|
private readonly ITrackedDownloadService trackedDownloadService; |
|
private readonly IModelIndexService modelIndexService; |
|
|
|
public SharedState SharedState { get; } |
|
|
|
public override string Title => "Settings"; |
|
public override IconSource IconSource => |
|
new SymbolIconSource { Symbol = Symbol.Settings, IsFilled = true }; |
|
|
|
// ReSharper disable once MemberCanBeMadeStatic.Global |
|
public string AppVersion => |
|
$"Version {Compat.AppVersion}" + (Program.IsDebugBuild ? " (Debug)" : ""); |
|
|
|
// Theme section |
|
[ObservableProperty] |
|
private string? selectedTheme; |
|
|
|
public IReadOnlyList<string> AvailableThemes { get; } = new[] { "Light", "Dark", "System", }; |
|
|
|
[ObservableProperty] |
|
private CultureInfo selectedLanguage; |
|
|
|
// ReSharper disable once MemberCanBeMadeStatic.Global |
|
public IReadOnlyList<CultureInfo> AvailableLanguages => Cultures.SupportedCultures; |
|
|
|
public IReadOnlyList<float> AnimationScaleOptions { get; } = |
|
new[] { 0f, 0.25f, 0.5f, 0.75f, 1f, 1.25f, 1.5f, 1.75f, 2f, }; |
|
|
|
[ObservableProperty] |
|
private float selectedAnimationScale; |
|
|
|
// Shared folder options |
|
[ObservableProperty] |
|
private bool removeSymlinksOnShutdown; |
|
|
|
// Inference UI section |
|
[ObservableProperty] |
|
private bool isPromptCompletionEnabled; |
|
|
|
[ObservableProperty] |
|
private IReadOnlyList<string> availableTagCompletionCsvs = Array.Empty<string>(); |
|
|
|
[ObservableProperty] |
|
private string? selectedTagCompletionCsv; |
|
|
|
[ObservableProperty] |
|
private bool isCompletionRemoveUnderscoresEnabled; |
|
|
|
[ObservableProperty] |
|
private bool isImageViewerPixelGridEnabled; |
|
|
|
// Integrations section |
|
[ObservableProperty] |
|
private bool isDiscordRichPresenceEnabled; |
|
|
|
// Debug section |
|
[ObservableProperty] |
|
private string? debugPaths; |
|
|
|
[ObservableProperty] |
|
private string? debugCompatInfo; |
|
|
|
[ObservableProperty] |
|
private string? debugGpuInfo; |
|
|
|
// Info section |
|
private const int VersionTapCountThreshold = 7; |
|
|
|
[ObservableProperty, NotifyPropertyChangedFor(nameof(VersionFlyoutText))] |
|
private int versionTapCount; |
|
|
|
[ObservableProperty] |
|
private bool isVersionTapTeachingTipOpen; |
|
public string VersionFlyoutText => |
|
$"You are {VersionTapCountThreshold - VersionTapCount} clicks away from enabling Debug options."; |
|
|
|
public SettingsViewModel( |
|
INotificationService notificationService, |
|
ISettingsManager settingsManager, |
|
IPrerequisiteHelper prerequisiteHelper, |
|
IPyRunner pyRunner, |
|
ServiceManager<ViewModelBase> dialogFactory, |
|
ITrackedDownloadService trackedDownloadService, |
|
SharedState sharedState, |
|
ICompletionProvider completionProvider, |
|
IModelIndexService modelIndexService |
|
) |
|
{ |
|
this.notificationService = notificationService; |
|
this.settingsManager = settingsManager; |
|
this.prerequisiteHelper = prerequisiteHelper; |
|
this.pyRunner = pyRunner; |
|
this.dialogFactory = dialogFactory; |
|
this.trackedDownloadService = trackedDownloadService; |
|
this.completionProvider = completionProvider; |
|
this.modelIndexService = modelIndexService; |
|
|
|
SharedState = sharedState; |
|
|
|
SelectedTheme = settingsManager.Settings.Theme ?? AvailableThemes[1]; |
|
SelectedLanguage = Cultures.GetSupportedCultureOrDefault(settingsManager.Settings.Language); |
|
RemoveSymlinksOnShutdown = settingsManager.Settings.RemoveFolderLinksOnShutdown; |
|
SelectedAnimationScale = settingsManager.Settings.AnimationScale; |
|
|
|
settingsManager.RelayPropertyFor(this, vm => vm.SelectedTheme, settings => settings.Theme); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
vm => vm.IsDiscordRichPresenceEnabled, |
|
settings => settings.IsDiscordRichPresenceEnabled |
|
); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
settings => settings.IsDiscordRichPresenceEnabled, |
|
true |
|
); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
vm => vm.SelectedAnimationScale, |
|
settings => settings.AnimationScale |
|
); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
vm => vm.SelectedTagCompletionCsv, |
|
settings => settings.TagCompletionCsv |
|
); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
vm => vm.IsPromptCompletionEnabled, |
|
settings => settings.IsPromptCompletionEnabled, |
|
true |
|
); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
vm => vm.IsCompletionRemoveUnderscoresEnabled, |
|
settings => settings.IsCompletionRemoveUnderscoresEnabled, |
|
true |
|
); |
|
|
|
settingsManager.RelayPropertyFor( |
|
this, |
|
vm => vm.IsImageViewerPixelGridEnabled, |
|
settings => settings.IsImageViewerPixelGridEnabled, |
|
true |
|
); |
|
|
|
DebugThrowAsyncExceptionCommand.WithNotificationErrorHandler( |
|
notificationService, |
|
LogLevel.Warn |
|
); |
|
ImportTagCsvCommand.WithNotificationErrorHandler(notificationService, LogLevel.Warn); |
|
} |
|
|
|
/// <inheritdoc /> |
|
public override void OnLoaded() |
|
{ |
|
base.OnLoaded(); |
|
|
|
UpdateAvailableTagCompletionCsvs(); |
|
} |
|
|
|
partial void OnSelectedThemeChanged(string? value) |
|
{ |
|
// In case design / tests |
|
if (Application.Current is null) |
|
return; |
|
// Change theme |
|
Application.Current.RequestedThemeVariant = value switch |
|
{ |
|
"Dark" => ThemeVariant.Dark, |
|
"Light" => ThemeVariant.Light, |
|
_ => ThemeVariant.Default |
|
}; |
|
} |
|
|
|
partial void OnSelectedLanguageChanged(CultureInfo? oldValue, CultureInfo newValue) |
|
{ |
|
if (oldValue is null || newValue.Name == Cultures.Current.Name) |
|
return; |
|
// Set locale |
|
if (AvailableLanguages.Contains(newValue)) |
|
{ |
|
Logger.Info("Changing language from {Old} to {New}", oldValue, newValue); |
|
|
|
Cultures.TrySetSupportedCulture(newValue); |
|
settingsManager.Transaction(s => s.Language = newValue.Name); |
|
|
|
var dialog = new BetterContentDialog |
|
{ |
|
Title = Resources.Label_RelaunchRequired, |
|
Content = Resources.Text_RelaunchRequiredToApplyLanguage, |
|
DefaultButton = ContentDialogButton.Primary, |
|
PrimaryButtonText = Resources.Action_Relaunch, |
|
CloseButtonText = Resources.Action_RelaunchLater |
|
}; |
|
|
|
Dispatcher.UIThread.InvokeAsync(async () => |
|
{ |
|
if (await dialog.ShowAsync() == ContentDialogResult.Primary) |
|
{ |
|
Process.Start(Compat.AppCurrentPath); |
|
App.Shutdown(); |
|
} |
|
}); |
|
} |
|
else |
|
{ |
|
Logger.Info( |
|
"Requested invalid language change from {Old} to {New}", |
|
oldValue, |
|
newValue |
|
); |
|
} |
|
} |
|
|
|
partial void OnRemoveSymlinksOnShutdownChanged(bool value) |
|
{ |
|
settingsManager.Transaction(s => s.RemoveFolderLinksOnShutdown = value); |
|
} |
|
|
|
public async Task ResetCheckpointCache() |
|
{ |
|
settingsManager.Transaction(s => s.InstalledModelHashes = new HashSet<string>()); |
|
await Task.Run(() => settingsManager.IndexCheckpoints()); |
|
notificationService.Show( |
|
"Checkpoint cache reset", |
|
"The checkpoint cache has been reset.", |
|
NotificationType.Success |
|
); |
|
} |
|
|
|
#region Package Environment |
|
|
|
[RelayCommand] |
|
private async Task OpenEnvVarsDialog() |
|
{ |
|
var viewModel = dialogFactory.Get<EnvVarsViewModel>(); |
|
|
|
// Load current settings |
|
var current = |
|
settingsManager.Settings.EnvironmentVariables ?? new Dictionary<string, string>(); |
|
viewModel.EnvVars = new ObservableCollection<EnvVarKeyPair>( |
|
current.Select(kvp => new EnvVarKeyPair(kvp.Key, kvp.Value)) |
|
); |
|
|
|
var dialog = new BetterContentDialog |
|
{ |
|
Content = new EnvVarsDialog { DataContext = viewModel }, |
|
PrimaryButtonText = "Save", |
|
IsPrimaryButtonEnabled = true, |
|
CloseButtonText = "Cancel", |
|
}; |
|
|
|
if (await dialog.ShowAsync() == ContentDialogResult.Primary) |
|
{ |
|
// Save settings |
|
var newEnvVars = viewModel.EnvVars |
|
.Where(kvp => !string.IsNullOrWhiteSpace(kvp.Key)) |
|
.GroupBy(kvp => kvp.Key, StringComparer.Ordinal) |
|
.ToDictionary(g => g.Key, g => g.First().Value, StringComparer.Ordinal); |
|
settingsManager.Transaction(s => s.EnvironmentVariables = newEnvVars); |
|
} |
|
} |
|
|
|
[RelayCommand] |
|
private async Task CheckPythonVersion() |
|
{ |
|
var isInstalled = prerequisiteHelper.IsPythonInstalled; |
|
Logger.Debug($"Check python installed: {isInstalled}"); |
|
// Ensure python installed |
|
if (!prerequisiteHelper.IsPythonInstalled) |
|
{ |
|
// Need 7z as well for site packages repack |
|
Logger.Debug("Python not installed, unpacking resources..."); |
|
await prerequisiteHelper.UnpackResourcesIfNecessary(); |
|
Logger.Debug("Unpacked resources, installing python..."); |
|
await prerequisiteHelper.InstallPythonIfNecessary(); |
|
} |
|
|
|
// Get python version |
|
await pyRunner.Initialize(); |
|
var result = await pyRunner.GetVersionInfo(); |
|
// Show dialog box |
|
var dialog = new ContentDialog |
|
{ |
|
Title = "Python version info", |
|
Content = result, |
|
PrimaryButtonText = "Ok", |
|
IsPrimaryButtonEnabled = true |
|
}; |
|
dialog.Title = "Python version info"; |
|
dialog.Content = result; |
|
dialog.PrimaryButtonText = "Ok"; |
|
await dialog.ShowAsync(); |
|
} |
|
|
|
#endregion |
|
|
|
#region Inference UI |
|
|
|
private void UpdateAvailableTagCompletionCsvs() |
|
{ |
|
if (!settingsManager.IsLibraryDirSet) |
|
return; |
|
|
|
var tagsDir = settingsManager.TagsDirectory; |
|
if (!tagsDir.Exists) |
|
return; |
|
|
|
var csvFiles = tagsDir.Info.EnumerateFiles("*.csv"); |
|
AvailableTagCompletionCsvs = csvFiles.Select(f => f.Name).ToImmutableArray(); |
|
|
|
// Set selected to current if exists |
|
var settingsCsv = settingsManager.Settings.TagCompletionCsv; |
|
if (settingsCsv is not null && AvailableTagCompletionCsvs.Contains(settingsCsv)) |
|
{ |
|
SelectedTagCompletionCsv = settingsCsv; |
|
} |
|
} |
|
|
|
[RelayCommand(FlowExceptionsToTaskScheduler = true)] |
|
private async Task ImportTagCsv() |
|
{ |
|
var storage = App.StorageProvider; |
|
var files = await storage.OpenFilePickerAsync( |
|
new FilePickerOpenOptions |
|
{ |
|
FileTypeFilter = new List<FilePickerFileType> |
|
{ |
|
new("CSV") { Patterns = new[] { "*.csv" }, } |
|
} |
|
} |
|
); |
|
|
|
if (files.Count == 0) |
|
return; |
|
|
|
var sourceFile = new FilePath(files[0].TryGetLocalPath()!); |
|
|
|
var tagsDir = settingsManager.TagsDirectory; |
|
tagsDir.Create(); |
|
|
|
// Copy to tags directory |
|
var targetFile = tagsDir.JoinFile(sourceFile.Name); |
|
await sourceFile.CopyToAsync(targetFile); |
|
|
|
// Update index |
|
UpdateAvailableTagCompletionCsvs(); |
|
|
|
// Trigger load |
|
completionProvider.BackgroundLoadFromFile(targetFile, true); |
|
|
|
notificationService.Show( |
|
$"Imported {sourceFile.Name}", |
|
$"The {sourceFile.Name} file has been imported.", |
|
NotificationType.Success |
|
); |
|
} |
|
#endregion |
|
|
|
#region System |
|
|
|
/// <summary> |
|
/// Adds Stability Matrix to Start Menu for the current user. |
|
/// </summary> |
|
[RelayCommand] |
|
private async Task AddToStartMenu() |
|
{ |
|
if (!Compat.IsWindows) |
|
{ |
|
notificationService.Show("Not supported", "This feature is only supported on Windows."); |
|
return; |
|
} |
|
|
|
await using var _ = new MinimumDelay(200, 300); |
|
|
|
var shortcutDir = new DirectoryPath( |
|
Environment.GetFolderPath(Environment.SpecialFolder.StartMenu), |
|
"Programs" |
|
); |
|
var shortcutLink = shortcutDir.JoinFile("Stability Matrix.lnk"); |
|
|
|
var appPath = Compat.AppCurrentPath; |
|
var iconPath = shortcutDir.JoinFile("Stability Matrix.ico"); |
|
await Assets.AppIcon.ExtractTo(iconPath); |
|
|
|
WindowsShortcuts.CreateShortcut(shortcutLink, appPath, iconPath, "Stability Matrix"); |
|
|
|
notificationService.Show( |
|
"Added to Start Menu", |
|
"Stability Matrix has been added to the Start Menu.", |
|
NotificationType.Success |
|
); |
|
} |
|
|
|
/// <summary> |
|
/// Add Stability Matrix to Start Menu for all users. |
|
/// <remarks>Requires Admin elevation.</remarks> |
|
/// </summary> |
|
[RelayCommand] |
|
private async Task AddToGlobalStartMenu() |
|
{ |
|
if (!Compat.IsWindows) |
|
{ |
|
notificationService.Show("Not supported", "This feature is only supported on Windows."); |
|
return; |
|
} |
|
|
|
// Confirmation dialog |
|
var dialog = new BetterContentDialog |
|
{ |
|
Title = |
|
"This will create a shortcut for Stability Matrix in the Start Menu for all users", |
|
Content = "You will be prompted for administrator privileges. Continue?", |
|
PrimaryButtonText = "Yes", |
|
CloseButtonText = "Cancel", |
|
DefaultButton = ContentDialogButton.Primary |
|
}; |
|
|
|
if (await dialog.ShowAsync() != ContentDialogResult.Primary) |
|
{ |
|
return; |
|
} |
|
|
|
await using var _ = new MinimumDelay(200, 300); |
|
|
|
var shortcutDir = new DirectoryPath( |
|
Environment.GetFolderPath(Environment.SpecialFolder.CommonStartMenu), |
|
"Programs" |
|
); |
|
var shortcutLink = shortcutDir.JoinFile("Stability Matrix.lnk"); |
|
|
|
var appPath = Compat.AppCurrentPath; |
|
var iconPath = shortcutDir.JoinFile("Stability Matrix.ico"); |
|
|
|
// We can't directly write to the targets, so extract to temporary directory first |
|
using var tempDir = new TempDirectoryPath(); |
|
|
|
await Assets.AppIcon.ExtractTo(tempDir.JoinFile("Stability Matrix.ico")); |
|
WindowsShortcuts.CreateShortcut( |
|
tempDir.JoinFile("Stability Matrix.lnk"), |
|
appPath, |
|
iconPath, |
|
"Stability Matrix" |
|
); |
|
|
|
// Move to target |
|
try |
|
{ |
|
var moveLinkResult = await WindowsElevated.MoveFiles( |
|
(tempDir.JoinFile("Stability Matrix.lnk"), shortcutLink), |
|
(tempDir.JoinFile("Stability Matrix.ico"), iconPath) |
|
); |
|
if (moveLinkResult != 0) |
|
{ |
|
notificationService.ShowPersistent( |
|
"Failed to create shortcut", |
|
$"Could not copy shortcut", |
|
NotificationType.Error |
|
); |
|
} |
|
} |
|
catch (Win32Exception e) |
|
{ |
|
// We'll get this exception if user cancels UAC |
|
Logger.Warn(e, "Could not create shortcut"); |
|
notificationService.Show("Could not create shortcut", "", NotificationType.Warning); |
|
return; |
|
} |
|
|
|
notificationService.Show( |
|
"Added to Start Menu", |
|
"Stability Matrix has been added to the Start Menu for all users.", |
|
NotificationType.Success |
|
); |
|
} |
|
|
|
public async Task PickNewDataDirectory() |
|
{ |
|
var viewModel = dialogFactory.Get<SelectDataDirectoryViewModel>(); |
|
var dialog = new BetterContentDialog |
|
{ |
|
IsPrimaryButtonEnabled = false, |
|
IsSecondaryButtonEnabled = false, |
|
IsFooterVisible = false, |
|
Content = new SelectDataDirectoryDialog { DataContext = viewModel } |
|
}; |
|
|
|
var result = await dialog.ShowAsync(); |
|
if (result == ContentDialogResult.Primary) |
|
{ |
|
// 1. For portable mode, call settings.SetPortableMode() |
|
if (viewModel.IsPortableMode) |
|
{ |
|
settingsManager.SetPortableMode(); |
|
} |
|
// 2. For custom path, call settings.SetLibraryPath(path) |
|
else |
|
{ |
|
settingsManager.SetLibraryPath(viewModel.DataDirectory); |
|
} |
|
|
|
// Restart |
|
var restartDialog = new BetterContentDialog |
|
{ |
|
Title = "Restart required", |
|
Content = "Stability Matrix must be restarted for the changes to take effect.", |
|
PrimaryButtonText = "Restart", |
|
DefaultButton = ContentDialogButton.Primary, |
|
IsSecondaryButtonEnabled = false, |
|
}; |
|
await restartDialog.ShowAsync(); |
|
|
|
Process.Start(Compat.AppCurrentPath); |
|
App.Shutdown(); |
|
} |
|
} |
|
|
|
#endregion |
|
|
|
#region Debug Section |
|
public void LoadDebugInfo() |
|
{ |
|
var assembly = Assembly.GetExecutingAssembly(); |
|
var appData = Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData); |
|
DebugPaths = $""" |
|
Current Working Directory [Environment.CurrentDirectory] |
|
"{Environment.CurrentDirectory}" |
|
App Directory [Assembly.GetExecutingAssembly().Location] |
|
"{assembly.Location}" |
|
App Directory [AppContext.BaseDirectory] |
|
"{AppContext.BaseDirectory}" |
|
AppData Directory [SpecialFolder.ApplicationData] |
|
"{appData}" |
|
"""; |
|
|
|
// 1. Check portable mode |
|
var appDir = Compat.AppCurrentDir; |
|
var expectedPortableFile = Path.Combine(appDir, "Data", ".sm-portable"); |
|
var isPortableMode = File.Exists(expectedPortableFile); |
|
|
|
DebugCompatInfo = $""" |
|
Platform: {Compat.Platform} |
|
AppData: {Compat.AppData} |
|
AppDataHome: {Compat.AppDataHome} |
|
AppCurrentDir: {Compat.AppCurrentDir} |
|
ExecutableName: {Compat.GetExecutableName()} |
|
-- Settings -- |
|
Expected Portable Marker file: {expectedPortableFile} |
|
Portable Marker file exists: {isPortableMode} |
|
IsLibraryDirSet = {settingsManager.IsLibraryDirSet} |
|
IsPortableMode = {settingsManager.IsPortableMode} |
|
"""; |
|
|
|
// Get Gpu info |
|
var gpuInfo = ""; |
|
foreach (var (i, gpu) in HardwareHelper.IterGpuInfo().Enumerate()) |
|
{ |
|
gpuInfo += $"[{i + 1}] {gpu}\n"; |
|
} |
|
DebugGpuInfo = gpuInfo; |
|
} |
|
|
|
// Debug buttons |
|
[RelayCommand] |
|
private void DebugNotification() |
|
{ |
|
notificationService.Show( |
|
new Notification( |
|
title: "Test Notification", |
|
message: "Here is some message", |
|
type: NotificationType.Information |
|
) |
|
); |
|
} |
|
|
|
[RelayCommand] |
|
private async Task DebugContentDialog() |
|
{ |
|
var dialog = new ContentDialog |
|
{ |
|
DefaultButton = ContentDialogButton.Primary, |
|
Title = "Test title", |
|
PrimaryButtonText = "OK", |
|
CloseButtonText = "Close" |
|
}; |
|
|
|
var result = await dialog.ShowAsync(); |
|
notificationService.Show(new Notification("Content dialog closed", $"Result: {result}")); |
|
} |
|
|
|
[RelayCommand] |
|
private void DebugThrowException() |
|
{ |
|
throw new OperationCanceledException("Example Message"); |
|
} |
|
|
|
[RelayCommand(FlowExceptionsToTaskScheduler = true)] |
|
private async Task DebugThrowAsyncException() |
|
{ |
|
await Task.Yield(); |
|
|
|
throw new ApplicationException("Example Message"); |
|
} |
|
|
|
[RelayCommand] |
|
private async Task DebugMakeImageGrid() |
|
{ |
|
var provider = App.StorageProvider; |
|
var files = await provider.OpenFilePickerAsync( |
|
new FilePickerOpenOptions() { AllowMultiple = true } |
|
); |
|
|
|
if (files.Count == 0) |
|
return; |
|
|
|
var images = await files.SelectAsync( |
|
async f => SKImage.FromEncodedData(await f.OpenReadAsync()) |
|
); |
|
|
|
var grid = ImageProcessor.CreateImageGrid(images.ToImmutableArray()); |
|
|
|
// Show preview |
|
|
|
using var peekPixels = grid.PeekPixels(); |
|
using var data = peekPixels.Encode(SKEncodedImageFormat.Jpeg, 100); |
|
await using var stream = data.AsStream(); |
|
|
|
var bitmap = WriteableBitmap.Decode(stream); |
|
|
|
var galleryImages = new List<ImageSource> { new(bitmap), }; |
|
galleryImages.AddRange(files.Select(f => new ImageSource(f.Path.ToString()))); |
|
|
|
var imageBox = new ImageGalleryCard |
|
{ |
|
Width = 1000, |
|
Height = 900, |
|
DataContext = dialogFactory.Get<ImageGalleryCardViewModel>(vm => |
|
{ |
|
vm.ImageSources.AddRange(galleryImages); |
|
}) |
|
}; |
|
|
|
var dialog = new BetterContentDialog |
|
{ |
|
MaxDialogWidth = 1000, |
|
MaxDialogHeight = 1000, |
|
FullSizeDesired = true, |
|
Content = imageBox, |
|
CloseButtonText = "Close", |
|
ContentVerticalScrollBarVisibility = ScrollBarVisibility.Disabled, |
|
}; |
|
|
|
await dialog.ShowAsync(); |
|
} |
|
|
|
[RelayCommand] |
|
private async Task DebugLoadCompletionCsv() |
|
{ |
|
var provider = App.StorageProvider; |
|
var files = await provider.OpenFilePickerAsync(new FilePickerOpenOptions()); |
|
|
|
if (files.Count == 0) |
|
return; |
|
|
|
await completionProvider.LoadFromFile(files[0].TryGetLocalPath()!, true); |
|
|
|
notificationService.Show("Loaded completion file", ""); |
|
} |
|
|
|
[RelayCommand] |
|
private async Task DebugImageMetadata() |
|
{ |
|
var provider = App.StorageProvider; |
|
var files = await provider.OpenFilePickerAsync(new FilePickerOpenOptions()); |
|
|
|
if (files.Count == 0) |
|
return; |
|
|
|
var metadata = ImageMetadata.ParseFile(files[0].TryGetLocalPath()!); |
|
var comfyJson = metadata.GetComfyMetadata(); |
|
|
|
if (comfyJson is null) |
|
{ |
|
notificationService.Show("No Comfy metadata found", ""); |
|
return; |
|
} |
|
|
|
var dialog = DialogHelper.CreateJsonDialog(comfyJson); |
|
await dialog.ShowAsync(); |
|
} |
|
|
|
[RelayCommand] |
|
private async Task DebugRefreshModelsIndex() |
|
{ |
|
await modelIndexService.RefreshIndex(); |
|
} |
|
|
|
[RelayCommand] |
|
private async Task DebugTrackedDownload() |
|
{ |
|
var textFields = new TextBoxField[] |
|
{ |
|
new() { Label = "Url", }, |
|
new() { Label = "File path" } |
|
}; |
|
|
|
var dialog = DialogHelper.CreateTextEntryDialog("Add download", "", textFields); |
|
|
|
if (await dialog.ShowAsync() == ContentDialogResult.Primary) |
|
{ |
|
var url = textFields[0].Text; |
|
var filePath = textFields[1].Text; |
|
var download = trackedDownloadService.NewDownload(new Uri(url), new FilePath(filePath)); |
|
download.Start(); |
|
} |
|
} |
|
#endregion |
|
|
|
#region Info Section |
|
|
|
public void OnVersionClick() |
|
{ |
|
// Ignore if already enabled |
|
if (SharedState.IsDebugMode) |
|
return; |
|
|
|
VersionTapCount++; |
|
|
|
switch (VersionTapCount) |
|
{ |
|
// Reached required threshold |
|
case >= VersionTapCountThreshold: |
|
{ |
|
IsVersionTapTeachingTipOpen = false; |
|
// Enable debug options |
|
SharedState.IsDebugMode = true; |
|
notificationService.Show( |
|
"Debug options enabled", |
|
"Warning: Improper use may corrupt application state or cause loss of data." |
|
); |
|
VersionTapCount = 0; |
|
break; |
|
} |
|
// Open teaching tip above 3rd click |
|
case >= 3: |
|
IsVersionTapTeachingTipOpen = true; |
|
break; |
|
} |
|
} |
|
|
|
[RelayCommand] |
|
private async Task ShowLicensesDialog() |
|
{ |
|
try |
|
{ |
|
var markdown = GetLicensesMarkdown(); |
|
|
|
var dialog = DialogHelper.CreateMarkdownDialog(markdown, "Licenses"); |
|
dialog.MaxDialogHeight = 600; |
|
await dialog.ShowAsync(); |
|
} |
|
catch (Exception e) |
|
{ |
|
notificationService.Show( |
|
"Failed to read licenses information", |
|
$"{e}", |
|
NotificationType.Error |
|
); |
|
} |
|
} |
|
|
|
private static string GetLicensesMarkdown() |
|
{ |
|
// Read licenses.json |
|
using var reader = new StreamReader(Assets.LicensesJson.Open()); |
|
var licenses = |
|
JsonSerializer.Deserialize<IReadOnlyList<LicenseInfo>>(reader.ReadToEnd()) |
|
?? throw new InvalidOperationException("Failed to read licenses.json"); |
|
|
|
// Generate markdown |
|
var builder = new StringBuilder(); |
|
foreach (var license in licenses) |
|
{ |
|
builder.AppendLine( |
|
$"## [{license.PackageName}]({license.PackageUrl}) by {string.Join(", ", license.Authors)}" |
|
); |
|
builder.AppendLine(); |
|
builder.AppendLine(license.Description); |
|
builder.AppendLine(); |
|
builder.AppendLine($"[{license.LicenseUrl}]({license.LicenseUrl})"); |
|
builder.AppendLine(); |
|
} |
|
|
|
return builder.ToString(); |
|
} |
|
|
|
#endregion |
|
}
|
|
|