FakeModRunner.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Diagnostics;
using System.Runtime.CompilerServices;
using System.IO;
 
using Barotrauma;
using Microsoft.Xna.Framework;
using Microsoft.Xna.Framework.Input;
using Microsoft.Xna.Framework.Graphics;
using HarmonyLib;
using CrabUI;
 
namespace CrabUITest
{
  public class FakeModRunner
  {
    public CUIComponent TestChamber => CUITest.TestChamber;
    public IFakeMod ActiveMod;
 
 
    public void RunTest(string name)
    {
      Assembly targetAssembly = Assembly.GetAssembly(typeof(FakeModRunner));
 
      Type type = targetAssembly.GetTypes().Find(T => T.Name == name);
      if (type == null)
      {
        CUITest.Log($"can't find fake mod {name}");
        return;
      }
 
      RunTest(type);
    }
 
    public void RunTest<RawType>() => RunTest(typeof(RawType));
 
    public void RunTest(Type type)
    {
      if (!type.IsAssignableTo(typeof(IFakeMod)))
      {
        CUITest.Log($"it's not a fake mod");
        return;
      }
 
      IFakeMod mod = (IFakeMod)Activator.CreateInstance(type);
 
      RunTest(mod);
    }
 
    public void RunTest(IFakeMod mod)
    {
      if (TestChamber == null) CUITest.CreateTestChamber();
      if (TestChamber.Parent == null) CUITest.OpenTestChamber();
 
 
      if (ActiveMod != null)
      {
        ActiveMod.Dispose();
      }
      ActiveMod = mod;
 
      CUITest.ClearTestChamber();
 
      try
      {
        mod.Initialize(TestChamber);
      }
      catch (Exception e) { CUITest.Log(e, Color.Orange); }
    }
 
    public void StopTest()
    {
      if (ActiveMod == null) return;
      ActiveMod.Dispose();
      CUITest.ClearTestChamber();
      if (TestChamber.Parent != null) CUITest.CloseTestChamber();
 
      ActiveMod = null;
    }
 
    public string[] FindAllFakeMods()
    {
      Assembly targetAssembly = Assembly.GetAssembly(typeof(FakeModRunner));
 
      IEnumerable<Type> fakeModTypes = targetAssembly.GetTypes().Where(T => T.IsAssignableTo(typeof(IFakeMod)) && T != typeof(IFakeMod));
 
      return fakeModTypes.Select(T => T.Name).ToArray();
    }
 
    public FakeModRunner()
    {
 
    }
 
  }
}