using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using System.Diagnostics.CodeAnalysis;
using EntityFrameworkCore.Projectables;
using Iceshrimp.Backend.Core.Configuration;
using Iceshrimp.Backend.Core.Helpers;
using Microsoft.EntityFrameworkCore;
using NpgsqlTypes;
using J = System.Text.Json.Serialization.JsonPropertyNameAttribute;
namespace Iceshrimp.Backend.Core.Database.Tables;
[Table("note")]
[Index(nameof(Uri), IsUnique = true)]
[Index(nameof(ReplyId))]
[Index(nameof(AttachedFileTypes))]
[Index(nameof(FileIds))]
[Index(nameof(RenoteId))]
[Index(nameof(Mentions))]
[Index(nameof(UserId))]
[Index(nameof(UserHost))]
[Index(nameof(VisibleUserIds))]
[Index(nameof(Tags))]
[Index(nameof(ThreadId))]
[Index(nameof(CreatedAt))]
[Index(nameof(ChannelId))]
[Index(nameof(CreatedAt), nameof(UserId))]
[Index(nameof(Id), nameof(UserHost))]
[Index(nameof(Url))]
[Index(nameof(UserId), nameof(Id))]
[Index(nameof(Visibility))]
[Index(nameof(ReplyUri))]
[Index(nameof(RenoteUri))]
public class Note : IEntity
{
[PgName("note_visibility_enum")]
public enum NoteVisibility
{
[PgName("public")] Public = 0,
[PgName("home")] Home = 1,
[PgName("followers")] Followers = 2,
[PgName("specified")] Specified = 3
}
///
/// The created date of the Note.
///
[Column("createdAt")]
public DateTime CreatedAt { get; set; }
///
/// The ID of reply target.
///
[Column("replyId")]
[StringLength(32)]
public string? ReplyId { get; set; }
///
/// The ID of renote target.
///
[Column("renoteId")]
[StringLength(32)]
public string? RenoteId { get; set; }
///
/// The URI of the reply target, if it couldn't be resolved at time of ingestion.
///
[Column("replyUri")]
[StringLength(512)]
public string? ReplyUri { get; set; }
///
/// The URI of the renote target, if it couldn't be resolved at time of ingestion.
///
[Column("renoteUri")]
[StringLength(512)]
public string? RenoteUri { get; set; }
[Column("text")] public string? Text { get; set; }
[Column("name")] [StringLength(256)] public string? Name { get; set; }
[Column("cw")] public string? Cw { get; set; }
///
/// The ID of author.
///
[Column("userId")]
[StringLength(32)]
public string UserId { get; set; } = null!;
[Column("visibility")] public NoteVisibility Visibility { get; set; }
[NotMapped]
[Projectable]
[SuppressMessage("ReSharper", "MergeIntoLogicalPattern",
Justification = "Projectable expression cannot contain patterns")]
public bool VisibilityIsPublicOrHome => Visibility == NoteVisibility.Public || Visibility == NoteVisibility.Home;
[Column("localOnly")] public bool LocalOnly { get; set; }
[Column("renoteCount")] public short RenoteCount { get; set; }
[Column("repliesCount")] public short RepliesCount { get; set; }
[Column("likeCount")] public int LikeCount { get; set; }
[Column("reactions", TypeName = "jsonb")]
public Dictionary Reactions { get; set; } = null!;
///
/// The URI of a note. it will be null when the note is local.
///
[Column("uri")]
[StringLength(512)]
public string? Uri { get; set; }
[Column("score")] public int Score { get; set; }
[Column("fileIds", TypeName = "character varying(32)[]")]
public List FileIds { get; set; } = [];
[Column("attachedFileTypes", TypeName = "character varying(256)[]")]
public List AttachedFileTypes { get; set; } = [];
[Column("visibleUserIds", TypeName = "character varying(32)[]")]
public List VisibleUserIds { get; set; } = [];
[Column("mentions", TypeName = "character varying(32)[]")]
public List Mentions { get; set; } = [];
[Column("mentionedRemoteUsers", TypeName = "jsonb")]
public List MentionedRemoteUsers { get; set; } = [];
[Column("emojis", TypeName = "character varying(128)[]")]
public List Emojis { get; set; } = [];
[Column("tags", TypeName = "character varying(128)[]")]
public List Tags { get; set; } = [];
[Column("hasPoll")] public bool HasPoll { get; set; }
///
/// [Denormalized]
///
[Column("userHost")]
[StringLength(512)]
public string? UserHost { get; set; }
///
/// [Denormalized]
///
[Column("replyUserId")]
[StringLength(32)]
public string? ReplyUserId { get; set; }
///
/// Mastodon requires a slightly differently computed replyUserId field. To save processing time, we do this ahead of
/// time.
///
[Column("mastoReplyUserId")]
[StringLength(32)]
public string? MastoReplyUserId { get; set; }
///
/// [Denormalized]
///
[Column("replyUserHost")]
[StringLength(512)]
public string? ReplyUserHost { get; set; }
///
/// [Denormalized]
///
[Column("renoteUserId")]
[StringLength(32)]
public string? RenoteUserId { get; set; }
///
/// [Denormalized]
///
[Column("renoteUserHost")]
[StringLength(512)]
public string? RenoteUserHost { get; set; }
///
/// The human readable url of a note. it will be null when the note is local.
///
[Column("url")]
[StringLength(512)]
public string? Url { get; set; }
///
/// The ID of source channel.
///
[Column("channelId")]
[StringLength(32)]
public string? ChannelId { get; set; }
[Column("threadId")]
[StringLength(256)]
public string? ThreadId { get; set; }
[Projectable] [NotMapped] public string ThreadIdOrId => ThreadId ?? Id;
///
/// The updated date of the Note.
///
[Column("updatedAt")]
public DateTime? UpdatedAt { get; set; }
[ForeignKey(nameof(ChannelId))]
[InverseProperty(nameof(Tables.Channel.Notes))]
public virtual Channel? Channel { get; set; }
[InverseProperty(nameof(ChannelNotePin.Note))]
public virtual ICollection ChannelNotePins { get; set; } = new List();
[InverseProperty(nameof(ClipNote.Note))]
public virtual ICollection ClipNotes { get; set; } = new List();
[InverseProperty(nameof(Renote))] public virtual ICollection InverseRenote { get; set; } = new List();
[InverseProperty(nameof(Reply))] public virtual ICollection InverseReply { get; set; } = new List();
[InverseProperty(nameof(NoteEdit.Note))]
public virtual ICollection NoteEdits { get; set; } = new List();
[InverseProperty(nameof(NoteBookmark.Note))]
public virtual ICollection NoteBookmarks { get; set; } = new List();
[InverseProperty(nameof(NoteReaction.Note))]
public virtual ICollection NoteReactions { get; set; } = new List();
[InverseProperty(nameof(NoteLike.Note))]
public virtual ICollection NoteLikes { get; set; } = new List();
[InverseProperty(nameof(NoteUnread.Note))]
public virtual ICollection NoteUnreads { get; set; } = new List();
[InverseProperty(nameof(NoteWatching.Note))]
public virtual ICollection NoteWatchings { get; set; } = new List();
[InverseProperty(nameof(Notification.Note))]
public virtual ICollection Notifications { get; set; } = new List();
[InverseProperty(nameof(Tables.Poll.Note))]
public virtual Poll? Poll { get; set; }
[InverseProperty(nameof(PollVote.Note))]
public virtual ICollection PollVotes { get; set; } = new List();
[InverseProperty(nameof(Tables.PromoNote.Note))]
public virtual PromoNote? PromoNote { get; set; }
[InverseProperty(nameof(PromoRead.Note))]
public virtual ICollection PromoReads { get; set; } = new List();
[ForeignKey(nameof(RenoteId))]
[InverseProperty(nameof(InverseRenote))]
public virtual Note? Renote { get; set; }
[ForeignKey(nameof(ReplyId))]
[InverseProperty(nameof(InverseReply))]
public virtual Note? Reply { get; set; }
[Projectable]
public string RawAttachments
=> InternalRawAttachments(Id);
[NotMapped] [Projectable] public bool IsPureRenote => (RenoteId != null || Renote != null) && !IsQuote;
[NotMapped]
[Projectable]
public bool IsQuote => (RenoteId != null || Renote != null) &&
(Text != null || Cw != null || HasPoll || FileIds.Count > 0);
[ForeignKey(nameof(UserId))]
[InverseProperty(nameof(Tables.User.Notes))]
public virtual User User { get; set; } = null!;
[InverseProperty(nameof(UserNotePin.Note))]
public virtual ICollection UserNotePins { get; set; } = new List();
[NotMapped] public bool? PrecomputedIsReplyVisible { get; private set; } = false;
[NotMapped] public bool? PrecomputedIsRenoteVisible { get; private set; } = false;
[Key]
[Column("id")]
[StringLength(32)]
public string Id { get; set; } = null!;
public static string InternalRawAttachments(string id)
=> throw new NotSupportedException();
[Projectable]
public bool TextContainsCaseInsensitive(string str) =>
Text != null && EF.Functions.ILike(Text, "%" + EfHelpers.EscapeLikeQuery(str) + "%", @"\");
[Projectable]
[SuppressMessage("ReSharper", "MergeIntoPattern", Justification = "Projectable chain must not contain patterns")]
public bool IsVisibleFor(User? user) =>
(VisibilityIsPublicOrHome && (!LocalOnly || (user != null && user.IsLocalUser))) ||
(user != null && CheckComplexVisibility(user));
[Projectable]
[SuppressMessage("ReSharper", "MemberCanBePrivate.Global", Justification = "Projectable chain must to be public")]
public bool CheckComplexVisibility(User user) => User == user ||
VisibleUserIds.Contains(user.Id) ||
Mentions.Contains(user.Id) ||
(Visibility == NoteVisibility.Followers &&
(User.IsFollowedBy(user) || ReplyUserId == user.Id));
public bool IsVisibleFor(User? user, IEnumerable followingIds) =>
VisibilityIsPublicOrHome || (user != null && CheckComplexVisibility(user, followingIds));
private bool CheckComplexVisibility(User user, IEnumerable followingIds)
=> User.Id == user.Id ||
VisibleUserIds.Contains(user.Id) ||
Mentions.Contains(user.Id) ||
(Visibility == NoteVisibility.Followers &&
(followingIds.Contains(User.Id) || ReplyUserId == user.Id));
public Note WithPrecomputedVisibilities(bool reply, bool renote, bool renoteRenote)
{
if (Reply != null)
PrecomputedIsReplyVisible = reply;
if (Renote != null)
PrecomputedIsRenoteVisible = renote;
if (Renote?.Renote != null)
Renote.PrecomputedIsRenoteVisible = renoteRenote;
return this;
}
[Projectable]
[SuppressMessage("ReSharper", "MergeIntoPattern", Justification = "Projectables")]
[SuppressMessage("ReSharper", "MergeSequentialChecks", Justification = "Projectables")]
public Note WithPrecomputedVisibilities(User user)
=> WithPrecomputedVisibilities(Reply != null && Reply.IsVisibleFor(user),
Renote != null && Renote.IsVisibleFor(user),
Renote != null && Renote.Renote != null && Renote.Renote.IsVisibleFor(user));
public string GetPublicUri(Config.InstanceSection config) => UserHost == null
? $"https://{config.WebDomain}/notes/{Id}"
: throw new Exception("Cannot access PublicUri for remote note");
public string? GetPublicUriOrNull(Config.InstanceSection config) => UserHost == null
? $"https://{config.WebDomain}/notes/{Id}"
: null;
public class MentionedUser
{
[J("uri")] public required string Uri { get; set; }
[J("url")] public string? Url { get; set; }
[J("username")] public required string Username { get; set; }
[J("host")] public required string? Host { get; set; }
}
}