PluralKit/PluralKit.Core/Models/Patch/GroupPatch.cs

#nullable enable
using Newtonsoft.Json.Linq;

using SqlKata;

namespace PluralKit.Core;

public class GroupPatch: PatchObject
{
    public Partial<string> Name { get; set; }
    public Partial<string> Hid { get; set; }
    public Partial<string?> DisplayName { get; set; }
    public Partial<string?> Description { get; set; }
    public Partial<string?> Icon { get; set; }
    public Partial<string?> BannerImage { get; set; }
    public Partial<string?> Color { get; set; }

    public Partial<PrivacyLevel> DescriptionPrivacy { get; set; }
    public Partial<PrivacyLevel> IconPrivacy { get; set; }
    public Partial<PrivacyLevel> ListPrivacy { get; set; }
    public Partial<PrivacyLevel> Visibility { get; set; }

    public override Query Apply(Query q) => q.ApplyPatch(wrapper => wrapper
        .With("name", Name)
        .With("hid", Hid)
        .With("display_name", DisplayName)
        .With("description", Description)
        .With("icon", Icon)
        .With("banner_image", BannerImage)
        .With("color", Color)
        .With("description_privacy", DescriptionPrivacy)
        .With("icon_privacy", IconPrivacy)
        .With("list_privacy", ListPrivacy)
        .With("visibility", Visibility)
    );

    public new void AssertIsValid()
    {
        if (Name.Value != null)
            AssertValid(Name.Value, "name", Limits.MaxGroupNameLength);
        if (DisplayName.Value != null)
            AssertValid(DisplayName.Value, "display_name", Limits.MaxGroupNameLength);
        if (Description.Value != null)
            AssertValid(Description.Value, "description", Limits.MaxDescriptionLength);
        if (Icon.Value != null)
            AssertValid(Icon.Value, "icon", 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}$");
    }
#nullable disable

    public static GroupPatch FromJson(JObject o)
    {
        var patch = new GroupPatch();

        if (o.ContainsKey("name"))
        {
            patch.Name = o.Value<string>("name").NullIfEmpty();
            if (patch.Name.Value == null)
                patch.Errors.Add(new ValidationError("name", "Group name can not be set to null."));
        }

        if (o.ContainsKey("display_name")) patch.DisplayName = o.Value<string>("display_name").NullIfEmpty();
        if (o.ContainsKey("description")) patch.Description = o.Value<string>("description").NullIfEmpty();
        if (o.ContainsKey("icon")) patch.Icon = o.Value<string>("icon").NullIfEmpty();
        if (o.ContainsKey("banner")) patch.BannerImage = o.Value<string>("banner").NullIfEmpty();
        if (o.ContainsKey("color")) patch.Color = o.Value<string>("color").NullIfEmpty()?.ToLower();

        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("icon_privacy"))
                patch.IconPrivacy = patch.ParsePrivacy(privacy, "icon_privacy");

            if (privacy.ContainsKey("list_privacy"))
                patch.ListPrivacy = patch.ParsePrivacy(privacy, "list_privacy");

            if (privacy.ContainsKey("visibility"))
                patch.Visibility = patch.ParsePrivacy(privacy, "visibility");
        }

        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 (DisplayName.IsPresent)
            o.Add("display_name", DisplayName.Value);
        if (Description.IsPresent)
            o.Add("description", Description.Value);
        if (Icon.IsPresent)
            o.Add("icon", Icon.Value);
        if (BannerImage.IsPresent)
            o.Add("banner", BannerImage.Value);
        if (Color.IsPresent)
            o.Add("color", Color.Value);

        if (
            DescriptionPrivacy.IsPresent
            || IconPrivacy.IsPresent
            || ListPrivacy.IsPresent
            || Visibility.IsPresent
        )
        {
            var p = new JObject();

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

            if (IconPrivacy.IsPresent)
                p.Add("icon_privacy", IconPrivacy.Value.ToJsonString());

            if (ListPrivacy.IsPresent)
                p.Add("list_privacy", ListPrivacy.Value.ToJsonString());

            if (Visibility.IsPresent)
                p.Add("visibility", Visibility.Value.ToJsonString());

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

        return o;
    }
}