More product setup

AobaCore/Models/Media.cs

@@ -0,0 +1,79 @@
+using MongoDB.Bson;
+using MongoDB.Bson.Serialization.Attributes;
+
+namespace AobaV2.Models;
+
+[BsonIgnoreExtraElements]
+public class Media
+{
+	[BsonId]
+	public ObjectId Id { get; set; }
+	public ObjectId MediaId { get; set; }
+	public required string Filename { get; set; }
+	public MediaType MediaType { get; set; }
+	public required string Ext { get; set; }
+	public int ViewCount { get; set; }
+	public ObjectId Owner { get; set; }
+
+
+	public static readonly Dictionary<string, MediaType> KnownTypes = new()
+		{
+			{ ".jpg", MediaType.Image },
+			{ ".avif", MediaType.Image },
+			{ ".jpeg", MediaType.Image },
+			{ ".png", MediaType.Image },
+			{ ".apng", MediaType.Image },
+			{ ".webp", MediaType.Image },
+			{ ".ico", MediaType.Image },
+			{ ".gif", MediaType.Image },
+			{ ".mp3", MediaType.Audio },
+			{ ".flac", MediaType.Audio },
+			{ ".alac", MediaType.Audio },
+			{ ".mp4", MediaType.Video },
+			{ ".webm", MediaType.Video },
+			{ ".mov", MediaType.Video },
+			{ ".avi", MediaType.Video },
+			{ ".mkv", MediaType.Video },
+			{ ".txt", MediaType.Text },
+			{ ".log", MediaType.Text },
+			{ ".css", MediaType.Code },
+			{ ".cs", MediaType.Code },
+			{ ".cpp", MediaType.Code },
+			{ ".lua", MediaType.Code },
+			{ ".js", MediaType.Code },
+			{ ".htm", MediaType.Code },
+			{ ".html", MediaType.Code },
+			{ ".cshtml", MediaType.Code },
+			{ ".xml", MediaType.Code },
+			{ ".json", MediaType.Code },
+			{ ".py", MediaType.Code },
+		};
+
+	public string GetMediaUrl()
+	{
+		return this switch
+		{
+			Media { MediaType: MediaType.Raw or MediaType.Text or MediaType.Code} => $"/i/dl/{Id}/{Filename}",
+			_ => $"/i/{Id}"
+		};
+	}
+
+	public static MediaType GetMediaType(string filename)
+	{
+		string ext = Path.GetExtension(filename);
+		if (KnownTypes.TryGetValue(ext, out MediaType mType))
+			return mType;
+		else
+			return MediaType.Raw;
+	}
+}
+
+public enum MediaType
+{
+	Image,
+	Audio,
+	Video,
+	Text,
+	Code,
+	Raw
+}