Can create couple

geneit_backend/src/services/couples_service.rs

@@ -0,0 +1,124 @@
+use crate::connections::db_connection;
+use crate::models::{Couple, CoupleID, FamilyID, MemberID, NewCouple};
+use crate::schema::couples;
+use crate::services::photos_service;
+use crate::utils::time_utils::time;
+use diesel::prelude::*;
+
+/// Create a new couple
+pub async fn create(family_id: FamilyID) -> anyhow::Result<Couple> {
+    db_connection::execute(|conn| {
+        let res: Couple = diesel::insert_into(couples::table)
+            .values(&NewCouple {
+                family_id: family_id.0,
+                time_create: time() as i64,
+                time_update: time() as i64,
+            })
+            .get_result(conn)?;
+
+        Ok(res)
+    })
+}
+
+/// Get the information of a couple
+pub async fn get_by_id(id: CoupleID) -> anyhow::Result<Couple> {
+    db_connection::execute(|conn| couples::table.filter(couples::dsl::id.eq(id.0)).first(conn))
+}
+
+/// Get all the couples of a family
+pub async fn get_all_of_family(id: FamilyID) -> anyhow::Result<Vec<Couple>> {
+    db_connection::execute(|conn| {
+        couples::table
+            .filter(couples::dsl::family_id.eq(id.0))
+            .get_results(conn)
+    })
+}
+
+/// Get all the couples associated to a member
+pub async fn get_all_of_member(id: MemberID) -> anyhow::Result<Vec<Couple>> {
+    db_connection::execute(|conn| {
+        couples::table
+            .filter(
+                couples::dsl::wife
+                    .eq(id.0)
+                    .or(couples::dsl::husband.eq(id.0)),
+            )
+            .get_results(conn)
+    })
+}
+
+/// Check whether a couple with a given id exists or not
+pub async fn exists(family_id: FamilyID, couple_id: CoupleID) -> anyhow::Result<bool> {
+    db_connection::execute(|conn| {
+        let count: i64 = couples::table
+            .filter(
+                couples::id
+                    .eq(couple_id.0)
+                    .and(couples::family_id.eq(family_id.0)),
+            )
+            .count()
+            .get_result(conn)?;
+
+        Ok(count != 0)
+    })
+}
+
+/// Update the information of a couple
+pub async fn update(couple: &mut Couple) -> anyhow::Result<()> {
+    couple.time_update = time() as i64;
+
+    db_connection::execute(|conn| {
+        diesel::update(couples::dsl::couples.filter(couples::dsl::id.eq(couple.id().0)))
+            .set((
+                couples::dsl::state.eq(couple.state().map(|c| c.as_str().to_string())),
+                couples::dsl::wife.eq(couple.wife().map(|m| m.0)),
+                couples::dsl::husband.eq(couple.husband().map(|m| m.0)),
+                couples::dsl::photo_id.eq(couple.photo_id().map(|p| p.0)),
+                couples::dsl::time_update.eq(couple.time_update),
+                couples::dsl::wedding_year.eq(couple.wedding_year),
+                couples::dsl::wedding_month.eq(couple.wedding_month),
+                couples::dsl::wedding_day.eq(couple.wedding_day),
+                couples::dsl::divorce_year.eq(couple.divorce_year),
+                couples::dsl::divorce_month.eq(couple.divorce_month),
+                couples::dsl::divorce_day.eq(couple.divorce_day),
+            ))
+            .execute(conn)
+    })?;
+
+    Ok(())
+}
+
+/// Delete a couple photo
+pub async fn remove_photo(couple: &mut Couple) -> anyhow::Result<()> {
+    match couple.photo_id() {
+        None => {}
+        Some(photo) => {
+            photos_service::delete(photo).await?;
+            couple.set_photo_id(None);
+            update(couple).await?;
+        }
+    }
+
+    Ok(())
+}
+
+/// Delete a couple
+pub async fn delete(couple: &mut Couple) -> anyhow::Result<()> {
+    remove_photo(couple).await?;
+
+    // Remove the couple
+    db_connection::execute(|conn| {
+        diesel::delete(couples::dsl::couples.filter(couples::dsl::id.eq(couple.id().0)))
+            .execute(conn)
+    })?;
+
+    Ok(())
+}
+
+/// Delete all the couples of a family
+pub async fn delete_all_family(family_id: FamilyID) -> anyhow::Result<()> {
+    for mut m in get_all_of_family(family_id).await? {
+        delete(&mut m).await?;
+    }
+    Ok(())
+}

geneit_backend/src/services/families_service.rs

@@ -5,7 +5,7 @@ use crate::models::{
     Family, FamilyID, FamilyMembership, Membership, NewFamily, NewMembership, UserID,
 };
 use crate::schema::{families, memberships};
-use crate::services::{members_service, users_service};
+use crate::services::{couples_service, members_service, users_service};
 use crate::utils::string_utils::rand_str;
 use crate::utils::time_utils::time;
 use diesel::prelude::*;
@@ -183,7 +183,8 @@ pub async fn update_family(family: &Family) -> anyhow::Result<()> {
 
 /// Delete a family
 pub async fn delete_family(family_id: FamilyID) -> anyhow::Result<()> {
-    // TODO : delete couples
+    // Delete all family couples
+    couples_service::delete_all_family(family_id).await?;
 
     // Remove all family members
     members_service::delete_all_family(family_id).await?;

geneit_backend/src/services/members_service.rs

@@ -1,7 +1,7 @@
 use crate::connections::db_connection;
 use crate::models::{FamilyID, Member, MemberID, NewMember};
 use crate::schema::members;
-use crate::services::photos_service;
+use crate::services::{couples_service, photos_service};
 use crate::utils::time_utils::time;
 use diesel::prelude::*;
 use diesel::RunQueryDsl;
@@ -102,7 +102,16 @@ pub async fn remove_photo(member: &mut Member) -> anyhow::Result<()> {
 
 /// Delete a member
 pub async fn delete(member: &mut Member) -> anyhow::Result<()> {
-    // TODO : remove associated couple
+    // Remove associated couple
+    for mut c in couples_service::get_all_of_member(member.id()).await? {
+        // Check if only one person is attached to the couple
+        // i.e. if the current member is the last person attached to the couple
+        if c.wife().is_some() && c.husband().is_some() {
+            continue;
+        }
+
+        couples_service::delete(&mut c).await?;
+    }
 
     remove_photo(member).await?;
 

geneit_backend/src/services/mod.rs

@@ -1,5 +1,6 @@
 //! # Backend services
 
+pub mod couples_service;
 pub mod families_service;
 pub mod login_token_service;
 pub mod mail_service;