Files
Foxel/Services/PictureService.cs

771 lines
26 KiB
C#
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
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 = null)
{
// 如果未指定存储类型,则从配置中获取默认存储类型
if (storageType == null)
{
var defaultStorageTypeStr = configuration["Storage:DefaultStorage"];
if (string.IsNullOrEmpty(defaultStorageTypeStr) || !Enum.TryParse<StorageType>(defaultStorageTypeStr, out var defaultStorageType))
{
defaultStorageType = StorageType.Local; // 如果配置中没有或解析失败,使用本地存储作为默认
}
storageType = defaultStorageType;
}
string fileExtension = Path.GetExtension(fileName);
string newFileName = $"{Guid.NewGuid()}{fileExtension}";
// 获取对应的存储提供者
var storageProvider = storageProviderFactory.GetProvider(storageType.Value);
// 使用存储提供者保存文件
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.Value,
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);
}
}