Refactor and stuff

This commit is contained in:
John Gross
2016-12-23 13:24:58 -08:00
parent c381439ea1
commit 2695cda5fa
29 changed files with 513 additions and 182 deletions

View File

@@ -1,6 +1,6 @@
using System; using System;
namespace Torch namespace Torch.API
{ {
/// <summary> /// <summary>
/// Identifies a player's current connection state. /// Identifies a player's current connection state.

View File

@@ -6,9 +6,10 @@ using System.Threading.Tasks;
namespace Torch.API namespace Torch.API
{ {
public interface ITorchServer public interface IChatItem
{ {
void Start(); IPlayer Player { get; }
void Stop(); string Message { get; }
DateTime Time { get; }
} }
} }

View File

@@ -1,13 +0,0 @@
namespace Torch.API
{
public interface IEnvironmentInfo
{
EnvironmentType Type { get; }
}
public enum EnvironmentType
{
DedicatedServer,
GameClient
}
}

18
Torch.API/IMultiplayer.cs Normal file
View File

@@ -0,0 +1,18 @@
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
namespace Torch.API
{
public interface IMultiplayer
{
event Action<IPlayer> PlayerJoined;
event Action<IPlayer> PlayerLeft;
event Action<IChatItem> MessageReceived;
Dictionary<ulong, IPlayer> Players { get; }
List<IChatItem> Chat { get; }
void SendMessage(string message);
void KickPlayer(ulong id);
void BanPlayer(ulong id, bool banned = true);
}
}

18
Torch.API/IPlayer.cs Normal file
View File

@@ -0,0 +1,18 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Torch.API
{
public interface IPlayer
{
ulong SteamId { get; }
List<ulong> IdentityIds { get; }
string Name { get; }
ConnectionState State { get; }
DateTime LastConnected { get; }
void SetConnectionState(ConnectionState state);
}
}

View File

@@ -0,0 +1,21 @@
using System;
using VRage.Collections;
using VRage.Plugins;
namespace Torch.API
{
public interface IPluginManager
{
ListReader<IPlugin> Plugins { get; }
string[] GetPluginFolders();
string GetPluginName(Type pluginType);
void LoadAllPlugins();
void LoadPlugin(IPlugin plugin);
void LoadPluginFolder(string folderName);
void ReloadAll();
void ReloadPlugin(IPlugin plugin, bool forceNonPiston = false);
bool UnblockDll(string fileName);
void UnloadPlugin(IPlugin plugin);
}
}

31
Torch.API/ITorchBase.cs Normal file
View File

@@ -0,0 +1,31 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Torch.API
{
public interface ITorchBase
{
event Action SessionLoaded;
IMultiplayer Multiplayer { get; }
IPluginManager Plugins { get; }
void GameAction(Action action);
void BeginGameAction(Action action, Action<object> callback = null, object state = null);
void Start();
void Stop();
void Init();
}
public interface ITorchServer : ITorchBase
{
bool IsRunning { get; }
string[] RunArgs { get; set; }
}
public interface ITorchClient : ITorchBase
{
}
}

View File

@@ -10,7 +10,7 @@ namespace Torch.API
{ {
public interface ITorchPlugin : IPlugin public interface ITorchPlugin : IPlugin
{ {
void Init(ITorchServer server); void Init(ITorchBase torch);
void Reload(); void Reload();
} }
} }

View File

@@ -45,13 +45,20 @@
<Reference Include="VRage"> <Reference Include="VRage">
<HintPath>C:\Program Files (x86)\Steam\steamapps\common\SpaceEngineers\Bin64\VRage.dll</HintPath> <HintPath>C:\Program Files (x86)\Steam\steamapps\common\SpaceEngineers\Bin64\VRage.dll</HintPath>
</Reference> </Reference>
<Reference Include="VRage.Library, Version=1.0.0.0, Culture=neutral, processorArchitecture=MSIL">
<SpecificVersion>False</SpecificVersion>
<HintPath>..\..\..\..\..\..\..\Program Files (x86)\Steam\steamapps\common\SpaceEngineers\Bin64\VRage.Library.dll</HintPath>
</Reference>
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>
<Compile Include="IEnvironmentInfo.cs" /> <Compile Include="ConnectionState.cs" />
<Compile Include="IChatItem.cs" />
<Compile Include="IMultiplayer.cs" />
<Compile Include="IPlayer.cs" />
<Compile Include="IPluginManager.cs" />
<Compile Include="ITorchPlugin.cs" /> <Compile Include="ITorchPlugin.cs" />
<Compile Include="IServerControls.cs" /> <Compile Include="IServerControls.cs" />
<Compile Include="ITorchServer.cs" /> <Compile Include="ITorchBase.cs" />
<Compile Include="TorchAPI.cs" />
<Compile Include="PluginAttribute.cs" /> <Compile Include="PluginAttribute.cs" />
<Compile Include="Properties\AssemblyInfo.cs" /> <Compile Include="Properties\AssemblyInfo.cs" />
</ItemGroup> </ItemGroup>

View File

@@ -1,11 +0,0 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Controls;
namespace Torch.API
{
}

View File

@@ -9,26 +9,29 @@ using System.Threading.Tasks;
using System.Windows; using System.Windows;
using System.Windows.Threading; using System.Windows.Threading;
using Torch; using Torch;
using Torch.API;
namespace Torch.Server namespace Torch.Server
{ {
public static class Program public static class Program
{ {
private static TorchServer _server = new TorchServer(); private static readonly ITorchServer _server = new TorchServer();
private static TorchUI _ui;
[STAThread] [STAThread]
public static void Main(string[] args) public static void Main(string[] args)
{ {
_server.Init(); _server.Init();
_server.Server.RunArgs = new[] { "-console" }; _server.RunArgs = new[] { "-console" };
_ui = new TorchUI(_server);
if (args.Contains("-nogui")) if (args.Contains("-nogui"))
_server.Server.StartServer(); _server.Start();
else else
StartUI(); StartUI();
if (args.Contains("-autostart") && !_server.Server.IsRunning) if (args.Contains("-autostart") && !_server.IsRunning)
_server.Server.StartServerThread(); new Thread(() => _server.Start()).Start();
Dispatcher.Run(); Dispatcher.Run();
} }
@@ -36,12 +39,12 @@ namespace Torch.Server
public static void StartUI() public static void StartUI()
{ {
Thread.CurrentThread.Name = "UI Thread"; Thread.CurrentThread.Name = "UI Thread";
_server.UI.Show(); _ui.Show();
} }
public static void FullRestart() public static void FullRestart()
{ {
_server.Server.StopServer(); _server.Stop();
Process.Start("TorchServer.exe", "-autostart"); Process.Start("TorchServer.exe", "-autostart");
Environment.Exit(1); Environment.Exit(1);
} }

View File

@@ -13,6 +13,7 @@ using Sandbox.Engine.Multiplayer;
using Sandbox.Game; using Sandbox.Game;
using Sandbox.Game.World; using Sandbox.Game.World;
using SpaceEngineers.Game; using SpaceEngineers.Game;
using Torch.API;
using VRage.Dedicated; using VRage.Dedicated;
using VRage.Game; using VRage.Game;
using VRage.Game.SessionComponents; using VRage.Game.SessionComponents;
@@ -20,7 +21,7 @@ using VRage.Profiler;
namespace Torch.Server namespace Torch.Server
{ {
public class ServerManager : IDisposable public class TorchServer : ITorchServer
{ {
public Thread ServerThread { get; private set; } public Thread ServerThread { get; private set; }
public string[] RunArgs { get; set; } = new string[0]; public string[] RunArgs { get; set; } = new string[0];
@@ -31,12 +32,12 @@ namespace Torch.Server
private readonly ManualResetEvent _stopHandle = new ManualResetEvent(false); private readonly ManualResetEvent _stopHandle = new ManualResetEvent(false);
internal ServerManager() internal TorchServer()
{ {
MySession.OnLoading += OnSessionLoading; MySession.OnLoading += OnSessionLoading;
} }
public void InitSandbox() public void Init()
{ {
SpaceEngineersGame.SetupBasicGameInfo(); SpaceEngineersGame.SetupBasicGameInfo();
SpaceEngineersGame.SetupPerGameSettings(); SpaceEngineersGame.SetupPerGameSettings();
@@ -53,7 +54,7 @@ namespace Torch.Server
SpaceEngineersGame.SetupBasicGameInfo(); SpaceEngineersGame.SetupBasicGameInfo();
SpaceEngineersGame.SetupPerGameSettings(); SpaceEngineersGame.SetupPerGameSettings();
}; };
int? gameVersion = MyPerGameSettings.BasicGameInfo.GameVersion; var gameVersion = MyPerGameSettings.BasicGameInfo.GameVersion;
MyFinalBuildConstants.APP_VERSION = gameVersion ?? 0; MyFinalBuildConstants.APP_VERSION = gameVersion ?? 0;
} }
@@ -150,14 +151,14 @@ namespace Torch.Server
return; return;
} }
ServerThread = new Thread(StartServer); ServerThread = new Thread(Start);
ServerThread.Start(); ServerThread.Start();
} }
/// <summary> /// <summary>
/// Start server on the current thread. /// Start server on the current thread.
/// </summary> /// </summary>
public void StartServer() public void Start()
{ {
IsRunning = true; IsRunning = true;
Logger.Write("Starting server."); Logger.Write("Starting server.");
@@ -171,12 +172,12 @@ namespace Torch.Server
/// <summary> /// <summary>
/// Stop the server. /// Stop the server.
/// </summary> /// </summary>
public void StopServer() public void Stop()
{ {
if (Thread.CurrentThread.ManagedThreadId != ServerThread?.ManagedThreadId) if (Thread.CurrentThread.ManagedThreadId != ServerThread?.ManagedThreadId)
{ {
Logger.Write("Requesting server stop."); Logger.Write("Requesting server stop.");
MySandboxGame.Static.Invoke(StopServer); MySandboxGame.Static.Invoke(Stop);
_stopHandle.WaitOne(); _stopHandle.WaitOne();
return; return;
} }
@@ -204,11 +205,5 @@ namespace Torch.Server
typeof(MyRenderProfiler).GetField("m_gpuProfiler", BindingFlags.Static | BindingFlags.NonPublic).SetValue(null, null); typeof(MyRenderProfiler).GetField("m_gpuProfiler", BindingFlags.Static | BindingFlags.NonPublic).SetValue(null, null);
(typeof(MyRenderProfiler).GetField("m_threadProfilers", BindingFlags.Static | BindingFlags.NonPublic).GetValue(null) as List<MyProfiler>).Clear(); (typeof(MyRenderProfiler).GetField("m_threadProfilers", BindingFlags.Static | BindingFlags.NonPublic).GetValue(null) as List<MyProfiler>).Clear();
} }
public void Dispose()
{
if (IsRunning)
StopServer();
}
} }
} }

View File

@@ -111,20 +111,17 @@
<Compile Include="Views\ChatControl.xaml.cs"> <Compile Include="Views\ChatControl.xaml.cs">
<DependentUpon>ChatControl.xaml</DependentUpon> <DependentUpon>ChatControl.xaml</DependentUpon>
</Compile> </Compile>
<Compile Include="ViewModels\ChatItemInfo.cs" />
<Compile Include="Views\ModsControl.xaml.cs"> <Compile Include="Views\ModsControl.xaml.cs">
<DependentUpon>ModsControl.xaml</DependentUpon> <DependentUpon>ModsControl.xaml</DependentUpon>
</Compile> </Compile>
<Compile Include="Views\PistonUI.xaml.cs"> <Compile Include="Views\TorchUI.xaml.cs">
<DependentUpon>PistonUI.xaml</DependentUpon> <DependentUpon>TorchUI.xaml</DependentUpon>
</Compile> </Compile>
<Compile Include="MultiplayerManager.cs" />
<Compile Include="TorchServer.cs" />
<Compile Include="Views\PlayerListControl.xaml.cs"> <Compile Include="Views\PlayerListControl.xaml.cs">
<DependentUpon>PlayerListControl.xaml</DependentUpon> <DependentUpon>PlayerListControl.xaml</DependentUpon>
</Compile> </Compile>
<Compile Include="Program.cs" /> <Compile Include="Program.cs" />
<Compile Include="ServerManager.cs" /> <Compile Include="TorchServer.cs" />
<Compile Include="Views\PropertyGrid.xaml.cs"> <Compile Include="Views\PropertyGrid.xaml.cs">
<DependentUpon>PropertyGrid.xaml</DependentUpon> <DependentUpon>PropertyGrid.xaml</DependentUpon>
</Compile> </Compile>
@@ -181,7 +178,7 @@
<SubType>Designer</SubType> <SubType>Designer</SubType>
<Generator>MSBuild:Compile</Generator> <Generator>MSBuild:Compile</Generator>
</Page> </Page>
<Page Include="Views\PistonUI.xaml"> <Page Include="Views\TorchUI.xaml">
<Generator>MSBuild:Compile</Generator> <Generator>MSBuild:Compile</Generator>
<SubType>Designer</SubType> <SubType>Designer</SubType>
</Page> </Page>

View File

@@ -1,75 +1,127 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Drawing; using System.IO;
using System.Linq; using System.Linq;
using System.Reflection;
using System.Text; using System.Text;
using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
using System.Windows;
using Torch; using Torch;
using Sandbox; using Sandbox;
using Sandbox.Engine.Multiplayer;
using Sandbox.Game; using Sandbox.Game;
using Sandbox.Game.World;
using SpaceEngineers.Game; using SpaceEngineers.Game;
using Torch.API; using Torch.API;
using Torch.Launcher;
using VRage.Dedicated; using VRage.Dedicated;
using VRage.Game; using VRage.Game;
using VRage.Game.SessionComponents; using VRage.Game.SessionComponents;
using VRage.Profiler;
namespace Torch.Server namespace Torch.Server
{ {
/// <summary> public class TorchServer : TorchBase, ITorchServer
/// Entry point for all Piston server functionality.
/// </summary>
public class TorchServer : ITorchServer
{ {
public ServerManager Server { get; private set; } public Thread ServerThread { get; private set; }
public MultiplayerManager Multiplayer { get; private set; } public string[] RunArgs { get; set; } = new string[0];
public PluginManager Plugins { get; private set; } public bool IsRunning { get; private set; }
public PistonUI UI { get; private set; }
private bool _init; public event Action SessionLoading;
public void Start() private readonly ManualResetEvent _stopHandle = new ManualResetEvent(false);
internal TorchServer()
{ {
if (!_init) MySession.OnLoading += OnSessionLoading;
Init();
Server.StartServer();
}
public void Stop()
{
Server.StopServer();
}
public void Init()
{
if (_init)
return;
Logger.Write("Initializing Torch");
_init = true;
Server = new ServerManager();
Multiplayer = new MultiplayerManager(this);
Plugins = new PluginManager(); Plugins = new PluginManager();
UI = new PistonUI();
Server.SessionLoaded += Plugins.LoadAllPlugins;
Server.InitSandbox();
SteamHelper.Init();
UI.PropGrid.SetObject(MySandboxGame.ConfigDedicated);
} }
public void Reset() public override void Init()
{ {
Logger.Write("Resetting Torch"); SpaceEngineersGame.SetupBasicGameInfo();
Server.Dispose(); SpaceEngineersGame.SetupPerGameSettings();
UI.Close(); MyPerGameSettings.SendLogToKeen = false;
MyPerServerSettings.GameName = MyPerGameSettings.GameName;
MyPerServerSettings.GameNameSafe = MyPerGameSettings.GameNameSafe;
MyPerServerSettings.GameDSName = MyPerServerSettings.GameNameSafe + "Dedicated";
MyPerServerSettings.GameDSDescription = "Your place for space engineering, destruction and exploring.";
MySessionComponentExtDebug.ForceDisable = true;
MyPerServerSettings.AppId = 244850u;
ConfigForm<MyObjectBuilder_SessionSettings>.GameAttributes = Game.SpaceEngineers;
ConfigForm<MyObjectBuilder_SessionSettings>.OnReset = delegate
{
SpaceEngineersGame.SetupBasicGameInfo();
SpaceEngineersGame.SetupPerGameSettings();
};
var gameVersion = MyPerGameSettings.BasicGameInfo.GameVersion;
MyFinalBuildConstants.APP_VERSION = gameVersion ?? 0;
}
Server = null; private void OnSessionLoading()
Multiplayer = null; {
Plugins = null; SessionLoading?.Invoke();
UI = null; MySession.Static.OnReady += OnSessionReady;
_init = false; }
private void OnSessionReady()
{
Plugins.LoadAllPlugins();
InvokeSessionLoaded();
}
/// <summary>
/// Start server on the current thread.
/// </summary>
public override void Start()
{
if (IsRunning)
throw new InvalidOperationException("Server is already running.");
IsRunning = true;
Logger.Write("Starting server.");
if (MySandboxGame.Log.LogEnabled)
MySandboxGame.Log.Close();
DedicatedServer.Run<MyObjectBuilder_SessionSettings>(RunArgs);
}
/// <summary>
/// Stop the server.
/// </summary>
public override void Stop()
{
if (Thread.CurrentThread.ManagedThreadId != ServerThread?.ManagedThreadId)
{
Logger.Write("Requesting server stop.");
MySandboxGame.Static.Invoke(Stop);
_stopHandle.WaitOne();
return;
}
Logger.Write("Stopping server.");
MySession.Static.Save();
MySession.Static.Unload();
MySandboxGame.Static.Exit();
//Unload all the static junk.
//TODO: Finish unloading all server data so it's in a completely clean state.
VRage.FileSystem.MyFileSystem.Reset();
VRage.Input.MyGuiGameControlsHelpers.Reset();
VRage.Input.MyInput.UnloadData();
CleanupProfilers();
Logger.Write("Server stopped.");
_stopHandle.Set();
IsRunning = false;
}
private void CleanupProfilers()
{
typeof(MyRenderProfiler).GetField("m_threadProfiler", BindingFlags.Static | BindingFlags.NonPublic).SetValue(null, null);
typeof(MyRenderProfiler).GetField("m_gpuProfiler", BindingFlags.Static | BindingFlags.NonPublic).SetValue(null, null);
(typeof(MyRenderProfiler).GetField("m_threadProfilers", BindingFlags.Static | BindingFlags.NonPublic).GetValue(null) as List<MyProfiler>).Clear();
} }
} }
} }

View File

@@ -18,6 +18,7 @@ using Sandbox;
using Sandbox.Engine.Multiplayer; using Sandbox.Engine.Multiplayer;
using Sandbox.Game.World; using Sandbox.Game.World;
using SteamSDK; using SteamSDK;
using Torch.API;
namespace Torch.Server namespace Torch.Server
{ {
@@ -26,10 +27,27 @@ namespace Torch.Server
/// </summary> /// </summary>
public partial class ChatControl : UserControl public partial class ChatControl : UserControl
{ {
private ITorchServer _server;
public ChatControl() public ChatControl()
{ {
InitializeComponent(); InitializeComponent();
ChatItems.ItemsSource = TorchServer.Multiplayer.ChatView; }
public void BindServer(ITorchServer server)
{
_server = server;
server.Multiplayer.MessageReceived += Refresh;
Refresh();
}
private void Refresh(IChatItem chatItem = null)
{
Dispatcher.Invoke(() =>
{
ChatItems.ItemsSource = null;
ChatItems.ItemsSource = _server.Multiplayer.Chat;
});
} }
private void SendButton_Click(object sender, RoutedEventArgs e) private void SendButton_Click(object sender, RoutedEventArgs e)
@@ -47,7 +65,7 @@ namespace Torch.Server
{ {
//Can't use Message.Text directly because of object ownership in WPF. //Can't use Message.Text directly because of object ownership in WPF.
var text = Message.Text; var text = Message.Text;
TorchServer.Multiplayer.SendMessage(text); _server.Multiplayer.SendMessage(text);
Message.Text = ""; Message.Text = "";
} }
} }

View File

@@ -18,6 +18,7 @@ using Sandbox.Engine.Multiplayer;
using Sandbox.Game.Multiplayer; using Sandbox.Game.Multiplayer;
using Sandbox.Game.World; using Sandbox.Game.World;
using SteamSDK; using SteamSDK;
using Torch.API;
namespace Torch.Server namespace Torch.Server
{ {
@@ -26,27 +27,45 @@ namespace Torch.Server
/// </summary> /// </summary>
public partial class PlayerListControl : UserControl public partial class PlayerListControl : UserControl
{ {
private ITorchServer _server;
public PlayerListControl() public PlayerListControl()
{ {
InitializeComponent(); InitializeComponent();
PlayerList.ItemsSource = TorchServer.Multiplayer.PlayersView; }
public void BindServer(ITorchServer server)
{
_server = server;
server.Multiplayer.PlayerJoined += Refresh;
server.Multiplayer.PlayerLeft += Refresh;
Refresh();
}
private void Refresh(IPlayer player = null)
{
Dispatcher.Invoke(() =>
{
PlayerList.ItemsSource = null;
PlayerList.ItemsSource = _server.Multiplayer.Players;
});
} }
private void KickButton_Click(object sender, RoutedEventArgs e) private void KickButton_Click(object sender, RoutedEventArgs e)
{ {
var player = PlayerList.SelectedItem as PlayerInfo; var player = PlayerList.SelectedItem as Player;
if (player != null) if (player != null)
{ {
TorchServer.Multiplayer.KickPlayer(player.SteamId); _server.Multiplayer.KickPlayer(player.SteamId);
} }
} }
private void BanButton_Click(object sender, RoutedEventArgs e) private void BanButton_Click(object sender, RoutedEventArgs e)
{ {
var player = PlayerList.SelectedItem as PlayerInfo; var player = PlayerList.SelectedItem as Player;
if (player != null) if (player != null)
{ {
TorchServer.Multiplayer.BanPlayer(player.SteamId); _server.Multiplayer.BanPlayer(player.SteamId);
} }
} }
} }

View File

@@ -1,4 +1,4 @@
<Window x:Class="Torch.Server.PistonUI" <Window x:Class="Torch.Server.TorchUI"
xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation" xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
xmlns:d="http://schemas.microsoft.com/expression/blend/2008" xmlns:d="http://schemas.microsoft.com/expression/blend/2008"

View File

@@ -5,6 +5,7 @@ using System.Diagnostics;
using System.IO; using System.IO;
using System.Linq; using System.Linq;
using System.Text; using System.Text;
using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
using System.Timers; using System.Timers;
using System.Windows; using System.Windows;
@@ -16,14 +17,17 @@ using System.Windows.Media;
using System.Windows.Media.Imaging; using System.Windows.Media.Imaging;
using System.Windows.Navigation; using System.Windows.Navigation;
using System.Windows.Shapes; using System.Windows.Shapes;
using Torch.API;
using Timer = System.Timers.Timer;
namespace Torch.Server namespace Torch.Server
{ {
/// <summary> /// <summary>
/// Interaction logic for PistonUI.xaml /// Interaction logic for TorchUI.xaml
/// </summary> /// </summary>
public partial class PistonUI : Window public partial class TorchUI : Window
{ {
private ITorchServer _server;
private DateTime _startTime; private DateTime _startTime;
private readonly Timer _uiUpdate = new Timer private readonly Timer _uiUpdate = new Timer
{ {
@@ -31,13 +35,15 @@ namespace Torch.Server
AutoReset = true, AutoReset = true,
}; };
public PistonUI() public TorchUI(ITorchServer server)
{ {
_server = server;
InitializeComponent(); InitializeComponent();
_startTime = DateTime.Now; _startTime = DateTime.Now;
_uiUpdate.Elapsed += UiUpdate_Elapsed; _uiUpdate.Elapsed += UiUpdate_Elapsed;
TabControl.Items.Add(new TabItem()); Chat.BindServer(server);
PlayerList.BindServer(server);
} }
private void UiUpdate_Elapsed(object sender, ElapsedEventArgs e) private void UiUpdate_Elapsed(object sender, ElapsedEventArgs e)
@@ -61,7 +67,7 @@ namespace Torch.Server
((Button) sender).IsEnabled = false; ((Button) sender).IsEnabled = false;
BtnStop.IsEnabled = true; BtnStop.IsEnabled = true;
_uiUpdate.Start(); _uiUpdate.Start();
TorchServer.Server.StartServerThread(); new Thread(() => _server.Start()).Start();
} }
private void BtnStop_Click(object sender, RoutedEventArgs e) private void BtnStop_Click(object sender, RoutedEventArgs e)
@@ -72,12 +78,12 @@ namespace Torch.Server
//HACK: Uncomment when restarting is possible. //HACK: Uncomment when restarting is possible.
//BtnStart.IsEnabled = true; //BtnStart.IsEnabled = true;
_uiUpdate.Stop(); _uiUpdate.Stop();
TorchServer.Server.StopServer(); _server.Stop();
} }
protected override void OnClosing(CancelEventArgs e) protected override void OnClosing(CancelEventArgs e)
{ {
TorchServer.Reset(); _server.Stop();
} }
private void BtnRestart_Click(object sender, RoutedEventArgs e) private void BtnRestart_Click(object sender, RoutedEventArgs e)

View File

@@ -20,12 +20,11 @@ namespace Torch
{ {
var dispObj = nh.Target as DispatcherObject; var dispObj = nh.Target as DispatcherObject;
Dispatcher dispatcher = dispObj?.Dispatcher; var dispatcher = dispObj?.Dispatcher;
if (dispatcher != null && !dispatcher.CheckAccess()) if (dispatcher != null && !dispatcher.CheckAccess())
{ {
dispatcher.BeginInvoke( dispatcher.BeginInvoke(
(Action)(() => nh.Invoke(this, (Action)(() => nh.Invoke(this, new NotifyCollectionChangedEventArgs(NotifyCollectionChangedAction.Reset))),
new NotifyCollectionChangedEventArgs(NotifyCollectionChangedAction.Reset))),
DispatcherPriority.DataBind); DispatcherPriority.DataBind);
continue; continue;
} }

View File

@@ -3,25 +3,25 @@ using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Text; using System.Text;
using System.Threading.Tasks; using System.Threading.Tasks;
using Torch.API;
namespace Torch namespace Torch
{ {
public static class PlayerInfoCache public static class PlayerInfoCache
{ {
private static readonly Dictionary<ulong, PlayerInfo> _cache = new Dictionary<ulong, PlayerInfo>(); private static readonly Dictionary<ulong, Player> _cache = new Dictionary<ulong, Player>();
public static PlayerInfo GetOrCreate(ulong steamId) public static Player GetOrCreate(ulong steamId)
{ {
PlayerInfo info; if (_cache.TryGetValue(steamId, out Player info))
if (_cache.TryGetValue(steamId, out info))
return info; return info;
info = new PlayerInfo(steamId) {State = ConnectionState.Unknown}; info = new Player(steamId) {State = ConnectionState.Unknown};
_cache.Add(steamId, info); _cache.Add(steamId, info);
return info; return info;
} }
public static void Add(PlayerInfo info) public static void Add(Player info)
{ {
if (_cache.ContainsKey(info.SteamId)) if (_cache.ContainsKey(info.SteamId))
return; return;

View File

@@ -5,6 +5,6 @@ namespace Torch.Commands
public class ChatCommandModule public class ChatCommandModule
{ {
public ITorchPlugin Plugin { get; set; } public ITorchPlugin Plugin { get; set; }
public ITorchServer Server { get; set; } public ITorchBase Server { get; set; }
} }
} }

View File

@@ -1,8 +1,26 @@
namespace Torch.Commands using System.Linq;
using System.Text.RegularExpressions;
namespace Torch.Commands
{ {
public struct CommandContext public struct CommandContext
{ {
public string Argument; public string Argument;
public ulong SteamId; public ulong SteamId;
/// <summary>
/// Splits the argument by single words and quoted blocks.
/// </summary>
/// <returns></returns>
public string[] SplitArgument()
{
var split = Regex.Split(Argument, "(\"[^\"]+\"|\\S+)");
for (var i = 0; i < split.Length; i++)
{
split[i] = Regex.Replace(split[i], "\"", "");
}
return split;
}
} }
} }

View File

@@ -7,12 +7,12 @@ namespace Torch.Commands
{ {
public class CommandSystem public class CommandSystem
{ {
public ITorchServer Server { get; } public ITorchBase Server { get; }
public char Prefix { get; set; } public char Prefix { get; set; }
public Dictionary<string, ChatCommand> Commands { get; } = new Dictionary<string, ChatCommand>(); public Dictionary<string, ChatCommand> Commands { get; } = new Dictionary<string, ChatCommand>();
public CommandSystem(ITorchServer server, char prefix = '/') public CommandSystem(ITorchBase server, char prefix = '/')
{ {
Server = server; Server = server;
Prefix = prefix; Prefix = prefix;
@@ -40,6 +40,12 @@ namespace Torch.Commands
if (commandAttrib == null) if (commandAttrib == null)
continue; continue;
if (Commands.ContainsKey(commandAttrib.Name))
{
Console.WriteLine($"[ERROR]: Command \"{method.Name}\" is already registered!");
continue;
}
var parameters = method.GetParameters(); var parameters = method.GetParameters();
if (parameters.Length != 1 || parameters[0].ParameterType != typeof(CommandContext)) if (parameters.Length != 1 || parameters[0].ParameterType != typeof(CommandContext))
{ {
@@ -82,8 +88,6 @@ namespace Torch.Commands
Commands[cmdName].Invoke(context); Commands[cmdName].Invoke(context);
//Regex.Matches(command, "(\"[^\"]+\"|\\S+)");
} }
} }
} }

View File

@@ -18,40 +18,43 @@ using Sandbox.Engine.Multiplayer;
using Sandbox.Game.Multiplayer; using Sandbox.Game.Multiplayer;
using Sandbox.Game.World; using Sandbox.Game.World;
using SteamSDK; using SteamSDK;
using Torch.Server.ViewModels; using Torch.API;
using Torch.ViewModels;
using VRage.Game; using VRage.Game;
using VRage.Library.Collections; using VRage.Library.Collections;
using VRage.Network; using VRage.Network;
using VRage.Utils; using VRage.Utils;
namespace Torch.Server namespace Torch
{ {
/// <summary> /// <summary>
/// Provides a proxy to the game's multiplayer-related functions. /// Provides a proxy to the game's multiplayer-related functions.
/// </summary> /// </summary>
public class MultiplayerManager public class MultiplayerManager : IMultiplayer
{ {
public event Action<PlayerInfo> PlayerJoined; public event Action<IPlayer> PlayerJoined;
public event Action<PlayerInfo> PlayerLeft; public event Action<IPlayer> PlayerLeft;
public event Action<ChatItemInfo> ChatMessageReceived; public event Action<IChatItem> MessageReceived;
public MTObservableCollection<PlayerInfo> PlayersView { get; } = new MTObservableCollection<PlayerInfo>(); //public MTObservableCollection<MyPlayer> PlayersView { get; } = new MTObservableCollection<MyPlayer>();
public MTObservableCollection<ChatItemInfo> ChatView { get; } = new MTObservableCollection<ChatItemInfo>(); //public MTObservableCollection<ChatItem> ChatView { get; } = new MTObservableCollection<ChatItem>();
public PlayerInfo LocalPlayer { get; private set; } public List<IChatItem> Chat { get; } = new List<IChatItem>();
public Dictionary<ulong, IPlayer> Players { get; } = new Dictionary<ulong, IPlayer>();
public Player LocalPlayer { get; private set; }
private TorchServer _server; private readonly ITorchBase _torch;
internal MultiplayerManager(TorchServer server) internal MultiplayerManager(ITorchBase torch)
{ {
_server = server; _torch = torch;
_server.Server.SessionLoaded += OnSessionLoaded; _torch.SessionLoaded += OnSessionLoaded;
} }
public void KickPlayer(ulong steamId) => _server.Server.BeginGameAction(() => MyMultiplayer.Static.KickClient(steamId)); public void KickPlayer(ulong steamId) => _torch.BeginGameAction(() => MyMultiplayer.Static.KickClient(steamId));
public void BanPlayer(ulong steamId, bool banned = true) public void BanPlayer(ulong steamId, bool banned = true)
{ {
_server.Server.BeginGameAction(() => _torch.BeginGameAction(() =>
{ {
MyMultiplayer.Static.BanClient(steamId, banned); MyMultiplayer.Static.BanClient(steamId, banned);
if (_gameOwnerIds.ContainsKey(steamId)) if (_gameOwnerIds.ContainsKey(steamId))
@@ -65,12 +68,12 @@ namespace Torch.Server
public void SendMessage(string message) public void SendMessage(string message)
{ {
MyMultiplayer.Static.SendChatMessage(message); MyMultiplayer.Static.SendChatMessage(message);
ChatView.Add(new ChatItemInfo(LocalPlayer, message)); //ChatView.Add(new ChatItem(LocalPlayer, message));
} }
private void OnSessionLoaded() private void OnSessionLoaded()
{ {
LocalPlayer = new PlayerInfo(MyMultiplayer.Static.ServerId) { Name = "Server", State = ConnectionState.Connected }; LocalPlayer = new Player(MyMultiplayer.Static.ServerId) { Name = "Server", State = ConnectionState.Connected };
MyMultiplayer.Static.ChatMessageReceived += OnChatMessage; MyMultiplayer.Static.ChatMessageReceived += OnChatMessage;
MyMultiplayer.Static.ClientKicked += OnClientKicked; MyMultiplayer.Static.ClientKicked += OnClientKicked;
@@ -79,21 +82,21 @@ namespace Torch.Server
//TODO: Move these with the methods? //TODO: Move these with the methods?
RemoveHandlers(); RemoveHandlers();
SteamSDK.SteamServerAPI.Instance.GameServer.ValidateAuthTicketResponse += ValidateAuthTicketResponse; SteamServerAPI.Instance.GameServer.ValidateAuthTicketResponse += ValidateAuthTicketResponse;
SteamSDK.SteamServerAPI.Instance.GameServer.UserGroupStatus += UserGroupStatus; SteamServerAPI.Instance.GameServer.UserGroupStatus += UserGroupStatus;
_members = (List<ulong>)typeof(MyDedicatedServerBase).GetField("m_members", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(MyMultiplayer.Static); _members = (List<ulong>)typeof(MyDedicatedServerBase).GetField("m_members", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(MyMultiplayer.Static);
_waitingForGroup = (HashSet<ulong>)typeof(MyDedicatedServerBase).GetField("m_waitingForGroup", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(MyMultiplayer.Static); _waitingForGroup = (HashSet<ulong>)typeof(MyDedicatedServerBase).GetField("m_waitingForGroup", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(MyMultiplayer.Static);
} }
private void OnChatMessage(ulong steamId, string message, ChatEntryTypeEnum chatType) private void OnChatMessage(ulong steamId, string message, ChatEntryTypeEnum chatType)
{ {
var player = PlayersView.FirstOrDefault(p => p.SteamId == steamId); var player = MyMultiplayer.Static.GetMemberName(steamId);
if (player == null || player == LocalPlayer) if (string.IsNullOrEmpty(player))
return; return;
var info = new ChatItemInfo(player, message); var info = new ChatItem(new Player(steamId), message);
ChatView.Add(info); Chat.Add(info);
ChatMessageReceived?.Invoke(info); MessageReceived?.Invoke(info);
} }
/// <summary> /// <summary>
@@ -102,8 +105,20 @@ namespace Torch.Server
private void OnPlayerRequesting(PlayerRequestArgs args) private void OnPlayerRequesting(PlayerRequestArgs args)
{ {
var steamId = args.PlayerId.SteamId; var steamId = args.PlayerId.SteamId;
var player = new PlayerInfo(steamId) {State = ConnectionState.Connected};
PlayersView.Add(player); IPlayer player;
if (!Players.ContainsKey(steamId))
{
player = new Player(steamId) { State = ConnectionState.Connected };
Players.Add(steamId, player);
}
else
{
player = Players[steamId];
player.SetConnectionState(ConnectionState.Connected);
}
Logger.Write($"{player.Name} connected.");
PlayerJoined?.Invoke(player); PlayerJoined?.Invoke(player);
} }
@@ -114,13 +129,12 @@ namespace Torch.Server
private void OnClientLeft(ulong steamId, ChatMemberStateChangeEnum stateChange) private void OnClientLeft(ulong steamId, ChatMemberStateChangeEnum stateChange)
{ {
var player = PlayersView.FirstOrDefault(p => p.SteamId == steamId); if (!Players.ContainsKey(steamId))
if (player == null)
return; return;
player.State = (ConnectionState)stateChange; var player = Players[steamId];
PlayersView.Remove(player); Logger.Write($"{player.Name} disconnected ({(ConnectionState)stateChange}).");
player.SetConnectionState((ConnectionState)stateChange);
PlayerLeft?.Invoke(player); PlayerLeft?.Invoke(player);
} }

View File

@@ -1,11 +1,14 @@
using Sandbox.Engine.Multiplayer; using System;
using System.Collections.Generic;
using Sandbox.Engine.Multiplayer;
using Torch.API;
namespace Torch namespace Torch
{ {
/// <summary> /// <summary>
/// Stores player information in an observable format. /// Stores player information in an observable format.
/// </summary> /// </summary>
public class PlayerInfo : ViewModel public class Player : ViewModel, IPlayer
{ {
private ulong _steamId; private ulong _steamId;
private string _name; private string _name;
@@ -23,17 +26,30 @@ namespace Torch
set { _name = value; OnPropertyChanged(); } set { _name = value; OnPropertyChanged(); }
} }
//TODO: track identity history
public List<ulong> IdentityIds { get; } = new List<ulong>();
public DateTime LastConnected { get; private set; }
public ConnectionState State public ConnectionState State
{ {
get { return _state; } get { return _state; }
set { _state = value; OnPropertyChanged(); } set { _state = value; OnPropertyChanged(); }
} }
public PlayerInfo(ulong steamId) public Player(ulong steamId)
{ {
_steamId = steamId; _steamId = steamId;
_name = MyMultiplayer.Static.GetMemberName(steamId); _name = MyMultiplayer.Static.GetMemberName(steamId);
_state = ConnectionState.Unknown; _state = ConnectionState.Unknown;
} }
public void SetConnectionState(ConnectionState state)
{
if (state == ConnectionState.Connected)
LastConnected = DateTime.Now;
State = state;
}
} }
} }

View File

@@ -15,7 +15,7 @@ using VRage.Library.Collections;
namespace Torch namespace Torch
{ {
public class PluginManager public class PluginManager : IPluginManager
{ {
//TODO: Disable reloading if the plugin has static elements because they prevent a full reload. //TODO: Disable reloading if the plugin has static elements because they prevent a full reload.

View File

@@ -126,12 +126,14 @@
<Compile Include="Commands\ChatCommandModule.cs" /> <Compile Include="Commands\ChatCommandModule.cs" />
<Compile Include="Commands\CommandContext.cs" /> <Compile Include="Commands\CommandContext.cs" />
<Compile Include="Commands\CommandSystem.cs" /> <Compile Include="Commands\CommandSystem.cs" />
<Compile Include="ConnectionState.cs" />
<Compile Include="Logger.cs" /> <Compile Include="Logger.cs" />
<Compile Include="MultiplayerManager.cs" />
<Compile Include="TorchBase.cs" />
<Compile Include="TorchPlugin.cs" /> <Compile Include="TorchPlugin.cs" />
<Compile Include="PlayerInfo.cs" /> <Compile Include="Player.cs" />
<Compile Include="Collections\PlayerInfoCache.cs" /> <Compile Include="Collections\PlayerInfoCache.cs" />
<Compile Include="SteamService.cs" /> <Compile Include="SteamService.cs" />
<Compile Include="ViewModels\ChatItem.cs" />
<Compile Include="ViewModels\ModViewModel.cs" /> <Compile Include="ViewModels\ModViewModel.cs" />
<Compile Include="Collections\MTObservableCollection.cs" /> <Compile Include="Collections\MTObservableCollection.cs" />
<Compile Include="Extensions\MyPlayerCollectionExtensions.cs" /> <Compile Include="Extensions\MyPlayerCollectionExtensions.cs" />

111
Torch/TorchBase.cs Normal file
View File

@@ -0,0 +1,111 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Sandbox;
using Torch.API;
namespace Torch
{
public abstract class TorchBase : ITorchBase
{
public IPluginManager Plugins { get; protected set; }
public IMultiplayer Multiplayer { get; protected set; }
public event Action SessionLoaded;
protected void InvokeSessionLoaded()
{
SessionLoaded?.Invoke();
}
protected TorchBase()
{
Plugins = new PluginManager();
Multiplayer = new MultiplayerManager(this);
}
/// <summary>
/// Invokes an action on the game thread and blocks until completion
/// </summary>
/// <param name="action"></param>
public void GameAction(Action action)
{
if (action == null)
return;
try
{
if (Thread.CurrentThread == MySandboxGame.Static.UpdateThread)
{
action();
}
else
{
AutoResetEvent e = new AutoResetEvent(false);
MySandboxGame.Static.Invoke(() =>
{
try
{
action();
}
catch (Exception ex)
{
//log
}
finally
{
e.Set();
}
});
//timeout so we don't accidentally hang the server
e.WaitOne(60000);
}
}
catch (Exception ex)
{
//we need a logger :(
}
}
/// <summary>
/// Queues an action for invocation on the game thread and optionally runs a callback on completion
/// </summary>
/// <param name="action"></param>
/// <param name="callback"></param>
/// <param name="state"></param>
public void BeginGameAction(Action action, Action<object> callback = null, object state = null)
{
if (action == null)
return;
try
{
if (Thread.CurrentThread == MySandboxGame.Static.UpdateThread)
{
action();
}
else
{
Task.Run(() =>
{
GameAction(action);
callback?.Invoke(state);
});
}
}
catch (Exception ex)
{
// log
}
}
public abstract void Start();
public abstract void Stop();
public abstract void Init();
}
}

View File

@@ -1,14 +1,15 @@
using System; using System;
using Torch.API;
namespace Torch.Server.ViewModels namespace Torch.ViewModels
{ {
public class ChatItemInfo : ViewModel public class ChatItem : ViewModel, IChatItem
{ {
private PlayerInfo _sender; private IPlayer _sender;
private string _message; private string _message;
private DateTime _timestamp; private DateTime _timestamp;
public PlayerInfo Sender public IPlayer Player
{ {
get { return _sender; } get { return _sender; }
set { _sender = value; OnPropertyChanged(); } set { _sender = value; OnPropertyChanged(); }
@@ -20,19 +21,23 @@ namespace Torch.Server.ViewModels
set { _message = value; OnPropertyChanged(); } set { _message = value; OnPropertyChanged(); }
} }
public DateTime Timestamp public DateTime Time
{ {
get { return _timestamp; } get { return _timestamp; }
set { _timestamp = value; OnPropertyChanged(); } set { _timestamp = value; OnPropertyChanged(); }
} }
public string Time => Timestamp.ToShortTimeString(); public string TimeString => Time.ToShortTimeString();
public ChatItemInfo(PlayerInfo sender, string message) public ChatItem(IPlayer sender, string message, DateTime timestamp = default(DateTime))
{ {
_sender = sender; _sender = sender;
_message = message; _message = message;
if (timestamp == default(DateTime))
_timestamp = DateTime.Now; _timestamp = DateTime.Now;
else
_timestamp = timestamp;
} }
} }
} }