using Dapper.Contrib.Extensions; using Newtonsoft.Json; using Newtonsoft.Json.Linq; using NodaTime; namespace PluralKit.Core; public readonly struct SystemId: INumericId { public int Value { get; } public SystemId(int value) { Value = value; } public bool Equals(SystemId other) => Value == other.Value; public override bool Equals(object obj) => obj is SystemId other && Equals(other); public override int GetHashCode() => Value; public static bool operator ==(SystemId left, SystemId right) => left.Equals(right); public static bool operator !=(SystemId left, SystemId right) => !left.Equals(right); public int CompareTo(SystemId other) => Value.CompareTo(other.Value); public override string ToString() => $"System #{Value}"; } public class PKSystem { [Key] public SystemId Id { get; } public string Hid { get; } public Guid Uuid { get; private set; } public string Name { get; } public string Description { get; } public string Tag { get; } public string Pronouns { get; } public string AvatarUrl { get; } public string BannerImage { get; } public string Color { get; } public string Token { get; } public string WebhookUrl { get; } public string WebhookToken { get; } public Instant Created { get; } public PrivacyLevel DescriptionPrivacy { get; } public PrivacyLevel MemberListPrivacy { get; } public PrivacyLevel FrontPrivacy { get; } public PrivacyLevel FrontHistoryPrivacy { get; } public PrivacyLevel GroupListPrivacy { get; } public PrivacyLevel PronounPrivacy { get; } } public static class PKSystemExt { public static string DescriptionFor(this PKSystem system, LookupContext ctx) => system.DescriptionPrivacy.Get(ctx, system.Description); public static JObject ToJson(this PKSystem system, LookupContext ctx, APIVersion v = APIVersion.V1) { var o = new JObject(); o.Add("id", system.Hid); if (v == APIVersion.V2) o.Add("uuid", system.Uuid.ToString()); o.Add("name", system.Name); o.Add("description", system.DescriptionFor(ctx)); o.Add("tag", system.Tag); if (v == APIVersion.V2) o.Add("pronouns", system.PronounPrivacy.Get(ctx, system.Pronouns)); o.Add("avatar_url", system.AvatarUrl.TryGetCleanCdnUrl()); o.Add("banner", system.DescriptionPrivacy.Get(ctx, system.BannerImage).TryGetCleanCdnUrl()); o.Add("color", system.Color); o.Add("created", system.Created.FormatExport()); switch (v) { case APIVersion.V1: { // this property was moved to SystemConfig // see notice in /api/legacy docs o.Add("tz", "UTC"); o.Add("description_privacy", ctx == LookupContext.ByOwner ? system.DescriptionPrivacy.ToJsonString() : null); o.Add("member_list_privacy", ctx == LookupContext.ByOwner ? system.MemberListPrivacy.ToJsonString() : null); o.Add("group_list_privacy", ctx == LookupContext.ByOwner ? system.GroupListPrivacy.ToJsonString() : null); o.Add("front_privacy", ctx == LookupContext.ByOwner ? system.FrontPrivacy.ToJsonString() : null); o.Add("front_history_privacy", ctx == LookupContext.ByOwner ? system.FrontHistoryPrivacy.ToJsonString() : null); break; } case APIVersion.V2: { if (ctx == LookupContext.ByOwner) { // todo: should this be moved to a different JSON model? o.Add("webhook_url", system.WebhookUrl); // o.Add("webhook_token", system.WebhookToken); var p = new JObject(); p.Add("description_privacy", system.DescriptionPrivacy.ToJsonString()); p.Add("pronoun_privacy", system.PronounPrivacy.ToJsonString()); p.Add("member_list_privacy", system.MemberListPrivacy.ToJsonString()); p.Add("group_list_privacy", system.GroupListPrivacy.ToJsonString()); p.Add("front_privacy", system.FrontPrivacy.ToJsonString()); p.Add("front_history_privacy", system.FrontHistoryPrivacy.ToJsonString()); o.Add("privacy", p); } else { o.Add("privacy", null); } break; } } return o; } }