DrakiaXYZ/BotGenerator
0
0
Fork 0
forked from chomp/BotGenerator
Code Pull Requests Activity

cleanup of file structure

Browse Source
This commit is contained in:
Chomp 2021-08-24 23:04:26 +01:00
parent a155b366d1
commit d0b6718c5c
2 changed files with 51 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
PMCGenerator/Models/Presets.cs
View File

@ -1,6 +1,4 @@
using System;
using System.Collections.Generic;
using System.Text;
using System.Collections.Generic;
namespace PMCGenerator
{

68
PMCGenerator/Program.cs
View File

@ -2,7 +2,6 @@
using Common.Extensions;
using Newtonsoft.Json;
using PMCGenerator.Models;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
@ -14,30 +13,24 @@ namespace PMCGenerator
{
static void Main(string[] args)
{
// take a list of preset files and spit out a drop-in-replacement mod list
var presetPath = CreateInputFolder();
var presetFiles = GetPresetFileList(presetPath);
// Parse into list of strongly typed objects
var parsedPresets = new List<Presets>();
foreach (var presetFile in presetFiles)
{
var json = File.ReadAllText(presetFile);
var parsedFile = JsonConvert.DeserializeObject<Presets>(json);
parsedPresets.Add(parsedFile);
}
var parsedPresets = ParsePresetJsons(presetFiles);
// Create flat lists of weapons + list of mods
var flatWeaponsList = GetWeaponsFromRawFile(parsedPresets);
var flatPrimaryWeaponsList = GetWeaponsFromRawFile(parsedPresets);
var flatSecondaryWeaponsList = GetSecondaryWeaponsFromRawFile(parsedPresets);
var flatModList = GetModsFromRawFile(parsedPresets);
// Add weapon mods to output
var output = new { FirstPrimaryWeapon = new List<string>(), mods = new Dictionary<string, Dictionary<string, List<string>>>() };
output.FirstPrimaryWeapon.AddRange(flatWeaponsList.Select(x => x.TemplateId).Distinct());
var output = new { FirstPrimaryWeapon = new List<string>(), Holster = new List<string>(), mods = new Dictionary<string, Dictionary<string, List<string>>>() };
output.FirstPrimaryWeapon.AddRange(flatPrimaryWeaponsList.Select(x => x.TemplateId).Distinct());
output.Holster.AddRange(flatSecondaryWeaponsList.Select(x => x.TemplateId).Distinct());
foreach (var weapon in flatWeaponsList)
foreach (var weapon in flatPrimaryWeaponsList)
{
// add weapon if its not already here
if (!output.mods.ContainsKey(weapon.TemplateId))
@ -69,7 +62,7 @@ namespace PMCGenerator
// Get mods where parent is not weapon and add to output
foreach (var mod in flatModList.Where(x => x.ParentId != null
&& !flatWeaponsList.Any(y => y.Id == x.ParentId)).ToList())
&& !flatPrimaryWeaponsList.Any(y => y.Id == x.ParentId)).ToList())
{
// No parent tempalte id found, create and add mods details
if (!output.mods.ContainsKey(mod.ParentTemplateId))
@ -101,6 +94,26 @@ namespace PMCGenerator
CreateJsonFile(outputPath, outputJson);
}
private static List<Presets> ParsePresetJsons(List<string> presetFiles)
{
var result = new List<Presets>();
foreach (var presetFile in presetFiles)
{
var json = File.ReadAllText(presetFile);
var parsedFile = JsonConvert.DeserializeObject<Presets>(json);
result.Add(parsedFile);
}
int count = 0;
foreach (var file in result)
{
count += file.weaponbuilds.Count;
}
LoggingHelpers.LogToConsole($"{count} presets parsed");
return result;
}
private static List<ModDetails> GetModsFromRawFile(List<Presets> parsedPresets)
{
List <ModDetails> result = new List<ModDetails>();
@ -120,7 +133,10 @@ namespace PMCGenerator
}
Module parentMod = GetModsParent(file, mod.parentId);
result.Add(new ModDetails(mod.slotId, mod._id, mod._tpl, mod.parentId, parentMod._tpl));
if (parentMod != null)
{
result.Add(new ModDetails(mod.slotId, mod._id, mod._tpl, mod.parentId, parentMod._tpl));
}
}
}
}
@ -152,7 +168,7 @@ namespace PMCGenerator
var result = new List<WeaponDetails>();
foreach (var file in parsedPresets)
{
foreach (var item in file.weaponbuilds)
foreach (var item in file.weaponbuilds.Where(x=> !x.Key.Contains("pistol")))
{
Weapon weapon = item.Value;
result.Add(new WeaponDetails(item.Key, weapon.items[0]._id, weapon.items[0]._tpl));
@ -163,6 +179,22 @@ namespace PMCGenerator
return result;
}
private static List<WeaponDetails> GetSecondaryWeaponsFromRawFile(List<Presets> parsedPresets)
{
var result = new List<WeaponDetails>();
foreach (var file in parsedPresets)
{
foreach (var item in file.weaponbuilds.Where(x => x.Key.Contains("pistol")))
{
Weapon weapon = item.Value;
result.Add(new WeaponDetails(item.Key, weapon.items[0]._id, weapon.items[0]._tpl));
}
}
return result;
}
// Write json to a file
private static void CreateJsonFile(string outputPath, string outputJson)
{
@ -177,7 +209,7 @@ namespace PMCGenerator
private static List<string> GetPresetFileList(string presetPath)
{
var presetFiles = Directory.GetFiles(presetPath, "*.json", SearchOption.TopDirectoryOnly).ToList();
Console.WriteLine($"{presetFiles.Count} preset files found");
LoggingHelpers.LogToConsole($"{presetFiles.Count} preset files found");
return presetFiles;
}