Amatsugu/AobaV2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects 1 Releases Wiki Activity

generate thumbnails

Browse Source
This commit is contained in:
Amatsugu
2025-05-26 21:36:40 -04:00
parent c40b6f0fa3
commit 1051799fb3
9 changed files with 53 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
AobaCore/Services/AobaService.cs
View File

@@ -18,7 +18,12 @@ public class AobaService(IMongoDatabase db)
return await _media.Find(m => m.Id == id).FirstOrDefaultAsync(cancellationToken);
}
public async Task<PagedResult<Media>> FindMediaAsync(string? query, int page = 1, int pageSize = 100)
public async Task<Media?> GetMediaFromFileAsync(ObjectId id, CancellationToken cancellationToken = default)
{
return await _media.Find(m => m.MediaId == 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);

37
AobaCore/Services/ThumbnailService.cs
View File

@@ -4,6 +4,9 @@ using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.GridFS;
using SixLabors.ImageSharp;
using SixLabors.ImageSharp.Processing;
using System;
using System.Collections.Generic;
using System.Linq;
@@ -11,18 +14,42 @@ using System.Text;
using System.Threading.Tasks;
namespace AobaCore.Services;
internal class ThumbnailService(IMongoDatabase db, AobaService aobaService)
public class ThumbnailService(IMongoDatabase db, AobaService aobaService)
{
private readonly GridFSBucket _gridfs = new GridFSBucket(db);
private readonly IMongoCollection<MeidaThumbnail> _thumbnails = db.GetCollection<MeidaThumbnail>("thumbs");
public async Task<MemoryStream> GetThumbnailAsync(ObjectId id)
public async Task<Stream?> GetThumbnailAsync(ObjectId id, CancellationToken cancellationToken = default)
{
throw new NotImplementedException();
var media = await aobaService.GetMediaAsync(id);
if (media == null)
return null;
if (media.MediaType != MediaType.Image)
return null;
using var file = await _gridfs.OpenDownloadStreamAsync(media.MediaId, new GridFSDownloadOptions { Seekable = true });
return await GenerateThumbnailAsync(file, cancellationToken);
}
public async Task<Stream?> GetThumbnailFromFileAsync(ObjectId id, CancellationToken cancellationToken = default)
{
var media = await aobaService.GetMediaFromFileAsync(id);
if (media == null)
return null;
if (media.MediaType != MediaType.Image)
return null;
using var file = await _gridfs.OpenDownloadStreamAsync(media.MediaId, new GridFSDownloadOptions { Seekable = true });
return await GenerateThumbnailAsync(file, cancellationToken);
}
public async Task GenerateThumbnailAsync(ObjectId id)
public async Task<Stream> GenerateThumbnailAsync(Stream stream, CancellationToken cancellationToken = default)
{
var img = Image.Load(stream);
img.Mutate(o =>
{
o.Resize(200, 200);
});
var result = new MemoryStream();
await img.SaveAsWebpAsync(result, cancellationToken);
result.Position = 0;
return result;
}
}