use crate::db::get_client; use crate::errors::{AppError, AppErrorType}; use deadpool_postgres::Pool; use serde::{Deserialize, Serialize}; use tokio_pg_mapper::FromTokioPostgresRow; use tokio_pg_mapper_derive::PostgresMapper; use uuid::Uuid; #[derive(Serialize, Deserialize, PostgresMapper)] #[pg_mapper(table = "branch")] /// Branch model pub struct Branch { pub id: Uuid, pub name: String, pub repository_id: Uuid, pub head: String, } /// Struct used for forms pub struct BranchData { pub name: String, pub repository_id: Uuid, pub head: String, } impl Branch { /// Find all branches pub async fn find_all(pool: Pool) -> Result, AppError> { let client = get_client(pool.clone()).await.unwrap(); let statement = client.prepare("SELECT * FROM branch").await?; let branches = client .query(&statement, &[]) .await? .iter() .map(|row| Branch::from_row_ref(row).unwrap()) .collect::>(); Ok(branches) } /// Find a branch with an `id` equals to an Uuid element pub async fn find(pool: Pool, id: &Uuid) -> Result { let client = get_client(pool.clone()).await.unwrap(); let statement = client.prepare("SELECT * FROM branch WHERE id = $1").await?; let branch = client .query_opt(&statement, &[&id]) .await? .map(|row| Branch::from_row_ref(&row).unwrap()); match branch { Some(branch) => Ok(branch), None => Err(AppError { error_type: AppErrorType::NotFoundError, cause: None, message: Some("Branch not found".to_string()), }), } } /// Find all branches of a repository pub async fn find_by_repo( pool: Pool, repo: &Uuid, ) -> Result, AppError> { let client = get_client(pool.clone()).await.unwrap(); let statement = client .prepare("SELECT * FROM branch WHERE repository_id=$1") .await?; let branches = client .query(&statement, &[&repo]) .await? .iter() .map(|row| Branch::from_row_ref(row).unwrap()) .collect::>(); Ok(branches) } /// Find a branch and delete it, but before check if "Authorization" /// matches with SECRET_KEY pub async fn delete(pool: Pool, id: &Uuid) -> Result { let client = get_client(pool.clone()).await.unwrap(); let statement = client .prepare( " DELETE FROM branch WHERE id=$1 RETURNING * ", ) .await?; let branch = client .query_opt(&statement, &[&id]) .await? .map(|row| Branch::from_row_ref(&row).unwrap()); match branch { Some(branch) => Ok(branch), None => Err(AppError { error_type: AppErrorType::NotFoundError, cause: None, message: Some("Branch not found".to_string()), }), } } /// Create a new branch pub async fn create( pool: Pool, data: &BranchData, ) -> Result { let client = get_client(pool.clone()).await.unwrap(); let statement = client .prepare( "INSERT INTO branch(id, name, repository_id, head) VALUES($1, $2, $3, $4) RETURNING *", ) .await?; // Create a new UUID v4 let uuid = Uuid::new_v4(); let branch = client .query_opt( &statement, &[&uuid, &data.name, &data.repository_id, &data.head], ) .await? .map(|row| Branch::from_row_ref(&row).unwrap()); match branch { Some(branch) => Ok(branch), None => Err(AppError { message: Some("Error creating a new branch".to_string()), cause: Some("Unknown error".to_string()), error_type: AppErrorType::DbError, }), } } }