using Newtonsoft.Json; using Newtonsoft.Json.Linq; using NodaTime; using NodaTime.Text; namespace PluralKit.Core; public readonly struct MemberId: INumericId { public int Value { get; } public MemberId(int value) { Value = value; } public bool Equals(MemberId other) => Value == other.Value; public override bool Equals(object obj) => obj is MemberId other && Equals(other); public override int GetHashCode() => Value; public static bool operator ==(MemberId left, MemberId right) => left.Equals(right); public static bool operator !=(MemberId left, MemberId right) => !left.Equals(right); public int CompareTo(MemberId other) => Value.CompareTo(other.Value); public override string ToString() => $"Member #{Value}"; } public class PKMember { // Dapper *can* figure out mapping to getter-only properties, but this doesn't work // when trying to map to *subclasses* (eg. ListedMember). Adding private setters makes it work anyway. public MemberId Id { get; private set; } public string Hid { get; private set; } public Guid Uuid { get; private set; } public SystemId System { get; private set; } public string Color { get; private set; } public string WebhookAvatarUrl { get; private set; } public string AvatarUrl { get; private set; } public string BannerImage { get; private set; } public string Name { get; private set; } public string DisplayName { get; private set; } public LocalDate? Birthday { get; private set; } public string Pronouns { get; private set; } public string Description { get; private set; } public ICollection ProxyTags { get; private set; } public bool KeepProxy { get; private set; } public Instant Created { get; private set; } public int MessageCount { get; private set; } public Instant? LastMessageTimestamp { get; private set; } public bool AllowAutoproxy { get; private set; } public PrivacyLevel MemberVisibility { get; private set; } public PrivacyLevel DescriptionPrivacy { get; private set; } public PrivacyLevel AvatarPrivacy { get; private set; } public PrivacyLevel NamePrivacy { get; private set; } //ignore setting if no display name is set public PrivacyLevel BirthdayPrivacy { get; private set; } public PrivacyLevel PronounPrivacy { get; private set; } public PrivacyLevel MetadataPrivacy { get; private set; } // public PrivacyLevel ColorPrivacy { get; private set; } /// Returns a formatted string representing the member's birthday, taking into account that a year of "0001" or "0004" is hidden /// Before Feb 10 2020, the sentinel year was 0001, now it is 0004. [JsonIgnore] public string BirthdayString { get { if (Birthday == null) return null; var format = LocalDatePattern.CreateWithInvariantCulture("MMM dd, yyyy"); if (Birthday?.Year == 1 || Birthday?.Year == 4) format = LocalDatePattern.CreateWithInvariantCulture("MMM dd"); return format.Format(Birthday.Value); } } [JsonIgnore] public bool HasProxyTags => ProxyTags.Count > 0; } public static class PKMemberExt { public static string NameFor(this PKMember member, LookupContext ctx) => member.NamePrivacy.Get(ctx, member.Name, member.DisplayName ?? member.Name); public static string AvatarFor(this PKMember member, LookupContext ctx) => member.AvatarPrivacy.Get(ctx, member.AvatarUrl.TryGetCleanCdnUrl()); public static string WebhookAvatarFor(this PKMember member, LookupContext ctx) => member.AvatarPrivacy.Get(ctx, (member.WebhookAvatarUrl ?? member.AvatarUrl).TryGetCleanCdnUrl()); public static string DescriptionFor(this PKMember member, LookupContext ctx) => member.DescriptionPrivacy.Get(ctx, member.Description); public static LocalDate? BirthdayFor(this PKMember member, LookupContext ctx) => member.BirthdayPrivacy.Get(ctx, member.Birthday); public static string PronounsFor(this PKMember member, LookupContext ctx) => member.PronounPrivacy.Get(ctx, member.Pronouns); public static Instant? CreatedFor(this PKMember member, LookupContext ctx) => member.MetadataPrivacy.Get(ctx, (Instant?)member.Created); public static int? MessageCountFor(this PKMember member, LookupContext ctx) => member.MetadataPrivacy.Get(ctx, member.MessageCount, fallback: null); public static Instant? LastMessageTimestampFor(this PKMember member, LookupContext ctx) => member.MetadataPrivacy.Get(ctx, member.LastMessageTimestamp); public static JObject ToJson(this PKMember member, LookupContext ctx, bool needsLegacyProxyTags = false, string systemStr = null) { var includePrivacy = ctx == LookupContext.ByOwner; var o = new JObject(); o.Add("id", member.Hid); o.Add("uuid", member.Uuid.ToString()); if (systemStr != null) o.Add("system", systemStr); o.Add("name", member.NameFor(ctx)); o.Add("display_name", member.NamePrivacy.CanAccess(ctx) ? member.DisplayName : null); // o.Add("color", member.ColorPrivacy.CanAccess(ctx) ? member.Color : null); o.Add("color", member.Color); o.Add("birthday", member.BirthdayFor(ctx)?.FormatExport()); o.Add("pronouns", member.PronounsFor(ctx)); o.Add("avatar_url", member.AvatarFor(ctx).TryGetCleanCdnUrl()); o.Add("webhook_avatar_url", member.WebhookAvatarFor(ctx).TryGetCleanCdnUrl()); o.Add("banner", member.DescriptionPrivacy.Get(ctx, member.BannerImage).TryGetCleanCdnUrl()); o.Add("description", member.DescriptionFor(ctx)); o.Add("created", member.CreatedFor(ctx)?.FormatExport()); o.Add("keep_proxy", member.KeepProxy); o.Add("autoproxy_enabled", ctx == LookupContext.ByOwner ? member.AllowAutoproxy : null); o.Add("message_count", member.MessageCountFor(ctx)); o.Add("last_message_timestamp", member.LastMessageTimestampFor(ctx)?.FormatExport()); var tagArray = new JArray(); foreach (var tag in member.ProxyTags) tagArray.Add(new JObject { { "prefix", tag.Prefix }, { "suffix", tag.Suffix } }); o.Add("proxy_tags", tagArray); if (includePrivacy) { var p = new JObject(); p.Add("visibility", member.MemberVisibility.ToJsonString()); p.Add("name_privacy", member.NamePrivacy.ToJsonString()); p.Add("description_privacy", member.DescriptionPrivacy.ToJsonString()); p.Add("birthday_privacy", member.BirthdayPrivacy.ToJsonString()); p.Add("pronoun_privacy", member.PronounPrivacy.ToJsonString()); p.Add("avatar_privacy", member.AvatarPrivacy.ToJsonString()); p.Add("metadata_privacy", member.MetadataPrivacy.ToJsonString()); o.Add("privacy", p); } else { o.Add("privacy", null); } return o; } }