Servers/MoonlightServers.ApiServer/Http/Controllers/Admin/Nodes/NodeAllocationsController.cs

using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.EntityFrameworkCore;
using MoonCore.Exceptions;
using MoonCore.Extended.Abstractions;
using MoonCore.Extended.Models;
using MoonCore.Models;
using MoonlightServers.ApiServer.Database.Entities;
using MoonlightServers.ApiServer.Mappers;
using MoonlightServers.Shared.Http.Requests.Admin.NodeAllocations;
using MoonlightServers.Shared.Http.Responses.Admin.NodeAllocations;

namespace MoonlightServers.ApiServer.Http.Controllers.Admin.Nodes;

[ApiController]
[Route("api/admin/servers/nodes/{nodeId:int}/allocations")]
public class NodeAllocationsController : Controller
{
    private readonly DatabaseRepository<Node> NodeRepository;
    private readonly DatabaseRepository<Allocation> AllocationRepository;

    public NodeAllocationsController(
        DatabaseRepository<Node> nodeRepository,
        DatabaseRepository<Allocation> allocationRepository
    )
    {
        NodeRepository = nodeRepository;
        AllocationRepository = allocationRepository;
    }

    [HttpGet]
    [Authorize(Policy = "permissions:admin.servers.nodes.get")]
    public async Task<ActionResult<IPagedData<NodeAllocationResponse>>> Get(
        [FromRoute] int nodeId,
        [FromQuery] PagedOptions options
    )
    {
        var count = await AllocationRepository
            .Get()
            .CountAsync(x => x.Node.Id == nodeId);

        var allocations = await AllocationRepository
            .Get()
            .OrderBy(x => x.Id)
            .Skip(options.Page * options.PageSize)
            .Take(options.PageSize)
            .Where(x => x.Node.Id == nodeId)
            .AsNoTracking()
            .ProjectToAdminResponse()
            .ToArrayAsync();

        return new PagedData<NodeAllocationResponse>()
        {
            Items = allocations,
            CurrentPage = options.Page,
            PageSize = options.PageSize,
            TotalItems = count,
            TotalPages = (int)Math.Ceiling(Math.Max(0, count) / (double)options.PageSize)
        };
    }

    [HttpGet("{id:int}")]
    [Authorize(Policy = "permissions:admin.servers.nodes.get")]
    public async Task<ActionResult<NodeAllocationResponse>> GetSingle([FromRoute] int nodeId, [FromRoute] int id)
    {
        var allocation = await AllocationRepository
            .Get()
            .Where(x => x.Node.Id == nodeId)
            .AsNoTracking()
            .ProjectToAdminResponse()
            .FirstOrDefaultAsync(x => x.Id == id);

        if (allocation == null)
            return Problem("No allocation with that id found", statusCode: 400);

        return allocation;
    }

    [HttpPost]
    [Authorize(Policy = "permissions:admin.servers.nodes.create")]
    public async Task<ActionResult<NodeAllocationResponse>> Create(
        [FromRoute] int nodeId,
        [FromBody] CreateNodeAllocationRequest request
    )
    {
        var node = await NodeRepository
            .Get()
            .FirstOrDefaultAsync(x => x.Id == nodeId);

        if (node == null)
            return Problem("No node with that id found", statusCode: 404);

        var allocation = AllocationMapper.ToAllocation(request);

        var finalAllocation = await AllocationRepository.Add(allocation);

        return AllocationMapper.ToNodeAllocation(finalAllocation);
    }

    [HttpPatch("{id:int}")]
    public async Task<ActionResult<NodeAllocationResponse>> Update(
        [FromRoute] int nodeId,
        [FromRoute] int id,
        [FromBody] UpdateNodeAllocationRequest request
    )
    {
        var allocation = await AllocationRepository
            .Get()
            .Where(x => x.Node.Id == nodeId)
            .FirstOrDefaultAsync(x => x.Id == id);

        if (allocation == null)
            return Problem("No allocation with that id found", statusCode: 404);

        AllocationMapper.Merge(request, allocation);
        await AllocationRepository.Update(allocation);

        return AllocationMapper.ToNodeAllocation(allocation);
    }

    [HttpDelete("{id:int}")]
    public async Task<ActionResult> Delete([FromRoute] int nodeId, [FromRoute] int id)
    {
        var allocation = await AllocationRepository
            .Get()
            .Where(x => x.Node.Id == nodeId)
            .FirstOrDefaultAsync(x => x.Id == id);

        if (allocation == null)
            return Problem("No allocation with that id found", statusCode: 404);

        await AllocationRepository.Remove(allocation);
        return NoContent();
    }

    [HttpPost("range")]
    public async Task<ActionResult> CreateRange(
        [FromRoute] int nodeId,
        [FromBody] CreateNodeAllocationRangeRequest request
    )
    {
        if (request.Start > request.End)
            return Problem("Invalid start and end specified", statusCode: 400);

        if (request.End - request.Start == 0)
            return Problem("Empty range specified", statusCode: 400);

        var node = await NodeRepository
            .Get()
            .FirstOrDefaultAsync(x => x.Id == nodeId);

        if (node == null)
            return Problem("No node with that id found", statusCode: 404);

        var existingAllocations = await AllocationRepository
            .Get()
            .Where(x => x.Port >= request.Start && x.Port <= request.End &&
                        x.IpAddress == request.IpAddress)
            .AsNoTracking()
            .ToArrayAsync();

        var ports = new List<int>();

        for (var i = request.Start; i < request.End; i++)
        {
            // Skip existing allocations
            if (existingAllocations.Any(x => x.Port == i))
                continue;

            ports.Add(i);
        }

        var allocations = ports
            .Select(port => new Allocation()
            {
                IpAddress = request.IpAddress,
                Port = port,
                Node = node
            })
            .ToArray();
        
        await AllocationRepository.RunTransaction(async set => { await set.AddRangeAsync(allocations); });
        return NoContent();
    }

    [HttpDelete("all")]
    public async Task<ActionResult> DeleteAll([FromRoute] int nodeId)
    {
        var allocations = AllocationRepository
            .Get()
            .Where(x => x.Node.Id == nodeId)
            .ToArray();

        await AllocationRepository.RunTransaction(set => { set.RemoveRange(allocations); });
        return NoContent();
    }

    [HttpGet("free")]
    [Authorize(Policy = "permissions:admin.servers.nodes.get")]
    public async Task<IPagedData<NodeAllocationResponse>> GetFree(
        [FromRoute] int nodeId,
        [FromQuery] PagedOptions options,
        [FromQuery] int serverId = -1
    )
    {
        var node = NodeRepository
            .Get()
            .FirstOrDefault(x => x.Id == nodeId);

        if (node == null)
            throw new HttpApiException("A node with this id could not be found", 404);

        var freeAllocationsQuery = AllocationRepository
            .Get()
            .Where(x => x.Node.Id == node.Id)
            .Where(x => x.Server == null || x.Server.Id == serverId);

        var count = await freeAllocationsQuery.CountAsync();

        var allocations = await freeAllocationsQuery
            .Skip(options.Page * options.PageSize)
            .Take(options.PageSize)
            .AsNoTracking()
            .ProjectToAdminResponse()
            .ToArrayAsync();

        return new PagedData<NodeAllocationResponse>()
        {
            Items = allocations,
            CurrentPage = options.Page,
            PageSize = options.PageSize,
            TotalItems = count,
            TotalPages = (int)Math.Ceiling(Math.Max(0, count) / (double)options.PageSize)
        };
    }
}