BridgeController.cs

Go to the documentation of this file.

using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Newtonsoft.Json;
using Serilog.Context;
using System;
using System.Net;
using System.Net.Mime;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Host.Components.Interop;
using Tgstation.Server.Host.Components.Interop.Bridge;

namespace Tgstation.Server.Host.Controllers
{
    [Route("Bridge")]
    [Produces(MediaTypeNames.Application.Json)]
    public class BridgeController : Controller
    {
        static long requestsProcessed;

        readonly IBridgeDispatcher bridgeDispatcher;

        readonly ILogger<BridgeController> logger;

        public BridgeController(IBridgeDispatcher bridgeDispatcher, IHostApplicationLifetime applicationLifetime, ILogger<BridgeController> logger)
        {
            this.bridgeDispatcher = bridgeDispatcher ?? throw new ArgumentNullException(nameof(bridgeDispatcher));
            if (applicationLifetime == null)
                throw new ArgumentNullException(nameof(applicationLifetime));

            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));

            applicationLifetime.ApplicationStopped.Register(() => requestsProcessed = 0);
        }

        [HttpGet]
        public async Task<IActionResult> Process([FromQuery]string data, CancellationToken cancellationToken)
        {
            // Nothing to see here
            if (!IPAddress.IsLoopback(Request.HttpContext.Connection.RemoteIpAddress))
                return NotFound();

            using (LogContext.PushProperty("Bridge", Interlocked.Increment(ref requestsProcessed)))
            {
                BridgeParameters request;
                try
                {
                    request = JsonConvert.DeserializeObject<BridgeParameters>(data, DMApiConstants.SerializerSettings);
                }
                catch
                {
                    logger.LogWarning("Error deserializing bridge request: {0}", data);
                    return BadRequest();
                }

                logger.LogTrace("Bridge Request: {0}", data);

                var response = await bridgeDispatcher.ProcessBridgeRequest(request, cancellationToken).ConfigureAwait(false);
                if (response == null)
                    Forbid();

                var responseJson = JsonConvert.SerializeObject(response, DMApiConstants.SerializerSettings);
                logger.LogTrace("Bridge Response: {0}", responseJson);
                return Content(responseJson, MediaTypeNames.Application.Json);
            }
        }
    }
}