mirror of
https://github.com/lennart-k/rustical.git
synced 2025-12-13 14:42:30 +00:00
dav: refactor overwrite header
This commit is contained in:
@@ -22,7 +22,7 @@ pub async fn route_import<C: CalendarStore, S: SubscriptionStore>(
|
||||
Path((principal, cal_id)): Path<(String, String)>,
|
||||
user: Principal,
|
||||
State(resource_service): State<CalendarResourceService<C, S>>,
|
||||
overwrite: Overwrite,
|
||||
Overwrite(overwrite): Overwrite,
|
||||
body: String,
|
||||
) -> Result<Response, Error> {
|
||||
if !user.is_principal(&principal) {
|
||||
@@ -103,7 +103,7 @@ pub async fn route_import<C: CalendarStore, S: SubscriptionStore>(
|
||||
|
||||
let cal_store = resource_service.cal_store;
|
||||
cal_store
|
||||
.import_calendar(new_cal, objects, overwrite.is_true())
|
||||
.import_calendar(new_cal, objects, overwrite)
|
||||
.await?;
|
||||
|
||||
Ok(StatusCode::OK.into_response())
|
||||
|
||||
@@ -14,16 +14,12 @@ impl IntoResponse for InvalidOverwriteHeader {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, PartialEq, Default)]
|
||||
pub enum Overwrite {
|
||||
#[default]
|
||||
T,
|
||||
F,
|
||||
}
|
||||
#[derive(Debug, PartialEq)]
|
||||
pub struct Overwrite(pub bool);
|
||||
|
||||
impl Overwrite {
|
||||
pub fn is_true(&self) -> bool {
|
||||
matches!(self, Self::T)
|
||||
impl Default for Overwrite {
|
||||
fn default() -> Self {
|
||||
Self(true)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -47,9 +43,48 @@ impl TryFrom<&[u8]> for Overwrite {
|
||||
|
||||
fn try_from(value: &[u8]) -> Result<Self, Self::Error> {
|
||||
match value {
|
||||
b"T" => Ok(Overwrite::T),
|
||||
b"F" => Ok(Overwrite::F),
|
||||
b"T" => Ok(Self(true)),
|
||||
b"F" => Ok(Self(false)),
|
||||
_ => Err(InvalidOverwriteHeader),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use axum::{extract::FromRequestParts, response::IntoResponse};
|
||||
use http::Request;
|
||||
|
||||
use crate::header::Overwrite;
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_overwrite_default() {
|
||||
let request = Request::put("asd").body(()).unwrap();
|
||||
let (mut parts, _) = request.into_parts();
|
||||
let overwrite = Overwrite::from_request_parts(&mut parts, &())
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(
|
||||
Overwrite(true),
|
||||
overwrite,
|
||||
"By default we want to overwrite!"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_overwrite() {
|
||||
assert_eq!(
|
||||
Overwrite(true),
|
||||
Overwrite::try_from(b"T".as_slice()).unwrap()
|
||||
);
|
||||
assert_eq!(
|
||||
Overwrite(false),
|
||||
Overwrite::try_from(b"F".as_slice()).unwrap()
|
||||
);
|
||||
if let Err(err) = Overwrite::try_from(b"aslkdjlad".as_slice()) {
|
||||
let _ = err.into_response();
|
||||
} else {
|
||||
unreachable!("should return error")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -17,7 +17,7 @@ pub(crate) async fn axum_route_copy<R: ResourceService>(
|
||||
State(resource_service): State<R>,
|
||||
depth: Option<Depth>,
|
||||
principal: R::Principal,
|
||||
overwrite: Overwrite,
|
||||
Overwrite(overwrite): Overwrite,
|
||||
matched_path: MatchedPath,
|
||||
header_map: HeaderMap,
|
||||
) -> Result<Response, R::Error> {
|
||||
@@ -39,7 +39,7 @@ pub(crate) async fn axum_route_copy<R: ResourceService>(
|
||||
.map_err(|_| crate::Error::Forbidden)?;
|
||||
|
||||
if resource_service
|
||||
.copy_resource(&path, &dest_path, &principal, overwrite.is_true())
|
||||
.copy_resource(&path, &dest_path, &principal, overwrite)
|
||||
.await?
|
||||
{
|
||||
// Overwritten
|
||||
|
||||
@@ -17,7 +17,7 @@ pub(crate) async fn axum_route_move<R: ResourceService>(
|
||||
State(resource_service): State<R>,
|
||||
depth: Option<Depth>,
|
||||
principal: R::Principal,
|
||||
overwrite: Overwrite,
|
||||
Overwrite(overwrite): Overwrite,
|
||||
matched_path: MatchedPath,
|
||||
header_map: HeaderMap,
|
||||
) -> Result<Response, R::Error> {
|
||||
@@ -39,7 +39,7 @@ pub(crate) async fn axum_route_move<R: ResourceService>(
|
||||
.map_err(|_| crate::Error::Forbidden)?;
|
||||
|
||||
if resource_service
|
||||
.copy_resource(&path, &dest_path, &principal, overwrite.is_true())
|
||||
.copy_resource(&path, &dest_path, &principal, overwrite)
|
||||
.await?
|
||||
{
|
||||
// Overwritten
|
||||
|
||||
Reference in New Issue
Block a user