SwarmNode.cs

Go to the documentation of this file.

using System;
using System.Linq;
 
using HotChocolate;
using HotChocolate.Types.Relay;
 
using Microsoft.Extensions.Options;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Internal;
using Tgstation.Server.Host.Configuration;
using Tgstation.Server.Host.GraphQL.Interfaces;
using Tgstation.Server.Host.Security;
using Tgstation.Server.Host.Swarm;
 
namespace Tgstation.Server.Host.GraphQL.Types
{
    [Node]
    public sealed class SwarmNode : IServerNode
    {
        [ID]
        public string NodeId => Identifier;
 
        public string Identifier { get; }
 
        public Uri Address { get; }
 
        public Uri? PublicAddress { get; }
 
        public bool Controller { get; }
 
        [TgsGraphQLAuthorize]
        public static SwarmNode? GetSwarmNode(
            string identifier,
            [Service] ISwarmService swarmService)
        {
            ArgumentNullException.ThrowIfNull(identifier);
            ArgumentNullException.ThrowIfNull(swarmService);
            var info = swarmService
                .GetSwarmServers()
                ?.FirstOrDefault(x => x.Identifier == identifier);
 
            if (info == null)
                return null;
 
            return new SwarmNode(info);
        }
 
        public SwarmNode(SwarmServerInformation? nodeInformation)
        {
            ArgumentNullException.ThrowIfNull(nodeInformation);
 
            Identifier = nodeInformation.Identifier!;
            Address = nodeInformation.Address!;
            PublicAddress = nodeInformation.PublicAddress;
            Controller = nodeInformation.Controller;
        }
 
        public IGateway Gateway([Service] IOptionsSnapshot<SwarmConfiguration> swarmConfigurationOptions)
        {
            ArgumentNullException.ThrowIfNull(swarmConfigurationOptions);
 
            bool local = Identifier == swarmConfigurationOptions.Value.Identifier;
            if (local)
                return new LocalGateway();
 
            throw new ErrorMessageException(ErrorCode.RemoteGatewaysNotImplemented);
 
            // return new RemoteGateway();
        }
    }
}