134 lines
3.6 KiB
C#
134 lines
3.6 KiB
C#
using System.IO;
|
|
using System.Reflection;
|
|
using System.Text.RegularExpressions;
|
|
using BepInEx;
|
|
using HarmonyLib;
|
|
using StationeersIC10Editor;
|
|
|
|
namespace Slang
|
|
{
|
|
class L
|
|
{
|
|
private static BepInEx.Logging.ManualLogSource? _logger;
|
|
|
|
public static void SetLogger(BepInEx.Logging.ManualLogSource logger)
|
|
{
|
|
_logger = logger;
|
|
}
|
|
|
|
public static void Debug(string message)
|
|
{
|
|
_logger?.LogDebug(message);
|
|
}
|
|
|
|
public static void Info(string message)
|
|
{
|
|
_logger?.LogInfo(message);
|
|
}
|
|
|
|
public static void Error(string message)
|
|
{
|
|
_logger?.LogError(message);
|
|
}
|
|
|
|
public static void Warning(string message)
|
|
{
|
|
_logger?.LogWarning(message);
|
|
}
|
|
}
|
|
|
|
[BepInPlugin(PluginGuid, PluginName, "0.1.0")]
|
|
[BepInDependency(StationeersIC10Editor.IC10EditorPlugin.PluginGuid)]
|
|
public class SlangPlugin : BaseUnityPlugin
|
|
{
|
|
public const string PluginGuid = "com.biddydev.slang";
|
|
public const string PluginName = "Slang";
|
|
|
|
private static Regex? _slangSourceCheck = null;
|
|
|
|
private static Regex SlangSourceCheck
|
|
{
|
|
get
|
|
{
|
|
if (_slangSourceCheck is null)
|
|
{
|
|
_slangSourceCheck = new Regex(@"[;{}()]|\b(let|fn|device)\b|\/\/");
|
|
}
|
|
|
|
return _slangSourceCheck;
|
|
}
|
|
}
|
|
|
|
public static unsafe string Compile(string source)
|
|
{
|
|
string compiled;
|
|
if (Marshal.CompileFromString(source, out compiled))
|
|
{
|
|
// TODO: handle saving the original source code
|
|
return compiled;
|
|
}
|
|
else
|
|
{
|
|
return compiled;
|
|
}
|
|
}
|
|
|
|
/// <summary>Take original slang source code and copies it to a file
|
|
/// for use in restoring later.
|
|
/// </summary>
|
|
public static bool CopySourceToFile(string source)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
public static bool IsSlangSource(ref string input)
|
|
{
|
|
return SlangSourceCheck.IsMatch(input);
|
|
}
|
|
|
|
private void Awake()
|
|
{
|
|
L.SetLogger(Logger);
|
|
|
|
if (ExtractNativeDll(Ffi.RustLib))
|
|
{
|
|
var harmony = new Harmony(PluginGuid);
|
|
harmony.PatchAll();
|
|
CodeFormatters.RegisterFormatter("slang", () => new SlangFormatter(), true);
|
|
}
|
|
}
|
|
|
|
private bool ExtractNativeDll(string fileName)
|
|
{
|
|
string destinationPath = Path.Combine(Path.GetDirectoryName(Info.Location), fileName);
|
|
|
|
Assembly assembly = Assembly.GetExecutingAssembly();
|
|
|
|
using (Stream stream = assembly.GetManifestResourceStream(fileName))
|
|
{
|
|
if (stream == null)
|
|
{
|
|
L.Error(
|
|
$"{Ffi.RustLib} not found. This means it was not embedded in the mod. Please contact the mod author!"
|
|
);
|
|
return false;
|
|
}
|
|
|
|
try
|
|
{
|
|
using (FileStream fileStream = new FileStream(destinationPath, FileMode.Create))
|
|
{
|
|
stream.CopyTo(fileStream);
|
|
}
|
|
return true;
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
L.Warning($"Could not overwrite {fileName} (it might be in use): {e.Message}");
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|