Iceshrimp.NET/Iceshrimp.Backend/Core/Database/Tables/DriveFile.cs

using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using Iceshrimp.Backend.Core.Helpers;
using Microsoft.EntityFrameworkCore;
using J = System.Text.Json.Serialization.JsonPropertyNameAttribute;

namespace Iceshrimp.Backend.Core.Database.Tables;

[Table("drive_file")]
[Index("IsLink")]
[Index("Sha256")]
[Index("UserId", "FolderId", "Id")]
[Index("UserId")]
[Index("UserHost")]
[Index("Type")]
[Index("IsSensitive")]
[Index("FolderId")]
[Index("WebpublicAccessKey")]
[Index("CreatedAt")]
[Index("AccessKey")]
[Index("Uri")]
[Index("ThumbnailAccessKey")]
public class DriveFile : IEntity {
	/// <summary>
	///     The created date of the DriveFile.
	/// </summary>
	[Column("createdAt")]
	public DateTime CreatedAt { get; set; } = DateTime.UtcNow;

	/// <summary>
	///     The owner ID.
	/// </summary>
	[Column("userId")]
	[StringLength(32)]
	public string? UserId { get; set; }

	/// <summary>
	///     The host of owner. It will be null if the user in local.
	/// </summary>
	[Column("userHost")]
	[StringLength(512)]
	public string? UserHost { get; set; }

	/// <summary>
	///     The SHA256 hash of the DriveFile.
	/// </summary>
	[Column("sha256")]
	[StringLength(64)]
	public string? Sha256 { get; set; }

	/// <summary>
	///     The file name of the DriveFile.
	/// </summary>
	[Column("name")]
	[StringLength(256)]
	public string Name { get; set; } = null!;

	/// <summary>
	///     The content type (MIME) of the DriveFile.
	/// </summary>
	[Column("type")]
	[StringLength(128)]
	public string Type { get; set; } = null!;

	/// <summary>
	///     The file size (bytes) of the DriveFile.
	/// </summary>
	[Column("size")]
	public int Size { get; set; }

	/// <summary>
	///     The comment of the DriveFile.
	/// </summary>
	[Column("comment")]
	public string? Comment { get; set; }

	/// <summary>
	///     The any properties of the DriveFile. For example, it includes image width/height.
	/// </summary>
	[Column("properties", TypeName = "jsonb")]
	public FileProperties Properties { get; set; } = null!;

	[Column("storedInternal")] public bool StoredInternal { get; set; }

	/// <summary>
	///     The URL of the DriveFile.
	/// </summary>
	[Column("url")]
	[StringLength(512)]
	public string Url { get; set; } = null!;

	/// <summary>
	///     The URL of the thumbnail of the DriveFile.
	/// </summary>
	[Column("thumbnailUrl")]
	[StringLength(512)]
	public string? ThumbnailUrl { get; set; }

	/// <summary>
	///     The URL of the webpublic of the DriveFile.
	/// </summary>
	[Column("webpublicUrl")]
	[StringLength(512)]
	public string? WebpublicUrl { get; set; }

	[Column("accessKey")]
	[StringLength(256)]
	public string? AccessKey { get; set; }

	[Column("thumbnailAccessKey")]
	[StringLength(256)]
	public string? ThumbnailAccessKey { get; set; }

	[Column("webpublicAccessKey")]
	[StringLength(256)]
	public string? WebpublicAccessKey { get; set; }

	/// <summary>
	///     The URI of the DriveFile. it will be null when the DriveFile is local.
	/// </summary>
	[Column("uri")]
	[StringLength(512)]
	public string? Uri { get; set; }

	[Column("src")] [StringLength(512)] public string? Src { get; set; }

	/// <summary>
	///     The parent folder ID. If null, it means the DriveFile is located in root.
	/// </summary>
	[Column("folderId")]
	[StringLength(32)]
	public string? FolderId { get; set; }

	/// <summary>
	///     Whether the DriveFile is NSFW.
	/// </summary>
	[Column("isSensitive")]
	public bool IsSensitive { get; set; }

	/// <summary>
	///     Whether the DriveFile is direct link to remote server.
	/// </summary>
	[Column("isLink")]
	public bool IsLink { get; set; }

	/// <summary>
	///     The BlurHash string.
	/// </summary>
	[Column("blurhash")]
	[StringLength(128)]
	public string? Blurhash { get; set; }

	[Column("webpublicType")]
	[StringLength(128)]
	public string? WebpublicType { get; set; }

	[Column("requestHeaders", TypeName = "jsonb")]
	public Dictionary<string, string>? RequestHeaders { get; set; }

	[Column("requestIp")]
	[StringLength(128)]
	public string? RequestIp { get; set; }

	[InverseProperty(nameof(Channel.Banner))]
	public virtual ICollection<Channel> Channels { get; set; } = new List<Channel>();

	[ForeignKey("FolderId")]
	[InverseProperty(nameof(DriveFolder.DriveFiles))]
	public virtual DriveFolder? Folder { get; set; }

	[InverseProperty(nameof(MessagingMessage.File))]
	public virtual ICollection<MessagingMessage> MessagingMessages { get; set; } = new List<MessagingMessage>();

	[InverseProperty(nameof(Page.EyeCatchingImage))]
	public virtual ICollection<Page> Pages { get; set; } = new List<Page>();

	[ForeignKey("UserId")]
	[InverseProperty(nameof(Tables.User.DriveFiles))]
	public virtual User? User { get; set; }

	[InverseProperty(nameof(Tables.User.Avatar))]
	public virtual User? UserAvatar { get; set; }

	[InverseProperty(nameof(Tables.User.Banner))]
	public virtual User? UserBanner { get; set; }

	[Key]
	[Column("id")]
	[StringLength(32)]
	public string Id { get; set; } = IdHelpers.GenerateSlowflakeId();

	public class FileProperties {
		[J("width")]       public int?    Width        { get; set; }
		[J("height")]      public int?    Height       { get; set; }
		[J("orientation")] public int?    Orientation  { get; set; }
		[J("avgColor")]    public string? AverageColor { get; set; }
	}
}