PluralKit/PluralKit.Core/Models/Patch/SystemPatch.cs

#nullable enable
using System;
using System.Text.RegularExpressions;

using Newtonsoft.Json.Linq;

using NodaTime;

using SqlKata;

namespace PluralKit.Core
{
    public class SystemPatch: PatchObject
    {
        public Partial<string?> Name { get; set; }
        public Partial<string?> Hid { get; set; }
        public Partial<string?> Description { get; set; }
        public Partial<string?> Tag { get; set; }
        public Partial<string?> AvatarUrl { get; set; }
        public Partial<string?> BannerImage { get; set; }
        public Partial<string?> Color { get; set; }
        public Partial<string?> Token { get; set; }
        public Partial<string?> WebhookUrl { get; set; }
        public Partial<string?> WebhookToken { get; set; }
        public Partial<string> UiTz { get; set; }
        public Partial<PrivacyLevel> DescriptionPrivacy { get; set; }
        public Partial<PrivacyLevel> MemberListPrivacy { get; set; }
        public Partial<PrivacyLevel> GroupListPrivacy { get; set; }
        public Partial<PrivacyLevel> FrontPrivacy { get; set; }
        public Partial<PrivacyLevel> FrontHistoryPrivacy { get; set; }
        public Partial<bool> PingsEnabled { get; set; }
        public Partial<int?> LatchTimeout { get; set; }
        public Partial<int?> MemberLimitOverride { get; set; }
        public Partial<int?> GroupLimitOverride { get; set; }

        public override Query Apply(Query q) => q.ApplyPatch(wrapper => wrapper
            .With("name", Name)
            .With("hid", Hid)
            .With("description", Description)
            .With("tag", Tag)
            .With("avatar_url", AvatarUrl)
            .With("banner_image", BannerImage)
            .With("color", Color)
            .With("token", Token)
            .With("webhook_url", WebhookUrl)
            .With("webhook_token", WebhookToken)
            .With("ui_tz", UiTz)
            .With("description_privacy", DescriptionPrivacy)
            .With("member_list_privacy", MemberListPrivacy)
            .With("group_list_privacy", GroupListPrivacy)
            .With("front_privacy", FrontPrivacy)
            .With("front_history_privacy", FrontHistoryPrivacy)
            .With("pings_enabled", PingsEnabled)
            .With("latch_timeout", LatchTimeout)
            .With("member_limit_override", MemberLimitOverride)
            .With("group_limit_override", GroupLimitOverride)
        );

        public new void AssertIsValid()
        {
            if (Name.Value != null)
                AssertValid(Name.Value, "name", Limits.MaxSystemNameLength);
            if (Description.Value != null)
                AssertValid(Description.Value, "description", Limits.MaxDescriptionLength);
            if (Tag.Value != null)
                AssertValid(Tag.Value, "tag", Limits.MaxSystemTagLength);
            if (AvatarUrl.Value != null)
                AssertValid(AvatarUrl.Value, "avatar_url", Limits.MaxUriLength,
                    s => MiscUtils.TryMatchUri(s, out var avatarUri));
            if (BannerImage.Value != null)
                AssertValid(BannerImage.Value, "banner", Limits.MaxUriLength,
                    s => MiscUtils.TryMatchUri(s, out var bannerUri));
            if (Color.Value != null)
                AssertValid(Color.Value, "color", "^[0-9a-fA-F]{6}$");
            if (UiTz.IsPresent && DateTimeZoneProviders.Tzdb.GetZoneOrNull(UiTz.Value) == null)
                Errors.Add(new ValidationError("timezone"));
        }

#nullable disable

        public static SystemPatch FromJSON(JObject o, APIVersion v = APIVersion.V1)
        {
            var patch = new SystemPatch();
            if (o.ContainsKey("name")) patch.Name = o.Value<string>("name").NullIfEmpty();
            if (o.ContainsKey("description")) patch.Description = o.Value<string>("description").NullIfEmpty();
            if (o.ContainsKey("tag")) patch.Tag = o.Value<string>("tag").NullIfEmpty();
            if (o.ContainsKey("avatar_url")) patch.AvatarUrl = o.Value<string>("avatar_url").NullIfEmpty();
            if (o.ContainsKey("banner")) patch.BannerImage = o.Value<string>("banner").NullIfEmpty();
            if (o.ContainsKey("color")) patch.Color = o.Value<string>("color").NullIfEmpty();
            if (o.ContainsKey("timezone")) patch.UiTz = o.Value<string>("timezone") ?? "UTC";

            switch (v)
            {
                case APIVersion.V1:
                    {
                        if (o.ContainsKey("tz")) patch.UiTz = o.Value<string>("tz") ?? "UTC";

                        if (o.ContainsKey("description_privacy")) patch.DescriptionPrivacy = patch.ParsePrivacy(o, "description_privacy");
                        if (o.ContainsKey("member_list_privacy")) patch.MemberListPrivacy = patch.ParsePrivacy(o, "member_list_privacy");
                        if (o.ContainsKey("front_privacy")) patch.FrontPrivacy = patch.ParsePrivacy(o, "front_privacy");
                        if (o.ContainsKey("front_history_privacy")) patch.FrontHistoryPrivacy = patch.ParsePrivacy(o, "front_history_privacy");

                        break;
                    }
                case APIVersion.V2:
                    {
                        if (o.ContainsKey("privacy") && o["privacy"].Type != JTokenType.Null)
                        {
                            var privacy = o.Value<JObject>("privacy");

                            if (privacy.ContainsKey("description_privacy"))
                                patch.DescriptionPrivacy = patch.ParsePrivacy(privacy, "description_privacy");

                            if (privacy.ContainsKey("member_list_privacy"))
                                patch.DescriptionPrivacy = patch.ParsePrivacy(privacy, "member_list_privacy");

                            if (privacy.ContainsKey("group_list_privacy"))
                                patch.GroupListPrivacy = patch.ParsePrivacy(privacy, "group_list_privacy");

                            if (privacy.ContainsKey("front_privacy"))
                                patch.DescriptionPrivacy = patch.ParsePrivacy(privacy, "front_privacy");

                            if (privacy.ContainsKey("front_history_privacy"))
                                patch.DescriptionPrivacy = patch.ParsePrivacy(privacy, "front_history_privacy");
                        }

                        break;
                    }
            }

            return patch;
        }

        public JObject ToJson()
        {
            var o = new JObject();

            if (Name.IsPresent)
                o.Add("name", Name.Value);
            if (Hid.IsPresent)
                o.Add("id", Hid.Value);
            if (Description.IsPresent)
                o.Add("description", Description.Value);
            if (Tag.IsPresent)
                o.Add("tag", Tag.Value);
            if (AvatarUrl.IsPresent)
                o.Add("avatar_url", AvatarUrl.Value);
            if (BannerImage.IsPresent)
                o.Add("banner", BannerImage.Value);
            if (Color.IsPresent)
                o.Add("color", Color.Value);
            if (UiTz.IsPresent)
                o.Add("timezone", UiTz.Value);

            if (
                   DescriptionPrivacy.IsPresent
                || MemberListPrivacy.IsPresent
                || GroupListPrivacy.IsPresent
                || FrontPrivacy.IsPresent
                || FrontHistoryPrivacy.IsPresent
            )
            {
                var p = new JObject();

                if (DescriptionPrivacy.IsPresent)
                    p.Add("description_privacy", DescriptionPrivacy.Value.ToJsonString());

                if (MemberListPrivacy.IsPresent)
                    p.Add("member_list_privacy", MemberListPrivacy.Value.ToJsonString());

                if (GroupListPrivacy.IsPresent)
                    p.Add("group_list_privacy", GroupListPrivacy.Value.ToJsonString());

                if (FrontPrivacy.IsPresent)
                    p.Add("front_privacy", FrontPrivacy.Value.ToJsonString());

                if (FrontHistoryPrivacy.IsPresent)
                    p.Add("front_history_privacy", FrontHistoryPrivacy.Value.ToJsonString());

                o.Add("privacy", p);
            }

            return o;
        }
    }
}