wabbajack/Wabbajack.DTOs/Game/GameRegistry.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Wabbajack.Paths;

namespace Wabbajack.DTOs
{
    public static class GameRegistry
    {
        public static IReadOnlyDictionary<Game, GameMetaData> Games = new Dictionary<Game, GameMetaData>
        {
            {
                Game.Morrowind, new GameMetaData
                {
                    Game = Game.Morrowind,
                    SteamIDs = new[] { 22320 },
                    GOGIDs = new[] { 1440163901, 1435828767 },
                    NexusName = "morrowind",
                    NexusGameId = 100,
                    MO2Name = "Morrowind",
                    MO2ArchiveName = "morrowind",
                    BethNetID = 31,
                    RegString =
                        @"SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall\The Elder Scrolls III: Morrowind Game of the Year Edition",
                    RequiredFiles = new[]
                    {
                        "Morrowind.exe".ToRelativePath()
                    },
                    MainExecutable = "Morrowind.exe".ToRelativePath()
                }
            },
            {
                Game.Oblivion, new GameMetaData
                {
                    Game = Game.Oblivion,
                    NexusName = "oblivion",
                    NexusGameId = 101,
                    MO2Name = "Oblivion",
                    MO2ArchiveName = "oblivion",
                    SteamIDs = new[] { 22330 },
                    GOGIDs = new[] { 1458058109 },
                    RequiredFiles = new[]
                    {
                        "oblivion.exe".ToRelativePath()
                    },
                    MainExecutable = "Oblivion.exe".ToRelativePath()
                }
            },

            {
                Game.Fallout3, new GameMetaData
                {
                    Game = Game.Fallout3,
                    NexusName = "fallout3",
                    NexusGameId = 120,
                    MO2Name = "Fallout 3",
                    MO2ArchiveName = "fallout3",
                    SteamIDs = new[] { 22300, 22370 }, // base game and GotY
                    GOGIDs = new[] { 1454315831 }, // GotY edition
                    RequiredFiles = new[]
                    {
                        "Fallout3.exe".ToRelativePath()
                    },
                    MainExecutable = "Fallout3.exe".ToRelativePath()
                }
            },
            {
                Game.FalloutNewVegas, new GameMetaData
                {
                    Game = Game.FalloutNewVegas,
                    NexusName = "newvegas",
                    NexusGameId = 130,
                    MO2Name = "New Vegas",
                    MO2ArchiveName = "falloutnv",
                    SteamIDs = new[] { 22380, 22490 }, // normal and RU version
                    GOGIDs = new[] { 1454587428 },
                    RequiredFiles = new[]
                    {
                        "FalloutNV.exe".ToRelativePath()
                    },
                    MainExecutable = "FalloutNV.exe".ToRelativePath()
                }
            },
            {
                Game.Skyrim, new GameMetaData
                {
                    Game = Game.Skyrim,
                    NexusName = "skyrim",
                    NexusGameId = 110,
                    MO2Name = "Skyrim",
                    MO2ArchiveName = "skyrim",
                    SteamIDs = new[] { 72850 },
                    RequiredFiles = new[]
                    {
                        "tesv.exe".ToRelativePath()
                    },
                    MainExecutable = "TESV.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.SkyrimSpecialEdition, Game.SkyrimVR }
                }
            },
            {
                Game.SkyrimSpecialEdition, new GameMetaData
                {
                    Game = Game.SkyrimSpecialEdition,
                    NexusName = "skyrimspecialedition",
                    NexusGameId = 1704,
                    MO2Name = "Skyrim Special Edition",
                    MO2ArchiveName = "skyrimse",
                    SteamIDs = new[] { 489830 },
                    RequiredFiles = new[]
                    {
                        "SkyrimSE.exe".ToRelativePath()
                    },
                    MainExecutable = "SkyrimSE.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.Skyrim, Game.SkyrimVR }
                }
            },
            {
                Game.Fallout4, new GameMetaData
                {
                    Game = Game.Fallout4,
                    NexusName = "fallout4",
                    NexusGameId = 1151,
                    MO2Name = "Fallout 4",
                    MO2ArchiveName = "fallout4",
                    SteamIDs = new[] { 377160 },
                    RequiredFiles = new[]
                    {
                        "Fallout4.exe".ToRelativePath()
                    },
                    MainExecutable = "Fallout4.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.Fallout4VR }
                }
            },
            {
                Game.SkyrimVR, new GameMetaData
                {
                    Game = Game.SkyrimVR,
                    NexusName = "skyrimspecialedition",
                    NexusGameId = 1704,
                    MO2Name = "Skyrim VR",
                    MO2ArchiveName = "skyrimse",
                    SteamIDs = new[] { 611670 },
                    RequiredFiles = new[]
                    {
                        "SkyrimVR.exe".ToRelativePath()
                    },
                    MainExecutable = "SkyrimVR.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.Skyrim, Game.SkyrimSpecialEdition },
                    CanSourceFrom = new[] { Game.SkyrimSpecialEdition }
                }
            },
            {
                Game.Enderal, new GameMetaData
                {
                    Game = Game.Enderal,
                    NexusName = "enderal",
                    NexusGameId = 2736,
                    MO2Name = "Enderal",
                    MO2ArchiveName = "enderal",
                    SteamIDs = new[] { 1027920, 933480 },
                    RequiredFiles = new[]
                    {
                        "TESV.exe".ToRelativePath()
                    },
                    MainExecutable = "TESV.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.EnderalSpecialEdition }
                }
            },
            {
                Game.EnderalSpecialEdition, new GameMetaData
                {
                    Game = Game.EnderalSpecialEdition,
                    NexusName = "enderalspecialedition",
                    NexusGameId = 3685,
                    MO2Name = "Enderal Special Edition",
                    MO2ArchiveName = "enderalse",
                    SteamIDs = new[] { 976620 },
                    RequiredFiles = new[]
                    {
                        "SkyrimSE.exe".ToRelativePath()
                    },
                    MainExecutable = "SkyrimSE.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.Enderal }
                }
            },
            {
                Game.Fallout4VR, new GameMetaData
                {
                    Game = Game.Fallout4VR,
                    NexusName = "fallout4",
                    MO2Name = "Fallout 4 VR",
                    MO2ArchiveName = "Fallout4",
                    SteamIDs = new[] { 611660 },
                    RequiredFiles = new[]
                    {
                        "Fallout4VR.exe".ToRelativePath()
                    },
                    MainExecutable = "Fallout4VR.exe".ToRelativePath(),
                    CommonlyConfusedWith = new[] { Game.Fallout4 },
                    CanSourceFrom = new[] { Game.Fallout4 }
                }
            },
            {
                Game.DarkestDungeon, new GameMetaData
                {
                    Game = Game.DarkestDungeon,
                    NexusName = "darkestdungeon",
                    MO2Name = "Darkest Dungeon",
                    NexusGameId = 804,
                    SteamIDs = new[] { 262060 },
                    GOGIDs = new[] { 1450711444 },
                    EpicGameStoreIDs = new[] { "b4eecf70e3fe4e928b78df7855a3fc2d" },
                    IsGenericMO2Plugin = true,
                    RequiredFiles = new[]
                    {
                        @"_windowsnosteam\Darkest.exe".ToRelativePath()
                    },
                    MainExecutable = @"_windowsnosteam\Darkest.exe".ToRelativePath()
                }
            },
            {
                Game.Dishonored, new GameMetaData
                {
                    Game = Game.Dishonored,
                    NexusName = "dishonored",
                    MO2Name = "Dishonored",
                    NexusGameId = 802,
                    SteamIDs = new[] { 205100 },
                    GOGIDs = new[] { 1701063787 },
                    RequiredFiles = new[]
                    {
                        @"Binaries\Win32\Dishonored.exe".ToRelativePath()
                    },
                    MainExecutable = @"Binaries\Win32\Dishonored.exe".ToRelativePath()
                }
            },
            {
                Game.Witcher3, new GameMetaData
                {
                    Game = Game.Witcher3,
                    NexusName = "witcher3",
                    NexusGameId = 952,
                    MO2Name = "The Witcher 3: Wild Hunt",
                    SteamIDs = new[] { 292030, 499450 }, // normal and GotY
                    GOGIDs = new[]
                        { 1207664643, 1495134320, 1207664663, 1640424747 }, // normal, GotY and both in packages
                    RequiredFiles = new[]
                    {
                        @"bin\x64\witcher3.exe".ToRelativePath()
                    },
                    MainExecutable = @"bin\x64\witcher3.exe".ToRelativePath()
                }
            },
            {
                Game.StardewValley, new GameMetaData
                {
                    Game = Game.StardewValley,
                    NexusName = "stardewvalley",
                    MO2Name = "Stardew Valley",
                    NexusGameId = 1303,
                    SteamIDs = new[] { 413150 },
                    GOGIDs = new[] { 1453375253 },
                    IsGenericMO2Plugin = true,
                    RequiredFiles = new[]
                    {
                        "Stardew Valley.exe".ToRelativePath()
                    },
                    MainExecutable = "Stardew Valley.exe".ToRelativePath()
                }
            },
            {
                Game.KingdomComeDeliverance, new GameMetaData
                {
                    Game = Game.KingdomComeDeliverance,
                    NexusName = "kingdomcomedeliverance",
                    MO2Name = "Kingdom Come: Deliverance",
                    MO2ArchiveName = "kingdomcomedeliverance",
                    NexusGameId = 2298,
                    SteamIDs = new[] { 379430 },
                    GOGIDs = new[] { 1719198803 },
                    IsGenericMO2Plugin = true,
                    RequiredFiles = new[]
                    {
                        @"bin\Win64\KingdomCome.exe".ToRelativePath()
                    },
                    MainExecutable = @"bin\Win64\KingdomCome.exe".ToRelativePath()
                }
            },
            {
                Game.MechWarrior5Mercenaries, new GameMetaData
                {
                    Game = Game.MechWarrior5Mercenaries,
                    NexusName = "mechwarrior5mercenaries",
                    MO2Name = "Mechwarrior 5: Mercenaries",
                    MO2ArchiveName = "mechwarrior5mercenaries",
                    NexusGameId = 3099,
                    EpicGameStoreIDs = new[] { "9fd39d8ac72946a2a10a887ce86e6c35" },
                    IsGenericMO2Plugin = true,
                    RequiredFiles = new[]
                    {
                        @"MW5Mercs\Binaries\Win64\MechWarrior-Win64-Shipping.exe".ToRelativePath()
                    },
                    MainExecutable = @"MW5Mercs\Binaries\Win64\MechWarrior-Win64-Shipping.exe".ToRelativePath()
                }
            },
            {
                Game.NoMansSky, new GameMetaData
                {
                    Game = Game.NoMansSky,
                    NexusName = "nomanssky",
                    NexusGameId = 1634,
                    MO2Name = "No Man's Sky",
                    SteamIDs = new[] { 275850 },
                    GOGIDs = new[] { 1446213994 },
                    RequiredFiles = new[]
                    {
                        @"Binaries\NMS.exe".ToRelativePath()
                    },
                    MainExecutable = @"Binaries\NMS.exe".ToRelativePath()
                }
            },
            {
                Game.DragonAgeOrigins, new GameMetaData
                {
                    Game = Game.DragonAgeOrigins,
                    NexusName = "dragonage",
                    NexusGameId = 140,
                    MO2Name = "Dragon Age: Origins",
                    SteamIDs = new[] { 47810 },
                    OriginIDs = new[] { "DR:169789300", "DR:208591800" },
                    GOGIDs = new[] { 1949616134 },
                    RequiredFiles = new[]
                    {
                        @"bin_ship\daorigins.exe".ToRelativePath()
                    },
                    MainExecutable = @"bin_ship\daorigins.exe".ToRelativePath()
                }
            },
            {
                Game.DragonAge2, new GameMetaData
                {
                    Game = Game.DragonAge2,
                    NexusName = "dragonage2",
                    NexusGameId = 141,
                    MO2Name = "Dragon Age 2", // Probably wrong
                    SteamIDs = new[] { 1238040 },
                    OriginIDs = new[] { "OFB-EAST:59474", "DR:201797000" },
                    RequiredFiles = new[]
                    {
                        @"bin_ship\DragonAge2.exe".ToRelativePath()
                    },
                    MainExecutable = @"bin_ship\DragonAge2.exe".ToRelativePath()
                }
            },
            {
                Game.DragonAgeInquisition, new GameMetaData
                {
                    Game = Game.DragonAgeInquisition,
                    NexusName = "dragonageinquisition",
                    NexusGameId = 728,
                    MO2Name = "Dragon Age: Inquisition", // Probably wrong
                    SteamIDs = new[] { 1222690 },
                    OriginIDs = new[] { "OFB-EAST:51937", "OFB-EAST:1000032" },
                    RequiredFiles = new[]
                    {
                        @"DragonAgeInquisition.exe".ToRelativePath()
                    },
                    MainExecutable = @"DragonAgeInquisition.exe".ToRelativePath()
                }
            },
            {
                Game.KerbalSpaceProgram, new GameMetaData
                {
                    Game = Game.KerbalSpaceProgram,
                    NexusName = "kerbalspaceprogram",
                    MO2Name = "Kerbal Space Program",
                    NexusGameId = 272,
                    SteamIDs = new[] { 220200 },
                    GOGIDs = new[] { 1429864849 },
                    IsGenericMO2Plugin = true,
                    RequiredFiles = new[]
                    {
                        @"KSP_x64.exe".ToRelativePath()
                    },
                    MainExecutable = @"KSP_x64.exe".ToRelativePath()
                }
            }
        };

        public static ILookup<string?, GameMetaData> ByNexusName = Games.Values.ToLookup(g => g.NexusName ?? "");

        public static GameMetaData? GetByMO2ArchiveName(string gameName)
        {
            gameName = gameName.ToLower();
            return Games.Values.FirstOrDefault(g => g.MO2ArchiveName?.ToLower() == gameName);
        }

        public static GameMetaData? GetByNexusName(string gameName)
        {
            return Games.Values.FirstOrDefault(g => g.NexusName == gameName.ToLower());
        }

        public static GameMetaData? GetBySteamID(int id)
        {
            return Games.Values
                .FirstOrDefault(g => g.SteamIDs.Length > 0 && g.SteamIDs.Any(i => i == id));
        }
        
        /// <summary>
        /// Parse game data from an arbitrary string. Tries first via parsing as a game Enum, then by Nexus name.
        /// <param nambe="someName">Name to query</param>
        /// <returns>GameMetaData found</returns>
        /// <exception cref="ArgumentNullException">If string could not be translated to a game</exception>
        /// </summary>
        public static GameMetaData GetByFuzzyName(string someName)
        {
            return TryGetByFuzzyName(someName) ?? throw new ArgumentNullException(nameof(someName), $"\"{someName}\" could not be translated to a game!");
        }
        
        private static GameMetaData? GetByMO2Name(string gameName)
        {
            gameName = gameName.ToLower();
            return Games.Values.FirstOrDefault(g => g.MO2Name?.ToLower() == gameName);
        }
        
        /// <summary>
        /// Tries to parse game data from an arbitrary string. Tries first via parsing as a game Enum, then by Nexus name.
        /// <param nambe="someName">Name to query</param>
        /// <returns>GameMetaData if found, otherwise null</returns>
        /// </summary>
        public static GameMetaData? TryGetByFuzzyName(string someName)
        {
            if (Enum.TryParse(typeof(Game), someName, true, out var metadata)) return ((Game)metadata!).MetaData();

            var result = GetByNexusName(someName);
            if (result != null) return result;

            result = GetByMO2ArchiveName(someName);
            if (result != null) return result;

            result = GetByMO2Name(someName);
            if (result != null) return result;


            return int.TryParse(someName, out int id) ? GetBySteamID(id) : null;
        }
        
        public static bool TryGetByFuzzyName(string someName, out GameMetaData gameMetaData)
        {
            var result = TryGetByFuzzyName(someName);
            if (result == null)
            {
                gameMetaData = Games.Values.First();
                return false;
            }

            gameMetaData = result;
            return true;
        }

        public static GameMetaData MetaData(this Game game)
        {
            return Games[game];
        }
    }
}