Chewing_Bever
/
rieter
1
0
Fork 0
Code Issues 11 Pull Requests Packages Releases Activity

feat: re-enable most repo functionality

agents
Jef Roosens 2024-06-26 21:10:04 +02:00
parent 9237add869
commit 042f1ecbd3
Signed by: Jef Roosens
GPG Key ID: B75D4F293C7052DB
3 changed files with 107 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/src/repo/actor.rs
View File

@ -27,6 +27,8 @@ use uuid::Uuid;
pub enum RepoCommand { pub enum RepoCommand {
ParsePkg(i32, PathBuf), ParsePkg(i32, PathBuf),
SyncRepo(i32),
Clean,
} }
pub struct RepoSharedState { pub struct RepoSharedState {
@ -98,6 +100,12 @@ impl RepoActor {
let _ = self.sync_repo(repo); let _ = self.sync_repo(repo);
} }
} }
RepoCommand::SyncRepo(repo) => {
let _ = self.sync_repo(repo);
}
RepoCommand::Clean => {
let _ = self.clean();
}
} }
} }
} }
@ -223,4 +231,8 @@ impl RepoActor {
Ok(()) Ok(())
} }
fn clean(&self) -> crate::Result<()> {
todo!()
}
} }

53
server/src/repo/handle.rs
View File

@ -14,6 +14,7 @@ use sea_orm::{
ActiveModelTrait, ColumnTrait, Condition, ConnectionTrait, DbConn, EntityTrait, JoinType, ActiveModelTrait, ColumnTrait, Condition, ConnectionTrait, DbConn, EntityTrait, JoinType,
ModelTrait, NotSet, QueryFilter, QuerySelect, Related, RelationTrait, Set, TransactionTrait, ModelTrait, NotSet, QueryFilter, QuerySelect, Related, RelationTrait, Set, TransactionTrait,
}; };
use sea_query::{Alias, Expr, Query};
use tokio::{ use tokio::{
runtime, runtime,
sync::mpsc::{unbounded_channel, UnboundedSender}, sync::mpsc::{unbounded_channel, UnboundedSender},
@ -117,6 +118,50 @@ impl Handle {
Ok(repo_id) Ok(repo_id)
} }
pub async fn get_repo(&self, distro: &str, repo: &str) -> crate::Result<Option<i32>> {
Ok(db::Repo::find()
.find_also_related(db::Distro)
.filter(
Condition::all()
.add(db::repo::Column::Name.eq(repo))
.add(db::distro::Column::Name.eq(distro)),
)
.one(&self.state.conn)
.await
.map(|res| res.map(|(repo, _)| repo.id))?)
}
pub async fn remove_repo(&self, repo: i32) -> crate::Result<()> {
self.state.repos.write().await.remove(&repo);
db::Repo::delete_by_id(repo).exec(&self.state.conn).await?;
let _ = tokio::fs::remove_dir_all(self.state.repos_dir.join(repo.to_string())).await;
Ok(())
}
/// Remove all packages in the repository that have a given arch. This method marks all
/// packages with the given architecture as "pending deletion", before performing a manual sync
/// & removal of stale packages.
pub async fn remove_repo_arch(&self, repo: i32, arch: &str) -> crate::Result<()> {
db::Package::update_many()
.col_expr(
db::package::Column::State,
Expr::value(db::PackageState::PendingDeletion),
)
.filter(
Condition::all()
.add(db::package::Column::RepoId.eq(repo))
.add(db::package::Column::Arch.eq(arch)),
)
.exec(&self.state.conn)
.await?;
self.queue_sync(repo).await;
self.queue_clean().await;
Ok(())
}
pub async fn queue_pkg(&self, repo: i32, path: PathBuf) { pub async fn queue_pkg(&self, repo: i32, path: PathBuf) {
self.state self.state
.tx .tx
@ -126,4 +171,12 @@ impl Handle {
n.0.fetch_add(1, Ordering::SeqCst); n.0.fetch_add(1, Ordering::SeqCst);
}); });
} }
async fn queue_sync(&self, repo: i32) {
self.state.tx.send(RepoCommand::SyncRepo(repo)).unwrap();
}
async fn queue_clean(&self) {
self.state.tx.send(RepoCommand::Clean).unwrap();
}
} }

100
server/src/repo/mod.rs
View File

@ -51,31 +51,30 @@ async fn get_file(
Path((distro, repo, arch, file_name)): Path<(String, String, String, String)>, Path((distro, repo, arch, file_name)): Path<(String, String, String, String)>,
req: Request<Body>, req: Request<Body>,
) -> crate::Result<impl IntoResponse> { ) -> crate::Result<impl IntoResponse> {
Ok(StatusCode::NOT_FOUND) if let Some(repo_id) = global.repo.get_repo(&distro, &repo).await? {
//if let Some(repo_id) = global.mgr.get_repo(&distro, &repo).await? { match global.config.fs {
// match global.config.fs { FsConfig::Local { data_dir } => {
// FsConfig::Local { data_dir } => { let repo_dir = data_dir.join("repos").join(repo_id.to_string());
// let repo_dir = data_dir.join("repos").join(repo_id.to_string());
// let file_name = if file_name == format!("{}.db", repo)
// let file_name = if file_name == format!("{}.db", repo) || file_name == format!("{}.db.tar.gz", repo)
// || file_name == format!("{}.db.tar.gz", repo) {
// { format!("{}.db.tar.gz", arch)
// format!("{}.db.tar.gz", arch) } else if file_name == format!("{}.files", repo)
// } else if file_name == format!("{}.files", repo) || file_name == format!("{}.files.tar.gz", repo)
// || file_name == format!("{}.files.tar.gz", repo) {
// { format!("{}.files.tar.gz", arch)
// format!("{}.files.tar.gz", arch) } else {
// } else { file_name
// file_name };
// };
// let path = repo_dir.join(file_name);
// let path = repo_dir.join(file_name); Ok(ServeFile::new(path).oneshot(req).await)
// Ok(ServeFile::new(path).oneshot(req).await) }
// } }
// } } else {
//} else { Err(StatusCode::NOT_FOUND.into())
// Err(StatusCode::NOT_FOUND.into()) }
//}
} }
async fn post_package_archive( async fn post_package_archive(
@ -99,45 +98,30 @@ async fn delete_repo(
State(global): State<crate::Global>, State(global): State<crate::Global>,
Path((distro, repo)): Path<(String, String)>, Path((distro, repo)): Path<(String, String)>,
) -> crate::Result<StatusCode> { ) -> crate::Result<StatusCode> {
Ok(StatusCode::NOT_FOUND) if let Some(repo) = global.repo.get_repo(&distro, &repo).await? {
//if let Some(repo) = global.mgr.get_repo(&distro, &repo).await? { global.repo.remove_repo(repo).await?;
// global.mgr.remove_repo(repo).await?;
// tracing::info!("Removed repository {repo}");
// tracing::info!("Removed repository {repo}");
// Ok(StatusCode::OK)
// Ok(StatusCode::OK) } else {
//} else { Ok(StatusCode::NOT_FOUND)
// Ok(StatusCode::NOT_FOUND) }
//}
} }
async fn delete_arch_repo( async fn delete_arch_repo(
State(global): State<crate::Global>, State(global): State<crate::Global>,
Path((distro, repo, arch)): Path<(String, String, String)>, Path((distro, repo, arch)): Path<(String, String, String)>,
) -> crate::Result<StatusCode> { ) -> crate::Result<StatusCode> {
Ok(StatusCode::NOT_FOUND) if let Some(repo) = global.repo.get_repo(&distro, &repo).await? {
//if let Some(repo) = global.mgr.get_repo(&distro, &repo).await? { global.repo.remove_repo_arch(repo, &arch).await?;
// global.mgr.remove_repo_arch(repo, &arch).await?;
// tracing::info!("Removed architecture '{arch}' from repository {repo}");
// tracing::info!("Removed architecture '{arch}' from repository {repo}");
// Ok(StatusCode::OK)
// Ok(StatusCode::OK) } else {
//} else { Ok(StatusCode::NOT_FOUND)
// Ok(StatusCode::NOT_FOUND) }
//}
//if let Some(mgr) = global.mgr.get_mgr(&distro).await {
// let repo_removed = mgr.remove_repo_arch(&repo, &arch).await?;
//
// if repo_removed {
// tracing::info!("Removed arch '{}' from repository '{}'", arch, repo);
//
// Ok(StatusCode::OK)
// } else {
// Ok(StatusCode::NOT_FOUND)
// }
//} else {
// Ok(StatusCode::NOT_FOUND)
//}
} }
async fn delete_package( async fn delete_package(