outsource package logic into package

backend/src/package/add.rs

@@ -0,0 +1,63 @@
+use crate::aur::aur::get_info_by_name;
+use crate::builder::types::Action;
+use crate::db::prelude::Packages;
+use crate::db::{packages, versions};
+use anyhow::anyhow;
+use rocket::State;
+use sea_orm::ColumnTrait;
+use sea_orm::QueryFilter;
+use sea_orm::{ActiveModelTrait, DatabaseConnection, EntityTrait, Set, TransactionTrait};
+use tokio::sync::broadcast::Sender;
+
+pub async fn package_add(
+    db: &DatabaseConnection,
+    pkg_name: String,
+    tx: &State<Sender<Action>>,
+) -> anyhow::Result<()> {
+    let txn = db.begin().await?;
+
+    // remove leading and trailing whitespaces
+    let pkg_name = pkg_name.trim();
+
+    if let Some(..) = Packages::find()
+        .filter(packages::Column::Name.eq(pkg_name))
+        .one(&txn)
+        .await?
+    {
+        return Err(anyhow!("Package already exists"));
+    }
+
+    let pkg = get_info_by_name(pkg_name).await?;
+
+    let new_package = packages::ActiveModel {
+        name: Set(pkg_name.to_string()),
+        status: Set(3),
+        latest_aur_version: Set(pkg.version.clone()),
+        ..Default::default()
+    };
+
+    let mut new_package = new_package.clone().save(&txn).await?;
+
+    let new_version = versions::ActiveModel {
+        version: Set(pkg.version.clone()),
+        package_id: new_package.id.clone(),
+        ..Default::default()
+    };
+
+    let new_version = new_version.clone().save(&txn).await?;
+
+    new_package.status = Set(3);
+    new_package.latest_version_id = Set(Some(new_version.id.clone().unwrap()));
+    new_package.save(&txn).await?;
+
+    let _ = tx.send(Action::Build(
+        pkg.name,
+        pkg.version,
+        pkg.url_path.unwrap(),
+        new_version,
+    ));
+
+    txn.commit().await?;
+
+    Ok(())
+}