DesignData.cs

using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.ComponentModel;
using System.Diagnostics.CodeAnalysis;
using System.IO;
using System.Linq;
using System.Net.Http;
using System.Text;
using AvaloniaEdit.Document;
using AvaloniaEdit.Utils;
using DynamicData.Binding;
using Microsoft.Extensions.DependencyInjection;
using NSubstitute;
using Semver;
using StabilityMatrix.Avalonia.Controls.CodeCompletion;
using StabilityMatrix.Avalonia.Models;
using StabilityMatrix.Avalonia.Models.TagCompletion;
using StabilityMatrix.Avalonia.Services;
using StabilityMatrix.Avalonia.ViewModels;
using StabilityMatrix.Avalonia.ViewModels.Base;
using StabilityMatrix.Avalonia.ViewModels.CheckpointBrowser;
using StabilityMatrix.Avalonia.ViewModels.CheckpointManager;
using StabilityMatrix.Avalonia.ViewModels.Controls;
using StabilityMatrix.Avalonia.ViewModels.Dialogs;
using StabilityMatrix.Avalonia.ViewModels.Inference;
using StabilityMatrix.Avalonia.ViewModels.Inference.Video;
using StabilityMatrix.Avalonia.ViewModels.OutputsPage;
using StabilityMatrix.Avalonia.ViewModels.PackageManager;
using StabilityMatrix.Avalonia.ViewModels.Progress;
using StabilityMatrix.Avalonia.ViewModels.Settings;
using StabilityMatrix.Avalonia.Views.Dialogs;
using StabilityMatrix.Core.Api;
using StabilityMatrix.Core.Database;
using StabilityMatrix.Core.Extensions;
using StabilityMatrix.Core.Helper;
using StabilityMatrix.Core.Helper.Cache;
using StabilityMatrix.Core.Helper.Factory;
using StabilityMatrix.Core.Models;
using StabilityMatrix.Core.Models.Api;
using StabilityMatrix.Core.Models.Api.CivitTRPC;
using StabilityMatrix.Core.Models.Api.Comfy;
using StabilityMatrix.Core.Models.Api.OpenArt;
using StabilityMatrix.Core.Models.Api.OpenModelsDb;
using StabilityMatrix.Core.Models.Database;
using StabilityMatrix.Core.Models.FileInterfaces;
using StabilityMatrix.Core.Models.PackageModification;
using StabilityMatrix.Core.Models.Packages;
using StabilityMatrix.Core.Models.Packages.Extensions;
using StabilityMatrix.Core.Models.Progress;
using StabilityMatrix.Core.Models.Update;
using StabilityMatrix.Core.Python;
using StabilityMatrix.Core.Services;
using StabilityMatrix.Core.Updater;
using CivitAiBrowserViewModel = StabilityMatrix.Avalonia.ViewModels.CheckpointBrowser.CivitAiBrowserViewModel;
using HuggingFacePageViewModel = StabilityMatrix.Avalonia.ViewModels.CheckpointBrowser.HuggingFacePageViewModel;
using MainPackageManagerViewModel = StabilityMatrix.Avalonia.ViewModels.PackageManager.MainPackageManagerViewModel;

namespace StabilityMatrix.Avalonia.DesignData;

[Localizable(false)]
[SuppressMessage("ReSharper", "MemberCanBePrivate.Global")]
public static class DesignData
{
    [NotNull]
    public static IServiceProvider? Services { get; set; }

    private static bool isInitialized;

    // This needs to be static method instead of static constructor
    // or else Avalonia analyzers won't work.
    public static void Initialize()
    {
        if (isInitialized)
            throw new InvalidOperationException("DesignData is already initialized.");

        var services = App.ConfigureServices(disableMessagePipeInterprocess: true);

        var activePackageId = Guid.NewGuid();
        services.AddSingleton<ISettingsManager, MockSettingsManager>(_ => new MockSettingsManager
        {
            Settings =
            {
                InstalledPackages = new List<InstalledPackage>
                {
                    new()
                    {
                        Id = activePackageId,
                        DisplayName = "My Installed Package",
                        PackageName = "framepack",
                        Version = new InstalledPackageVersion { InstalledReleaseVersion = "v1.0.0" },
                        LibraryPath = $"Packages{Path.DirectorySeparatorChar}example-webui",
                        LastUpdateCheck = DateTimeOffset.Now,
                    },
                    new()
                    {
                        Id = Guid.NewGuid(),
                        DisplayName = "Comfy Diffusion WebUI Dev Branch Long Name",
                        PackageName = "ComfyUI",
                        Version = new InstalledPackageVersion
                        {
                            InstalledBranch = "master",
                            InstalledCommitSha = "abc12uwu345568972abaedf7g7e679a98879e879f87ga8",
                        },
                        LibraryPath = $"Packages{Path.DirectorySeparatorChar}example-webui",
                        LastUpdateCheck = DateTimeOffset.Now,
                    },
                    new()
                    {
                        Id = Guid.NewGuid(),
                        DisplayName = "Running Comfy",
                        PackageName = "ComfyUI",
                        Version = new InstalledPackageVersion
                        {
                            InstalledBranch = "master",
                            InstalledCommitSha = "abc12uwu345568972abaedf7g7e679a98879e879f87ga8",
                        },
                        LibraryPath = $"Packages{Path.DirectorySeparatorChar}example-webui",
                        LastUpdateCheck = DateTimeOffset.Now,
                    },
                },
                ActiveInstalledPackageId = activePackageId,
            },
        });

        // General services
        services
            .AddLogging()
            .AddSingleton<IPackageFactory, PackageFactory>()
            .AddSingleton<IUpdateHelper, UpdateHelper>()
            .AddSingleton<ModelFinder>()
            .AddSingleton<SharedState>();

        // Mock services
        services
            .AddSingleton(Substitute.For<INotificationService>())
            .AddSingleton(Substitute.For<ISharedFolders>())
            .AddSingleton(Substitute.For<IDownloadService>())
            .AddSingleton(Substitute.For<IHttpClientFactory>())
            .AddSingleton(Substitute.For<IApiFactory>())
            .AddSingleton(Substitute.For<IDiscordRichPresenceService>())
            .AddSingleton(Substitute.For<ITrackedDownloadService>())
            .AddSingleton(Substitute.For<ILiteDbContext>())
            .AddSingleton(Substitute.For<IAccountsService>())
            .AddSingleton<IInferenceClientManager, MockInferenceClientManager>()
            .AddSingleton<ICompletionProvider, MockCompletionProvider>()
            .AddSingleton<IModelIndexService, MockModelIndexService>()
            .AddSingleton<IImageIndexService, MockImageIndexService>()
            .AddSingleton<IMetadataImportService, MetadataImportService>();

        // Placeholder services that nobody should need during design time
        services
            .AddSingleton<IPyRunner>(_ => null!)
            .AddSingleton<ILiteDbContext>(_ => null!)
            .AddSingleton<ICivitApi>(_ => null!)
            .AddSingleton<IGithubApiCache>(_ => null!)
            .AddSingleton<ITokenizerProvider>(_ => null!)
            .AddSingleton<IPrerequisiteHelper>(_ => null!)
            .AddSingleton<IPyPiApi>(_ => null!)
            .AddSingleton<IPyPiCache>(_ => null!);

        // Override Launch page with mock
        services.Remove(ServiceDescriptor.Singleton<LaunchPageViewModel, LaunchPageViewModel>());
        services.AddSingleton<LaunchPageViewModel, MockLaunchPageViewModel>();

        Services = services.BuildServiceProvider();

        var dialogFactory = Services.GetRequiredService<IServiceManager<ViewModelBase>>();
        var settingsManager = Services.GetRequiredService<ISettingsManager>();
        var downloadService = Services.GetRequiredService<IDownloadService>();
        var modelFinder = Services.GetRequiredService<ModelFinder>();
        var packageFactory = Services.GetRequiredService<IPackageFactory>();
        var notificationService = Services.GetRequiredService<INotificationService>();
        var modelImportService = Services.GetRequiredService<IMetadataImportService>();

        LaunchOptionsViewModel = Services.GetRequiredService<LaunchOptionsViewModel>();
        LaunchOptionsViewModel.Cards = new[]
        {
            LaunchOptionCard.FromDefinition(
                new LaunchOptionDefinition
                {
                    Name = "Host",
                    Type = LaunchOptionType.String,
                    Description = "The host name for the Web UI",
                    DefaultValue = "localhost",
                    Options = { "--host" },
                }
            ),
            LaunchOptionCard.FromDefinition(
                new LaunchOptionDefinition
                {
                    Name = "API",
                    Type = LaunchOptionType.Bool,
                    Options = { "--api" },
                }
            ),
        };
        LaunchOptionsViewModel.UpdateFilterCards();

        NewInstallerDialogViewModel = Services.GetRequiredService<PackageInstallBrowserViewModel>();
        // NewInstallerDialogViewModel.InferencePackages = new ObservableCollectionExtended<BasePackage>(
        //     packageFactory.GetPackagesByType(PackageType.SdInference).OrderBy(p => p.InstallerSortOrder)
        // );
        // NewInstallerDialogViewModel.TrainingPackages = new ObservableCollection<BasePackage>(
        //     packageFactory.GetPackagesByType(PackageType.SdTraining).OrderBy(p => p.InstallerSortOrder)
        // );

        PackageInstallDetailViewModel = new PackageInstallDetailViewModel(
            packageFactory.GetAllAvailablePackages().FirstOrDefault() as BaseGitPackage,
            settingsManager,
            notificationService,
            null,
            null,
            null,
            packageFactory,
            null,
            null
        );

        PackageInstallDetailViewModel.AvailablePythonVersions = new ObservableCollection<UvPythonInfo>(
            [
                new UvPythonInfo(
                    PyInstallationManager.Python_3_10_17,
                    "C:\\SMData\\Data\\Data\\Assets\\Python\\cpython-3.11.12-windows-x86_64-none",
                    true,
                    "cpython",
                    "x86_64",
                    "windows",
                    "cpython-3.10.17-windows-x86_64-none",
                    "default",
                    "none"
                ),
                new UvPythonInfo(
                    PyInstallationManager.Python_3_12_10,
                    null,
                    false,
                    "cpython",
                    "x86_64",
                    "windows",
                    "guh I can't be bothered",
                    "freethreaded",
                    "none"
                ),
            ]
        );

        /*ObservableCacheEx.AddOrUpdate(
            OldCheckpointsPageViewModel.CheckpointFoldersCache,
            new CheckpointFolder[]
            {
                new(settingsManager, downloadService, modelFinder, notificationService, modelImportService)
                {
                    DirectoryPath = "Models/StableDiffusion",
                    DisplayedCheckpointFiles = new ObservableCollectionExtended<CheckpointFile>()
                    {
                        new()
                        {
                            FilePath = "~/Models/StableDiffusion/electricity-light.safetensors",
                            Title = "Auroral Background",
                            PreviewImagePath =
                                "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/"
                                + "78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
                            UpdateAvailable = true,
                            ConnectedModel = new ConnectedModelInfo
                            {
                                VersionName = "Lightning Auroral",
                                BaseModel = "SD 1.5",
                                ModelName = "Auroral Background",
                                ModelType = CivitModelType.Model,
                                FileMetadata = new CivitFileMetadata
                                {
                                    Format = CivitModelFormat.SafeTensor,
                                    Fp = CivitModelFpType.fp16,
                                    Size = CivitModelSize.pruned,
                                },
                                TrainedWords = ["aurora", "lightning"]
                            }
                        },
                        new() { FilePath = "~/Models/Lora/model.safetensors", Title = "Some model" },
                    },
                },
                new(settingsManager, downloadService, modelFinder, notificationService, modelImportService)
                {
                    Title = "Lora",
                    DirectoryPath = "Packages/Lora",
                    DisplayedCheckpointFiles = new ObservableCollectionExtended<CheckpointFile>
                    {
                        new() { FilePath = "~/Models/Lora/lora_v2.pt", Title = "Best Lora v2", }
                    }
                }
            }
        );*/

        CivitAiBrowserViewModel.ModelCards = new ObservableCollectionExtended<CheckpointBrowserCardViewModel>
        {
            dialogFactory.Get<CheckpointBrowserCardViewModel>(vm =>
            {
                vm.CivitModel = new CivitModel
                {
                    Name = "BB95 Furry Mix",
                    Description = "A furry mix of BB95",
                    Stats = new CivitModelStats { Rating = 3.5, RatingCount = 24 },
                    ModelVersions = [new() { Name = "v1.2.2-Inpainting" }],
                    Creator = new CivitCreator
                    {
                        Image = "https://gravatar.com/avatar/fe74084ae8a081dc2283f5bde4736756ad?f=y&d=retro",
                        Username = "creator-1",
                    },
                };
            }),
            dialogFactory.Get<CheckpointBrowserCardViewModel>(vm =>
            {
                vm.CivitModel = new CivitModel
                {
                    Name = "Another Model",
                    Description = "A mix of example",
                    Stats = new CivitModelStats { Rating = 5, RatingCount = 3500 },
                    ModelVersions =
                    [
                        new()
                        {
                            Name = "v1.2.2-Inpainting",
                            Images = new List<CivitImage>
                            {
                                new()
                                {
                                    NsfwLevel = 1,
                                    Url =
                                        "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/"
                                        + "78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
                                },
                            },
                        },
                    ],
                    Creator = new CivitCreator
                    {
                        Image = "https://gravatar.com/avatar/205e460b479e2e5b48aec07710c08d50?f=y&d=retro",
                        Username = "creator-2",
                    },
                };
            }),
        };

        CheckpointsPageViewModel.Categories = new ObservableCollectionExtended<CheckpointCategory>
        {
            new()
            {
                Name = "Category 1",
                Path = "path1",
                SubDirectories = [new CheckpointCategory { Name = "SubCategory 1", Path = "path3" }],
            },
            new() { Name = "Category 2", Path = "path2" },
        };

        CheckpointsPageViewModel.Models = new ObservableCollectionExtended<CheckpointFileViewModel>()
        {
            new(
                settingsManager,
                new MockModelIndexService(),
                notificationService,
                downloadService,
                dialogFactory,
                null,
                new LocalModelFile
                {
                    SharedFolderType = SharedFolderType.StableDiffusion,
                    RelativePath = "~/Models/StableDiffusion/electricity-light.safetensors",
                    PreviewImageFullPath =
                        "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/"
                        + "78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
                    HasUpdate = true,
                    ConnectedModelInfo = new ConnectedModelInfo
                    {
                        VersionName = "Lightning Auroral",
                        BaseModel = "SD 1.5",
                        ModelName = "Auroral Background",
                        ModelType = CivitModelType.Model,
                        FileMetadata = new CivitFileMetadata
                        {
                            Format = CivitModelFormat.SafeTensor,
                            Fp = "fp16",
                            Size = "pruned",
                        },
                        TrainedWords = ["aurora", "lightning"],
                    },
                }
            ),
            new(
                settingsManager,
                new MockModelIndexService(),
                notificationService,
                downloadService,
                dialogFactory,
                null,
                new LocalModelFile
                {
                    RelativePath = "~/Models/Lora/model.safetensors",
                    SharedFolderType = SharedFolderType.StableDiffusion,
                }
            ),
        };

        var packageInstall = new PackageInstallProgressItemViewModel(
            new PackageModificationRunner
            {
                CurrentProgress = new ProgressReport(0.5f, "Installing package...", "Installing... 50%"),
                ModificationCompleteMessage = "Package installed successfully",
            }
        )
        {
            Progress = new ContentDialogProgressViewModelBase
            {
                Value = 50,
                IsIndeterminate = false,
                Text = "UwU Install",
                Description = "Installing...",
            },
        };

        ProgressManagerViewModel.ProgressItems.AddRange(
            [
                new ProgressItemViewModel(
                    new ProgressItem(
                        Guid.NewGuid(),
                        "Test File.exe",
                        new ProgressReport(0.5f, "Downloading...")
                    )
                ),
                new MockDownloadProgressItemViewModel(
                    "Very Long Test File Name Need Even More Longness Thanks That's pRobably good 2.exe"
                ),
                new MockDownloadProgressItemViewModel(
                    "Very Long Test File Name Need Even More Longness Thanks That's pRobably good 2.exe"
                )
                {
                    Progress = new ContentDialogProgressViewModelBase
                    {
                        Value = 50,
                        IsIndeterminate = false,
                        Text = "Waiting on other downloads to finish",
                        Description = "Waiting on other downloads to finish",
                    },
                },
                packageInstall,
            ]
        );

        UpdateViewModel = Services.GetRequiredService<UpdateViewModel>();
        UpdateViewModel.CurrentVersionText = "v2.0.0";
        UpdateViewModel.NewVersionText = "v2.0.1";
        UpdateViewModel.ReleaseNotes =
            "## v2.0.1\n- Fixed a bug\n- Added a feature\n- Removed a feature\n - Did some `--code` stuff";

        isInitialized = true;
    }

    [NotNull]
    public static PackageInstallBrowserViewModel? NewInstallerDialogViewModel { get; private set; }

    [NotNull]
    public static PackageInstallDetailViewModel? PackageInstallDetailViewModel { get; private set; }

    [NotNull]
    public static LaunchOptionsViewModel? LaunchOptionsViewModel { get; private set; }

    [NotNull]
    public static UpdateViewModel? UpdateViewModel { get; private set; }

    public static IServiceManager<ViewModelBase> DialogFactory =>
        Services.GetRequiredService<IServiceManager<ViewModelBase>>();

    public static MainWindowViewModel MainWindowViewModel =>
        Services.GetRequiredService<MainWindowViewModel>();

    public static FirstLaunchSetupViewModel FirstLaunchSetupViewModel =>
        Services.GetRequiredService<FirstLaunchSetupViewModel>();

    public static NotificationBannerViewModel NotificationBannerViewModel =>
        DialogFactory.Get<NotificationBannerViewModel>(vm =>
        {
            vm.Show(
                new Core.Models.Notifications.AppNotification
                {
                    Id = "test",
                    Type = Core.Models.Notifications.AppNotificationType.Banner,
                    Priority = Core.Models.Notifications.AppNotificationPriority.Normal,
                    Message = new Dictionary<string, string> { { "en", "This is a test notification." } },
                    Style = new Core.Models.Notifications.AppNotificationStyle { Variant = "info" },
                    Action = new Core.Models.Notifications.AppNotificationAction
                    {
                        Type = Core.Models.Notifications.AppNotificationActionType.Url,
                        Label = new Dictionary<string, string> { { "en", "Learn More" } },
                        Url = "https://example.com",
                    },
                }
            );
        });

    public static LaunchPageViewModel LaunchPageViewModel =>
        Services.GetRequiredService<LaunchPageViewModel>();

    public static HuggingFacePageViewModel HuggingFacePageViewModel =>
        Services.GetRequiredService<HuggingFacePageViewModel>();

    public static DirectUrlImportViewModel DirectUrlImportViewModel =>
        Services.GetRequiredService<DirectUrlImportViewModel>();

    public static NewOneClickInstallViewModel NewOneClickInstallViewModel =>
        Services.GetRequiredService<NewOneClickInstallViewModel>();

    public static RecommendedModelsViewModel RecommendedModelsViewModel =>
        DialogFactory.Get<RecommendedModelsViewModel>(vm =>
        {
            // Populate the single RecommendedModels collection for design time
            vm.RecommendedModels.AddRange(
                [
                    new RecommendedModelItemViewModel
                    {
                        CivitModel = new CivitModel { Name = "BB95 Furry Mix", Id = 1 }, // Added Id for clarity
                        ModelVersion = new CivitModelVersion
                        {
                            Id = 101, // Added Id for clarity
                            Name = "v1.0", // Example version name
                            BaseModel = "SD 1.5", // Example base model
                            Stats = new CivitModelStats { Rating = 4.5, RatingCount = 124 },
                            Files = [new CivitFile { Type = CivitFileType.Model }], // Example file
                            Images =
                            [
                                new CivitImage
                                {
                                    Type = "image", // Ensure type is set
                                    Url =
                                        "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
                                },
                            ],
                        },
                        Author = "by bb95",
                    },
                    new RecommendedModelItemViewModel
                    {
                        CivitModel = new CivitModel { Name = "DreamShaper XL", Id = 2 },
                        ModelVersion = new CivitModelVersion
                        {
                            Id = 201,
                            Name = "v2.1 Turbo",
                            BaseModel = "SDXL 1.0",
                            Stats = new CivitModelStats { Rating = 4.8, RatingCount = 589 },
                            Files = [new CivitFile { Type = CivitFileType.Model, IsPrimary = true }],
                            Images =
                            [
                                new CivitImage
                                {
                                    Type = "image",
                                    // Placeholder - replace with an actual relevant image URL if possible
                                    Url =
                                        "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0cf3e133-4dde-458b-8a70-7451a3361472/width=450/00016-3919014893.jpeg",
                                },
                            ],
                        },
                        Author = "by Lykon",
                    },
                    new RecommendedModelItemViewModel
                    {
                        CivitModel = new CivitModel { Name = "Another Model SD1.5", Id = 3 },
                        ModelVersion = new CivitModelVersion
                        {
                            Id = 301,
                            Name = "Final",
                            BaseModel = "SD 1.5",
                            Stats = new CivitModelStats { Rating = 4.2, RatingCount = 99 },
                            Files = [new CivitFile { Type = CivitFileType.Model }],
                            Images = [new CivitImage { Type = "image", Url = Assets.NoImage.ToString() }], // Use placeholder
                        },
                        Author = "by Creator3",
                    },
                    // Add more items as needed for design-time preview
                ]
            );
        });
    public static OutputsPageViewModel OutputsPageViewModel
    {
        get
        {
            var vm = Services.GetRequiredService<OutputsPageViewModel>();
            vm.Outputs = new ObservableCollectionExtended<OutputImageViewModel>
            {
                new(
                    new LocalImageFile
                    {
                        AbsolutePath =
                            "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
                        ImageType = LocalImageFileType.TextToImage,
                    }
                ),
            };
            vm.Categories = new ObservableCollectionExtended<TreeViewDirectory>
            {
                new()
                {
                    Name = "Category 1",
                    Path = "path1",
                    SubDirectories = [new TreeViewDirectory { Name = "SubCategory 1", Path = "path3" }],
                },
                new() { Name = "Category 2", Path = "path2" },
            };
            return vm;
        }
    }

    public static MainPackageManagerViewModel MainPackageManagerViewModel
    {
        get
        {
            var settings = Services.GetRequiredService<ISettingsManager>();
            var vm = Services.GetRequiredService<MainPackageManagerViewModel>();

            vm.SetPackages(settings.Settings.InstalledPackages);
            vm.SetUnknownPackages(
                new InstalledPackage[]
                {
                    UnknownInstalledPackage.FromDirectoryName("sd-unknown-with-long-name"),
                }
            );

            vm.PackageCards[0].IsUpdateAvailable = true;

            return vm;
        }
    }

    public static PackageExtensionBrowserViewModel PackageExtensionBrowserViewModel =>
        DialogFactory.Get<PackageExtensionBrowserViewModel>(vm =>
        {
            vm.AddExtensions(
                [
                    new PackageExtension
                    {
                        Author = "123",
                        Title = "Cool Extension",
                        Description = "This is an interesting extension",
                        Reference = new Uri("https://github.com/LykosAI/StabilityMatrix"),
                        Files = [new Uri("https://github.com/LykosAI/StabilityMatrix")],
                    },
                    new PackageExtension
                    {
                        Author = "123",
                        Title = "Cool Extension",
                        Description = "This is an interesting extension",
                        Reference = new Uri("https://github.com/LykosAI/StabilityMatrix"),
                        Files = [new Uri("https://github.com/LykosAI/StabilityMatrix")],
                    },
                ],
                [
                    new InstalledPackageExtension
                    {
                        GitRepositoryUrl = "https://github.com/LykosAI/StabilityMatrix",
                        Paths = [new DirectoryPath("example-dir")],
                    },
                    new InstalledPackageExtension { Paths = [new DirectoryPath("example-dir-2")] },
                ]
            );
            vm.AddExtensionPacks(
                [
                    new ExtensionPack
                    {
                        Name = "Test Pack",
                        PackageType = "ComfyUI",
                        Extensions =
                        [
                            new SavedPackageExtension
                            {
                                PackageExtension = new PackageExtension
                                {
                                    Author = "TestAuthor",
                                    Title = "Test",
                                    Reference = new Uri("https://github.com/LykosAI/StabilityMatrix"),
                                    Files = [new Uri("https://github.com/LykosAI/StabilityMatrix")],
                                },
                                Version = new PackageExtensionVersion
                                {
                                    Branch = "main",
                                    CommitSha = "abcd123",
                                },
                            },
                        ],
                    },
                ]
            );
        });

    public static CheckpointsPageViewModel CheckpointsPageViewModel =>
        Services.GetRequiredService<CheckpointsPageViewModel>();

    public static SettingsViewModel SettingsViewModel => Services.GetRequiredService<SettingsViewModel>();

    public static PackageManagerViewModel PackageManagerViewModel =>
        Services.GetRequiredService<PackageManagerViewModel>();

    public static InferenceSettingsViewModel InferenceSettingsViewModel =>
        Services.GetRequiredService<InferenceSettingsViewModel>();

    public static MainSettingsViewModel MainSettingsViewModel
    {
        get
        {
            var vm = Services.GetRequiredService<MainSettingsViewModel>();
            vm.AllBaseModelTypes = new List<string>()
            {
                CivitBaseModelType.WanVideo.GetStringValue(),
                CivitBaseModelType.Sdxl10.GetStringValue(),
                CivitBaseModelType.Flux1D.GetStringValue(),
                "Flux 1. Kontext",
            }
                .Select(s => new BaseModelOptionViewModel { ModelType = s, IsSelected = true })
                .ToList();
            return vm;
        }
    }

    public static AccountSettingsViewModel AccountSettingsViewModel =>
        Services.GetRequiredService<AccountSettingsViewModel>();

    public static NotificationSettingsViewModel NotificationSettingsViewModel =>
        Services.GetRequiredService<NotificationSettingsViewModel>();

    public static UpdateSettingsViewModel UpdateSettingsViewModel
    {
        get
        {
            var vm = Services.GetRequiredService<UpdateSettingsViewModel>();

            var update = new UpdateInfo
            {
                Version = SemVersion.Parse("2.0.1"),
                ReleaseDate = DateTimeOffset.Now,
                Url = new Uri("https://example.org"),
                Changelog = new Uri("https://example.org"),
                HashBlake3 = "",
                Signature = "",
            };

            vm.UpdateStatus = new UpdateStatusChangedEventArgs
            {
                LatestUpdate = update,
                UpdateChannels = new Dictionary<UpdateChannel, UpdateInfo>
                {
                    [UpdateChannel.Stable] = update,
                    [UpdateChannel.Preview] = update,
                    [UpdateChannel.Development] = update,
                },
                CheckedAt = DateTimeOffset.UtcNow,
            };
            return vm;
        }
    }

    public static CivitAiBrowserViewModel CivitAiBrowserViewModel =>
        Services.GetRequiredService<CivitAiBrowserViewModel>();

    public static CheckpointBrowserViewModel CheckpointBrowserViewModel =>
        Services.GetRequiredService<CheckpointBrowserViewModel>();

    public static CivitDetailsPageViewModel CivitDetailsPageViewModel =>
        DialogFactory.Get<CivitDetailsPageViewModel>(vm =>
        {
            vm.CivitModel = new CivitModel
            {
                Name = "BB95 Furry Mix",
                Description = "A furry mix of BB95",
                Stats = new CivitModelStats
                {
                    Rating = 3.5,
                    RatingCount = 24,
                    ThumbsUpCount = 1337,
                    DownloadCount = 100_000,
                },
                Tags = ["base model", "furry", "animals", "photorealistic", "highly detailed", "yiff"],
                ModelVersions =
                [
                    new CivitModelVersion
                    {
                        Name = "v1.2.2-Inpainting",
                        PublishedAt = DateTimeOffset.Now,
                        Images =
                        [
                            new CivitImage
                            {
                                Url =
                                    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/244bc929-9bbc-44b2-8a26-7622a1669f4a/original=true,quality=90/00123-3430906941-1girl,%20hatsune%20miku,%20white%20pupils,%20power%20elements,%20microphone,%20vibrant%20blue%20color%20palette,%20abstract,abstract%20background,%20dreamli.jpeg",
                            },
                            new CivitImage
                            {
                                Url =
                                    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/aa671302-496f-4dcc-839a-b75a70a7665e/original=true,quality=90/00136-4204868169-1girl,%20solo,%20long%20hair,%20city,%20boots,%20hands%20in%20pockets,%20coat,%20blonde%20hair,%20sky,%20planet,%20night,%20knee%20boots,%20building,%20star%20_(sky_).jpeg",
                            },
                        ],
                        Files =
                        [
                            new CivitFile
                            {
                                Name = "bb95-v100-uwu-reallylongfilename-v1234576802.safetensors",
                                Type = CivitFileType.Model,
                                Metadata = new CivitFileMetadata
                                {
                                    Format = CivitModelFormat.SafeTensor,
                                    Fp = "fp16",
                                    Size = "pruned",
                                },
                            },
                            new CivitFile
                            {
                                Name = "bb95-v100-uwu-reallylongfilename-v1234576802-fp32.safetensors",
                                Type = CivitFileType.Model,
                                Metadata = new CivitFileMetadata
                                {
                                    Format = CivitModelFormat.SafeTensor,
                                    Fp = "fp32",
                                    Size = "full",
                                },
                                Hashes = new CivitFileHashes
                                {
                                    BLAKE3 =
                                        "A7383E54F2E4570678B0F18545B2EB8FD95325DA76CCBA8467DBDBD481CF6B99",
                                    SHA256 =
                                        "BDB59BAC77D94AE7A55FF893170F9554C3F349E48A1B73C0C17C0B7C6F4D41A2",
                                },
                            },
                        ],
                    },
                    new CivitModelVersion { Name = "v1.2.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v1.1.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v1.0.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v0.9.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v0.8.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v0.7.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v0.6.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v0.5.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                    new CivitModelVersion { Name = "v0.4.0", PublishedAt = DateTimeOffset.Now.AddDays(-3) },
                ],
                Creator = new CivitCreator
                {
                    Image = "https://gravatar.com/avatar/fe74084ae8a081dc2283f5bde4736756ad?f=y&d=retro",
                    Username = "creator-1",
                },
            };
        });

    public static SelectModelVersionViewModel SelectModelVersionViewModel =>
        DialogFactory.Get<SelectModelVersionViewModel>(vm =>
        {
            // Sample data
            var sampleCivitVersions = new List<CivitModelVersion>
            {
                new()
                {
                    Name = "BB95 Furry Mix",
                    Description =
                        @"Introducing SnoutMix
A Mix of non-Furry and Furry models such as Furtastic and BB95Furry to create a great variety of anthro AI generation options, but bringing out more detail, still giving a lot of freedom to customise the human aspects, and having great backgrounds, with a focus on something more realistic. Works well with realistic character loras.
The gallery images are often inpainted, but you will get something very similar if copying their data directly. They are inpainted using the same model, therefore all results are possible without anything custom/hidden-away. Controlnet Tiled is applied to enhance them further afterwards. Gallery images were made with same model but before it was renamed",
                    BaseModel = "SD 1.5",
                    Files = new List<CivitFile>
                    {
                        new()
                        {
                            Name = "bb95-v100-uwu-reallylongfilename-v1234576802.safetensors",
                            Type = CivitFileType.Model,
                            Metadata = new CivitFileMetadata
                            {
                                Format = CivitModelFormat.SafeTensor,
                                Fp = "fp16",
                                Size = "pruned",
                            },
                        },
                        new()
                        {
                            Name = "bb95-v100-uwu-reallylongfilename-v1234576802-fp32.safetensors",
                            Type = CivitFileType.Model,
                            Metadata = new CivitFileMetadata
                            {
                                Format = CivitModelFormat.SafeTensor,
                                Fp = "fp32",
                                Size = "full",
                            },
                            Hashes = new CivitFileHashes { BLAKE3 = "ABCD" },
                        },
                    },
                },
            };
            var sampleViewModel = new ModelVersionViewModel(
                Services.GetRequiredService<IModelIndexService>(),
                sampleCivitVersions[0]
            );

            // Sample data for dialogs
            vm.Versions = new List<ModelVersionViewModel> { sampleViewModel };
            vm.Title = sampleCivitVersions[0].Name;
            vm.Description = sampleCivitVersions[0].Description;
            vm.SelectedVersionViewModel = sampleViewModel;
        });

    public static OneClickInstallViewModel OneClickInstallViewModel =>
        Services.GetRequiredService<OneClickInstallViewModel>();

    public static InferenceViewModel InferenceViewModel => Services.GetRequiredService<InferenceViewModel>();

    public static SelectDataDirectoryViewModel SelectDataDirectoryViewModel =>
        Services.GetRequiredService<SelectDataDirectoryViewModel>();

    public static ProgressManagerViewModel ProgressManagerViewModel =>
        Services.GetRequiredService<ProgressManagerViewModel>();

    public static ExceptionViewModel ExceptionViewModel =>
        DialogFactory.Get<ExceptionViewModel>(viewModel =>
        {
            // Use try-catch to generate traceback information
            try
            {
                try
                {
                    throw new OperationCanceledException("Example");
                }
                catch (OperationCanceledException e)
                {
                    throw new AggregateException(e);
                }
            }
            catch (AggregateException e)
            {
                viewModel.Exception = e;
            }
        });

    public static EnvVarsViewModel EnvVarsViewModel =>
        DialogFactory.Get<EnvVarsViewModel>(viewModel =>
        {
            viewModel.EnvVars = new ObservableCollection<EnvVarKeyPair> { new("UWU", "TRUE") };
        });

    public static PythonPackagesViewModel PythonPackagesViewModel =>
        DialogFactory.Get<PythonPackagesViewModel>(vm =>
        {
            vm.AddPackages(new PipPackageInfo("pip", "1.0.0"), new PipPackageInfo("torch", "2.1.0+cu121"));
        });

    public static LykosLoginViewModel LykosLoginViewModel => DialogFactory.Get<LykosLoginViewModel>();

    public static OAuthConnectViewModel OAuthConnectViewModel =>
        DialogFactory.Get<OAuthConnectViewModel>(vm =>
        {
            vm.Url =
                "https://www.example.org/oauth2/authorize?"
                + "client_id=66ad566552679cb6e650be01ed6f8d2ae9a0f803c0369850a5c9ee82a2396062&"
                + "scope=identity%20identity.memberships&"
                + "response_type=code&state=test%40example.org&"
                + "redirect_uri=http://localhost:5022/api/oauth/patreon/callback";
        });

    public static OAuthLoginViewModel OAuthLoginViewModel =>
        DialogFactory.Get<OAuthLoginViewModel>(vm =>
        {
            vm.Url =
                "https://www.example.org/oauth2/authorize?"
                + "client_id=66ad566552679cb6e650be01ed6f8d2ae9a0f803c0369850a5c9ee82a2396062&"
                + "scope=identity%20identity.memberships&"
                + "response_type=code&state=test%40example.org&"
                + "redirect_uri=http://localhost:5022/api/oauth/patreon/callback";
        });

    public static OAuthDeviceAuthViewModel OAuthDeviceAuthViewModel =>
        DialogFactory.Get<OAuthDeviceAuthViewModel>(vm =>
        {
            vm.VerificationUri = new Uri("https://example.org/connect/verify");
            vm.UserCode = "AB23-CD56";
        });

    public static PythonPackageSpecifiersViewModel PythonPackageSpecifiersViewModel =>
        DialogFactory.Get<PythonPackageSpecifiersViewModel>();

    public static MaskEditorViewModel MaskEditorViewModel => DialogFactory.Get<MaskEditorViewModel>();

    public static InferenceTextToImageViewModel InferenceTextToImageViewModel =>
        DialogFactory.Get<InferenceTextToImageViewModel>(vm =>
        {
            vm.OutputProgress.Value = 10;
            vm.OutputProgress.Maximum = 30;
            vm.OutputProgress.Text = "Sampler 10/30";
        });

    public static InferenceImageToVideoViewModel InferenceImageToVideoViewModel =>
        DialogFactory.Get<InferenceImageToVideoViewModel>(vm =>
        {
            vm.OutputProgress.Value = 10;
            vm.OutputProgress.Maximum = 30;
            vm.OutputProgress.Text = "Sampler 10/30";
        });

    public static InferenceImageToImageViewModel InferenceImageToImageViewModel =>
        DialogFactory.Get<InferenceImageToImageViewModel>();

    public static InferenceImageUpscaleViewModel InferenceImageUpscaleViewModel =>
        DialogFactory.Get<InferenceImageUpscaleViewModel>();

    public static InferenceWanTextToVideoViewModel InferenceWanTextToVideoViewModel =>
        DialogFactory.Get<InferenceWanTextToVideoViewModel>();
    public static InferenceWanImageToVideoViewModel InferenceWanImageToVideoViewModel =>
        DialogFactory.Get<InferenceWanImageToVideoViewModel>();

    public static PackageImportViewModel PackageImportViewModel =>
        DialogFactory.Get<PackageImportViewModel>();

    public static RefreshBadgeViewModel RefreshBadgeViewModel => new() { State = ProgressState.Success };

    public static PropertyGridViewModel PropertyGridViewModel =>
        DialogFactory.Get<PropertyGridViewModel>(vm =>
        {
            vm.SelectedObject = new INotifyPropertyChanged[]
            {
                new MockPropertyGridObject(),
                new MockPropertyGridObjectAlt(),
            };
            vm.ExcludeCategories = ["Excluded Category"];
        });

    public static SeedCardViewModel SeedCardViewModel => new();
    public static SvdImgToVidConditioningViewModel SvdImgToVidConditioningViewModel => new();
    public static VideoOutputSettingsCardViewModel VideoOutputSettingsCardViewModel => new();

    public static SamplerCardViewModel SamplerCardViewModel =>
        DialogFactory.Get<SamplerCardViewModel>(vm =>
        {
            vm.Steps = 20;
            vm.CfgScale = 7;
            vm.IsCfgScaleEnabled = true;
            vm.IsSamplerSelectionEnabled = true;
            vm.IsDimensionsEnabled = true;
            vm.IsLengthEnabled = true;
            vm.SelectedSampler = new ComfySampler("euler");
        });

    public static SamplerCardViewModel SamplerCardViewModelScaleMode =>
        DialogFactory.Get<SamplerCardViewModel>(vm =>
        {
            vm.IsDenoiseStrengthEnabled = true;
        });

    public static SamplerCardViewModel SamplerCardViewModelRefinerMode =>
        DialogFactory.Get<SamplerCardViewModel>(vm =>
        {
            vm.IsCfgScaleEnabled = true;
            vm.IsSamplerSelectionEnabled = true;
            vm.IsDimensionsEnabled = true;
            vm.IsRefinerStepsEnabled = true;
        });

    public static ModelCardViewModel ModelCardViewModel => DialogFactory.Get<ModelCardViewModel>();
    public static WanModelCardViewModel WanModelCardViewModel => DialogFactory.Get<WanModelCardViewModel>();

    public static ImgToVidModelCardViewModel ImgToVidModelCardViewModel =>
        DialogFactory.Get<ImgToVidModelCardViewModel>();

    public static PlasmaNoiseCardViewModel PlasmaNoiseCardViewModel =>
        DialogFactory.Get<PlasmaNoiseCardViewModel>();

    public static ImageGalleryCardViewModel ImageGalleryCardViewModel =>
        DialogFactory.Get<ImageGalleryCardViewModel>(vm =>
        {
            vm.ImageSources.AddRange(
                new ImageSource[]
                {
                    new(
                        new Uri(
                            "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4a7e00a7-6f18-42d4-87c0-10e792df2640/width=1152"
                        )
                    ),
                    new(
                        new Uri(
                            "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a318ac1f-3ad0-48ac-98cc-79126febcc17/width=1024"
                        )
                    ),
                    new(
                        new Uri(
                            "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/16588c94-6595-4be9-8806-d7e6e22d198c/width=1152"
                        )
                    ),
                }
            );
        });

    public static ImageFolderCardViewModel ImageFolderCardViewModel =>
        DialogFactory.Get<ImageFolderCardViewModel>();

    public static FreeUCardViewModel FreeUCardViewModel => DialogFactory.Get<FreeUCardViewModel>();

    public static NrsCardViewModel NrsCardViewModel => DialogFactory.Get<NrsCardViewModel>();

    public static PromptCardViewModel PromptCardViewModel =>
        DialogFactory.Get<PromptCardViewModel>(vm =>
        {
            var builder = new StringBuilder();
            builder.AppendLine("house, (high quality), [example], BREAK");
            builder.AppendLine("# this is a comment");
            builder.AppendLine(
                "(detailed), [purple and orange lighting], looking pleased, (cinematic, god rays:0.8), "
                    + "(8k, 4k, high res:1), (intricate), (unreal engine:1.2), (shaded:1.1), "
                    + "(soft focus, detailed background), (horizontal lens flare), "
                    + "(clear eyes)"
            );
            builder.AppendLine("<lora:details:0.8>, <lyco:some_model>");

            vm.PromptDocument.Text = builder.ToString();
            vm.NegativePromptDocument.Text = "embedding:EasyNegative, blurry, jpeg artifacts";
        });

    public static StackCardViewModel StackCardViewModel =>
        DialogFactory.Get<StackCardViewModel>(vm =>
        {
            vm.AddCards(SamplerCardViewModel, SeedCardViewModel);
        });

    public static StackEditableCardViewModel StackEditableCardViewModel =>
        DialogFactory.Get<StackEditableCardViewModel>(vm =>
        {
            vm.AddCards(StackExpanderViewModel, StackExpanderViewModel2);
        });

    public static StackExpanderViewModel StackExpanderViewModel =>
        DialogFactory.Get<StackExpanderViewModel>(vm =>
        {
            vm.Title = "Hires Fix";
            vm.AddCards(UpscalerCardViewModel, SamplerCardViewModel);
            vm.OnContainerIndexChanged(0);
        });

    public static StackExpanderViewModel StackExpanderViewModel2 =>
        DialogFactory.Get<StackExpanderViewModel>(vm =>
        {
            vm.Title = "Hires Fix";
            vm.IsSettingsEnabled = true;
            vm.AddCards(UpscalerCardViewModel, SamplerCardViewModel);
            vm.OnContainerIndexChanged(1);
        });

    public static UpscalerCardViewModel UpscalerCardViewModel => DialogFactory.Get<UpscalerCardViewModel>();

    public static BatchSizeCardViewModel BatchSizeCardViewModel =>
        DialogFactory.Get<BatchSizeCardViewModel>();

    public static BatchSizeCardViewModel BatchSizeCardViewModelWithIndexOption =>
        DialogFactory.Get<BatchSizeCardViewModel>(vm =>
        {
            vm.IsBatchIndexEnabled = true;
        });

    public static LayerDiffuseCardViewModel LayerDiffuseCardViewModel =>
        DialogFactory.Get<LayerDiffuseCardViewModel>();

    public static ExtraNetworkCardViewModel ExtraNetworkCardViewModel =>
        DialogFactory.Get<ExtraNetworkCardViewModel>();

    public static ConfirmPackageDeleteDialogViewModel ConfirmPackageDeleteDialogViewModel =>
        DialogFactory.Get<ConfirmPackageDeleteDialogViewModel>(vm =>
            vm.Package = new InstalledPackage
            {
                PackageName = "a1111",
                DisplayName = "Automatic1111 WebUI",
                LibraryPath = "packages\\a1111",
            }
        );

    public static InstalledWorkflowsViewModel InstalledWorkflowsViewModel
    {
        get
        {
            var vm = Services.GetRequiredService<InstalledWorkflowsViewModel>();
            vm.DisplayedWorkflows = new ObservableCollectionExtended<OpenArtMetadata>
            {
                new()
                {
                    Workflow = new()
                    {
                        Name = "Test Workflow",
                        Creator = new OpenArtCreator { Name = "Test Creator" },
                        Thumbnails =
                        [
                            new OpenArtThumbnail
                            {
                                Url = new Uri(
                                    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/dd9b038c-bd15-43ab-86ab-66e145ad7ff2/width=512"
                                ),
                            },
                        ],
                    },
                },
            };

            return vm;
        }
    }

    public static OpenModelDbKeyedModel SampleOpenModelDbKeyedModel =>
        new()
        {
            Id = "test-id",
            Name = "16x PSNR Prretrained Model",
            Author = "author123",
            License = "Apache-2.0",
            Tags = ["pretrained"],
            Description =
                "Pretrained: RRDB_PSNR_x4.pth\n\nThe original RRDB_PSNR_x4.pth model converted to 1x, 2x, 8x and 16x scales, intended to be used as pretrained models for new models at those scales. These are compatible with victor's 4xESRGAN.pth conversions",
            Date = new DateOnly(2020, 4, 20),
            Architecture = "esrgan",
            Size = ["64nf", "23nb"],
            Scale = 16,
            InputChannels = 3,
            OutputChannels = 3,
            Resources =
            [
                new OpenModelDbResource
                {
                    Platform = "pytorch",
                    Type = "pth",
                    Size = 67254807,
                    Sha256 = "d7ae3b9a3572a01d1ddfc788ebca253c872d959d3765bcb3b48c65a3ab2f9aba",
                    Urls = ["https://drive.google.com/open?id=0"],
                },
                new OpenModelDbResource
                {
                    Platform = "onnx",
                    Type = "onnx",
                    Size = 20098601,
                    Sha256 = "d7ae3b9a3572a01d1ddfc788ebca253c872d959d3765bcb3b48c65a3ab2f9aba",
                    Urls = ["https://drive.google.com/open?id=0"],
                },
            ],
            Images =
            [
                new OpenModelDbImage.Paired
                {
                    Lr = new Uri("https://imgsli.com/i/c9978a74-32ee-478b-b228-10744521fc21.jpg"),
                    Sr = new Uri("https://imgsli.com/i/c90377d5-287d-4f1c-ab4c-96267bfaa04e.jpg"),
                },
                new OpenModelDbImage.Paired
                {
                    Lr = new Uri("https://imgsli.com/i/7c22ba75-01ca-407a-98b9-16e27ad6de55.jpg"),
                    Sr = new Uri("https://imgsli.com/i/12aa0959-36db-49a1-8a0c-e58f99226395.jpg"),
                },
                new OpenModelDbImage.Paired
                {
                    Lr = new Uri("https://imgsli.com/i/83e9600a-a8ee-4f2b-96e7-f42dc7e5ab12.jpg"),
                    Sr = new Uri("https://imgsli.com/i/a611b9a9-422f-44ed-882a-3ada8f478625.jpg"),
                },
            ],
        };

    public static OpenModelDbBrowserViewModel OpenModelDbBrowserViewModel
    {
        get
        {
            var vm = Services.GetRequiredService<OpenModelDbBrowserViewModel>();
            return vm;
        }
    }

    public static OpenModelDbModelDetailsViewModel OpenModelDbModelDetailsViewModel =>
        DialogFactory.Get<OpenModelDbModelDetailsViewModel>(vm =>
        {
            vm.Model = SampleOpenModelDbKeyedModel;
        });

    public static IList<ICompletionData> SampleCompletionData =>
        new List<ICompletionData>
        {
            new TagCompletionData("test1", TagType.General),
            new TagCompletionData("test2", TagType.Artist),
            new TagCompletionData("test3", TagType.Character),
            new TagCompletionData("test4", TagType.Copyright),
            new TagCompletionData("test5", TagType.Species),
            new TagCompletionData("test_unknown", TagType.Invalid),
        };

    public static CompletionList SampleCompletionList
    {
        get
        {
            var list = new CompletionList { IsFiltering = true };
            ExtensionMethods.AddRange(list.CompletionData, SampleCompletionData);
            ExtensionMethods.AddRange(list.FilteredCompletionData, list.CompletionData);
            list.SelectItem("te", true);
            return list;
        }
    }

    public static IEnumerable<HybridModelFile> SampleHybridModels { get; } =
        [
            HybridModelFile.FromLocal(
                new LocalModelFile
                {
                    SharedFolderType = SharedFolderType.StableDiffusion,
                    RelativePath = "art_shaper_v8.safetensors",
                    PreviewImageFullPath =
                        "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/dd9b038c-bd15-43ab-86ab-66e145ad7ff2/width=512",
                    ConnectedModelInfo = new ConnectedModelInfo
                    {
                        ModelName = "Art Shaper (very long name example)",
                        VersionName = "Style v8 (very long name)",
                        ModelId = 0,
                        VersionId = 0,
                        ModelDescription =
                            "This is a very long description for the Art Shaper model, which is used to demonstrate how long descriptions can be handled in the UI. It should be able to display multiple lines and still look good.",
                        ModelType = CivitModelType.Checkpoint,
                        BaseModel = "SD 1.5",
                    },
                }
            ),
            HybridModelFile.FromLocal(
                new LocalModelFile
                {
                    SharedFolderType = SharedFolderType.StableDiffusion,
                    RelativePath = "background_arts.safetensors",
                    PreviewImageFullPath =
                        "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/71c81ddf-d8c3-46b4-843d-9f8f20a9254a/width=512",
                    ConnectedModelInfo = new ConnectedModelInfo
                    {
                        ModelName = "Background Arts",
                        VersionName = "Anime Style v10",
                        ModelId = 0,
                        VersionId = 0,
                        ModelDescription =
                            "This is a very long description for the Art Shaper model, which is used to demonstrate how long descriptions can be handled in the UI. It should be able to display multiple lines and still look good.",
                        ModelType = CivitModelType.Checkpoint,
                        BaseModel = "SDXL 1.0",
                    },
                }
            ),
            HybridModelFile.FromRemote("v1-5-pruned-emaonly.safetensors"),
            HybridModelFile.FromRemote("sample-model.pt"),
        ];

    public static HybridModelFile SampleHybridModel => SampleHybridModels.First();

    public static ImageViewerViewModel ImageViewerViewModel =>
        DialogFactory.Get<ImageViewerViewModel>(vm =>
        {
            vm.ImageSource = new ImageSource(
                new Uri(
                    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a318ac1f-3ad0-48ac-98cc-79126febcc17/width=1500"
                )
            );
            vm.FileNameText = "TextToImage_00041.png";
            vm.FileSizeText = "2.4 MB";
            vm.ImageSizeText = "1280 x 1792";

            vm.CivitImageMetadata = new CivitImageGenerationDataResponse
            {
                Metadata = new CivitImageMetadata
                {
                    Prompt =
                        "closeup photp of a red haired anthro wolf female,\n holding an apple, wearing medieval drees is eating a apple, wolf ears, wolf tail with white tip\n,anthro,furry",
                    NegativePrompt = "Bad quality , watermark",
                    CfgScale = 2.5d,
                    Steps = 30,
                    Sampler = "DPM++ SDE",
                    Seed = 255842256659122,
                    Model = "RatatoskrIllustriousV2.3",
                    Height = 1152,
                    Width = 768,
                    Scheduler = "normal",
                },
                Resources =
                [
                    new CivitImageResource
                    {
                        ModelName = "noobAI XL (NAI-XL) really long name example with even more words",
                        ModelId = 1337,
                        VersionId = 1234,
                        VersionName = "Epsilon-pred 1.1-Version",
                        ModelType = "Checkpoint",
                    },
                ],
            };

            vm.CivitImageMetadata.OtherMetadata = new Dictionary<string, string>
            {
                ["CFG"] = "2.5",
                ["Steps"] = "30",
                ["Sampler"] = "DPM++ SDE",
                ["Seed"] = "255842256659122",
            };
        });

    public static DownloadResourceViewModel DownloadResourceViewModel =>
        DialogFactory.Get<DownloadResourceViewModel>(vm =>
        {
            vm.FileName = ComfyUpscaler.DefaultDownloadableModels[0].Name;
            vm.FileSize = Convert.ToInt64(2 * Size.GiB);
            vm.Resource = ComfyUpscaler.DefaultDownloadableModels[0].DownloadableResource!.Value;
        });

    public static SharpenCardViewModel SharpenCardViewModel => DialogFactory.Get<SharpenCardViewModel>();

    public static InferenceConnectionHelpViewModel InferenceConnectionHelpViewModel =>
        DialogFactory.Get<InferenceConnectionHelpViewModel>();

    public static SelectImageCardViewModel SelectImageCardViewModel =>
        DialogFactory.Get<SelectImageCardViewModel>();

    public static SelectImageCardViewModel SelectImageCardViewModel_WithImage =>
        DialogFactory.Get<SelectImageCardViewModel>(vm =>
        {
            vm.ImageSource = new ImageSource(
                new Uri(
                    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a318ac1f-3ad0-48ac-98cc-79126febcc17/width=1500"
                )
            );
        });

    public static PaintCanvasViewModel PaintCanvasViewModel => DialogFactory.Get<PaintCanvasViewModel>();

    public static ImageSource SampleImageSource =>
        new(
            new Uri(
                "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a318ac1f-3ad0-48ac-98cc-79126febcc17/width=1500"
            )
        )
        {
            Label = "Test Image",
        };

    public static ControlNetCardViewModel ControlNetCardViewModel =>
        DialogFactory.Get<ControlNetCardViewModel>();

    public static ConfirmDeleteDialogViewModel ConfirmDeleteDialogViewModel =>
        DialogFactory.Get<ConfirmDeleteDialogViewModel>(vm =>
        {
            vm.IsRecycleBinAvailable = true;
            vm.PathsToDelete = Enumerable
                .Range(1, 64)
                .Select(i => $"C:/Users/ExampleUser/Data/ExampleFile{i}.txt")
                .ToArray();
        });

    public static ConfirmBulkDownloadDialogViewModel ConfirmBulkDownloadDialogViewModel =>
        DialogFactory.Get<ConfirmBulkDownloadDialogViewModel>(vm =>
        {
            vm.Model = new CivitModel
            {
                Name = "Test Model",
                ModelVersions = Enumerable
                    .Range(1, 64)
                    .Select(i => new CivitModelVersion
                    {
                        Name = $"Version {i}",
                        Files =
                        [
                            new CivitFile
                            {
                                Name = $"test-file-{i}.safetensors",
                                Type = CivitFileType.Model,
                                Metadata = new CivitFileMetadata
                                {
                                    Format = CivitModelFormat.SafeTensor,
                                    Fp = "fp16",
                                    Size = "pruned",
                                },
                                SizeKb = new Random().Next(1, 10) * 1024 * 1024,
                            },
                        ],
                    })
                    .ToList(),
            };

            vm.FpTypePreference = CivitModelFpType.fp16;
            vm.IncludeVae = true;

            return vm;
        });

    public static SponsorshipPromptViewModel SponsorshipPromptViewModel =>
        DialogFactory.Get<SponsorshipPromptViewModel>(vm => { });

    public static OpenArtWorkflowViewModel OpenArtWorkflowViewModel =>
        new(Services.GetRequiredService<ISettingsManager>(), Services.GetRequiredService<IPackageFactory>())
        {
            Workflow = new OpenArtSearchResult
            {
                Name = "Test Workflow",
                Creator = new OpenArtCreator
                {
                    Name = "Test Creator Name",
                    Username = "Test Creator Username",
                },
                Thumbnails =
                [
                    new OpenArtThumbnail
                    {
                        Url = new Uri(
                            "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a318ac1f-3ad0-48ac-98cc-79126febcc17/width=1500"
                        ),
                    },
                ],
                NodesIndex =
                [
                    "Anything Everywhere",
                    "Reroute",
                    "Note",
                    ".",
                    "ComfyUI's ControlNet Auxiliary Preprocessors",
                    "DWPreprocessor",
                    "PixelPerfectResolution",
                    "AIO_Preprocessor",
                    ",",
                    "ComfyUI",
                    "PreviewImage",
                    "CLIPTextEncode",
                    "EmptyLatentImage",
                    "SplitImageWithAlpha",
                    "ControlNetApplyAdvanced",
                    "JoinImageWithAlpha",
                    "LatentUpscaleBy",
                    "VAEEncode",
                    "LoadImage",
                    "ControlNetLoader",
                    "CLIPVisionLoader",
                    "SaveImage",
                    ",",
                    "ComfyUI Impact Pack",
                    "SAMLoader",
                    "UltralyticsDetectorProvider",
                    "FaceDetailer",
                    ",",
                ],
            },
        };

    public static SafetensorMetadataViewModel SafetensorMetadataViewModel =>
        DialogFactory.Get<SafetensorMetadataViewModel>(vm =>
        {
            vm.Metadata = new SafetensorMetadata
            {
                TagFrequency = Enumerable
                    .Range(1, 100)
                    .Select(i => new SafetensorMetadata.Tag("tag" + i, i))
                    .ToList(),
                OtherMetadata = new List<SafetensorMetadata.Metadata>
                {
                    new("Name1", "Value1"),
                    new("Name2", "Value2"),
                    new("Name3", "Value3"),
                },
            };
        });

    public static ModelMetadataEditorDialogViewModel MetadataEditorDialogViewModel =>
        DialogFactory.Get<ModelMetadataEditorDialogViewModel>(vm =>
        {
            vm.ThumbnailFilePath = Assets.NoImage.ToString();
            vm.Tags = "tag1, tag2, tag3";
            vm.ModelDescription = "This is a description";
            vm.ModelName = "Model Name";
            vm.VersionName = "1.0.0";
            vm.TrainedWords = "word1, word2, word3";
            vm.ModelType = CivitModelType.Checkpoint;
            vm.BaseModelType = "Pony";
        });

    public static PackageInstallProgressItemViewModel PackageInstallProgressItemViewModel =>
        new(
            new PackageModificationRunner
            {
                CurrentProgress = new ProgressReport(50, "Installing Package", "Description"),
                ModificationCompleteMessage = "Install Complete",
            }
        )
        {
            Progress = new ContentDialogProgressViewModelBase
            {
                Value = 50,
                CloseWhenFinished = true,
                Text = "Installing Package",
                Description = "Description",
                Console = { Document = new TextDocument("Hello world") },
            },
        };

    public static MockGitVersionProvider MockGitVersionProvider => new();

    public static string CurrentDirectory => Directory.GetCurrentDirectory();

    public static Indexer Types { get; } = new();

    public class Indexer
    {
        private List<Type> types = new();

        public object? this[string typeName]
        {
            get
            {
                var type = Type.GetType(typeName);

                type ??= typeof(DesignData)
                    .Assembly.GetTypes()
                    .FirstOrDefault(t => (t.FullName ?? t.Name).EndsWith(typeName));

                if (type is null)
                {
                    throw new ArgumentException($"Type {typeName} not found");
                }

                try
                {
                    return Services.GetRequiredService(type);
                }
                catch (InvalidOperationException)
                {
                    return Activator.CreateInstance(type);
                }
            }
        }
    }
}