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

libertas_infra/src/repositories/album_repository.rs

@@ -166,4 +166,21 @@ impl AlbumRepository for PostgresAlbumRepository {
 
         Ok(result.exists)
     }
+
+    async fn set_thumbnail_media_id(&self, album_id: Uuid, media_id: Uuid) -> CoreResult<()> {
+        sqlx::query!(
+            r#"
+            UPDATE albums
+            SET thumbnail_media_id = $1
+            WHERE id = $2
+            "#,
+            media_id,
+            album_id
+        )
+        .execute(&self.pool)
+        .await
+        .map_err(|e| CoreError::Database(e.to_string()))?;
+
+        Ok(())
+    }
 }

libertas_infra/src/repositories/face_embedding_repository.rs

@@ -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))
+    }
+}

libertas_infra/src/repositories/mod.rs

@@ -1,5 +1,6 @@
 pub mod album_repository;
 pub mod album_share_repository;
+pub mod face_embedding_repository;
 pub mod face_region_repository;
 pub mod media_import_repository;
 pub mod media_metadata_repository;

libertas_infra/src/repositories/person_repository.rs

@@ -1,5 +1,9 @@
 use async_trait::async_trait;
-use libertas_core::{error::{CoreError, CoreResult}, models::Person, repositories::PersonRepository};
+use libertas_core::{
+    error::{CoreError, CoreResult},
+    models::Person,
+    repositories::PersonRepository,
+};
 use sqlx::PgPool;
 use uuid::Uuid;
 
@@ -95,4 +99,20 @@ impl PersonRepository for PostgresPersonRepository {
         .map_err(|e| CoreError::Database(e.to_string()))?;
         Ok(())
     }
-}
+
+    async fn set_thumbnail_media_id(&self, person_id: Uuid, media_id: Uuid) -> CoreResult<()> {
+        sqlx::query!(
+            r#"
+            UPDATE people
+            SET thumbnail_media_id = $1
+            WHERE id = $2
+            "#,
+            media_id,
+            person_id
+        )
+        .execute(&self.pool)
+        .await
+        .map_err(|e| CoreError::Database(e.to_string()))?;
+        Ok(())
+    }
+}