OOC <-> Discord link. (#201)
This commit is contained in:
committed by
GitHub
parent
903961771b
commit
747cb15888
154
Content.Server/MoMMILink.cs
Normal file
154
Content.Server/MoMMILink.cs
Normal file
@@ -0,0 +1,154 @@
|
||||
using System;
|
||||
using System.IO;
|
||||
using System.Net;
|
||||
using System.Net.Http;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
using Content.Server.Interfaces;
|
||||
using Content.Server.Interfaces.Chat;
|
||||
using Newtonsoft.Json;
|
||||
using Robust.Server.Interfaces.ServerStatus;
|
||||
using Robust.Server.ServerStatus;
|
||||
using Robust.Shared.Asynchronous;
|
||||
using Robust.Shared.Interfaces.Configuration;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Log;
|
||||
using Robust.Shared.Utility;
|
||||
|
||||
namespace Content.Server
|
||||
{
|
||||
internal sealed class MoMMILink : IMoMMILink, IPostInjectInit
|
||||
{
|
||||
#pragma warning disable 649
|
||||
[Dependency] private readonly IConfigurationManager _configurationManager;
|
||||
[Dependency] private readonly IStatusHost _statusHost;
|
||||
[Dependency] private readonly IChatManager _chatManager;
|
||||
[Dependency] private readonly ITaskManager _taskManager;
|
||||
#pragma warning restore 649
|
||||
|
||||
private readonly HttpClient _httpClient = new HttpClient();
|
||||
|
||||
void IPostInjectInit.PostInject()
|
||||
{
|
||||
_configurationManager.RegisterCVar<string>("status.mommiurl", null);
|
||||
_configurationManager.RegisterCVar<string>("status.mommipassword", null);
|
||||
|
||||
_statusHost.AddHandler(_handleChatPost);
|
||||
}
|
||||
|
||||
public async void SendOOCMessage(string sender, string message)
|
||||
{
|
||||
var sentMessage = new MoMMIMessageOOC
|
||||
{
|
||||
Sender = sender,
|
||||
Contents = message
|
||||
};
|
||||
|
||||
await _sendMessageInternal("ooc", sentMessage);
|
||||
}
|
||||
|
||||
private async Task _sendMessageInternal(string type, object messageObject)
|
||||
{
|
||||
var url = _configurationManager.GetCVar<string>("status.mommiurl");
|
||||
var password = _configurationManager.GetCVar<string>("status.mommipassword");
|
||||
if (string.IsNullOrWhiteSpace(url))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (string.IsNullOrWhiteSpace(password))
|
||||
{
|
||||
Logger.WarningS("mommi", "MoMMI URL specified but not password!");
|
||||
return;
|
||||
}
|
||||
|
||||
var sentMessage = new MoMMIMessageBase
|
||||
{
|
||||
Password = password,
|
||||
Type = type,
|
||||
Contents = messageObject
|
||||
};
|
||||
|
||||
var jsonMessage = JsonConvert.SerializeObject(sentMessage);
|
||||
var request =
|
||||
await _httpClient.PostAsync(url, new StringContent(jsonMessage, Encoding.UTF8, "application/json"));
|
||||
|
||||
if (!request.IsSuccessStatusCode)
|
||||
{
|
||||
throw new Exception($"MoMMI returned bad status code: {request.StatusCode}");
|
||||
}
|
||||
}
|
||||
|
||||
private bool _handleChatPost(HttpMethod method, HttpListenerRequest request, HttpListenerResponse response)
|
||||
{
|
||||
if (method != HttpMethod.Post || request.Url.AbsolutePath != "/ooc")
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
var password = _configurationManager.GetCVar<string>("status.mommipassword");
|
||||
OOCPostMessage message;
|
||||
|
||||
using (var streamReader = new StreamReader(request.InputStream, EncodingHelpers.UTF8))
|
||||
using (var jsonReader = new JsonTextReader(streamReader))
|
||||
{
|
||||
var serializer = new JsonSerializer();
|
||||
try
|
||||
{
|
||||
message = serializer.Deserialize<OOCPostMessage>(jsonReader);
|
||||
}
|
||||
catch (JsonSerializationException)
|
||||
{
|
||||
response.Respond(method, "400 Bad Request", HttpStatusCode.BadRequest, "text/plain");
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
if (message == null)
|
||||
{
|
||||
response.Respond(method, "400 Bad Request", HttpStatusCode.BadRequest, "text/plain");
|
||||
return true;
|
||||
}
|
||||
|
||||
if (message.Password != password)
|
||||
{
|
||||
response.Respond(method, "Incorrect password", HttpStatusCode.Forbidden, "text/plain");
|
||||
return true;
|
||||
}
|
||||
|
||||
_taskManager.RunOnMainThread(() => _chatManager.SendHookOOC(message.Sender, message.Contents));
|
||||
|
||||
response.Respond(method, "Message received", HttpStatusCode.OK, "text/plain");
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
[JsonObject(MemberSerialization.Fields)]
|
||||
private class MoMMIMessageBase
|
||||
{
|
||||
[JsonProperty("password")] public string Password;
|
||||
|
||||
[JsonProperty("type")] public string Type;
|
||||
|
||||
[JsonProperty("contents")] public object Contents;
|
||||
}
|
||||
|
||||
[JsonObject(MemberSerialization.Fields)]
|
||||
private class MoMMIMessageOOC
|
||||
{
|
||||
[JsonProperty("sender")] public string Sender;
|
||||
|
||||
[JsonProperty("contents")] public string Contents;
|
||||
}
|
||||
|
||||
[JsonObject(MemberSerialization.Fields, ItemRequired = Required.Always)]
|
||||
private class OOCPostMessage
|
||||
{
|
||||
[JsonProperty("password")] public string Password;
|
||||
|
||||
[JsonProperty("sender")] public string Sender;
|
||||
|
||||
[JsonProperty("contents")] public string Contents;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user