BotGenerator/Generator/BotParser.cs

using System.Collections.Concurrent;
using Common.Models.Input;
using Newtonsoft.Json.Linq;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text.Json;
using System.Threading.Tasks;
using Common.Models.Output;
using Common.Models;
using Generator;
using Generator.Helpers.Gear;

namespace Common.Bots;

public static class BotParser
{
    private static readonly JsonSerializerOptions serialiserOptions = new() { };

    public static List<Bot> Parse(string dumpPath, HashSet<string> botTypes)
    {
        var stopwatch = Stopwatch.StartNew();

        // Build the list of base bot data
        var baseBots = new HashSet<Bot>();
        foreach (var botType in botTypes)
        {
            var typeToAdd = (BotType)Enum.Parse(typeof(BotType), botType);
            baseBots.Add(new Bot(typeToAdd));
        }

        DiskHelpers.CreateDirIfDoesntExist(dumpPath);

        var botFiles = Directory.GetFiles(dumpPath, "*.json", SearchOption.TopDirectoryOnly);
        LoggingHelpers.LogToConsole($"{botFiles.Length.ToString()} bot dump files found");

        // Store a list of parsed bots so we don't parse the same bot twice
        int totalDupeCount = 0;
        var parsedBotIds = new HashSet<string>();
        int i = 0;
        foreach (var filePath in botFiles)
        {
            i++;
            if (i % 500 == 0) Console.WriteLine($"Processing file {i.ToString()}");
            ProcessBotFileSync(baseBots, filePath, parsedBotIds, totalDupeCount);
        }

        // Handle things we can only do once all data has been processed, or only needs to run once per bot type
        foreach (var bot in baseBots)
        {
            // Skip any bots we didn't handle
            if (bot.botCount == 0) continue;

            BaseBotGenerator.AddDifficulties(bot);
            GearChanceHelpers.CalculateModChances(bot);
            GearChanceHelpers.CalculateEquipmentModChances(bot);
            GearChanceHelpers.CalculateEquipmentChances(bot);
            GearChanceHelpers.ApplyModChanceOverrides(bot);
            GearChanceHelpers.ApplyEquipmentChanceOverrides(bot);

            GearHelpers.ReduceAmmoWeightValues(bot);
            GearHelpers.ReduceEquipmentWeightValues(bot.inventory.equipment);
            GearHelpers.ReduceWeightValues(bot.appearance.voice);
            GearHelpers.ReduceWeightValues(bot.appearance.feet);
            GearHelpers.ReduceWeightValues(bot.appearance.body);
            GearHelpers.ReduceWeightValues(bot.appearance.head);
            GearHelpers.ReduceWeightValues(bot.appearance.hands);
        }

        stopwatch.Stop();
        LoggingHelpers.LogToConsole($"{totalDupeCount.ToString()} dupes were ignored. Took {LoggingHelpers.LogTimeTaken(stopwatch.Elapsed.TotalSeconds)} seconds");

        return baseBots.ToList();
    }

    private static void ProcessBotFileSync(
        HashSet<Bot> baseBots,
        string filePath,
        HashSet<string> parsedBotIds,
        int totalDupeCount)
    {
        var splitFilePath = filePath.Split("\\");

        int dupeCount = 0;

        List<Datum> bots = new List<Datum>();
        try
        {
            // Parse the bots inside the json file
            using (var reader = new StreamReader(filePath))
            {
                var deSerialisedObject = JsonSerializer.Deserialize<Root>(reader.ReadToEnd(), serialiserOptions);
                foreach (var botData in deSerialisedObject.data.ToList())
                {
                    // Bot fucks up something, never allow it in
                    if (botData._id == "6483938c53cc9087c70eae86")
                    {
                        Console.WriteLine("oh no");
                        continue;
                    }

                    var role = botData.Info.Settings.Role;
                    var botType = Enum.Parse<BotType>(role, true);
                    
                    Bot baseBot = null;
                    foreach (var bot in baseBots)
                    {
                        if (bot.botType == botType)
                        {
                            baseBot = bot;
                            break;
                        }
                    }

                    if (baseBot == null)
                    {
                        Console.WriteLine($"Skipping {botData._id} due to unknown role {botData.Info.Settings.Role}");
                        continue;
                    }

                    // Add bot if not already added
                    if (!parsedBotIds.Add(botData._id))
                    {
                        dupeCount++;
                    }
                    baseBot.botCount += 1;
                    BaseBotGenerator.UpdateBaseDetails(baseBot, botData);
                    BotGearGenerator.AddGear(baseBot, botData);
                    BotLootGenerator.AddLoot(baseBot, botData);
                    BotChancesGenerator.AddChances(baseBot, botData);
                }
            }
        }
        catch (Exception)
        {
            Console.WriteLine($"File parse fucked up: {filePath}");
            throw;
        }

        totalDupeCount += dupeCount;
    }

    public static async Task<List<Datum>> ParseAsync(string dumpPath, HashSet<string> botTypes)
    {
        var stopwatch = Stopwatch.StartNew();

        DiskHelpers.CreateDirIfDoesntExist(dumpPath);

        var botFiles = Directory.GetFiles(dumpPath, "*.json", SearchOption.TopDirectoryOnly);
        LoggingHelpers.LogToConsole($"{botFiles.Length.ToString()} bot dump files found");

        // key = bot type
        // Store bots keyed against their ID so we never get duplicates
        var parsedBotsDict = new ConcurrentDictionary<string, Datum>();

        int totalDupeCount = 0;
        var tasks = new List<Task>();
        foreach (var filePath in botFiles)
        {
            tasks.Add(ProcessBotFile(botTypes, filePath, parsedBotsDict, totalDupeCount));
        }

        await Task.WhenAll(tasks.ToArray());
        stopwatch.Stop();

        LoggingHelpers.LogToConsole($"Cleaned and Parsed: {parsedBotsDict.Count.ToString()} bots. {totalDupeCount.ToString()} dupes were ignored. Took {LoggingHelpers.LogTimeTaken(stopwatch.Elapsed.TotalSeconds)} seconds");

        return parsedBotsDict.Values.ToList();
    }

    private static async Task<int> ProcessBotFile(
        HashSet<string> botTypes,
        string filePath,
        ConcurrentDictionary<string, Datum> parsedBotsDict,
        int totalDupeCount)
    {
        var splitFilePath = filePath.Split("\\");

        int dupeCount = 0;

        List<Datum> bots = new List<Datum>();
        try
        {
            // Parse the bots inside the json file
            using var reader = new StreamReader(filePath);
            var deSerialisedObject = await JsonSerializer.DeserializeAsync<Root>(reader.BaseStream, serialiserOptions);

            var botTypesLower = new HashSet<string>(botTypes, StringComparer.OrdinalIgnoreCase);
            var filteredBots = new List<Datum>();
            foreach (var botData in deSerialisedObject.data.ToList())
            {
                var roleLower = botData.Info.Settings.Role.ToLower();
                if (botTypesLower.Contains(roleLower))
                {
                    filteredBots.Add(botData);
                }
            }

            bots.AddRange(filteredBots);
        }
        catch (Exception)
        {
            Console.WriteLine($"File parse fucked up: {filePath}");
            throw;
        }

        if (bots == null || bots.Count == 0)
        {
            Console.WriteLine($"Skipping file: {splitFilePath.Last()}. no bots found, ");
            return totalDupeCount;
        }

        //Console.WriteLine($"parsing: {bots.Count} bots in file {splitFilePath.Last()}");
        foreach (var bot in bots)
        {
            // Bot fucks up something, never allow it in
            if (bot._id == "6483938c53cc9087c70eae86")
            {
                Console.WriteLine("oh no");
                continue;
            }

            // null out unnecessary data to save ram
            bot.Stats = null;
            bot.Encyclopedia = null;
            bot.Hideout = null;
            bot.TaskConditionCounters = null;
            bot.Bonuses = null;
            bot.InsuredItems = null;

            // Add bot if not already added
            if (!parsedBotsDict.TryAdd(bot._id, bot))
            {
                dupeCount++;
            }
        }

        totalDupeCount += dupeCount;
        //Console.WriteLine($"Parsed file: {filePath}");
        return totalDupeCount;
    }

    private static IEnumerable<Datum> ParseJson(string json)
    {
        var deSerialisedObject = JsonSerializer.Deserialize<Root>(json, serialiserOptions);
        return deSerialisedObject.data;
    }

    private static string PruneMalformedBsgJson(string json, string fileName)
    {
        // Bsg send json where an item has a location of 1 but it should be an object with x/y/z coords
        var o = JObject.Parse(json);
        var jItemsToReplace = o.SelectTokens("$.data[*].Inventory.items[?(@.location == 1)].location");
        //var jItemsToReplace = o.SelectTokens("$.data[*].Inventory.items[?(@.location == 1 && @.slotId == 'cartridges')].location");

        if (jItemsToReplace != null && jItemsToReplace.Any())
        {
            LoggingHelpers.LogToConsole($"file {fileName} has {jItemsToReplace.Count().ToString()} json issues, cleaning up.", ConsoleColor.Yellow);
            var jItemsToReplaceList = jItemsToReplace.ToList();
            foreach (var item in jItemsToReplaceList)
            {
                var obj = new { x = 1, y = 0, r = 0 };
                item.Replace(JToken.FromObject(obj));
            }
        }
        var returnString = o.ToString();

        o = null;
        jItemsToReplace = null;

        return returnString;
    }
}