mirror of
https://github.com/DrizzleTime/Foxel.git
synced 2026-08-28 19:46:41 +08:00
Initial commit
This commit is contained in:
shiyu
@@ -0,0 +1,760 @@
|
||||
using Foxel.Models;
|
||||
using Foxel.Models.DataBase;
|
||||
using Foxel.Services.Interface;
|
||||
using Foxel.Utils;
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using Pgvector;
|
||||
using Pgvector.EntityFrameworkCore;
|
||||
using System.Text.Json;
|
||||
using Foxel.Models.Response.Picture;
|
||||
|
||||
namespace Foxel.Services;
|
||||
|
||||
public class PictureService(
|
||||
IDbContextFactory<MyDbContext> contextFactory,
|
||||
IAiService embeddingService,
|
||||
IConfigService configuration,
|
||||
IBackgroundTaskQueue backgroundTaskQueue,
|
||||
IStorageProviderFactory storageProviderFactory)
|
||||
: IPictureService
|
||||
{
|
||||
private readonly string _serverUrl = configuration["AppSettings:ServerUrl"];
|
||||
|
||||
public async Task<PaginatedResult<PictureResponse>> GetPicturesAsync(
|
||||
int page = 1,
|
||||
int pageSize = 8,
|
||||
string? searchQuery = null,
|
||||
List<string>? tags = null,
|
||||
DateTime? startDate = null,
|
||||
DateTime? endDate = null,
|
||||
int? userId = null,
|
||||
string? sortBy = "newest",
|
||||
bool? onlyWithGps = false,
|
||||
bool useVectorSearch = false,
|
||||
double similarityThreshold = 0.36,
|
||||
int? excludeAlbumId = null,
|
||||
int? albumId = null,
|
||||
bool onlyFavorites = false,
|
||||
int? ownerId = null,
|
||||
bool includeAllPublic = false
|
||||
)
|
||||
{
|
||||
if (page < 1) page = 1;
|
||||
if (pageSize < 1) pageSize = 8;
|
||||
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
|
||||
// 决定是使用向量搜索还是普通搜索
|
||||
if (useVectorSearch && !string.IsNullOrWhiteSpace(searchQuery))
|
||||
{
|
||||
return await PerformVectorSearchAsync(
|
||||
dbContext, page, pageSize, searchQuery, tags,
|
||||
startDate, endDate, userId, onlyWithGps, similarityThreshold,
|
||||
excludeAlbumId, albumId, onlyFavorites, ownerId, includeAllPublic);
|
||||
}
|
||||
else
|
||||
{
|
||||
return await PerformStandardSearchAsync(
|
||||
dbContext, page, pageSize, searchQuery, tags,
|
||||
startDate, endDate, userId, sortBy, onlyWithGps,
|
||||
excludeAlbumId, albumId, onlyFavorites, ownerId, includeAllPublic);
|
||||
}
|
||||
}
|
||||
|
||||
// 执行向量搜索
|
||||
private async Task<PaginatedResult<PictureResponse>> PerformVectorSearchAsync(
|
||||
MyDbContext dbContext,
|
||||
int page,
|
||||
int pageSize,
|
||||
string searchQuery,
|
||||
List<string>? tags,
|
||||
DateTime? startDate,
|
||||
DateTime? endDate,
|
||||
int? userId,
|
||||
bool? onlyWithGps,
|
||||
double similarityThreshold,
|
||||
int? excludeAlbumId,
|
||||
int? albumId,
|
||||
bool onlyFavorites,
|
||||
int? ownerId,
|
||||
bool includeAllPublic)
|
||||
{
|
||||
var queryEmbedding = await embeddingService.GetEmbeddingAsync(searchQuery);
|
||||
var queryVector = new Vector(queryEmbedding);
|
||||
|
||||
// 构建基础查询
|
||||
var query = dbContext.Pictures
|
||||
.Include(p => p.Tags)
|
||||
.Include(p => p.User)
|
||||
.Where(p => p.Embedding != null);
|
||||
|
||||
// 应用共通的查询条件
|
||||
query = ApplyCommonFilters(query, tags, startDate, endDate, userId, onlyWithGps,
|
||||
excludeAlbumId, albumId, onlyFavorites, ownerId, includeAllPublic);
|
||||
|
||||
// 执行向量搜索
|
||||
var allResults = await query
|
||||
.Select(p => new
|
||||
{
|
||||
Picture = p,
|
||||
Similarity = 1.0 - p.Embedding!.CosineDistance(queryVector)
|
||||
})
|
||||
.Where(p => p.Similarity >= similarityThreshold)
|
||||
.OrderByDescending(p => p.Similarity)
|
||||
.ToListAsync();
|
||||
|
||||
// 计算总数并分页
|
||||
var totalCount = allResults.Count;
|
||||
|
||||
var paginatedResults = allResults
|
||||
.Skip((page - 1) * pageSize)
|
||||
.Take(pageSize)
|
||||
.Select(r => MapPictureToResponse(r.Picture, _serverUrl))
|
||||
.ToList();
|
||||
|
||||
// 处理收藏信息
|
||||
await PopulateFavoriteInfo(dbContext, paginatedResults, userId);
|
||||
|
||||
// 为当前用户的图片添加相册信息
|
||||
if (userId.HasValue)
|
||||
{
|
||||
await PopulateAlbumInfo(dbContext, paginatedResults, userId.Value);
|
||||
}
|
||||
|
||||
return new PaginatedResult<PictureResponse>
|
||||
{
|
||||
Data = paginatedResults,
|
||||
Page = page,
|
||||
PageSize = pageSize,
|
||||
TotalCount = totalCount
|
||||
};
|
||||
}
|
||||
|
||||
// 执行标准搜索
|
||||
private async Task<PaginatedResult<PictureResponse>> PerformStandardSearchAsync(
|
||||
MyDbContext dbContext,
|
||||
int page,
|
||||
int pageSize,
|
||||
string? searchQuery,
|
||||
List<string>? tags,
|
||||
DateTime? startDate,
|
||||
DateTime? endDate,
|
||||
int? userId,
|
||||
string? sortBy,
|
||||
bool? onlyWithGps,
|
||||
int? excludeAlbumId,
|
||||
int? albumId,
|
||||
bool onlyFavorites,
|
||||
int? ownerId,
|
||||
bool includeAllPublic)
|
||||
{
|
||||
// 构建基础查询
|
||||
IQueryable<Picture> query = dbContext.Pictures
|
||||
.Include(p => p.Tags)
|
||||
.Include(p => p.User);
|
||||
|
||||
// 应用文本搜索条件
|
||||
if (!string.IsNullOrWhiteSpace(searchQuery))
|
||||
{
|
||||
var searchTerm = searchQuery.ToLower();
|
||||
query = query.Where(p =>
|
||||
(p.Name.ToLower().Contains(searchTerm)) ||
|
||||
(p.Description.ToLower().Contains(searchTerm)));
|
||||
}
|
||||
|
||||
// 应用共通的查询条件
|
||||
query = ApplyCommonFilters(query, tags, startDate, endDate, userId, onlyWithGps,
|
||||
excludeAlbumId, albumId, onlyFavorites, ownerId, includeAllPublic);
|
||||
|
||||
// 应用排序
|
||||
query = ApplySorting(query, sortBy);
|
||||
|
||||
// 获取总记录数
|
||||
var totalCount = await query.CountAsync();
|
||||
|
||||
// 获取分页数据
|
||||
var picturesData = await query
|
||||
.Skip((page - 1) * pageSize)
|
||||
.Take(pageSize)
|
||||
.ToListAsync();
|
||||
|
||||
// 转换为响应格式
|
||||
var pictures = picturesData
|
||||
.Select(p => MapPictureToResponse(p, _serverUrl))
|
||||
.ToList();
|
||||
|
||||
// 处理收藏信息
|
||||
await PopulateFavoriteInfo(dbContext, pictures, userId);
|
||||
|
||||
// 为当前用户的图片添加相册信息
|
||||
if (userId.HasValue)
|
||||
{
|
||||
await PopulateAlbumInfo(dbContext, pictures, userId.Value);
|
||||
}
|
||||
|
||||
return new PaginatedResult<PictureResponse>
|
||||
{
|
||||
Data = pictures,
|
||||
Page = page,
|
||||
PageSize = pageSize,
|
||||
TotalCount = totalCount
|
||||
};
|
||||
}
|
||||
|
||||
// 应用共通的过滤条件
|
||||
private IQueryable<Picture> ApplyCommonFilters(
|
||||
IQueryable<Picture> query,
|
||||
List<string>? tags,
|
||||
DateTime? startDate,
|
||||
DateTime? endDate,
|
||||
int? userId,
|
||||
bool? onlyWithGps,
|
||||
int? excludeAlbumId,
|
||||
int? albumId,
|
||||
bool onlyFavorites,
|
||||
int? ownerId,
|
||||
bool includeAllPublic)
|
||||
{
|
||||
// 应用标签筛选
|
||||
if (tags != null && tags.Any())
|
||||
{
|
||||
foreach (var tag in tags)
|
||||
{
|
||||
var tagName = tag.Trim();
|
||||
if (!string.IsNullOrEmpty(tagName))
|
||||
{
|
||||
var normalizedTagName = tagName.ToLower();
|
||||
query = query.Where(p => p.Tags!.Any(t => t.Name.ToLower().Equals(normalizedTagName)));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 应用日期范围筛选
|
||||
if (startDate.HasValue)
|
||||
{
|
||||
DateTime utcStartDate = startDate.Value.ToUniversalTime();
|
||||
query = query.Where(p =>
|
||||
(p.TakenAt.HasValue && p.TakenAt >= utcStartDate) ||
|
||||
(!p.TakenAt.HasValue && p.CreatedAt >= utcStartDate));
|
||||
}
|
||||
|
||||
if (endDate.HasValue)
|
||||
{
|
||||
DateTime utcEndDate = endDate.Value.ToUniversalTime().AddDays(1).AddMilliseconds(-1);
|
||||
query = query.Where(p =>
|
||||
(p.TakenAt.HasValue && p.TakenAt <= utcEndDate) ||
|
||||
(!p.TakenAt.HasValue && p.CreatedAt <= utcEndDate));
|
||||
}
|
||||
|
||||
// 应用用户筛选和权限过滤逻辑
|
||||
if (ownerId.HasValue)
|
||||
{
|
||||
if (userId.HasValue && userId.Value == ownerId.Value)
|
||||
{
|
||||
query = query.Where(p => p.User != null && p.User.Id == ownerId.Value);
|
||||
}
|
||||
else
|
||||
{
|
||||
query = query.Where(p =>
|
||||
p.User != null && p.User.Id == ownerId.Value && p.Permission == PermissionType.Public);
|
||||
}
|
||||
}
|
||||
else if (userId.HasValue)
|
||||
{
|
||||
if (includeAllPublic)
|
||||
{
|
||||
query = query.Where(p =>
|
||||
(p.User != null && p.User.Id == userId.Value) ||
|
||||
(p.User != null && p.User.Id != userId.Value &&
|
||||
p.Permission == PermissionType.Public)
|
||||
);
|
||||
}
|
||||
else
|
||||
{
|
||||
query = query.Where(p => p.User != null && p.User.Id == userId.Value);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
query = query.Where(p => p.Permission == PermissionType.Public);
|
||||
}
|
||||
|
||||
// 筛选有GPS信息的图片
|
||||
if (onlyWithGps == true)
|
||||
{
|
||||
query = query.Where(p =>
|
||||
p.ExifInfo != null &&
|
||||
!string.IsNullOrEmpty(p.ExifInfo.GpsLatitude) &&
|
||||
!string.IsNullOrEmpty(p.ExifInfo.GpsLongitude));
|
||||
}
|
||||
|
||||
// 排除指定相册的图片
|
||||
if (excludeAlbumId.HasValue)
|
||||
{
|
||||
query = query.Where(p => p.AlbumId != excludeAlbumId.Value || p.AlbumId == null);
|
||||
}
|
||||
|
||||
// 筛选指定相册的图片
|
||||
if (albumId.HasValue)
|
||||
{
|
||||
query = query.Where(p => p.AlbumId == albumId.Value);
|
||||
}
|
||||
|
||||
// 筛选收藏的图片
|
||||
if (onlyFavorites && userId.HasValue)
|
||||
{
|
||||
query = query.Where(p => p.Favorites!.Any(f => f.User.Id == userId.Value));
|
||||
}
|
||||
|
||||
return query;
|
||||
}
|
||||
|
||||
// 应用排序
|
||||
private IQueryable<Picture> ApplySorting(IQueryable<Picture> query, string? sortBy)
|
||||
{
|
||||
return sortBy?.ToLower() switch
|
||||
{
|
||||
// 拍摄时间排序
|
||||
"takenat_desc" or "newest" => query.OrderByDescending(p => p.TakenAt ?? p.CreatedAt),
|
||||
"takenat_asc" or "oldest" => query.OrderBy(p => p.TakenAt ?? p.CreatedAt),
|
||||
|
||||
// 上传时间排序
|
||||
"uploaddate_desc" => query.OrderByDescending(p => p.CreatedAt),
|
||||
"uploaddate_asc" => query.OrderBy(p => p.CreatedAt),
|
||||
|
||||
// 名称排序
|
||||
"name_asc" or "name" => query.OrderBy(p => p.Name),
|
||||
"name_desc" => query.OrderByDescending(p => p.Name),
|
||||
|
||||
// 默认排序
|
||||
_ => query.OrderByDescending(p => p.TakenAt ?? p.CreatedAt)
|
||||
};
|
||||
}
|
||||
|
||||
// 将数据库实体映射到响应对象
|
||||
private PictureResponse MapPictureToResponse(Picture picture, string serverUrl)
|
||||
{
|
||||
var storageProvider = storageProviderFactory.GetProvider(picture.StorageType);
|
||||
|
||||
return new PictureResponse
|
||||
{
|
||||
Id = picture.Id,
|
||||
Name = picture.Name,
|
||||
Path = storageProvider.GetUrl(picture.Path),
|
||||
ThumbnailPath = storageProvider.GetUrl(picture.ThumbnailPath),
|
||||
Description = picture.Description,
|
||||
CreatedAt = picture.CreatedAt,
|
||||
Tags = picture.Tags != null ? picture.Tags.Select(t => t.Name).ToList() : new List<string>(),
|
||||
TakenAt = picture.TakenAt,
|
||||
ExifInfo = picture.ExifInfo ?? new ExifInfo(),
|
||||
UserId = picture.UserId,
|
||||
Username = picture.User?.UserName,
|
||||
AlbumId = picture.AlbumId,
|
||||
Permission = picture.Permission
|
||||
};
|
||||
}
|
||||
|
||||
// 填充收藏信息
|
||||
private async Task PopulateFavoriteInfo(MyDbContext dbContext, List<PictureResponse> pictures, int? userId)
|
||||
{
|
||||
if (userId.HasValue && pictures.Any())
|
||||
{
|
||||
var pictureIds = pictures.Select(p => p.Id).ToList();
|
||||
|
||||
// 获取用户收藏的图片ID
|
||||
var favoritedPictureIds = await dbContext.Favorites
|
||||
.Where(f => f.User.Id == userId.Value && pictureIds.Contains(f.PictureId))
|
||||
.Select(f => f.PictureId)
|
||||
.ToHashSetAsync(); // 使用 ToHashSetAsync 提高查找效率
|
||||
|
||||
// 一次性获取所有相关图片的收藏总数
|
||||
var favoriteCounts = await dbContext.Favorites
|
||||
.Where(f => pictureIds.Contains(f.PictureId))
|
||||
.GroupBy(f => f.PictureId)
|
||||
.Select(g => new { PictureId = g.Key, Count = g.Count() })
|
||||
.ToDictionaryAsync(x => x.PictureId, x => x.Count);
|
||||
|
||||
foreach (var picture in pictures)
|
||||
{
|
||||
picture.IsFavorited = favoritedPictureIds.Contains(picture.Id);
|
||||
picture.FavoriteCount = favoriteCounts.GetValueOrDefault(picture.Id, 0);
|
||||
}
|
||||
}
|
||||
else if (pictures.Any()) // 如果用户未登录，仍然需要获取收藏总数
|
||||
{
|
||||
var pictureIds = pictures.Select(p => p.Id).ToList();
|
||||
var favoriteCounts = await dbContext.Favorites
|
||||
.Where(f => pictureIds.Contains(f.PictureId))
|
||||
.GroupBy(f => f.PictureId)
|
||||
.Select(g => new { PictureId = g.Key, Count = g.Count() })
|
||||
.ToDictionaryAsync(x => x.PictureId, x => x.Count);
|
||||
|
||||
foreach (var picture in pictures)
|
||||
{
|
||||
picture.IsFavorited = false; // 用户未登录，不可能收藏
|
||||
picture.FavoriteCount = favoriteCounts.GetValueOrDefault(picture.Id, 0);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 填充相册信息
|
||||
private async Task PopulateAlbumInfo(MyDbContext dbContext, List<PictureResponse> pictures, int userId)
|
||||
{
|
||||
if (!pictures.Any())
|
||||
return;
|
||||
|
||||
// 获取当前用户拥有的图片ID列表
|
||||
var userPictureIds = pictures
|
||||
.Where(p => p.UserId == userId)
|
||||
.Select(p => p.Id)
|
||||
.ToList();
|
||||
|
||||
if (!userPictureIds.Any())
|
||||
return;
|
||||
|
||||
// 获取相册信息
|
||||
var pictureAlbums = await dbContext.Pictures
|
||||
.Where(p => userPictureIds.Contains(p.Id) && p.AlbumId.HasValue)
|
||||
.Select(p => new { p.Id, p.AlbumId, AlbumName = p.Album!.Name })
|
||||
.ToDictionaryAsync(p => p.Id, p => new { p.AlbumId, p.AlbumName });
|
||||
|
||||
// 填充相册信息到图片响应中
|
||||
foreach (var picture in pictures)
|
||||
{
|
||||
if (picture.UserId == userId && pictureAlbums.TryGetValue(picture.Id, out var albumInfo))
|
||||
{
|
||||
picture.AlbumId = albumInfo.AlbumId;
|
||||
picture.AlbumName = albumInfo.AlbumName;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public async Task<(PictureResponse Picture, int Id)> UploadPictureAsync(
|
||||
string fileName,
|
||||
Stream fileStream,
|
||||
string contentType,
|
||||
int? userId,
|
||||
PermissionType permission = PermissionType.Public,
|
||||
int? albumId = null,
|
||||
StorageType storageType = StorageType.Local)
|
||||
{
|
||||
string fileExtension = Path.GetExtension(fileName);
|
||||
string newFileName = $"{Guid.NewGuid()}{fileExtension}";
|
||||
|
||||
// 获取对应的存储提供者
|
||||
var storageProvider = storageProviderFactory.GetProvider(storageType);
|
||||
|
||||
// 使用存储提供者保存文件
|
||||
string relativePath = await storageProvider.SaveAsync(fileStream, fileName, contentType);
|
||||
|
||||
// 创建基本的Picture对象，使用文件名作为标题和描述
|
||||
string initialTitle = Path.GetFileNameWithoutExtension(fileName);
|
||||
string initialDescription = $"Uploaded on {DateTime.UtcNow}";
|
||||
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
|
||||
// 获取用户
|
||||
User? user = null;
|
||||
if (userId is not null)
|
||||
{
|
||||
user = await dbContext.Users.FirstOrDefaultAsync(u => u.Id == userId);
|
||||
if (user == null)
|
||||
{
|
||||
throw new Exception("找不到指定的用户");
|
||||
}
|
||||
}
|
||||
|
||||
// 检查相册是否存在并且属于当前用户
|
||||
Album? album = null;
|
||||
if (albumId.HasValue)
|
||||
{
|
||||
album = await dbContext.Albums.Include(a => a.User)
|
||||
.FirstOrDefaultAsync(a => a.Id == albumId.Value);
|
||||
|
||||
if (album == null)
|
||||
{
|
||||
throw new KeyNotFoundException($"找不到ID为{albumId.Value}的相册");
|
||||
}
|
||||
|
||||
if (album.User.Id != userId)
|
||||
{
|
||||
throw new Exception("您无权将图片添加到此相册");
|
||||
}
|
||||
}
|
||||
|
||||
bool isAnonymous = userId == null;
|
||||
|
||||
// 创建图片对象并保存到数据库
|
||||
var picture = new Picture
|
||||
{
|
||||
Name = initialTitle,
|
||||
Description = initialDescription,
|
||||
Path = relativePath,
|
||||
User = user,
|
||||
Permission = permission,
|
||||
AlbumId = albumId,
|
||||
StorageType = storageType,
|
||||
ProcessingStatus = isAnonymous ? ProcessingStatus.Completed : ProcessingStatus.Pending,
|
||||
ThumbnailPath = isAnonymous ? relativePath : null
|
||||
};
|
||||
|
||||
dbContext.Pictures.Add(picture);
|
||||
await dbContext.SaveChangesAsync();
|
||||
|
||||
if (!isAnonymous)
|
||||
{
|
||||
await backgroundTaskQueue.QueuePictureProcessingTaskAsync(picture.Id, relativePath);
|
||||
}
|
||||
|
||||
// 返回图片基本信息
|
||||
var pictureResponse = new PictureResponse
|
||||
{
|
||||
Id = picture.Id,
|
||||
Name = picture.Name,
|
||||
Path = storageProvider.GetUrl(relativePath),
|
||||
ThumbnailPath = isAnonymous ? storageProvider.GetUrl(relativePath) : null,
|
||||
Description = picture.Description,
|
||||
CreatedAt = picture.CreatedAt,
|
||||
Tags = new List<string>(),
|
||||
Permission = permission,
|
||||
AlbumId = albumId,
|
||||
AlbumName = album?.Name,
|
||||
ProcessingStatus = picture.ProcessingStatus
|
||||
};
|
||||
|
||||
return (pictureResponse, picture.Id);
|
||||
}
|
||||
|
||||
public async Task<ExifInfo> GetPictureExifInfoAsync(int pictureId)
|
||||
{
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
var picture = await dbContext.Pictures.FindAsync(pictureId);
|
||||
|
||||
if (picture == null)
|
||||
throw new KeyNotFoundException($"找不到ID为{pictureId}的图片");
|
||||
|
||||
// 如果已有保存的EXIF信息，则直接返回
|
||||
if (!string.IsNullOrEmpty(picture.ExifInfoJson))
|
||||
{
|
||||
var exifInfo = JsonSerializer.Deserialize<ExifInfo>(picture.ExifInfoJson);
|
||||
return exifInfo ?? new ExifInfo { ErrorMessage = "无法解析EXIF信息" };
|
||||
}
|
||||
|
||||
// 否则从文件中提取
|
||||
string fullPath = Path.Combine(Directory.GetCurrentDirectory(), picture.Path.TrimStart('/'));
|
||||
if (!File.Exists(fullPath))
|
||||
{
|
||||
return new ExifInfo { ErrorMessage = "找不到图片文件" };
|
||||
}
|
||||
|
||||
return await ImageHelper.ExtractExifInfoAsync(fullPath);
|
||||
}
|
||||
|
||||
public async Task<Dictionary<int, (bool Success, string? ErrorMessage, int? UserId)>> DeleteMultiplePicturesAsync(
|
||||
List<int> pictureIds)
|
||||
{
|
||||
var results = new Dictionary<int, (bool Success, string? ErrorMessage, int? UserId)>();
|
||||
if (pictureIds.Count == 0)
|
||||
return results;
|
||||
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
|
||||
var picturesToDelete = await dbContext.Pictures
|
||||
.Include(p => p.User)
|
||||
.Where(p => pictureIds.Contains(p.Id))
|
||||
.ToListAsync();
|
||||
|
||||
var foundPictureIds = picturesToDelete.Select(p => p.Id).ToHashSet();
|
||||
foreach (var id in pictureIds.Where(id => !foundPictureIds.Contains(id)))
|
||||
{
|
||||
results[id] = (false, "找不到此图片", null);
|
||||
}
|
||||
|
||||
var filesToDelete =
|
||||
new List<(int PictureId, string Path, string ThumbnailPath, int? UserId, StorageType StorageType)>();
|
||||
foreach (var picture in picturesToDelete)
|
||||
{
|
||||
filesToDelete.Add((picture.Id, picture.Path, picture.ThumbnailPath, picture.User?.Id, picture.StorageType));
|
||||
}
|
||||
|
||||
if (picturesToDelete.Any())
|
||||
{
|
||||
dbContext.Pictures.RemoveRange(picturesToDelete);
|
||||
await dbContext.SaveChangesAsync();
|
||||
}
|
||||
|
||||
foreach (var (pictureId, path, thumbnailPath, userId, storageType) in filesToDelete)
|
||||
{
|
||||
try
|
||||
{
|
||||
string? errorMsg = null;
|
||||
|
||||
try
|
||||
{
|
||||
// 根据存储类型获取相应的存储提供者并删除文件
|
||||
var storageProvider = storageProviderFactory.GetProvider(storageType);
|
||||
await storageProvider.DeleteAsync(path);
|
||||
|
||||
// 删除缩略图
|
||||
if (storageType == StorageType.Local)
|
||||
{
|
||||
// 对于本地存储，使用本地存储提供者删除缩略图
|
||||
await storageProvider.DeleteAsync(thumbnailPath);
|
||||
}
|
||||
else
|
||||
{
|
||||
// 对于其他存储类型(如Telegram)，使用相同的存储提供者删除缩略图
|
||||
// 因为缩略图元数据格式与原文件相同
|
||||
await storageProvider.DeleteAsync(thumbnailPath);
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
errorMsg = $"数据库记录已删除，但删除文件失败: {ex.Message}";
|
||||
Console.WriteLine($"删除图片文件时出错：{ex.Message}");
|
||||
}
|
||||
|
||||
results[pictureId] = (true, errorMsg, userId);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
results[pictureId] = (false, $"处理图片删除时出错: {ex.Message}", userId);
|
||||
}
|
||||
}
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
public async Task<(PictureResponse Picture, int? UserId)> UpdatePictureAsync(
|
||||
int pictureId,
|
||||
string? name = null,
|
||||
string? description = null,
|
||||
List<string>? tags = null)
|
||||
{
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
|
||||
var picture = await dbContext.Pictures
|
||||
.Include(p => p.User)
|
||||
.Include(p => p.Tags)
|
||||
.FirstOrDefaultAsync(p => p.Id == pictureId);
|
||||
|
||||
if (picture == null)
|
||||
throw new KeyNotFoundException($"找不到ID为{pictureId}的图片");
|
||||
|
||||
var userId = picture.User?.Id;
|
||||
|
||||
if (!string.IsNullOrWhiteSpace(name))
|
||||
{
|
||||
picture.Name = name.Trim();
|
||||
}
|
||||
|
||||
if (!string.IsNullOrWhiteSpace(description))
|
||||
{
|
||||
picture.Description = description.Trim();
|
||||
}
|
||||
|
||||
if (!string.IsNullOrWhiteSpace(name) || !string.IsNullOrWhiteSpace(description))
|
||||
{
|
||||
var combinedText = $"{picture.Name}. {picture.Description}";
|
||||
var embedding = await embeddingService.GetEmbeddingAsync(combinedText);
|
||||
picture.Embedding = new Vector(embedding);
|
||||
}
|
||||
|
||||
if (tags != null)
|
||||
{
|
||||
picture.Tags?.Clear();
|
||||
|
||||
foreach (var tagName in tags.Where(t => !string.IsNullOrWhiteSpace(t)))
|
||||
{
|
||||
var tag = await dbContext.Tags.FirstOrDefaultAsync(t => t.Name.ToLower() == tagName.ToLower().Trim());
|
||||
|
||||
if (tag == null)
|
||||
{
|
||||
tag = new Tag { Name = tagName.Trim() };
|
||||
dbContext.Tags.Add(tag);
|
||||
}
|
||||
|
||||
picture.Tags?.Add(tag);
|
||||
}
|
||||
}
|
||||
|
||||
picture.UpdatedAt = DateTime.UtcNow;
|
||||
|
||||
await dbContext.SaveChangesAsync();
|
||||
|
||||
var pictureResponse = new PictureResponse
|
||||
{
|
||||
Id = picture.Id,
|
||||
Name = picture.Name,
|
||||
Path = storageProviderFactory.GetProvider(picture.StorageType).GetUrl(picture.Path),
|
||||
ThumbnailPath = storageProviderFactory.GetProvider(picture.StorageType).GetUrl(picture.ThumbnailPath),
|
||||
Description = picture.Description,
|
||||
CreatedAt = picture.CreatedAt,
|
||||
Tags = picture.Tags?.Select(t => t.Name).ToList() ?? new List<string>(),
|
||||
TakenAt = picture.TakenAt,
|
||||
ExifInfo = picture.ExifInfo
|
||||
};
|
||||
|
||||
return (pictureResponse, userId);
|
||||
}
|
||||
|
||||
public async Task<bool> FavoritePictureAsync(int pictureId, int userId)
|
||||
{
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
|
||||
// 检查图片是否存在
|
||||
var picture = await dbContext.Pictures.FindAsync(pictureId);
|
||||
if (picture == null)
|
||||
throw new KeyNotFoundException($"找不到ID为{pictureId}的图片");
|
||||
|
||||
// 检查用户是否存在
|
||||
var user = await dbContext.Users.FindAsync(userId);
|
||||
if (user == null)
|
||||
throw new KeyNotFoundException($"找不到ID为{userId}的用户");
|
||||
|
||||
// 检查是否已经收藏
|
||||
var existingFavorite = await dbContext.Favorites
|
||||
.FirstOrDefaultAsync(f => f.PictureId == pictureId && f.User.Id == userId);
|
||||
|
||||
if (existingFavorite != null)
|
||||
throw new InvalidOperationException("您已经收藏过此图片");
|
||||
|
||||
// 创建新收藏
|
||||
var favorite = new Favorite
|
||||
{
|
||||
PictureId = pictureId,
|
||||
User = user,
|
||||
CreatedAt = DateTime.UtcNow
|
||||
};
|
||||
|
||||
dbContext.Favorites.Add(favorite);
|
||||
await dbContext.SaveChangesAsync();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public async Task<bool> UnfavoritePictureAsync(int pictureId, int userId)
|
||||
{
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
|
||||
// 查找收藏记录
|
||||
var favorite = await dbContext.Favorites
|
||||
.FirstOrDefaultAsync(f => f.PictureId == pictureId && f.User.Id == userId);
|
||||
|
||||
if (favorite == null)
|
||||
throw new KeyNotFoundException($"未找到该图片的收藏记录");
|
||||
|
||||
// 移除收藏
|
||||
dbContext.Favorites.Remove(favorite);
|
||||
await dbContext.SaveChangesAsync();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public async Task<bool> IsPictureFavoritedByUserAsync(int pictureId, int userId)
|
||||
{
|
||||
await using var dbContext = await contextFactory.CreateDbContextAsync();
|
||||
return await dbContext.Favorites
|
||||
.AnyAsync(f => f.PictureId == pictureId && f.User.Id == userId);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user