mirror of
https://github.com/wabbajack-tools/wabbajack.git
synced 2024-08-30 18:42:17 +00:00
Discord integration
This commit is contained in:
parent
a494839a09
commit
d93367d351
25
Wabbajack.Server.Test/DiscordFrontentTests.cs
Normal file
25
Wabbajack.Server.Test/DiscordFrontentTests.cs
Normal file
@ -0,0 +1,25 @@
|
||||
using System;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Wabbajack.BuildServer.Test;
|
||||
using Wabbajack.Server.Services;
|
||||
using Xunit;
|
||||
using Xunit.Abstractions;
|
||||
|
||||
namespace Wabbajack.Server.Test
|
||||
{
|
||||
public class DiscordFrontentTests: ABuildServerSystemTest
|
||||
{
|
||||
public DiscordFrontentTests(ITestOutputHelper output, SingletonAdaptor<BuildServerFixture> fixture) : base(output, fixture)
|
||||
{
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task CanLogIn()
|
||||
{
|
||||
var frontend = Fixture.GetService<DiscordFrontend>();
|
||||
frontend.Start();
|
||||
}
|
||||
|
||||
}
|
||||
}
|
121
Wabbajack.Server/Services/DiscordFrontend.cs
Normal file
121
Wabbajack.Server/Services/DiscordFrontend.cs
Normal file
@ -0,0 +1,121 @@
|
||||
using System;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using Discord;
|
||||
using Discord.WebSocket;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using OMODFramework;
|
||||
using Wabbajack.BuildServer;
|
||||
using Wabbajack.Server.DataLayer;
|
||||
using Utils = Wabbajack.Common.Utils;
|
||||
|
||||
namespace Wabbajack.Server.Services
|
||||
{
|
||||
public class DiscordFrontend : IStartable
|
||||
{
|
||||
private ILogger<DiscordFrontend> _logger;
|
||||
private AppSettings _settings;
|
||||
private QuickSync _quickSync;
|
||||
private DiscordSocketClient _client;
|
||||
private SqlService _sql;
|
||||
|
||||
public DiscordFrontend(ILogger<DiscordFrontend> logger, AppSettings settings, QuickSync quickSync, SqlService sql)
|
||||
{
|
||||
_logger = logger;
|
||||
_settings = settings;
|
||||
_quickSync = quickSync;
|
||||
|
||||
_client = new DiscordSocketClient();
|
||||
|
||||
_client.Log += LogAsync;
|
||||
_client.Ready += ReadyAsync;
|
||||
_client.MessageReceived += MessageReceivedAsync;
|
||||
|
||||
_sql = sql;
|
||||
}
|
||||
|
||||
private async Task MessageReceivedAsync(SocketMessage arg)
|
||||
{
|
||||
_logger.LogInformation(arg.Content);
|
||||
if (arg.Content.StartsWith("!dervenin"))
|
||||
{
|
||||
var parts = arg.Content.Split(" ", StringSplitOptions.RemoveEmptyEntries);
|
||||
if (parts[0] != "!dervenin")
|
||||
return;
|
||||
|
||||
if (parts.Length == 1)
|
||||
{
|
||||
await ReplyTo(arg, "Wat?");
|
||||
}
|
||||
|
||||
if (parts[1] == "purge-nexus-cache")
|
||||
{
|
||||
if (parts.Length != 3)
|
||||
{
|
||||
await ReplyTo(arg, "Welp you did that wrong, gotta give me a mod-id or url");
|
||||
return;
|
||||
}
|
||||
await PurgeNexusCache(arg, parts[2]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private async Task PurgeNexusCache(SocketMessage arg, string mod)
|
||||
{
|
||||
if (Uri.TryCreate(mod, UriKind.Absolute, out var url))
|
||||
{
|
||||
mod = url.AbsolutePath.Split("/", StringSplitOptions.RemoveEmptyEntries).Last();
|
||||
}
|
||||
|
||||
if (int.TryParse(mod, out var mod_id))
|
||||
{
|
||||
await _sql.PurgeNexusCache(mod_id);
|
||||
await _quickSync.Notify<ListValidator>();
|
||||
await ReplyTo(arg, $"It is done, {mod_id} has been purged, list validation has been triggered");
|
||||
}
|
||||
}
|
||||
|
||||
private async Task ReplyTo(SocketMessage socketMessage, string message)
|
||||
{
|
||||
await socketMessage.Channel.SendMessageAsync(message);
|
||||
}
|
||||
|
||||
private async Task ReadyAsync()
|
||||
{
|
||||
}
|
||||
|
||||
private async Task LogAsync(LogMessage arg)
|
||||
{
|
||||
switch (arg.Severity)
|
||||
{
|
||||
case LogSeverity.Info:
|
||||
_logger.LogInformation(arg.Message);
|
||||
break;
|
||||
case LogSeverity.Warning:
|
||||
_logger.LogWarning(arg.Message);
|
||||
break;
|
||||
case LogSeverity.Critical:
|
||||
_logger.LogCritical(arg.Message);
|
||||
break;
|
||||
case LogSeverity.Error:
|
||||
_logger.LogError(arg.Exception, arg.Message);
|
||||
break;
|
||||
case LogSeverity.Verbose:
|
||||
_logger.LogTrace(arg.Message);
|
||||
break;
|
||||
case LogSeverity.Debug:
|
||||
_logger.LogDebug(arg.Message);
|
||||
break;
|
||||
default:
|
||||
throw new ArgumentOutOfRangeException();
|
||||
}
|
||||
}
|
||||
|
||||
public void Start()
|
||||
{
|
||||
_client.LoginAsync(TokenType.Bot, Utils.FromEncryptedJson<string>("discord-key").Result).Wait();
|
||||
_client.StartAsync().Wait();
|
||||
}
|
||||
|
||||
}
|
||||
}
|
@ -72,6 +72,7 @@ namespace Wabbajack.Server
|
||||
services.AddSingleton<MirrorUploader>();
|
||||
services.AddSingleton<MirrorQueueService>();
|
||||
services.AddSingleton<Watchdog>();
|
||||
services.AddSingleton<DiscordFrontend>();
|
||||
|
||||
services.AddMvc();
|
||||
services.AddControllers()
|
||||
@ -131,6 +132,7 @@ namespace Wabbajack.Server
|
||||
app.UseService<MirrorUploader>();
|
||||
app.UseService<MirrorQueueService>();
|
||||
app.UseService<Watchdog>();
|
||||
app.UseService<DiscordFrontend>();
|
||||
|
||||
app.Use(next =>
|
||||
{
|
||||
|
@ -15,6 +15,7 @@
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="Dapper" Version="2.0.35" />
|
||||
<PackageReference Include="Discord.Net.WebSocket" Version="2.2.0" />
|
||||
<PackageReference Include="FluentFTP" Version="33.0.2" />
|
||||
<PackageReference Include="Microsoft.AspNetCore.Authentication.Core" Version="2.2.0" />
|
||||
<PackageReference Include="Microsoft.AspNetCore.Mvc.NewtonsoftJson" Version="3.1.9" />
|
||||
|
Loading…
Reference in New Issue
Block a user