refactor(server): separate web logic into separate module
parent
1b80bcd757
commit
80fb6d22f8
|
@ -1636,6 +1636,7 @@ dependencies = [
|
||||||
"chrono",
|
"chrono",
|
||||||
"clap",
|
"clap",
|
||||||
"futures",
|
"futures",
|
||||||
|
"hyper",
|
||||||
"libarchive",
|
"libarchive",
|
||||||
"sea-orm",
|
"sea-orm",
|
||||||
"sea-orm-migration",
|
"sea-orm-migration",
|
||||||
|
|
|
@ -8,6 +8,7 @@ authors = ["Jef Roosens"]
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
axum = { version = "0.6.18", features = ["http2"] }
|
axum = { version = "0.6.18", features = ["http2"] }
|
||||||
|
hyper = "*"
|
||||||
chrono = { version = "0.4.26", features = ["serde"] }
|
chrono = { version = "0.4.26", features = ["serde"] }
|
||||||
clap = { version = "4.3.12", features = ["env", "derive"] }
|
clap = { version = "4.3.12", features = ["env", "derive"] }
|
||||||
futures = "0.3.28"
|
futures = "0.3.28"
|
||||||
|
|
|
@ -2,12 +2,10 @@ use crate::repo::RepoGroupManager;
|
||||||
use crate::{Config, Global};
|
use crate::{Config, Global};
|
||||||
|
|
||||||
use axum::extract::FromRef;
|
use axum::extract::FromRef;
|
||||||
use axum::Router;
|
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use std::io;
|
use std::io;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use std::sync::{Arc, RwLock};
|
use std::sync::{Arc, RwLock};
|
||||||
use tower_http::trace::TraceLayer;
|
|
||||||
use tracing::debug;
|
use tracing::debug;
|
||||||
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
||||||
|
|
||||||
|
@ -95,18 +93,9 @@ impl Cli {
|
||||||
};
|
};
|
||||||
|
|
||||||
// build our application with a single route
|
// build our application with a single route
|
||||||
let app = Router::new()
|
let app = crate::web::app(global, &self.api_key);
|
||||||
.nest("/api", crate::api::router())
|
Ok(crate::web::serve(app, self.port)
|
||||||
.merge(crate::repo::router(&self.api_key))
|
.await
|
||||||
.with_state(global)
|
.map_err(|err| io::Error::new(io::ErrorKind::Other, err))?)
|
||||||
.layer(TraceLayer::new_for_http());
|
|
||||||
|
|
||||||
// run it with hyper on localhost:3000
|
|
||||||
Ok(
|
|
||||||
axum::Server::bind(&format!("0.0.0.0:{}", self.port).parse().unwrap())
|
|
||||||
.serve(app.into_make_service())
|
|
||||||
.await
|
|
||||||
.map_err(|err| io::Error::new(io::ErrorKind::Other, err))?,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,8 +1,8 @@
|
||||||
mod api;
|
|
||||||
mod cli;
|
mod cli;
|
||||||
pub mod db;
|
pub mod db;
|
||||||
mod error;
|
mod error;
|
||||||
mod repo;
|
mod repo;
|
||||||
|
mod web;
|
||||||
|
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
pub use error::{Result, ServerError};
|
pub use error::{Result, ServerError};
|
||||||
|
|
|
@ -1,255 +1,4 @@
|
||||||
mod manager;
|
pub mod manager;
|
||||||
pub mod package;
|
pub mod package;
|
||||||
|
|
||||||
pub use manager::RepoGroupManager;
|
pub use manager::RepoGroupManager;
|
||||||
|
|
||||||
use std::path::PathBuf;
|
|
||||||
|
|
||||||
use axum::body::Body;
|
|
||||||
use axum::extract::{BodyStream, Path, State};
|
|
||||||
use axum::http::Request;
|
|
||||||
use axum::http::StatusCode;
|
|
||||||
use axum::response::IntoResponse;
|
|
||||||
use axum::routing::{delete, post};
|
|
||||||
use axum::Router;
|
|
||||||
use futures::StreamExt;
|
|
||||||
use sea_orm::ModelTrait;
|
|
||||||
use std::sync::Arc;
|
|
||||||
use tokio::{fs, io::AsyncWriteExt};
|
|
||||||
use tower::util::ServiceExt;
|
|
||||||
use tower_http::services::{ServeDir, ServeFile};
|
|
||||||
use tower_http::validate_request::ValidateRequestHeaderLayer;
|
|
||||||
use uuid::Uuid;
|
|
||||||
|
|
||||||
const DB_FILE_EXTS: [&str; 4] = [".db", ".files", ".db.tar.gz", ".files.tar.gz"];
|
|
||||||
|
|
||||||
pub fn router(api_key: &str) -> Router<crate::Global> {
|
|
||||||
Router::new()
|
|
||||||
.route(
|
|
||||||
"/:repo",
|
|
||||||
post(post_package_archive)
|
|
||||||
.delete(delete_repo)
|
|
||||||
.route_layer(ValidateRequestHeaderLayer::bearer(api_key)),
|
|
||||||
)
|
|
||||||
.route(
|
|
||||||
"/:repo/:arch",
|
|
||||||
delete(delete_arch_repo).route_layer(ValidateRequestHeaderLayer::bearer(api_key)),
|
|
||||||
)
|
|
||||||
// Routes added after the layer do not get that layer applied, so the GET requests will not
|
|
||||||
// be authorized
|
|
||||||
.route(
|
|
||||||
"/:repo/:arch/:filename",
|
|
||||||
delete(delete_package)
|
|
||||||
.route_layer(ValidateRequestHeaderLayer::bearer(api_key))
|
|
||||||
.get(get_file),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Serve the package archive files and database archives. If files are requested for an
|
|
||||||
/// architecture that does not have any explicit packages, a repository containing only "any" files
|
|
||||||
/// is returned.
|
|
||||||
async fn get_file(
|
|
||||||
State(global): State<crate::Global>,
|
|
||||||
Path((repo, arch, mut file_name)): Path<(String, String, String)>,
|
|
||||||
req: Request<Body>,
|
|
||||||
) -> crate::Result<impl IntoResponse> {
|
|
||||||
let repo_dir = global.config.repo_dir.join(&repo).join(&arch);
|
|
||||||
let repo_exists = tokio::fs::try_exists(&repo_dir).await?;
|
|
||||||
|
|
||||||
let res = if DB_FILE_EXTS.iter().any(|ext| file_name.ends_with(ext)) {
|
|
||||||
// Append tar extension to ensure we find the file
|
|
||||||
if !file_name.ends_with(".tar.gz") {
|
|
||||||
file_name.push_str(".tar.gz");
|
|
||||||
};
|
|
||||||
|
|
||||||
if repo_exists {
|
|
||||||
ServeFile::new(repo_dir.join(file_name)).oneshot(req).await
|
|
||||||
} else {
|
|
||||||
let path = global
|
|
||||||
.config
|
|
||||||
.repo_dir
|
|
||||||
.join(repo)
|
|
||||||
.join(manager::ANY_ARCH)
|
|
||||||
.join(file_name);
|
|
||||||
|
|
||||||
ServeFile::new(path).oneshot(req).await
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
let any_file = global
|
|
||||||
.config
|
|
||||||
.pkg_dir
|
|
||||||
.join(repo)
|
|
||||||
.join(manager::ANY_ARCH)
|
|
||||||
.join(file_name);
|
|
||||||
|
|
||||||
if repo_exists {
|
|
||||||
ServeDir::new(global.config.pkg_dir)
|
|
||||||
.fallback(ServeFile::new(any_file))
|
|
||||||
.oneshot(req)
|
|
||||||
.await
|
|
||||||
} else {
|
|
||||||
ServeFile::new(any_file).oneshot(req).await
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(res)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn post_package_archive(
|
|
||||||
State(global): State<crate::Global>,
|
|
||||||
Path(repo): Path<String>,
|
|
||||||
mut body: BodyStream,
|
|
||||||
) -> crate::Result<()> {
|
|
||||||
// We first stream the uploaded file to disk
|
|
||||||
let uuid: uuid::fmt::Simple = Uuid::new_v4().into();
|
|
||||||
let path = global.config.pkg_dir.join(uuid.to_string());
|
|
||||||
let mut f = fs::File::create(&path).await?;
|
|
||||||
|
|
||||||
while let Some(chunk) = body.next().await {
|
|
||||||
f.write_all(&chunk?).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
let clone = Arc::clone(&global.repo_manager);
|
|
||||||
let path_clone = path.clone();
|
|
||||||
let repo_clone = repo.clone();
|
|
||||||
let res = tokio::task::spawn_blocking(move || {
|
|
||||||
clone
|
|
||||||
.write()
|
|
||||||
.unwrap()
|
|
||||||
.add_pkg_from_path(&repo_clone, &path_clone)
|
|
||||||
})
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
match res {
|
|
||||||
// Insert the newly added package into the database
|
|
||||||
Ok(pkg) => {
|
|
||||||
tracing::info!("Added '{}' to repository '{}'", pkg.file_name(), repo);
|
|
||||||
|
|
||||||
// Query the repo for its ID, or create it if it does not already exist
|
|
||||||
let res = global.db.repo.by_name(&repo).await?;
|
|
||||||
|
|
||||||
let repo_id = if let Some(repo_entity) = res {
|
|
||||||
repo_entity.id
|
|
||||||
} else {
|
|
||||||
global.db.repo.insert(&repo, None).await?.last_insert_id
|
|
||||||
};
|
|
||||||
|
|
||||||
// If the package already exists in the database, we remove it first
|
|
||||||
let res = global
|
|
||||||
.db
|
|
||||||
.pkg
|
|
||||||
.by_fields(repo_id, &pkg.info.name, None, &pkg.info.arch)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if let Some(entry) = res {
|
|
||||||
entry.delete(&global.db).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
global.db.pkg.insert(repo_id, pkg).await?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
// Remove the uploaded file and return the error
|
|
||||||
Err(err) => {
|
|
||||||
tokio::fs::remove_file(path).await?;
|
|
||||||
|
|
||||||
Err(err.into())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_repo(
|
|
||||||
State(global): State<crate::Global>,
|
|
||||||
Path(repo): Path<String>,
|
|
||||||
) -> crate::Result<StatusCode> {
|
|
||||||
let clone = Arc::clone(&global.repo_manager);
|
|
||||||
|
|
||||||
let repo_clone = repo.clone();
|
|
||||||
let repo_removed =
|
|
||||||
tokio::task::spawn_blocking(move || clone.write().unwrap().remove_repo(&repo_clone))
|
|
||||||
.await??;
|
|
||||||
|
|
||||||
if repo_removed {
|
|
||||||
let res = global.db.repo.by_name(&repo).await?;
|
|
||||||
|
|
||||||
if let Some(repo_entry) = res {
|
|
||||||
repo_entry.delete(&global.db).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
tracing::info!("Removed repository '{}'", repo);
|
|
||||||
|
|
||||||
Ok(StatusCode::OK)
|
|
||||||
} else {
|
|
||||||
Ok(StatusCode::NOT_FOUND)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_arch_repo(
|
|
||||||
State(global): State<crate::Global>,
|
|
||||||
Path((repo, arch)): Path<(String, String)>,
|
|
||||||
) -> crate::Result<StatusCode> {
|
|
||||||
let clone = Arc::clone(&global.repo_manager);
|
|
||||||
|
|
||||||
let arch_clone = arch.clone();
|
|
||||||
let repo_clone = repo.clone();
|
|
||||||
let repo_removed = tokio::task::spawn_blocking(move || {
|
|
||||||
clone
|
|
||||||
.write()
|
|
||||||
.unwrap()
|
|
||||||
.remove_repo_arch(&repo_clone, &arch_clone)
|
|
||||||
})
|
|
||||||
.await??;
|
|
||||||
|
|
||||||
if repo_removed {
|
|
||||||
let res = global.db.repo.by_name(&repo).await?;
|
|
||||||
|
|
||||||
if let Some(repo_entry) = res {
|
|
||||||
global.db.pkg.delete_with_arch(repo_entry.id, &arch).await?;
|
|
||||||
}
|
|
||||||
tracing::info!("Removed architecture '{}' from repository '{}'", arch, repo);
|
|
||||||
|
|
||||||
Ok(StatusCode::OK)
|
|
||||||
} else {
|
|
||||||
Ok(StatusCode::NOT_FOUND)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_package(
|
|
||||||
State(global): State<crate::Global>,
|
|
||||||
Path((repo, arch, file_name)): Path<(String, String, String)>,
|
|
||||||
) -> crate::Result<StatusCode> {
|
|
||||||
let clone = Arc::clone(&global.repo_manager);
|
|
||||||
let path = PathBuf::from(&repo).join(arch).join(&file_name);
|
|
||||||
|
|
||||||
let res = tokio::task::spawn_blocking(move || {
|
|
||||||
clone.write().unwrap().remove_pkg_from_path(path, true)
|
|
||||||
})
|
|
||||||
.await??;
|
|
||||||
|
|
||||||
if let Some((name, version, release, arch)) = res {
|
|
||||||
let res = global.db.repo.by_name(&repo).await?;
|
|
||||||
|
|
||||||
if let Some(repo_entry) = res {
|
|
||||||
let res = global
|
|
||||||
.db
|
|
||||||
.pkg
|
|
||||||
.by_fields(
|
|
||||||
repo_entry.id,
|
|
||||||
&name,
|
|
||||||
Some(&format!("{}-{}", version, release)),
|
|
||||||
&arch,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if let Some(entry) = res {
|
|
||||||
entry.delete(&global.db).await?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tracing::info!("Removed '{}' from repository '{}'", file_name, repo);
|
|
||||||
|
|
||||||
Ok(StatusCode::OK)
|
|
||||||
} else {
|
|
||||||
Ok(StatusCode::NOT_FOUND)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
@ -0,0 +1,19 @@
|
||||||
|
mod api;
|
||||||
|
mod repo;
|
||||||
|
|
||||||
|
use axum::{Router, Server};
|
||||||
|
use tower_http::trace::TraceLayer;
|
||||||
|
|
||||||
|
pub fn app(global: crate::Global, api_key: &str) -> Router {
|
||||||
|
Router::new()
|
||||||
|
.nest("/api", api::router())
|
||||||
|
.merge(repo::router(api_key))
|
||||||
|
.with_state(global)
|
||||||
|
.layer(TraceLayer::new_for_http())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn serve(app: Router, port: u16) -> Result<(), hyper::Error> {
|
||||||
|
Server::bind(&format!("0.0.0.0:{}", port).parse().unwrap())
|
||||||
|
.serve(app.into_make_service())
|
||||||
|
.await
|
||||||
|
}
|
|
@ -0,0 +1,250 @@
|
||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
|
use axum::body::Body;
|
||||||
|
use axum::extract::{BodyStream, Path, State};
|
||||||
|
use axum::http::Request;
|
||||||
|
use axum::http::StatusCode;
|
||||||
|
use axum::response::IntoResponse;
|
||||||
|
use axum::routing::{delete, post};
|
||||||
|
use axum::Router;
|
||||||
|
use futures::StreamExt;
|
||||||
|
use sea_orm::ModelTrait;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::{fs, io::AsyncWriteExt};
|
||||||
|
use tower::util::ServiceExt;
|
||||||
|
use tower_http::services::{ServeDir, ServeFile};
|
||||||
|
use tower_http::validate_request::ValidateRequestHeaderLayer;
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
const DB_FILE_EXTS: [&str; 4] = [".db", ".files", ".db.tar.gz", ".files.tar.gz"];
|
||||||
|
|
||||||
|
pub fn router(api_key: &str) -> Router<crate::Global> {
|
||||||
|
Router::new()
|
||||||
|
.route(
|
||||||
|
"/:repo",
|
||||||
|
post(post_package_archive)
|
||||||
|
.delete(delete_repo)
|
||||||
|
.route_layer(ValidateRequestHeaderLayer::bearer(api_key)),
|
||||||
|
)
|
||||||
|
.route(
|
||||||
|
"/:repo/:arch",
|
||||||
|
delete(delete_arch_repo).route_layer(ValidateRequestHeaderLayer::bearer(api_key)),
|
||||||
|
)
|
||||||
|
// Routes added after the layer do not get that layer applied, so the GET requests will not
|
||||||
|
// be authorized
|
||||||
|
.route(
|
||||||
|
"/:repo/:arch/:filename",
|
||||||
|
delete(delete_package)
|
||||||
|
.route_layer(ValidateRequestHeaderLayer::bearer(api_key))
|
||||||
|
.get(get_file),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Serve the package archive files and database archives. If files are requested for an
|
||||||
|
/// architecture that does not have any explicit packages, a repository containing only "any" files
|
||||||
|
/// is returned.
|
||||||
|
async fn get_file(
|
||||||
|
State(global): State<crate::Global>,
|
||||||
|
Path((repo, arch, mut file_name)): Path<(String, String, String)>,
|
||||||
|
req: Request<Body>,
|
||||||
|
) -> crate::Result<impl IntoResponse> {
|
||||||
|
let repo_dir = global.config.repo_dir.join(&repo).join(&arch);
|
||||||
|
let repo_exists = tokio::fs::try_exists(&repo_dir).await?;
|
||||||
|
|
||||||
|
let res = if DB_FILE_EXTS.iter().any(|ext| file_name.ends_with(ext)) {
|
||||||
|
// Append tar extension to ensure we find the file
|
||||||
|
if !file_name.ends_with(".tar.gz") {
|
||||||
|
file_name.push_str(".tar.gz");
|
||||||
|
};
|
||||||
|
|
||||||
|
if repo_exists {
|
||||||
|
ServeFile::new(repo_dir.join(file_name)).oneshot(req).await
|
||||||
|
} else {
|
||||||
|
let path = global
|
||||||
|
.config
|
||||||
|
.repo_dir
|
||||||
|
.join(repo)
|
||||||
|
.join(crate::repo::manager::ANY_ARCH)
|
||||||
|
.join(file_name);
|
||||||
|
|
||||||
|
ServeFile::new(path).oneshot(req).await
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let any_file = global
|
||||||
|
.config
|
||||||
|
.pkg_dir
|
||||||
|
.join(repo)
|
||||||
|
.join(crate::repo::manager::ANY_ARCH)
|
||||||
|
.join(file_name);
|
||||||
|
|
||||||
|
if repo_exists {
|
||||||
|
ServeDir::new(global.config.pkg_dir)
|
||||||
|
.fallback(ServeFile::new(any_file))
|
||||||
|
.oneshot(req)
|
||||||
|
.await
|
||||||
|
} else {
|
||||||
|
ServeFile::new(any_file).oneshot(req).await
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(res)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn post_package_archive(
|
||||||
|
State(global): State<crate::Global>,
|
||||||
|
Path(repo): Path<String>,
|
||||||
|
mut body: BodyStream,
|
||||||
|
) -> crate::Result<()> {
|
||||||
|
// We first stream the uploaded file to disk
|
||||||
|
let uuid: uuid::fmt::Simple = Uuid::new_v4().into();
|
||||||
|
let path = global.config.pkg_dir.join(uuid.to_string());
|
||||||
|
let mut f = fs::File::create(&path).await?;
|
||||||
|
|
||||||
|
while let Some(chunk) = body.next().await {
|
||||||
|
f.write_all(&chunk?).await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
let clone = Arc::clone(&global.repo_manager);
|
||||||
|
let path_clone = path.clone();
|
||||||
|
let repo_clone = repo.clone();
|
||||||
|
let res = tokio::task::spawn_blocking(move || {
|
||||||
|
clone
|
||||||
|
.write()
|
||||||
|
.unwrap()
|
||||||
|
.add_pkg_from_path(&repo_clone, &path_clone)
|
||||||
|
})
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
match res {
|
||||||
|
// Insert the newly added package into the database
|
||||||
|
Ok(pkg) => {
|
||||||
|
tracing::info!("Added '{}' to repository '{}'", pkg.file_name(), repo);
|
||||||
|
|
||||||
|
// Query the repo for its ID, or create it if it does not already exist
|
||||||
|
let res = global.db.repo.by_name(&repo).await?;
|
||||||
|
|
||||||
|
let repo_id = if let Some(repo_entity) = res {
|
||||||
|
repo_entity.id
|
||||||
|
} else {
|
||||||
|
global.db.repo.insert(&repo, None).await?.last_insert_id
|
||||||
|
};
|
||||||
|
|
||||||
|
// If the package already exists in the database, we remove it first
|
||||||
|
let res = global
|
||||||
|
.db
|
||||||
|
.pkg
|
||||||
|
.by_fields(repo_id, &pkg.info.name, None, &pkg.info.arch)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if let Some(entry) = res {
|
||||||
|
entry.delete(&global.db).await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
global.db.pkg.insert(repo_id, pkg).await?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
// Remove the uploaded file and return the error
|
||||||
|
Err(err) => {
|
||||||
|
tokio::fs::remove_file(path).await?;
|
||||||
|
|
||||||
|
Err(err.into())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn delete_repo(
|
||||||
|
State(global): State<crate::Global>,
|
||||||
|
Path(repo): Path<String>,
|
||||||
|
) -> crate::Result<StatusCode> {
|
||||||
|
let clone = Arc::clone(&global.repo_manager);
|
||||||
|
|
||||||
|
let repo_clone = repo.clone();
|
||||||
|
let repo_removed =
|
||||||
|
tokio::task::spawn_blocking(move || clone.write().unwrap().remove_repo(&repo_clone))
|
||||||
|
.await??;
|
||||||
|
|
||||||
|
if repo_removed {
|
||||||
|
let res = global.db.repo.by_name(&repo).await?;
|
||||||
|
|
||||||
|
if let Some(repo_entry) = res {
|
||||||
|
repo_entry.delete(&global.db).await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
tracing::info!("Removed repository '{}'", repo);
|
||||||
|
|
||||||
|
Ok(StatusCode::OK)
|
||||||
|
} else {
|
||||||
|
Ok(StatusCode::NOT_FOUND)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn delete_arch_repo(
|
||||||
|
State(global): State<crate::Global>,
|
||||||
|
Path((repo, arch)): Path<(String, String)>,
|
||||||
|
) -> crate::Result<StatusCode> {
|
||||||
|
let clone = Arc::clone(&global.repo_manager);
|
||||||
|
|
||||||
|
let arch_clone = arch.clone();
|
||||||
|
let repo_clone = repo.clone();
|
||||||
|
let repo_removed = tokio::task::spawn_blocking(move || {
|
||||||
|
clone
|
||||||
|
.write()
|
||||||
|
.unwrap()
|
||||||
|
.remove_repo_arch(&repo_clone, &arch_clone)
|
||||||
|
})
|
||||||
|
.await??;
|
||||||
|
|
||||||
|
if repo_removed {
|
||||||
|
let res = global.db.repo.by_name(&repo).await?;
|
||||||
|
|
||||||
|
if let Some(repo_entry) = res {
|
||||||
|
global.db.pkg.delete_with_arch(repo_entry.id, &arch).await?;
|
||||||
|
}
|
||||||
|
tracing::info!("Removed architecture '{}' from repository '{}'", arch, repo);
|
||||||
|
|
||||||
|
Ok(StatusCode::OK)
|
||||||
|
} else {
|
||||||
|
Ok(StatusCode::NOT_FOUND)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn delete_package(
|
||||||
|
State(global): State<crate::Global>,
|
||||||
|
Path((repo, arch, file_name)): Path<(String, String, String)>,
|
||||||
|
) -> crate::Result<StatusCode> {
|
||||||
|
let clone = Arc::clone(&global.repo_manager);
|
||||||
|
let path = PathBuf::from(&repo).join(arch).join(&file_name);
|
||||||
|
|
||||||
|
let res = tokio::task::spawn_blocking(move || {
|
||||||
|
clone.write().unwrap().remove_pkg_from_path(path, true)
|
||||||
|
})
|
||||||
|
.await??;
|
||||||
|
|
||||||
|
if let Some((name, version, release, arch)) = res {
|
||||||
|
let res = global.db.repo.by_name(&repo).await?;
|
||||||
|
|
||||||
|
if let Some(repo_entry) = res {
|
||||||
|
let res = global
|
||||||
|
.db
|
||||||
|
.pkg
|
||||||
|
.by_fields(
|
||||||
|
repo_entry.id,
|
||||||
|
&name,
|
||||||
|
Some(&format!("{}-{}", version, release)),
|
||||||
|
&arch,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if let Some(entry) = res {
|
||||||
|
entry.delete(&global.db).await?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tracing::info!("Removed '{}' from repository '{}'", file_name, repo);
|
||||||
|
|
||||||
|
Ok(StatusCode::OK)
|
||||||
|
} else {
|
||||||
|
Ok(StatusCode::NOT_FOUND)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue