64 lines
1.6 KiB
Rust
64 lines
1.6 KiB
Rust
|
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(())
|
||
|
|
}
|