From 9c9921d9151f813e30aed42cf2b0ae3ae775cc52 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Wed, 17 Mar 2021 23:19:06 +0100
Subject: feat: branch CRUD

---
 src/branch/models.rs | 105 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 src/branch/models.rs

(limited to 'src/branch/models.rs')

diff --git a/src/branch/models.rs b/src/branch/models.rs
new file mode 100644
index 0000000..b631162
--- /dev/null
+++ b/src/branch/models.rs
@@ -0,0 +1,105 @@
+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,
+}
+
+impl Branch {
+    /// Find all branches
+    pub async fn find_all(pool: Pool) -> Result<Vec<Branch>, 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::<Vec<Branch>>();
+
+        Ok(branches)
+    }
+
+    /// Find a branch with an `id` equals to an Uuid element
+    pub async fn find(pool: Pool, id: &Uuid) -> Result<Branch, AppError> {
+        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<Vec<Branch>, 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::<Vec<Branch>>();
+
+        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<Branch, AppError> {
+        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()),
+            }),
+        }
+    }
+}
-- 
cgit v1.2.3-18-g5258


From 3dff10c09ef6db02360a2d2e914e8c2d96ef7a89 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Wed, 17 Mar 2021 23:26:07 +0100
Subject: feat: method for create a new branch

Actually is unusued until #15
---
 src/branch/models.rs | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

(limited to 'src/branch/models.rs')

diff --git a/src/branch/models.rs b/src/branch/models.rs
index b631162..c8a8d65 100644
--- a/src/branch/models.rs
+++ b/src/branch/models.rs
@@ -17,6 +17,13 @@ pub struct Branch {
     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<Vec<Branch>, AppError> {
@@ -102,4 +109,40 @@ impl Branch {
             }),
         }
     }
+
+    /// Create a new branch
+    pub async fn create(
+        pool: Pool,
+        data: &BranchData,
+    ) -> Result<Branch, AppError> {
+        let client = get_client(pool.clone()).await.unwrap();
+
+        let statement = client
+            .prepare(
+                "INSERT INTO repository(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,
+            }),
+        }
+    }
 }
-- 
cgit v1.2.3-18-g5258