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

feat(server): authorized requests

Browse source
This commit is contained in:
Jef Roosens 2023-08-03 09:34:33 +02:00
parent bc19158747
commit a7e0c03b58
Signed by: Jef Roosens
GPG key ID: B75D4F293C7052DB
3 changed files with 29 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

21
server/src/repo/mod.rs
View file

@ -17,15 +17,28 @@ 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;
pub fn router() -> Router<crate::Global> {
Router::new()
.route("/:repo", post(post_package_archive).delete(delete_repo))
.route("/:repo/:arch", delete(delete_arch_repo))
.route(
"/:repo",
post(post_package_archive)
.delete(delete_repo)
.layer(ValidateRequestHeaderLayer::bearer("test")),
)
.route(
"/:repo/:arch",
delete(delete_arch_repo).layer(ValidateRequestHeaderLayer::bearer("test")),
)
// 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).get(get_file),
delete(delete_package)
.layer(ValidateRequestHeaderLayer::bearer("test"))
.get(get_file),
)
}
@ -57,6 +70,8 @@ async fn post_package_archive(
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 repo_entity = db_repo::Entity::find()
.filter(db_repo::Column::Name.eq(&repo))