242 lines
21 KiB
C#
242 lines
21 KiB
C#
using HarmonyLib;
|
|
using KMod;
|
|
using Newtonsoft.Json;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Specialized;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Text;
|
|
using UnityEngine;
|
|
|
|
namespace ONIAgentBridge
|
|
{
|
|
public class Mod : UserMod2
|
|
{
|
|
private HttpListener _listener;
|
|
internal static ConcurrentQueue<System.Action> cmdQueue = new ConcurrentQueue<System.Action>();
|
|
internal static List<GameEvent> eventLog = new List<GameEvent>();
|
|
internal static int eventSeq = 0;
|
|
internal static object eventLock = new object();
|
|
|
|
public class QueueProcessor : UnityEngine.MonoBehaviour
|
|
{
|
|
public void Update()
|
|
{
|
|
System.Action a;
|
|
while (cmdQueue.TryDequeue(out a))
|
|
try { a(); } catch (System.Exception ex) { lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[Q] "+ex.Message}); } }
|
|
}
|
|
}
|
|
|
|
public override void OnLoad(Harmony harmony)
|
|
{
|
|
base.OnLoad(harmony);
|
|
var go = new GameObject("ONI_Processor");
|
|
go.AddComponent<QueueProcessor>();
|
|
UnityEngine.Object.DontDestroyOnLoad(go);
|
|
int port = 23876;
|
|
_listener = new HttpListener();
|
|
_listener.Prefixes.Add($"http://127.0.0.1:{port}/");
|
|
_listener.Start();
|
|
_listener.BeginGetContext(OnRequest, null);
|
|
}
|
|
|
|
void OnRequest(IAsyncResult ar)
|
|
{
|
|
try
|
|
{
|
|
var ctx = _listener.EndGetContext(ar);
|
|
_listener.BeginGetContext(OnRequest, null);
|
|
Process(ctx);
|
|
}
|
|
catch { }
|
|
}
|
|
|
|
void Process(HttpListenerContext ctx)
|
|
{
|
|
try
|
|
{
|
|
var path = ctx.Request.Url.AbsolutePath.TrimEnd('/');
|
|
var method = ctx.Request.HttpMethod;
|
|
var q = ctx.Request.QueryString;
|
|
string json = null;
|
|
|
|
System.Action respond = () => {
|
|
var b = Encoding.UTF8.GetBytes(json ?? "{}");
|
|
ctx.Response.ContentType = "application/json";
|
|
ctx.Response.OutputStream.Write(b, 0, b.Length);
|
|
ctx.Response.OutputStream.Close();
|
|
};
|
|
|
|
if (path == "/api/screenshot/latest" && method == "GET") { ServeScreenshot(ctx); return; }
|
|
else if (path == "/health" && method == "GET") json = J(new { status = "ok" });
|
|
else if (path == "/api/state/game" && method == "GET") json = ReadGame();
|
|
else if (path == "/api/state/resources" && method == "GET") json = ReadResources();
|
|
else if (path == "/api/state/buildings" && method == "GET") json = ReadBuildings();
|
|
else if (path == "/api/state/duplicants" && method == "GET") json = ReadDupes();
|
|
else if (path == "/api/state/rooms" && method == "GET") json = ReadRooms();
|
|
else if (path == "/api/state/cell" && method == "GET") json = ReadCell(q);
|
|
else if (path == "/api/state/cells" && method == "GET") json = ReadCells(q);
|
|
else if (path == "/api/state/gas" && method == "GET") json = ReadGas(q);
|
|
else if (path == "/api/state/co2" && method == "GET") json = ReadCO2();
|
|
else if (path == "/api/state/temperature/zones" && method == "GET") json = ReadTemp();
|
|
else if (path == "/api/state/power" && method == "GET") json = ReadPower();
|
|
else if (path == "/api/state/pipes" && method == "GET") json = ReadPipes(q);
|
|
else if (path == "/api/state/events" && method == "GET") json = ReadEvents(q);
|
|
else if (path == "/api/state/alert" && method == "GET") json = ReadAlerts();
|
|
else if (path == "/api/state/storage" && method == "GET") json = ReadStorage();
|
|
else if (path == "/api/state/saves" && method == "GET") json = ReadSaves();
|
|
else if (path == "/api/state/camera" && method == "GET") json = ReadCamera();
|
|
else if (path == "/api/registry/buildings" && method == "GET") json = ReadBuildingReg();
|
|
else if (path == "/api/registry/elements" && method == "GET") json = ReadElementReg();
|
|
else if (path == "/api/action/pause" && method == "POST") { cmdQueue.Enqueue(() => { try { SpeedControlScreen.Instance.Pause(false, true); } catch { } }); json = Ok("paused"); }
|
|
else if (path == "/api/action/unpause" && method == "POST") { cmdQueue.Enqueue(() => { try { SpeedControlScreen.Instance.Unpause(true); SpeedControlScreen.Instance.SetSpeed(1); } catch { } }); json = Ok("unpaused"); }
|
|
else if (path == "/api/action/dig" && method == "POST") json = QueueDig(ctx);
|
|
else if (path == "/api/action/build" && method == "POST") json = QueueBuild(ctx);
|
|
else if (path == "/api/action/test" && method == "POST") { cmdQueue.Enqueue(() => { lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[Test] queue works!"}); } }); json = Ok("test_queued"); }
|
|
else json = J(new { error = "not_found", path });
|
|
|
|
respond();
|
|
}
|
|
catch { }
|
|
}
|
|
|
|
string J(object o) => JsonConvert.SerializeObject(o);
|
|
string Ok(string r, object d = null) => J(new { success = true, result = r, data = d ?? new { } });
|
|
string Fail(string e, string m = null) => J(new { success = false, error = e, errorMessage = m ?? e });
|
|
|
|
void PushEvent(string t, string s, string title, string msg, string cat = "general")
|
|
{
|
|
lock (eventLock) { eventLog.Add(new GameEvent { id = eventSeq++, type = t, severity = s, timestamp = DateTimeOffset.UtcNow.ToUnixTimeSeconds(), cycle = 0, category = cat, title = title, message = msg }); if (eventLog.Count > 500) eventLog.RemoveRange(0, eventLog.Count - 500); }
|
|
}
|
|
|
|
// === READERS (safe from background threads) ===
|
|
string ReadGame() => J(new { cycle = GameClock.Instance?.GetCycle() ?? 0, duplicantCount = Components.MinionIdentities.Count(), worldSize = Grid.CellCount, gridWidth = Grid.WidthInCells, gridHeight = Grid.HeightInCells, isPaused = SpeedControlScreen.Instance.IsPaused, gameSpeed = SpeedControlScreen.Instance.IsPaused ? 0 : SpeedControlScreen.Instance.GetSpeed() });
|
|
string ReadResources() { try { var l = new List<object>(); foreach (var e in ElementLoader.elements) { float a = 0; try { var w = ClusterManager.Instance?.activeWorld; if (w != null) a = w.worldInventory.GetAmount(e.tag, false); } catch { } if (a > 0) l.Add(new { id = e.id.ToString(), name = e.name, tag = e.tag.ToString(), amount = a, unit = "kg", state = e.IsGas ? "gas" : e.IsLiquid ? "liquid" : "solid" }); } return J(l); } catch { return J(new List<object>()); } }
|
|
string ReadBuildings() { try { var l = new List<object>(); foreach (var i in Components.BuildingCompletes) { var b = (BuildingComplete)i; var d = b.Def; l.Add(new { id = d.PrefabID, name = d.Name, x = (int)b.transform.position.x, y = (int)b.transform.position.y, width = d.WidthInCells, height = d.HeightInCells }); } return J(l); } catch { return J(new List<object>()); } }
|
|
string ReadDupes() { try { var l = new List<object>(); foreach (var i in Components.MinionIdentities) { var m = (MinionIdentity)i; l.Add(new { name = m.GetProperName(), x = (int)m.transform.position.x, y = (int)m.transform.position.y, health = m.gameObject.GetComponent<Health>()?.hitPoints ?? 100 }); } return J(l); } catch { return J(new List<object>()); } }
|
|
string ReadRooms() { try { var l = new List<object>(); var rp = Game.Instance.roomProber; if (rp != null) foreach (var r in rp.rooms) l.Add(new { id = r.roomType?.Id ?? "?", name = r.roomType?.Name ?? "?" }); return J(l); } catch { return J(new List<object>()); } }
|
|
string ReadCell(NameValueCollection q) { try { int x = int.Parse(q["x"] ?? "-1"), y = int.Parse(q["y"] ?? "-1"); int c = Grid.XYToCell(x, y); if (c < 0 || c >= Grid.CellCount) return J(new { error = "bounds" }); var el = Grid.Element[c]; return J(new { x, y, cell = c, element = el?.name ?? "Vacuum", massKg = Grid.Mass[c], temperatureC = Grid.Temperature[c] > 0 ? Grid.Temperature[c] - 273.15f : -273.15f, isSolid = Grid.Solid[c], isVisible = true, hasBuilding = Grid.Objects[c, (int)ObjectLayer.Building] != null, buildingName = Grid.Objects[c, (int)ObjectLayer.Building]?.name, hasDuplicant = Grid.Objects[c, (int)ObjectLayer.Minion] != null, isVacuum = el == null, isDiggable = Grid.Solid[c] && el != null && el.id != SimHashes.Unobtanium }); } catch { return J(new { error = "err" }); } }
|
|
string ReadCells(NameValueCollection q) { try { int x = int.Parse(q["x"] ?? "0"), y = int.Parse(q["y"] ?? "0"), w = int.Parse(q["width"] ?? "10"), h = int.Parse(q["height"] ?? "10"); var cs = new List<object>(); for (int cy = y; cy < y + h; cy++) for (int cx = x; cx < x + w; cx++) { int c = Grid.XYToCell(cx, cy); if (c >= 0 && c < Grid.CellCount) cs.Add(new { x = cx, y = cy, element = Grid.Element[c]?.name ?? "Vacuum", isSolid = Grid.Solid[c], hasBuilding = Grid.Objects[c, (int)ObjectLayer.Building] != null, hasDuplicant = Grid.Objects[c, (int)ObjectLayer.Minion] != null }); } return J(new { cells = cs }); } catch { return J(new { error = "err" }); } }
|
|
string ReadGas(NameValueCollection q) { try { int x = int.Parse(q["x"] ?? "0"), y = int.Parse(q["y"] ?? "0"), r = int.Parse(q["radius"] ?? "20"); var g = new Dictionary<string, GasEntry>(); for (int cy = Math.Max(0, y - r); cy <= Math.Min(Grid.HeightInCells - 1, y + r); cy++) for (int cx = Math.Max(0, x - r); cx <= Math.Min(Grid.WidthInCells - 1, x + r); cx++) { int c = Grid.XYToCell(cx, cy); if (c < 0) continue; var el = Grid.Element[c]; if (el != null && el.IsGas) { float m = Grid.Mass[c]; if (g.ContainsKey(el.name)) { g[el.name].mass += m; g[el.name].count++; } else g[el.name] = new GasEntry { gas = el.name, mass = m, count = 1 }; } } return J(new { gases = g.Values }); } catch { return J(new { error = "err" }); } }
|
|
string ReadCO2() { try { var p = new List<object>(); int step = Math.Max(1, Grid.CellCount / 500); for (int i = 0; i < Grid.CellCount; i += step) { var el = Grid.Element[i]; if (el != null && el.id == SimHashes.CarbonDioxide && Grid.Mass[i] > 0.5f) { int x, y; Grid.CellToXY(i, out x, out y); p.Add(new { x, y, mass = Grid.Mass[i] }); } } return J(new { pockets = p }); } catch { return J(new { error = "err" }); } }
|
|
string ReadTemp() { try { int s = Math.Max(1, Grid.CellCount / 300); float mn = float.MaxValue, mx = float.MinValue, sum = 0; int n = 0; for (int i = 0; i < Grid.CellCount; i += s) { float t = Grid.Temperature[i]; if (t <= 0) continue; float tc = t - 273.15f; sum += tc; n++; if (tc < mn) mn = tc; if (tc > mx) mx = tc; } return J(new { averageC = n > 0 ? sum / n : 0, minC = mn, maxC = mx }); } catch { return J(new { error = "err" }); } }
|
|
string ReadPower() { try { var c = new List<object>(); var m = Game.Instance.circuitManager; if (m != null) for (ushort i = 1; i <= 16; i++) { float u = m.GetWattsUsedByCircuit(i); if (u > 0 || m.HasGenerators(i)) c.Add(new { id = (int)i, wattsUsed = u, maxWatts = m.GetMaxSafeWattageForCircuit(i), isOverloaded = u > m.GetMaxSafeWattageForCircuit(i) }); } return J(new { circuits = c }); } catch { return J(new { error = "err" }); } }
|
|
string ReadPipes(NameValueCollection q) { try { var s = new List<object>(); string t = q["type"] ?? "all"; foreach (var f in new[] { new { n = "liquid", fl = Game.Instance.liquidConduitFlow }, new { n = "gas", fl = Game.Instance.gasConduitFlow } }) { if (t != "all" && t != f.n) continue; if (f.fl == null) continue; int cnt = 0; foreach (var i in Components.BuildingCompletes) { if (cnt > 100) break; var b = (BuildingComplete)i; var pid = b.Def.PrefabID; if (pid != (f.n == "gas" ? "GasConduit" : "LiquidConduit") && pid != (f.n == "gas" ? "GasConduitBridge" : "LiquidConduitBridge")) continue; var ct = f.fl.GetContents(b.GetCell()); if (ct.mass > 0) { s.Add(new { type = f.n, element = ct.element.ToString(), mass = ct.mass }); cnt++; } } } return J(new { segments = s }); } catch { return J(new { error = "err" }); } }
|
|
string ReadEvents(NameValueCollection q) { int s = int.Parse(q["since"] ?? "-1"), l = Math.Min(int.Parse(q["limit"] ?? "50"), 200); lock (eventLock) { var ev = eventLog.Where(e => e.id > s).Take(l).ToList(); return J(new { events = ev, next_seq = ev.Any() ? ev.Last().id : s }); } }
|
|
string ReadAlerts() { try { var l = new List<object>(); var nm = global::NotificationManager.Instance; if (nm != null) { foreach (var fn in new[] { "notifications", "pendingNotifications" }) { try { var f = typeof(global::NotificationManager).GetField(fn, System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.NonPublic); if (f == null) continue; var items = f.GetValue(nm) as System.Collections.IEnumerable; if (items != null) foreach (global::Notification n in items) l.Add(new { title = n.titleText, severity = n.Type.ToString() }); } catch { } } } return J(l); } catch { return J(new List<object>()); } }
|
|
string ReadStorage() { try { var l = new List<object>(); foreach (var i in Components.BuildingCompletes) { var b = (BuildingComplete)i; var s = b.gameObject.GetComponent<Storage>(); if (s == null || s.MassStored() <= 0) continue; l.Add(new { building = b.Def?.Name ?? b.name, x = (int)b.transform.position.x, y = (int)b.transform.position.y, mass = s.MassStored(), capacity = s.capacityKg }); } return J(new { storages = l.Take(50).ToList() }); } catch { return J(new { error = "err" }); } }
|
|
string ReadSaves() { try { var l = new List<object>(); string sp = SaveLoader.GetActiveSaveFilePath(); var d = Path.GetDirectoryName(sp); if (d != null && Directory.Exists(d)) foreach (var f in Directory.GetFiles(d, "*.sav")) { var fi = new FileInfo(f); l.Add(new { name = Path.GetFileNameWithoutExtension(f) }); } return J(new { saves = l }); } catch { return J(new { error = "err" }); } }
|
|
string ReadCamera() { try { var c = CameraController.Instance; var p = c.transform.position; return J(new { x = p.x, y = p.y }); } catch { return J(new { error = "err" }); } }
|
|
string ReadBuildingReg() { try { var l = new List<object>(); foreach (var i in Assets.BuildingDefs) { var d = (BuildingDef)i; l.Add(new { id = d.PrefabID, name = d.Name, width = d.WidthInCells, height = d.HeightInCells }); } return J(l); } catch { return J(new { error = "err" }); } }
|
|
string ReadElementReg() { try { var l = new List<object>(); foreach (var e in ElementLoader.elements) l.Add(new { id = e.id.ToString(), name = e.name }); return J(l); } catch { return J(new { error = "err" }); } }
|
|
|
|
string QueueDig(HttpListenerContext c)
|
|
{
|
|
try
|
|
{
|
|
var d = JsonConvert.DeserializeObject<DigReq>(new StreamReader(c.Request.InputStream).ReadToEnd());
|
|
if (d == null) return Fail("invalid");
|
|
int x = d.x, y = d.y, w = d.width, h = d.height;
|
|
cmdQueue.Enqueue(() => {
|
|
for (int dy = 0; dy < h; dy++) for (int dx = 0; dx < w; dx++) { int cell = Grid.XYToCell(x + dx, y + dy); if (cell >= 0 && cell < Grid.CellCount) try { DigTool.PlaceDig(cell, 0); } catch { } }
|
|
});
|
|
return Ok("dig_queued");
|
|
}
|
|
catch { return Fail("error"); }
|
|
}
|
|
|
|
string QueueBuild(HttpListenerContext c)
|
|
{
|
|
try
|
|
{
|
|
var d = JsonConvert.DeserializeObject<BuildReq>(new StreamReader(c.Request.InputStream).ReadToEnd());
|
|
if (d == null) return Fail("invalid");
|
|
int x = d.x, y = d.y; string bid = d.buildingId;
|
|
// Find available materials on HTTP thread (WorldInventory works from background)
|
|
var availableTags = new List<Tag>();
|
|
try
|
|
{
|
|
var worldInv = ClusterManager.Instance?.activeWorld?.worldInventory;
|
|
if (worldInv != null)
|
|
{
|
|
var def = Assets.GetBuildingDef(bid);
|
|
if (def != null)
|
|
foreach (var cat in def.MaterialCategory ?? new string[0])
|
|
foreach (var elem in ElementLoader.elements)
|
|
{
|
|
if (worldInv.GetAmount(elem.tag, false) <= 0) continue;
|
|
bool match = (cat == "RawMineral" && elem.HasTag(GameTags.ConsumableOre));
|
|
match = match || (cat == "Metal" && elem.HasTag(GameTags.Metal));
|
|
if (match) availableTags.Add(elem.tag);
|
|
}
|
|
}
|
|
}
|
|
catch { }
|
|
// If no materials found, use default sand
|
|
if (availableTags.Count == 0) availableTags.Add(new Tag("SandStone"));
|
|
List<Tag> capturedTags = new List<Tag>(availableTags);
|
|
cmdQueue.Enqueue(() => {
|
|
lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[Build] START "+bid}); }
|
|
try
|
|
{
|
|
var def = Assets.GetBuildingDef(bid);
|
|
if (def == null) { lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] def null"}); } return; }
|
|
int cell = Grid.XYToCell(x, y);
|
|
if (cell < 0 || cell >= Grid.CellCount) { lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] bounds"}); } return; }
|
|
lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] cell="+cell}); }
|
|
var world = ClusterManager.Instance?.activeWorld;
|
|
lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] world ok"}); }
|
|
for (int dy = 0; dy < def.HeightInCells; dy++)
|
|
for (int dx = 0; dx < def.WidthInCells; dx++) {
|
|
int c2 = Grid.XYToCell(x+dx, y+dy);
|
|
if (c2 < 0 || c2 >= Grid.CellCount) continue;
|
|
if (Grid.Objects[c2, (int)ObjectLayer.Building] != null) { lock(eventLock){eventLog.Add(new GameEvent{id=eventSeq++,title="[B] occupied"});} return; }
|
|
}
|
|
lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] cells ok"}); }
|
|
var pos = Grid.CellToPos(cell);
|
|
// Use game's build system (handles all initialization)
|
|
lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] cells ok"}); }
|
|
try {
|
|
var loader = UnityEngine.Object.FindObjectOfType<BuildingLoader>();
|
|
GameObject go = null;
|
|
if (loader != null) go = loader.CreateBuildingUnderConstruction(def);
|
|
if (go == null) {
|
|
// Fallback: create complete building directly
|
|
go = def.Instantiate(Grid.CellToPos(cell), Orientation.Neutral, null, (int)def.SceneLayer);
|
|
}
|
|
if (go != null) go.SetActive(true);
|
|
lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[B] "+(go==null?"null":"ok")}); }
|
|
} catch (Exception ex2) { lock(eventLock){eventLog.Add(new GameEvent{id=eventSeq++,title="[B] ex:"+ex2.Message});} }
|
|
}
|
|
catch (Exception ex) { lock(eventLock) { eventLog.Add(new GameEvent{id=eventSeq++,title="[Build] EX:"+ex.Message}); } }
|
|
});
|
|
return Ok("build_queued");
|
|
}
|
|
catch { return Fail("error"); }
|
|
}
|
|
|
|
static string _ss = null;
|
|
void ServeScreenshot(HttpListenerContext ctx)
|
|
{
|
|
try
|
|
{
|
|
if (_ss != null && File.Exists(_ss)) { var b = File.ReadAllBytes(_ss); ctx.Response.ContentType = "image/png"; ctx.Response.ContentLength64 = b.Length; ctx.Response.OutputStream.Write(b, 0, b.Length); return; }
|
|
}
|
|
catch { }
|
|
ctx.Response.StatusCode = 404;
|
|
ctx.Response.OutputStream.Close();
|
|
}
|
|
}
|
|
|
|
|
|
|
|
public class GameEvent { public int id; public string type; public string severity; public long timestamp; public int cycle; public string category; public string title; public string message; }
|
|
public class GasEntry { public string gas; public float mass; public int count; }
|
|
public class DigReq { public int x; public int y; public int width; public int height; }
|
|
public class BuildReq { public string buildingId; public int x; public int y; }
|
|
}
|