Got at least one data fetching method working; turns out, we can't use a patched LogicStack to get the data
This commit is contained in:
246
Scripts/Remotecontrol.cs
Normal file
246
Scripts/Remotecontrol.cs
Normal file
@@ -0,0 +1,246 @@
|
||||
using System;
|
||||
using UnityEngine;
|
||||
using System.Collections.Generic;
|
||||
using System.Diagnostics.CodeAnalysis;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Assets.Scripts;
|
||||
using Assets.Scripts.Networks;
|
||||
using Assets.Scripts.Objects.Electrical;
|
||||
using Assets.Scripts.Objects.Motherboards;
|
||||
using BepInEx;
|
||||
using BepInEx.Configuration;
|
||||
using Cysharp.Threading.Tasks;
|
||||
using HarmonyLib;
|
||||
using EmbedIO;
|
||||
using EmbedIO.Routing;
|
||||
using EmbedIO.WebApi;
|
||||
using RemoteControl.Message;
|
||||
using RemoteControl.Utils;
|
||||
using Swan;
|
||||
using GameDevice = Assets.Scripts.Objects.Pipes.Device;
|
||||
|
||||
|
||||
namespace RemoteControl
|
||||
{
|
||||
[BepInPlugin(pluginGuid, pluginName, pluginVersion)]
|
||||
[SuppressMessage("ReSharper", "MemberCanBePrivate.Global")]
|
||||
public class RemoteControl : BepInEx.BaseUnityPlugin
|
||||
{
|
||||
// ReSharper disable once MemberCanBePrivate.Global
|
||||
|
||||
// ReSharper disable once MemberCanBePrivate.Global
|
||||
public static ConfigEntry<int> Port;
|
||||
|
||||
// ReSharper disable once MemberCanBePrivate.Global
|
||||
public static ConfigEntry<bool> ListenOnAllInterfaces;
|
||||
|
||||
private readonly Harmony _harmony = new Harmony(pluginGuid);
|
||||
|
||||
public WebServer WebServer { get; private set; }
|
||||
|
||||
|
||||
private CancellationTokenSource _modLifecycle = new();
|
||||
|
||||
public const string pluginGuid = "com.thequux.stationeers.RemoteControl";
|
||||
public const string pluginName = "RemoteControl";
|
||||
public const string pluginVersion = "1.0";
|
||||
|
||||
public static void Log(string line)
|
||||
{
|
||||
Debug.Log("[" + pluginName + "]: " + line);
|
||||
}
|
||||
|
||||
private void StartRC()
|
||||
{
|
||||
if (!_modLifecycle.IsCancellationRequested)
|
||||
{
|
||||
_modLifecycle.Cancel();
|
||||
}
|
||||
|
||||
_modLifecycle = new CancellationTokenSource();
|
||||
|
||||
Port = Config.Bind(new ConfigDefinition("Server", "Port"), 39125,
|
||||
new ConfigDescription("The port to listen on"));
|
||||
ListenOnAllInterfaces = Config.Bind(
|
||||
new ConfigDefinition("Server", "ListenOnAllInterfaces"),
|
||||
false,
|
||||
new ConfigDescription("If set, listen on all interfaces. Otherwise, listen only on localhost"));
|
||||
|
||||
|
||||
var subscriptionModule = new SubscriptionModule("/subscribe");
|
||||
WebServer = new WebServer(o =>
|
||||
o.WithUrlPrefix($"http://{(ListenOnAllInterfaces.Value ? "0.0.0.0" : "localhost")}:{Port.Value}/")
|
||||
.WithEmbedIOHttpListener()
|
||||
)
|
||||
.WithWebApi("/api/v1", m => m.WithController<ApiController>())
|
||||
.WithModule(subscriptionModule);
|
||||
WebServer.Start(_modLifecycle.Token);
|
||||
|
||||
_harmony.PatchAll();
|
||||
foreach (var patchedMethod in _harmony.GetPatchedMethods())
|
||||
{
|
||||
Log($"Patched {patchedMethod.FullDescription()}");
|
||||
}
|
||||
Log($"Patched {_harmony.GetPatchedMethods().Count()} methods total");
|
||||
|
||||
|
||||
}
|
||||
|
||||
public void OnLoaded(List<GameObject> prefabs)
|
||||
{
|
||||
// Start();
|
||||
#if DEVELOPMENT_BUILD
|
||||
Debug.Log($"Loaded {prefabs.Count} prefabs");
|
||||
#endif
|
||||
}
|
||||
|
||||
public void OnUnloaded(List<GameObject> prefabs)
|
||||
{
|
||||
// Stop();
|
||||
}
|
||||
|
||||
public void StopRC()
|
||||
{
|
||||
_modLifecycle.Cancel();
|
||||
_harmony.UnpatchSelf();
|
||||
}
|
||||
|
||||
public void Awake()
|
||||
{
|
||||
Log("Starting RemoteControl");
|
||||
StartRC();
|
||||
}
|
||||
|
||||
private void OnDestroy()
|
||||
{
|
||||
Log("Tearing down RemoteControl");
|
||||
StopRC();
|
||||
}
|
||||
}
|
||||
|
||||
internal class ApiController : WebApiController
|
||||
{
|
||||
private static readonly Dictionary<string, LogicType> LtByName = EnumCollections.LogicTypes.AsLookupDict(true);
|
||||
private static readonly Dictionary<string, LogicSlotType> StByName = EnumCollections.LogicSlotTypes.AsLookupDict(true);
|
||||
|
||||
private static Device GetDeviceJson(GameDevice device) =>
|
||||
new()
|
||||
{
|
||||
ReferenceId = device.ReferenceId,
|
||||
Name = device.DisplayName,
|
||||
NameHash = device.GetNameHash(),
|
||||
PrefabHash = device.PrefabHash,
|
||||
PrefabName = device.PrefabName,
|
||||
LogicValues = EnumCollections.LogicTypes.Values.Where(ty => device.CanLogicRead(ty))
|
||||
.ToDictionary(ty => ty, ty => device.GetLogicValue(ty)),
|
||||
Slots = Enumerable.Range(0, device.TotalSlots)
|
||||
.Select(slot => EnumCollections.LogicSlotTypes.Values.Where(sty => device.CanLogicRead(sty, slot))
|
||||
.ToDictionary(ty => ty, sty=> device.GetLogicValue(sty, slot)))
|
||||
.ToList()
|
||||
};
|
||||
|
||||
private static Device? GetDeviceJson(long referenceId)
|
||||
{
|
||||
var dev = Referencable.Find<GameDevice>(referenceId);
|
||||
return dev == null ? null : GetDeviceJson(dev);
|
||||
}
|
||||
|
||||
[Route(HttpVerbs.Get, "/networks")]
|
||||
public Task<IList<string>> ListNetworks() => Task.FromResult(SubscriptionManager.GetProbes());
|
||||
|
||||
[Route(HttpVerbs.Get, "/networks/{networkId}")]
|
||||
public Task<IDictionary<long, Device>> ListDevices(string networkId)
|
||||
{
|
||||
// lock (SubscriptionManager.Lock)
|
||||
using (new LogTimer("ListDevices"))
|
||||
{
|
||||
var networks = new HashSet<CableNetwork>();
|
||||
GameDevice.AllDevices.ForEach(dev =>
|
||||
{
|
||||
if (dev is CableAnalyser analyser)
|
||||
{
|
||||
Debug.Log($"Found CA {analyser.DisplayName}: {analyser.ReferenceId}");
|
||||
networks.Add(analyser.CableNetwork);
|
||||
|
||||
}
|
||||
});
|
||||
|
||||
var devices = new Dictionary<long, Device>();
|
||||
foreach (var network in networks)
|
||||
{
|
||||
foreach (var device in network.DeviceList)
|
||||
{
|
||||
RemoteControl.Log($"Found {device.PrefabName}: {device.DisplayName}({device.ReferenceId})");
|
||||
if (devices.ContainsKey(device.ReferenceId)) continue;
|
||||
devices.Add(device.ReferenceId, GetDeviceJson(device));
|
||||
}
|
||||
}
|
||||
return Task.FromResult<IDictionary<long, Device>>(devices);
|
||||
}
|
||||
}
|
||||
|
||||
[Route(HttpVerbs.Post, "/networks/{networkId}/device/{deviceId}/logic/{varId}")]
|
||||
public async Task SetLogic(string networkId, long deviceId, string varId)
|
||||
{
|
||||
LogicType ltype;
|
||||
if (!LtByName.TryGetValue(varId.ToLowerInvariant(), out ltype))
|
||||
{
|
||||
throw HttpException.NotFound();
|
||||
}
|
||||
|
||||
var value = await HttpContext.GetRequestDataAsync<double>();
|
||||
|
||||
SubscriptionManager.SetLogic(networkId, deviceId, ltype, value);
|
||||
}
|
||||
|
||||
[Route(HttpVerbs.Post, "/networks/{networkId}/device/{deviceId}/slot/{slotId}/{varId}")]
|
||||
public async Task SetSlot(string networkId, long deviceId, int slotId, string varId)
|
||||
{
|
||||
LogicSlotType ltype;
|
||||
if (!StByName.TryGetValue(varId.ToLowerInvariant(), out ltype))
|
||||
{
|
||||
throw HttpException.NotFound();
|
||||
}
|
||||
|
||||
var value = await HttpContext.GetRequestDataAsync<double>();
|
||||
SubscriptionManager.SetSlot(networkId, deviceId, slotId, ltype, value);
|
||||
}
|
||||
|
||||
[Route(HttpVerbs.Patch, "/networks/{networkId}/device/{deviceId}")]
|
||||
public async Task PatchDevice(string networkId, long deviceId)
|
||||
{
|
||||
var values = await HttpContext.GetRequestDataAsync<Dictionary<string, double>>();
|
||||
if (values.Keys.Any(key => !LtByName.ContainsKey(key.ToLowerInvariant())))
|
||||
{
|
||||
throw HttpException.BadRequest();
|
||||
}
|
||||
|
||||
foreach (var keyValuePair in values)
|
||||
{
|
||||
SubscriptionManager.SetLogic(networkId, deviceId, LtByName[keyValuePair.Key.ToLowerInvariant()], keyValuePair.Value);
|
||||
}
|
||||
}
|
||||
|
||||
[Route(HttpVerbs.Get, "/networks/{networkId}/device/{deviceId}/code")]
|
||||
public async UniTask<string> GetCode(string networkId, long deviceId)
|
||||
{
|
||||
await UniTask.SwitchToMainThread(HttpContext.CancellationToken);
|
||||
|
||||
string source;
|
||||
// lock (SubscriptionManager.Lock)
|
||||
{
|
||||
var device = SubscriptionManager.GetDevice(networkId, deviceId) ?? throw HttpException.NotFound();
|
||||
if (device is not ICircuitHolder)
|
||||
{
|
||||
throw HttpException.NotFound();
|
||||
}
|
||||
|
||||
source = ((ICircuitHolder)device).GetSourceCode();
|
||||
}
|
||||
|
||||
return source;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user