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()
            }
        },
        {
            Game.Terraria, new GameMetaData
            {
                Game = Game.Terraria,
                SteamIDs = new[] {1281930},
                MO2Name = "Terraria",
                IsGenericMO2Plugin = true,
                RequiredFiles = new[]
                {
                    @"tModLoader.exe".ToRelativePath()
                },
                MainExecutable = @"tModLoader.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 var 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];
    }
}