Kalospacer/WulaFallenEmpireRW
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

zc

Browse Source
This commit is contained in:
ProjectKoi-Kalo\Kalo
2025-12-12 16:20:52 +08:00
parent 4417a26650
commit 968268b368
15 changed files with 1518 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

228
Source/WulaFallenEmpire/EventSystem/AI/AIHistoryManager.cs Normal file
View File

@@ -0,0 +1,228 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using RimWorld.Planet;
using Verse;
namespace WulaFallenEmpire.EventSystem.AI
{
public class AIHistoryManager : WorldComponent
{
private string _saveId;
private Dictionary<string, List<(string role, string message)>> _cache = new Dictionary<string, List<(string role, string message)>>();
public AIHistoryManager(World world) : base(world)
{
}
private string GetSaveDirectory()
{
string path = Path.Combine(GenFilePaths.SaveDataFolderPath, "WulaAIHistory");
if (!Directory.Exists(path))
{
Directory.CreateDirectory(path);
}
return path;
}
private string GetFilePath(string eventDefName)
{
if (string.IsNullOrEmpty(_saveId))
{
_saveId = Guid.NewGuid().ToString();
}
return Path.Combine(GetSaveDirectory(), $"{_saveId}_{eventDefName}.json");
}
public List<(string role, string message)> GetHistory(string eventDefName)
{
if (_cache.TryGetValue(eventDefName, out var cachedHistory))
{
return cachedHistory;
}
string path = GetFilePath(eventDefName);
if (File.Exists(path))
{
try
{
string json = File.ReadAllText(path);
var history = SimpleJsonParser.Deserialize(json);
if (history != null)
{
_cache[eventDefName] = history;
return history;
}
}
catch (Exception ex)
{
Log.Error($"[WulaFallenEmpire] Failed to load AI history from {path}: {ex}");
}
}
return new List<(string role, string message)>();
}
public void SaveHistory(string eventDefName, List<(string role, string message)> history)
{
_cache[eventDefName] = history;
string path = GetFilePath(eventDefName);
try
{
string json = SimpleJsonParser.Serialize(history);
File.WriteAllText(path, json);
}
catch (Exception ex)
{
Log.Error($"[WulaFallenEmpire] Failed to save AI history to {path}: {ex}");
}
}
public override void ExposeData()
{
base.ExposeData();
Scribe_Values.Look(ref _saveId, "WulaAIHistoryId");
if (Scribe.mode == LoadSaveMode.PostLoadInit && string.IsNullOrEmpty(_saveId))
{
_saveId = Guid.NewGuid().ToString();
}
}
}
public static class SimpleJsonParser
{
public static string Serialize(List<(string role, string message)> history)
{
StringBuilder sb = new StringBuilder();
sb.Append("[");
for (int i = 0; i < history.Count; i++)
{
var item = history[i];
sb.Append("{");
sb.Append($"\"role\":\"{Escape(item.role)}\",");
sb.Append($"\"message\":\"{Escape(item.message)}\"");
sb.Append("}");
if (i < history.Count - 1) sb.Append(",");
}
sb.Append("]");
return sb.ToString();
}
public static List<(string role, string message)> Deserialize(string json)
{
var result = new List<(string role, string message)>();
if (string.IsNullOrEmpty(json)) return result;
// Very basic parser, assumes standard format produced by Serialize
// Remove outer brackets
json = json.Trim();
if (json.StartsWith("[") && json.EndsWith("]"))
{
json = json.Substring(1, json.Length - 2);
}
if (string.IsNullOrEmpty(json)) return result;
// Split by objects
// This is fragile if objects contain nested objects or escaped braces, but for this specific structure it's fine
// We are splitting by "},{" which is risky. Better to iterate.
int depth = 0;
int start = 0;
for (int i = 0; i < json.Length; i++)
{
if (json[i] == '{')
{
if (depth == 0) start = i;
depth++;
}
else if (json[i] == '}')
{
depth--;
if (depth == 0)
{
string obj = json.Substring(start, i - start + 1);
var parsed = ParseObject(obj);
if (parsed.role != null) result.Add(parsed);
}
}
}
return result;
}
private static (string role, string message) ParseObject(string json)
{
string role = null;
string message = null;
// Remove braces
json = json.Trim('{', '}');
// Split by key-value pairs, respecting quotes
// Again, simple parsing
var parts = SplitByComma(json);
foreach (var part in parts)
{
var kv = SplitByColon(part);
if (kv.Length == 2)
{
string key = Unescape(kv[0].Trim().Trim('"'));
string val = Unescape(kv[1].Trim().Trim('"'));
if (key == "role") role = val;
if (key == "message") message = val;
}
}
return (role, message);
}
private static string[] SplitByComma(string input)
{
// Split by comma but ignore commas inside quotes
var list = new List<string>();
bool inQuote = false;
int start = 0;
for (int i = 0; i < input.Length; i++)
{
if (input[i] == '"' && (i == 0 || input[i-1] != '\\')) inQuote = !inQuote;
if (input[i] == ',' && !inQuote)
{
list.Add(input.Substring(start, i - start));
start = i + 1;
}
}
list.Add(input.Substring(start));
return list.ToArray();
}
private static string[] SplitByColon(string input)
{
// Split by first colon outside quotes
bool inQuote = false;
for (int i = 0; i < input.Length; i++)
{
if (input[i] == '"' && (i == 0 || input[i-1] != '\\')) inQuote = !inQuote;
if (input[i] == ':' && !inQuote)
{
return new[] { input.Substring(0, i), input.Substring(i + 1) };
}
}
return new[] { input };
}
private static string Escape(string s)
{
if (s == null) return "";
return s.Replace("\\", "\\\\").Replace("\"", "\\\"").Replace("\n", "\\n").Replace("\r", "\\r");
}
private static string Unescape(string s)
{
if (s == null) return "";
return s.Replace("\\r", "\r").Replace("\\n", "\n").Replace("\\\"", "\"").Replace("\\\\", "\\");
}
}
}