ModItemUpdateInitializer.cs

using System;
using System.IO;
using Timberborn.CoreUI;
using Timberborn.MainMenuModdingUI;
using Timberborn.Modding;
using Timberborn.ModdingUI;
using Timberborn.SingletonSystem;
using Timberborn.TooltipSystem;
using UnityEngine;
using UnityEngine.UIElements;
using TimberApi.UIBuilderSystem;
using Mods.SteamUpdateButtons.MainMenuModdingUI;
using Mods.SteamUpdateButtons.Modding;
using Mods.SteamUpdateButtons.ModdingUI;
using Mods.SteamUpdateButtons.SteamWorkshopModDownloading;
using System.Collections.Generic;

namespace Mods.SteamUpdateButtons {
  internal class ModItemUpdateInitializer : ILoadableSingleton, IUpdatableSingleton {

    private readonly UIBuilder _uiBuilder;
    private readonly ModManagerBox _modManagerBox;
    private readonly ModRepository _modRepository;
    private readonly SteamWorkshopModsProvider _steamWorkshopModsProvider;
    private readonly ITooltipRegistrar _tooltipRegistrar;

    private readonly List<ModItem> _updatingMods = [];

    public ModItemUpdateInitializer(UIBuilder uiBuilder,
                                    ModManagerBox modManagerBox,
                                    ModRepository modRepository,
                                    SteamWorkshopModsProvider steamWorkshopModsProvider,
                                    ITooltipRegistrar tooltipRegistrar) {
      _uiBuilder = uiBuilder;
      _modManagerBox = modManagerBox;
      _modRepository = modRepository;
      _steamWorkshopModsProvider = steamWorkshopModsProvider;
      _tooltipRegistrar = tooltipRegistrar;
    }

    public void Load() {
      foreach (var kv in _modManagerBox.GetModListView().GetModItems()) {
        if (kv.Key.ModDirectory.IsUserMod) {
          continue;
        }
        Initialize(kv.Value);
      }
    }

    public void UpdateSingleton() {
      foreach (var modItem in _updatingMods) {
        if (_steamWorkshopModsProvider.GetDownloadProgress(modItem.Mod.ModDirectory, out var downloaded, out var total) && total > 0) {
          var version = modItem.ModManifest.Version.Formatted;
          version += $" → {(float)downloaded / total:0%}";
          modItem.Root.Q<Label>("ModVersion").text = version;
        }
      }
    }

    private void Initialize(ModItem modItem) {
      var unavailableImage = _uiBuilder.Build<UnavailableImage>("UnavailableImage");
      modItem.Root.Add(unavailableImage);
      unavailableImage.RegisterCallback<AttachToPanelEvent>(
          _ => unavailableImage.ToggleDisplayStyle(
              !_steamWorkshopModsProvider.IsAvailable(modItem.Mod.ModDirectory)));
      var downloadPendingImage = _uiBuilder.Build<DownloadPendingImage>("DownloadPendingImage");
      modItem.Root.Add(downloadPendingImage);
      downloadPendingImage.RegisterCallback<AttachToPanelEvent>(
          _ => downloadPendingImage.ToggleDisplayStyle(
              _steamWorkshopModsProvider.IsDownloadPending(modItem.Mod.ModDirectory)));
      var button = _uiBuilder.Build<UpdateButton>("UpdateModButton");
      _tooltipRegistrar.RegisterLocalizable(button, "SteamUpdateButtons.UpdateMod");
      modItem.Root.Add(button);
      button.RegisterCallback<AttachToPanelEvent>(
          _ => button.ToggleDisplayStyle(
              !_steamWorkshopModsProvider.IsDownloadPending(modItem.Mod.ModDirectory) &&
              _steamWorkshopModsProvider.IsUpdatable(modItem.Mod.ModDirectory)));
      button.RegisterCallback<ClickEvent>(ce => {
        UpdateMod(modItem);
      });
    }

    public void UpdateMod(ModItem modItem) {
      Debug.Log(DateTime.Now.ToString("HH:mm:ss ") + "Steam Update Buttons: Updating: " + modItem.Mod.DisplayName);
      if (_steamWorkshopModsProvider.UpdateModDirectory(modItem.Mod.ModDirectory, r => {
        Debug.Log(DateTime.Now.ToString("HH:mm:ss ") + "Steam Update Buttons: No longer updating: " + modItem.Mod.DisplayName);
        _updatingMods.Remove(modItem);
        _modManagerBox.GetModListView().OnModToggled(this, EventArgs.Empty);  // show restartWarning
        Update(modItem);
      })) {
        var downloadPendingImage = modItem.Root.Q<VisualElement>("DownloadPendingImage");
        var button = modItem.Root.Q<VisualElement>("UpdateModButton");
        button.ToggleDisplayStyle(false);
        downloadPendingImage.ToggleDisplayStyle(true);
        _updatingMods.Add(modItem);
      }
    }

    private void Update(ModItem modItem) {
      var modDirectory = modItem.Mod.ModDirectory;
      var directory = new ModDirectory(
        new DirectoryInfo(modDirectory.OriginPath),
        modDirectory.IsUserMod,
        modDirectory.DisplaySource,
        modDirectory.GameVersion,
        isSubdirectory: false);
      if (_modRepository.TryGetModDirectory(directory, out var versionedDirectory)) {
        directory = versionedDirectory;
      }
      if (_steamWorkshopModsProvider.TryLoadModManifest(directory, out var manifest)) {
        var version = modItem.ModManifest.Version.Formatted;
        version += " → " + manifest.Version.Formatted;
        modItem.Root.Q<Label>("ModVersion").text = version;
      }
      var unavailableImage = modItem.Root.Q<VisualElement>("UnavailableImage");
      unavailableImage.ToggleDisplayStyle(
          !_steamWorkshopModsProvider.IsAvailable(modItem.Mod.ModDirectory));
      var downloadPendingImage = modItem.Root.Q<VisualElement>("DownloadPendingImage");
      bool downloadPending = _steamWorkshopModsProvider.IsDownloadPending(modItem.Mod.ModDirectory);
      downloadPendingImage.ToggleDisplayStyle(
          downloadPending);
      var button = modItem.Root.Q<VisualElement>("UpdateModButton");
      button.ToggleDisplayStyle(
          !downloadPending &&
          _steamWorkshopModsProvider.IsUpdatable(modItem.Mod.ModDirectory));
    }
  }
}