PluralKit/PluralKit.API/Controllers/v2/DiscordControllerV2.cs

using System;
using System.Threading.Tasks;

using Microsoft.AspNetCore.Mvc;

using Newtonsoft.Json.Linq;

using NodaTime;

using PluralKit.Core;

namespace PluralKit.API
{
    [ApiController]
    [ApiVersion("2.0")]
    [Route("v{version:apiVersion}")]
    public class GuildControllerV2: PKControllerBase
    {
        public GuildControllerV2(IServiceProvider svc) : base(svc) { }


        [HttpGet("systems/@me/guilds/{guild_id}")]
        public async Task<IActionResult> SystemGuildGet(ulong guild_id)
        {
            var system = await ResolveSystem("@me");
            var settings = await _repo.GetSystemGuild(guild_id, system.Id, defaultInsert: false);
            if (settings == null)
                throw Errors.SystemGuildNotFound;

            PKMember member = null;
            if (settings.AutoproxyMember != null)
                member = await _repo.GetMember(settings.AutoproxyMember.Value);

            return Ok(settings.ToJson(member?.Hid));
        }

        [HttpPatch("systems/@me/guilds/{guild_id}")]
        public async Task<IActionResult> DoSystemGuildPatch(ulong guild_id, [FromBody] JObject data)
        {
            var system = await ResolveSystem("@me");
            var settings = await _repo.GetSystemGuild(guild_id, system.Id, defaultInsert: false);
            if (settings == null)
                throw Errors.SystemGuildNotFound;

            MemberId? memberId = null;
            if (data.ContainsKey("autoproxy_member"))
            {
                if (data["autoproxy_member"].Type != JTokenType.Null)
                {
                    var member = await ResolveMember(data.Value<string>("autoproxy_member"));
                    if (member == null)
                        throw Errors.MemberNotFound;

                    memberId = member.Id;
                }
            }
            else
                memberId = settings.AutoproxyMember;

            var patch = SystemGuildPatch.FromJson(data, memberId);

            patch.AssertIsValid();
            if (patch.Errors.Count > 0)
                throw new ModelParseError(patch.Errors);

            // this is less than great, but at least it's legible
            if (patch.AutoproxyMember.Value == null)
                if (patch.AutoproxyMode.IsPresent)
                {
                    if (patch.AutoproxyMode.Value == AutoproxyMode.Member)
                        throw Errors.MissingAutoproxyMember;
                }
                else if (settings.AutoproxyMode == AutoproxyMode.Member)
                    throw Errors.MissingAutoproxyMember;

            var newSettings = await _repo.UpdateSystemGuild(system.Id, guild_id, patch);

            PKMember? newMember = null;
            if (newSettings.AutoproxyMember != null)
                newMember = await _repo.GetMember(newSettings.AutoproxyMember.Value);
            return Ok(newSettings.ToJson(newMember?.Hid));
        }

        [HttpGet("members/{memberRef}/guilds/{guild_id}")]
        public async Task<IActionResult> MemberGuildGet(string memberRef, ulong guild_id)
        {
            var system = await ResolveSystem("@me");
            var member = await ResolveMember(memberRef);
            if (member == null)
                throw Errors.MemberNotFound;
            if (member.System != system.Id)
                throw Errors.NotOwnMemberError;

            var settings = await _repo.GetMemberGuild(guild_id, member.Id, defaultInsert: false);
            if (settings == null)
                throw Errors.MemberGuildNotFound;

            return Ok(settings.ToJson());
        }

        [HttpPatch("members/{memberRef}/guilds/{guild_id}")]
        public async Task<IActionResult> DoMemberGuildPatch(string memberRef, ulong guild_id, [FromBody] JObject data)
        {
            var system = await ResolveSystem("@me");
            var member = await ResolveMember(memberRef);
            if (member == null)
                throw Errors.MemberNotFound;
            if (member.System != system.Id)
                throw Errors.NotOwnMemberError;

            var settings = await _repo.GetMemberGuild(guild_id, member.Id, defaultInsert: false);
            if (settings == null)
                throw Errors.MemberGuildNotFound;

            var patch = MemberGuildPatch.FromJson(data);

            patch.AssertIsValid();
            if (patch.Errors.Count > 0)
                throw new ModelParseError(patch.Errors);

            var newSettings = await _repo.UpdateMemberGuild(member.Id, guild_id, patch);
            return Ok(newSettings.ToJson());
        }

        [HttpGet("messages/{messageId}")]
        public async Task<ActionResult<MessageReturn>> MessageGet(ulong messageId)
        {
            var msg = await _db.Execute(c => _repo.GetMessage(c, messageId));
            if (msg == null)
                throw Errors.MessageNotFound;

            var ctx = this.ContextFor(msg.System);

            // todo: don't rely on v1 stuff
            return new MessageReturn
            {
                Timestamp = Instant.FromUnixTimeMilliseconds((long)(msg.Message.Mid >> 22) + 1420070400000),
                Id = msg.Message.Mid.ToString(),
                Channel = msg.Message.Channel.ToString(),
                Sender = msg.Message.Sender.ToString(),
                System = msg.System.ToJson(ctx, v: APIVersion.V2),
                Member = msg.Member.ToJson(ctx, v: APIVersion.V2),
                Original = msg.Message.OriginalMid?.ToString()
            };
        }
    }
}