BeatSaber-IPA-Reloaded/BSIPA-ModList/UI/ViewControllers/ModListController.cs

using System;
using System.Linq;
using System.Collections.Generic;
using CustomUI.BeatSaber;
using CustomUI.Utilities;
using HMUI;
using IPA.Loader;
using IPA.Old;
using UnityEngine;
using VRUI;
using IPA.Loader.Features;
using TMPro;

namespace BSIPA_ModList.UI
{
    internal class ModListController : CustomListViewController
    {
        private interface IClickableCell
        {
            void OnSelect(ModListController cntrl);
        }

        private class BSIPAModCell : CustomCellInfo, IClickableCell
        {
            private static Sprite _defaultIcon;
            public static Sprite DefaultIcon
            {
                get
                {
                    if (_defaultIcon == null)
                        _defaultIcon = UIUtilities.LoadSpriteFromResources("BSIPA_ModList.Icons.mod_bsipa.png");
                    return _defaultIcon;
                }
            }

            internal PluginLoader.PluginInfo Plugin;
            private ModListController list;

            public BSIPAModCell(ModListController list, PluginLoader.PluginInfo plugin) 
                : base($"{plugin.Metadata.Name} <size=60%>v{plugin.Metadata.Version}", plugin.Metadata.Manifest.Author, null)
            {
                Plugin = plugin;
                this.list = list;

                if (string.IsNullOrWhiteSpace(subtext))
                    subtext = "<color=#BFBFBF><i>Unspecified Author</i>";

                if (plugin.Metadata.Manifest.IconPath != null)
                {
                    try
                    {
                        icon = UIUtilities.LoadSpriteRaw(UIUtilities.GetResource(plugin.Metadata.Assembly, plugin.Metadata.Manifest.IconPath));
                    }
                    catch (Exception e)
                    {
                        Logger.log.Error($"Error loading icon for {plugin.Metadata.Name}");
                        Logger.log.Error(e);
                    }
                }

                if (icon == null)
                    icon = DefaultIcon;

                Logger.log.Debug($"BSIPAModCell {plugin.Metadata.Name} {plugin.Metadata.Version}");
            }

            private ModInfoViewController infoView;

            public void OnSelect(ModListController cntrl)
            {
                Logger.log.Debug($"Selected BSIPAModCell {Plugin.Metadata.Name} {Plugin.Metadata.Version}");

                if (infoView == null)
                {
                    var desc = Plugin.Metadata.Manifest.Description;
                    if (string.IsNullOrWhiteSpace(desc))
                        desc = "<color=#BFBFBF><i>No description</i>";

                    infoView = BeatSaberUI.CreateViewController<ModInfoViewController>();
                    infoView.Init(icon, Plugin.Metadata.Name, "v" + Plugin.Metadata.Version.ToString(), subtext,
                        desc, Plugin.Metadata.Features.FirstOrDefault(f => f is NoUpdateFeature) != null ? Plugin.Metadata : null);
                }

                list.flow.SetSelected(infoView, immediate: list.flow.HasSelected);
            }
        }

        private class BSIPAIgnoredModCell : CustomCellInfo, IClickableCell
        {
            internal PluginLoader.PluginMetadata Plugin;
            private ModListController list;

            private const string authorFormat = "{0}  <color=#BFBFBF>- <i>Not loaded</i>";

            private string authorText;

            public BSIPAIgnoredModCell(ModListController list, PluginLoader.PluginMetadata plugin)
                : base($"<color=#878787>{plugin.Name} <size=60%>v{plugin.Version}", "", BSIPAModCell.DefaultIcon)
            {
                Plugin = plugin;
                this.list = list;

                if (string.IsNullOrWhiteSpace(plugin.Manifest.Author))
                    authorText = "<color=#BFBFBF><i>Unspecified Author</i>";
                else
                    authorText = plugin.Manifest.Author;
                subtext = string.Format(authorFormat, authorText);

                Logger.log.Debug($"BSIPAIgnoredModCell {plugin.Name} {plugin.Version}");
            }

            private ModInfoViewController infoView;

            public void OnSelect(ModListController cntrl)
            {
                Logger.log.Debug($"Selected BSIPAIgnoredModCell {Plugin.Name} {Plugin.Version}");

                if (infoView == null)
                {
                    var desc = Plugin.Manifest.Description;
                    if (string.IsNullOrWhiteSpace(desc))
                        desc = "<color=#BFBFBF><i>No description</i>";

                    infoView = BeatSaberUI.CreateViewController<ModInfoViewController>();
                    infoView.Init(icon, Plugin.Name, "v" + Plugin.Version.ToString(), authorText,
                        desc, Plugin.Features.FirstOrDefault(f => f is NoUpdateFeature) != null ? Plugin : null);
                }

                list.flow.SetSelected(infoView, immediate: list.flow.HasSelected);
            }
        }
        private class LibraryModCell : CustomCellInfo, IClickableCell
        {
            private static Sprite _defaultIcon;
            public static Sprite DefaultIcon
            {
                get
                {
                    if (_defaultIcon == null)
                        _defaultIcon = UIUtilities.LoadSpriteFromResources("BSIPA_ModList.Icons.library.png");
                    return _defaultIcon;
                }
            }

            internal PluginLoader.PluginInfo Plugin;
            private ModListController list;

            public LibraryModCell(ModListController list, PluginLoader.PluginInfo plugin)
                : base($"{plugin.Metadata.Name} <size=60%>v{plugin.Metadata.Version}", plugin.Metadata.Manifest.Author, null)
            {
                Plugin = plugin;
                this.list = list;

                if (string.IsNullOrWhiteSpace(subtext))
                    subtext = "<color=#BFBFBF><i>Unspecified Author</i>";

                icon = DefaultIcon;

                Logger.log.Debug($"LibraryModCell {plugin.Metadata.Name} {plugin.Metadata.Version}");
            }

            private ModInfoViewController infoView;

            public void OnSelect(ModListController cntrl)
            {
                Logger.log.Debug($"Selected LibraryModCell {Plugin.Metadata.Name} {Plugin.Metadata.Version}");

                if (infoView == null)
                {
                    var desc = Plugin.Metadata.Manifest.Description;
                    if (string.IsNullOrWhiteSpace(desc))
                        desc = "<color=#BFBFBF><i>No description</i>";

                    infoView = BeatSaberUI.CreateViewController<ModInfoViewController>();
                    infoView.Init(icon, Plugin.Metadata.Name, "v" + Plugin.Metadata.Version.ToString(), subtext,
                        desc, Plugin.Metadata.Features.FirstOrDefault(f => f is NoUpdateFeature) != null ? Plugin.Metadata : null);
                }

                list.flow.SetSelected(infoView, immediate: list.flow.HasSelected);
            }
        }

#pragma warning disable CS0618
        private class IPAModCell : CustomCellInfo, IClickableCell
        {
            private static Sprite _defaultIcon;
            public static Sprite DefaultIcon
            {
                get
                {
                    if (_defaultIcon == null)
                        _defaultIcon = UIUtilities.LoadSpriteFromResources("BSIPA_ModList.Icons.mod_ipa.png");
                    return _defaultIcon;
                }
            }

            internal IPlugin Plugin;
            private ModListController list;

            public IPAModCell(ModListController list, IPlugin plugin)
                : base($"{plugin.Name} <size=60%>{plugin.Version}", "<color=#BFBFBF><i>Legacy</i>", DefaultIcon)
            {
                Plugin = plugin;
                this.list = list;

                Logger.log.Debug($"IPAModCell {plugin.Name} {plugin.Version}");
            }

            private ModInfoViewController infoView;

            public void OnSelect(ModListController cntrl)
            {
                Logger.log.Debug($"Selected IPAModCell {Plugin.Name} {Plugin.Version}");

                if (infoView == null)
                {
                    infoView = BeatSaberUI.CreateViewController<ModInfoViewController>();
                    infoView.Init(icon, Plugin.Name, "v" + Plugin.Version.ToString(), "<color=#BFBFBF><i>Unknown Author</i>",
                        "<color=#A0A0A0>This mod was written for IPA Reloaded. No metadata is avaliable for this mod. " +
                        "Please contact the mod author and ask them to port it to BSIPA to provide more information.", null);
                }

                list.flow.SetSelected(infoView, immediate: list.flow.HasSelected);
            }
        }
#pragma warning restore

        public override TableCell CellForIdx(int idx)
        {
            var cell = base.CellForIdx(idx) as LevelListTableCell;
            var nameText = cell.GetPrivateField<TextMeshProUGUI>("_songNameText");
            nameText.overflowMode = TextOverflowModes.Overflow;
            return cell;
        }

        private ModListFlowCoordinator flow;

#pragma warning disable CS0618
        public void Init(ModListFlowCoordinator flow, IEnumerable<PluginLoader.PluginInfo> bsipaPlugins, IEnumerable<PluginLoader.PluginMetadata> ignoredPlugins, IEnumerable<IPlugin> ipaPlugins)
        {
            Logger.log.Debug("List Controller Init");

            DidActivateEvent = DidActivate;
            DidSelectRowEvent = DidSelectRow;

            rectTransform.anchorMin = new Vector2(0f, 0f);
            rectTransform.anchorMax = new Vector2(.4f, 1f);

            includePageButtons = true;
            this.flow = flow;

            reuseIdentifier = "BSIPAModListTableCell";

            foreach (var plugin in bsipaPlugins.Where(p => !p.Metadata.IsBare))
                Data.Add(new BSIPAModCell(this, plugin));
            foreach (var plugin in ignoredPlugins)
                Data.Add(new BSIPAIgnoredModCell(this, plugin));
            foreach (var plugin in bsipaPlugins.Where(p => p.Metadata.IsBare))
                Data.Add(new LibraryModCell(this, plugin));
            foreach (var plugin in ipaPlugins)
                Data.Add(new IPAModCell(this, plugin));
        }

#pragma warning restore

        private void DidSelectRow(TableView view, int index)
        {
            Debug.Assert(ReferenceEquals(view.dataSource, this));
            (Data[index] as IClickableCell)?.OnSelect(this);
        }

        private new void DidActivate(bool first, ActivationType type)
        {
            var rt = _customListTableView.transform as RectTransform;
            rt.anchorMin = new Vector2(.1f, 0f);
            rt.anchorMax = new Vector2(.9f, 1f);
        }
    }
}