public static bool DoesSheetContainImages(CastleDBParser.RootNode rootNode, CastleDBParser.SheetNode sheetNode)
{
Debug.Log(sheetNode.Name);
for (int i = 0; i < sheetNode.Columns.Count; i++)
{
CastleDBParser.ColumnNode column = sheetNode.Columns[i];
string typeNum = GetTypeNumFromCastleDBTypeString(column.TypeStr);
if (typeNum == "7")
{
return(true);
}
else if (typeNum == "6")
{
string refType = GetTypeFromCastleDBColumn(column);
var refSheet = rootNode.GetSheetWithName(refType);
Debug.Log(refType);
return(DoesSheetContainImages(rootNode, refSheet));
}
else if (typeNum == "8")
{
var refSheet = rootNode.GetSheetWithName(column.Name);
Debug.Log(column.Name);
return(DoesSheetContainImages(rootNode, refSheet));
}
}
return(false);
}
public static void GenerateTypes(CastleDBParser.RootNode root, CastleDBConfig configFile)
{
// Create scripts
List <string> scripts = new List <string>();
CastleDBConfig config = configFile;
InitTypePath(config);
foreach (CastleDBParser.SheetNode sheet in root.Sheets)
{
Debug.Log("Doinog: " + sheet.Name);
string scriptPath = $"Assets/{config.GeneratedTypesLocation}/{sheet.Name}.cs";
scripts.Add(scriptPath);
//generate fields
string fieldText = "";
for (int i = 0; i < sheet.Columns.Count; i++)
{
CastleDBParser.ColumnNode column = sheet.Columns[i];
string fieldType = CastleDBUtils.GetTypeFromCastleDBColumn(column);
if (fieldType != "Enum") //non-enum, normal field
{
if (CastleDBUtils.GetTypeNumFromCastleDBTypeString(column.TypeStr) == "8")
{
fieldText += ($"public List<{fieldType}> {column.Name}List = new List<{fieldType}>();\n\t\t");
}
else
{
fieldText += ($"public {fieldType} {column.Name};\n\t\t");
}
}
else //enum type
{
string[] enumValueNames = CastleDBUtils.GetEnumValuesFromTypeString(column.TypeStr);
string enumEntries = "";
if (CastleDBUtils.GetTypeNumFromCastleDBTypeString(column.TypeStr) == "10") //flag
{
fieldText += ($"public {column.Name}Flag {column.Name};\n");
for (int val = 0; val < enumValueNames.Length; val++)
{
enumEntries += (enumValueNames[val] + " = " + (int)Math.Pow(2, val));
if (val + 1 < enumValueNames.Length)
{
enumEntries += ",";
}
;
}
fieldText += ($"[FlagsAttribute] public enum {column.Name}Flag {{ {enumEntries} }}");
}
else
{
fieldText += ($"public {column.Name}Enum {column.Name};\n");
for (int val = 0; val < enumValueNames.Length; val++)
{
enumEntries += (enumValueNames[val] + " = " + val);
if (val + 1 < enumValueNames.Length)
{
enumEntries += ",";
}
;
}
fieldText += ($"public enum {column.Name}Enum {{ {enumEntries} }}");
}
}
}
//generate the constructor that sets the fields based on the passed in value
StringBuilder constructorText = new StringBuilder();
for (int i = 0; i < sheet.Columns.Count; i++)
{
CastleDBParser.ColumnNode column = sheet.Columns[i];
string castText = CastleDBUtils.GetCastStringFromCastleDBTypeStr(column.TypeStr);
string enumCast = "";
string typeNum = CastleDBUtils.GetTypeNumFromCastleDBTypeString(column.TypeStr);
if (typeNum == "8")
{
//list type
if (CastleDBUtils.DoesSheetContainImages(root, root.GetSheetWithName(column.Name)))
{
constructorText.Append($"\t\t\tforeach(var item in node[\"{column.Name}\"]) {{ {column.Name}List.Add(new {column.Name}(root, item, DatabaseImages));}}\n");
}
else
{
constructorText.Append($"\t\t\tforeach(var item in node[\"{column.Name}\"]) {{ {column.Name}List.Add(new {column.Name}(root, item));}}\n");
}
}
else if (typeNum == "6")
{
//working area:
//ref type
string refType = CastleDBUtils.GetTypeFromCastleDBColumn(column);
//look up the line based on the passed in row
if (CastleDBUtils.DoesSheetContainImages(root, sheet))
{
constructorText.Append(
$"{column.Name} = new {config.GeneratedTypesNamespace}.{refType}(root, root.GetSheetWithName(\"{refType}\")." +
$"Rows.Find( pred => pred[\"{config.GUIDColumnName}\"] == node[\"{column.Name}\"]), DatabaseImages);\n"
);
}
else
{
constructorText.Append(
$"{column.Name} = new {config.GeneratedTypesNamespace}.{refType}(root, root.GetSheetWithName(\"{refType}\")." +
$"Rows.Find( pred => pred[\"{config.GUIDColumnName}\"] == node[\"{column.Name}\"]));\n"
);
}
}
else if (typeNum == "7") // Image type
{
constructorText.Append($"\t\t\t{column.Name} = DatabaseImages[node[\"{column.Name}\"]];\n");
}
else
{
if (typeNum == "10")
{
enumCast = $"({column.Name}Flag)";
}
else if (typeNum == "5")
{
enumCast = $"({column.Name}Enum)";
}
constructorText.Append($"\t\t\t{column.Name} = {enumCast}node[\"{column.Name}\"]{castText};\n");
}
}
string ctor = "";
if (CastleDBUtils.DoesSheetContainImages(root, sheet))
{
ctor = $"public {sheet.Name} (CastleDBParser.RootNode root, SimpleJSON.JSONNode node, Dictionary<string, Texture> DatabaseImages)";
}
else
{
ctor = $"public {sheet.Name} (CastleDBParser.RootNode root, SimpleJSON.JSONNode node)";
}
string fullClassText = $@"
using UnityEngine;
using System;
using System.Collections.Generic;
using SimpleJSON;
using CastleDBImporter;
namespace {config.GeneratedTypesNamespace}
{{
public class {sheet.Name}
{{
{fieldText}
{ctor}
{{
{constructorText}
}}
}}
}}";
Debug.Log("Generating CDB Class: " + sheet.Name);
File.WriteAllText(scriptPath, fullClassText);
}
//build the CastleDB file
string cdbscriptPath = $"Assets/{config.GeneratedTypesLocation}/CastleDB.cs";
scripts.Add(cdbscriptPath);
//fields
string cdbfields = "";
string cdbconstructorBody = "";
foreach (CastleDBParser.SheetNode sheet in root.Sheets)
{
if (sheet.NestedType)
{
continue;
} //only write main types to CastleDB
cdbfields += $"public Dictionary<string, {sheet.Name}> {sheet.Name};\n";
cdbconstructorBody += $"{sheet.Name} = new Dictionary<string, {sheet.Name}>();\n";
//get a list of all the row names
cdbconstructorBody += $"\t\t\tforeach( var row in root.GetSheetWithName(\"{sheet.Name}\").Rows ) {{\n";
if (CastleDBUtils.DoesSheetContainImages(root, sheet))
{
cdbconstructorBody += $"\t\t\t\t{sheet.Name}[row[\"id\"]] = new {sheet.Name}(root, row, parsedDB.DatabaseImages);\n\t\t\t}}";
}
else
{
cdbconstructorBody += $"\t\t\t\t{sheet.Name}[row[\"id\"]] = new {sheet.Name}(root, row);\n\t\t\t}}";
}
}
string fullCastle = $@"
using UnityEngine;
using CastleDBImporter;
using System.Collections.Generic;
using System;
namespace {config.GeneratedTypesNamespace}
{{
public class CastleDB
{{
static CastleDBParser parsedDB;
{cdbfields}
public CastleDB(TextAsset castleDBAsset, TextAsset castleDBImagesAsset = null)
{{
parsedDB = new CastleDBParser(castleDBAsset, castleDBImagesAsset);
CastleDBParser.RootNode root = parsedDB.Root;
{ cdbconstructorBody }
}}
}}
}}";
Debug.Log("Generating CastleDB class");
File.WriteAllText(cdbscriptPath, fullCastle);
AssetDatabase.Refresh();
}