thumbnail service wip

AobaCore/Services/AobaService.cs

@@ -0,0 +1,90 @@
+using AobaCore.Models;
+
+using MaybeError.Errors;
+
+using MongoDB.Bson;
+using MongoDB.Driver;
+using MongoDB.Driver.GridFS;
+
+namespace AobaCore.Services;
+
+public class AobaService(IMongoDatabase db)
+{
+	private readonly IMongoCollection<Media> _media = db.GetCollection<Media>("media");
+	private readonly GridFSBucket _gridFs = new(db);
+
+	public async Task<Media?> GetMediaAsync(ObjectId id, CancellationToken cancellationToken = default)
+	{
+		return await _media.Find(m => m.Id == id).FirstOrDefaultAsync(cancellationToken);
+	}
+
+	public async Task<PagedResult<Media>> FindMediaAsync(string? query, int page = 1, int pageSize = 100)
+	{
+		var filter = string.IsNullOrWhiteSpace(query) ? "{}" : Builders<Media>.Filter.Text(query);
+		var sort = Builders<Media>.Sort.Descending(m => m.UploadDate);
+		var find = _media.Find(filter);
+
+		var total = await find.CountDocumentsAsync();
+		page -= 1;
+		var items = await find.Sort(sort).Skip(page * pageSize).Limit(pageSize).ToListAsync();
+		return new PagedResult<Media>(items, page, pageSize, total);
+	}
+
+
+	public Task AddMediaAsync(Media media, CancellationToken cancellationToken = default)
+	{
+		return _media.InsertOneAsync(media, null, cancellationToken);
+	}
+
+	public Task IncrementViewCountAsync(ObjectId id, CancellationToken cancellationToken = default)
+	{
+		return _media.UpdateOneAsync(m => m.Id == id, Builders<Media>.Update.Inc(m => m.ViewCount, 1), cancellationToken: cancellationToken);
+	}
+
+	public Task IncrementFileViewCountAsync(ObjectId fileId, CancellationToken cancellationToken = default)
+	{
+		return _media.UpdateOneAsync(m => m.MediaId == fileId, Builders<Media>.Update.Inc(m => m.ViewCount, 1), cancellationToken: cancellationToken);
+	}
+
+
+	public async Task<Maybe<Media>> UploadFileAsync(Stream data, string filename, ObjectId owner, CancellationToken cancellationToken = default)
+	{
+		try
+		{
+			var fileId = await _gridFs.UploadFromStreamAsync(filename, data, cancellationToken: cancellationToken);
+			var media = new Media(fileId, filename, owner);
+			await AddMediaAsync(media, cancellationToken);
+			return media;
+		}
+		catch (Exception ex)
+		{
+			return ex;
+		}
+	}
+
+	public async Task<MaybeEx<GridFSDownloadStream, GridFSException>> GetFileStreamAsync(ObjectId id, bool seekable = false, CancellationToken cancellationToken = default)
+	{
+		try
+		{
+			return await _gridFs.OpenDownloadStreamAsync(id, new GridFSDownloadOptions { Seekable = seekable }, cancellationToken);
+		}
+		catch (GridFSException ex)
+		{
+			return ex;
+		}
+	}
+
+	public async Task DeleteFileAsync(ObjectId fileId, CancellationToken cancellationToken = default)
+	{
+		try
+		{
+			cancellationToken.ThrowIfCancellationRequested();
+			await _gridFs.DeleteAsync(fileId, CancellationToken.None);
+			await _media.DeleteOneAsync(m => m.MediaId == fileId, CancellationToken.None);
+		}
+		catch (GridFSFileNotFoundException)
+		{
+			//ignore if file was not found
+		}
+	}
+}