using System.Net.Http;
using System.Text;
using System.Text.Json;
using System.Text.Json.Serialization;
using Content.Server.Maps;
using Content.Shared.Corvax.CCCVars;
using Content.Shared.GameTicking;
using Robust.Shared.Configuration;
namespace Content.Server.Corvax.RoundNotifications;
///
/// Listen game events and send notifications to Discord
///
public sealed class RoundNotificationsSystem : EntitySystem
{
[Dependency] private readonly IConfigurationManager _config = default!;
[Dependency] private readonly IGameMapManager _gameMapManager = default!;
private ISawmill _sawmill = default!;
private readonly HttpClient _httpClient = new();
private string _webhookUrl = String.Empty;
private string _roleId = String.Empty;
private bool _roundStartOnly;
///
public override void Initialize()
{
SubscribeLocalEvent(OnRoundRestart);
SubscribeLocalEvent(OnRoundStarted);
SubscribeLocalEvent(OnRoundEnded);
_config.OnValueChanged(CCCVars.DiscordRoundWebhook, value => _webhookUrl = value, true);
_config.OnValueChanged(CCCVars.DiscordRoundRoleId, value => _roleId = value, true);
_config.OnValueChanged(CCCVars.DiscordRoundStartOnly, value => _roundStartOnly = value, true);
_sawmill = IoCManager.Resolve().GetSawmill("notifications");
}
private void OnRoundRestart(RoundRestartCleanupEvent e)
{
if (String.IsNullOrEmpty(_webhookUrl))
return;
var payload = new WebhookPayload()
{
Content = Loc.GetString("discord-round-new"),
};
if (!String.IsNullOrEmpty(_roleId))
{
payload = new WebhookPayload()
{
Content = $"<@&{_roleId}> {Loc.GetString("discord-round-new")}",
AllowedMentions = new Dictionary
{
{ "roles", new []{ _roleId } }
},
};
}
SendDiscordMessage(payload);
}
private void OnRoundStarted(RoundStartedEvent e)
{
if (String.IsNullOrEmpty(_webhookUrl))
return;
var map = _gameMapManager.GetSelectedMap();
var mapName = map?.MapName ?? Loc.GetString("discord-round-unknown-map");
var text = Loc.GetString("discord-round-start",
("id", e.RoundId),
("map", mapName));
var payload = new WebhookPayload() { Content = text };
SendDiscordMessage(payload);
}
private void OnRoundEnded(RoundEndedEvent e)
{
if (String.IsNullOrEmpty(_webhookUrl) || _roundStartOnly)
return;
var text = Loc.GetString("discord-round-end",
("id", e.RoundId),
("hours", e.RoundDuration.Hours),
("minutes", e.RoundDuration.Minutes),
("seconds", e.RoundDuration.Seconds));
var payload = new WebhookPayload() { Content = text };
SendDiscordMessage(payload);
}
private async void SendDiscordMessage(WebhookPayload payload)
{
var request = await _httpClient.PostAsync(_webhookUrl,
new StringContent(JsonSerializer.Serialize(payload), Encoding.UTF8, "application/json"));
var content = await request.Content.ReadAsStringAsync();
if (!request.IsSuccessStatusCode)
{
_sawmill.Log(LogLevel.Error, $"Discord returned bad status code when posting message: {request.StatusCode}\nResponse: {content}");
return;
}
}
private struct WebhookPayload
{
[JsonPropertyName("content")]
public string Content { get; set; } = "";
[JsonPropertyName("allowed_mentions")]
public Dictionary AllowedMentions { get; set; } =
new()
{
{ "parse", Array.Empty() }
};
public WebhookPayload()
{
}
}
}