mirror of
https://github.com/lennart-k/rustical.git
synced 2025-12-13 22:52:22 +00:00
91 lines
2.9 KiB
Rust
91 lines
2.9 KiB
Rust
use std::str::FromStr;
|
|
|
|
use super::resource::AddressObjectPathComponents;
|
|
use crate::Error;
|
|
use crate::address_object::resource::AddressObjectResourceService;
|
|
use crate::addressbook::resource::AddressbookResource;
|
|
use axum::body::Body;
|
|
use axum::extract::{Path, State};
|
|
use axum::response::{IntoResponse, Response};
|
|
use axum_extra::TypedHeader;
|
|
use axum_extra::headers::{ContentType, ETag, HeaderMapExt, IfNoneMatch};
|
|
use http::{HeaderMap, StatusCode};
|
|
use rustical_dav::privileges::UserPrivilege;
|
|
use rustical_dav::resource::Resource;
|
|
use rustical_ical::AddressObject;
|
|
use rustical_store::AddressbookStore;
|
|
use rustical_store::auth::User;
|
|
use tracing::instrument;
|
|
|
|
#[instrument(skip(addr_store))]
|
|
pub async fn get_object<AS: AddressbookStore>(
|
|
Path(AddressObjectPathComponents {
|
|
principal,
|
|
addressbook_id,
|
|
object_id,
|
|
}): Path<AddressObjectPathComponents>,
|
|
State(AddressObjectResourceService { addr_store }): State<AddressObjectResourceService<AS>>,
|
|
user: User,
|
|
) -> Result<Response, Error> {
|
|
if !user.is_principal(&principal) {
|
|
return Err(Error::Unauthorized);
|
|
}
|
|
|
|
let addressbook = addr_store
|
|
.get_addressbook(&principal, &addressbook_id, false)
|
|
.await?;
|
|
let addressbook_resource = AddressbookResource(addressbook);
|
|
if !addressbook_resource
|
|
.get_user_privileges(&user)?
|
|
.has(&UserPrivilege::Read)
|
|
{
|
|
return Err(Error::Unauthorized);
|
|
}
|
|
|
|
let object = addr_store
|
|
.get_object(&principal, &addressbook_id, &object_id, false)
|
|
.await?;
|
|
|
|
let mut resp = Response::builder().status(StatusCode::OK);
|
|
let hdrs = resp.headers_mut().unwrap();
|
|
hdrs.typed_insert(ETag::from_str(&object.get_etag()).unwrap());
|
|
hdrs.typed_insert(ContentType::from_str("text/vcard").unwrap());
|
|
Ok(resp.body(Body::new(object.get_vcf().to_owned())).unwrap())
|
|
}
|
|
|
|
#[instrument(skip(addr_store, body))]
|
|
pub async fn put_object<AS: AddressbookStore>(
|
|
Path(AddressObjectPathComponents {
|
|
principal,
|
|
addressbook_id,
|
|
object_id,
|
|
}): Path<AddressObjectPathComponents>,
|
|
State(AddressObjectResourceService { addr_store }): State<AddressObjectResourceService<AS>>,
|
|
user: User,
|
|
mut if_none_match: Option<TypedHeader<IfNoneMatch>>,
|
|
header_map: HeaderMap,
|
|
body: String,
|
|
) -> Result<Response, Error> {
|
|
if !user.is_principal(&principal) {
|
|
return Err(Error::Unauthorized);
|
|
}
|
|
|
|
// https://github.com/hyperium/headers/issues/204
|
|
if !header_map.contains_key("If-None-Match") {
|
|
if_none_match = None;
|
|
}
|
|
|
|
let overwrite = if let Some(TypedHeader(if_none_match)) = if_none_match {
|
|
if_none_match == IfNoneMatch::any()
|
|
} else {
|
|
true
|
|
};
|
|
|
|
let object = AddressObject::from_vcf(object_id, body)?;
|
|
addr_store
|
|
.put_object(principal, addressbook_id, object, overwrite)
|
|
.await?;
|
|
|
|
Ok(StatusCode::CREATED.into_response())
|
|
}
|