feat: Add thumbnail management for albums and people, implement face embedding functionality

This commit is contained in:
2025-11-15 22:50:53 +01:00
parent 98f56e4f1e
commit 0f3e098d6d
28 changed files with 560 additions and 26 deletions

View File

@@ -0,0 +1,61 @@
use async_trait::async_trait;
use libertas_core::{
error::{CoreError, CoreResult},
models::FaceEmbedding,
repositories::FaceEmbeddingRepository,
};
use sqlx::PgPool;
use uuid::Uuid;
use crate::db_models::PostgresFaceEmbedding;
#[derive(Clone)]
pub struct PostgresFaceEmbeddingRepository {
pool: PgPool,
}
impl PostgresFaceEmbeddingRepository {
pub fn new(pool: PgPool) -> Self {
Self { pool }
}
}
#[async_trait]
impl FaceEmbeddingRepository for PostgresFaceEmbeddingRepository {
async fn create(&self, embedding: &FaceEmbedding) -> CoreResult<()> {
sqlx::query!(
r#"
INSERT INTO face_embeddings (id, face_region_id, model_id, embedding)
VALUES ($1, $2, $3, $4)
"#,
embedding.id,
embedding.face_region_id,
embedding.model_id,
embedding.embedding
)
.execute(&self.pool)
.await
.map_err(|e| CoreError::Database(e.to_string()))?;
Ok(())
}
async fn find_by_face_region_id(
&self,
face_region_id: Uuid,
) -> CoreResult<Option<FaceEmbedding>> {
let pg_embedding = sqlx::query_as!(
PostgresFaceEmbedding,
r#"
SELECT id, face_region_id, model_id, embedding
FROM face_embeddings
WHERE face_region_id = $1
"#,
face_region_id
)
.fetch_optional(&self.pool)
.await
.map_err(|e| CoreError::Database(e.to_string()))?;
Ok(pg_embedding.map(FaceEmbedding::from))
}
}