Added external discord bot api

2023-05-23 21:03:09 +02:00
parent 732fbdd46a
commit cd4d278ceb
3 changed files with 161 additions and 0 deletions
--- a/Moonlight/App/Http/Controllers/Api/Moonlight/DiscordBotController.cs
+++ b/Moonlight/App/Http/Controllers/Api/Moonlight/DiscordBotController.cs
@@ -0,0 +1,149 @@
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.EntityFrameworkCore;
+using Moonlight.App.ApiClients.Wings;
+using Moonlight.App.ApiClients.Wings.Resources;
+using Moonlight.App.Database.Entities;
+using Moonlight.App.Http.Requests.DiscordBot.Requests;
+using Moonlight.App.Repositories;
+using Moonlight.App.Services;
+
+namespace Moonlight.App.Http.Controllers.Api.Moonlight;
+
+[ApiController]
+[Route("api/moonlight/discordbot")]
+public class DiscordBotController : Controller
+{
+    private readonly Repository<User> UserRepository;
+    private readonly Repository<Server> ServerRepository;
+    private readonly ServerService ServerService;
+    private readonly string Token = "";
+    private readonly bool Enable;
+
+    public DiscordBotController(
+        Repository<User> userRepository,
+        Repository<Server> serverRepository,
+        ServerService serverService,
+        ConfigService configService)
+    {
+        UserRepository = userRepository;
+        ServerRepository = serverRepository;
+        ServerService = serverService;
+
+        var config = configService
+            .GetSection("Moonlight")
+            .GetSection("DiscordBotApi");
+
+        Enable = config.GetValue<bool>("Enable");
+
+        if (Enable)
+        {
+            Token = config.GetValue<string>("Token");
+        }
+    }
+    
+    [HttpGet("{id}/link")]
+    public async Task<ActionResult> GetLink(ulong id)
+    {
+        if (!await IsAuth(Request))
+            return StatusCode(403);
+        
+        if (await GetUserFromDiscordId(id) == null)
+        {
+            return BadRequest();
+        }
+
+        return Ok();
+    }
+
+    [HttpGet("{id}/servers")]
+    public async Task<ActionResult<Server[]>> GetServers(ulong id)
+    {
+        if (!await IsAuth(Request))
+            return StatusCode(403);
+        
+        var user = await GetUserFromDiscordId(id);
+
+        if (user == null)
+            return BadRequest();
+
+        return ServerRepository
+            .Get()
+            .Include(x => x.Owner)
+            .Include(x => x.Image)
+            .Where(x => x.Owner.Id == user.Id)
+            .ToArray();
+    }
+
+    [HttpPost("{id}/servers/{uuid}")]
+    public async Task<ActionResult> SetPowerState(ulong id, Guid uuid, [FromBody] SetPowerSignal signal)
+    {
+        if (!await IsAuth(Request))
+            return StatusCode(403);
+        
+        var user = await GetUserFromDiscordId(id);
+
+        if (user == null)
+            return BadRequest();
+
+        var server = ServerRepository
+            .Get()
+            .Include(x => x.Owner)
+            .FirstOrDefault(x => x.Owner.Id == user.Id && x.Uuid == uuid);
+
+        if (server == null)
+            return NotFound();
+
+        if (Enum.TryParse(signal.Signal, true, out PowerSignal powerSignal))
+        {
+            await ServerService.SetPowerState(server, powerSignal);
+            return Ok();
+        }
+        else
+            return BadRequest();
+    }
+
+    [HttpGet("{id}/servers/{uuid}")]
+    public async Task<ActionResult<ServerDetails>> GetServerDetails(ulong id, Guid uuid)
+    {
+        if (!await IsAuth(Request))
+            return StatusCode(403);
+        
+        var user = await GetUserFromDiscordId(id);
+
+        if (user == null)
+            return BadRequest();
+
+        var server = ServerRepository
+            .Get()
+            .Include(x => x.Owner)
+            .FirstOrDefault(x => x.Owner.Id == user.Id && x.Uuid == uuid);
+
+        if (server == null)
+            return NotFound();
+
+        return await ServerService.GetDetails(server);
+    }
+
+    private Task<User?> GetUserFromDiscordId(ulong discordId)
+    {
+        var user = UserRepository
+            .Get()
+            .FirstOrDefault(x => x.DiscordId == discordId);
+
+        return Task.FromResult(user);
+    }
+
+    private Task<bool> IsAuth(HttpRequest request)
+    {
+        if (!Enable)
+            return Task.FromResult(false);
+        
+        if (string.IsNullOrEmpty(request.Headers.Authorization))
+            return Task.FromResult(false);
+        
+        if(request.Headers.Authorization == Token)
+            return Task.FromResult(true);
+        
+        return Task.FromResult(false);
+    }
+}