mirror of
https://github.com/lennart-k/rustical.git
synced 2025-12-14 14:02:29 +00:00
migrate birthday store to sqlite
This commit is contained in:
2
Cargo.lock
generated
2
Cargo.lock
generated
@@ -3227,7 +3227,6 @@ dependencies = [
|
|||||||
"rustical_store_sqlite",
|
"rustical_store_sqlite",
|
||||||
"rustical_xml",
|
"rustical_xml",
|
||||||
"serde",
|
"serde",
|
||||||
"sha2",
|
|
||||||
"thiserror 2.0.17",
|
"thiserror 2.0.17",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tower",
|
"tower",
|
||||||
@@ -3250,6 +3249,7 @@ dependencies = [
|
|||||||
"rustical_ical",
|
"rustical_ical",
|
||||||
"rustical_store",
|
"rustical_store",
|
||||||
"serde",
|
"serde",
|
||||||
|
"sha2",
|
||||||
"sqlx",
|
"sqlx",
|
||||||
"thiserror 2.0.17",
|
"thiserror 2.0.17",
|
||||||
"tokio",
|
"tokio",
|
||||||
|
|||||||
@@ -11,7 +11,6 @@ publish = false
|
|||||||
anyhow = { workspace = true }
|
anyhow = { workspace = true }
|
||||||
async-trait = { workspace = true }
|
async-trait = { workspace = true }
|
||||||
serde = { workspace = true }
|
serde = { workspace = true }
|
||||||
sha2 = { workspace = true }
|
|
||||||
ical = { workspace = true }
|
ical = { workspace = true }
|
||||||
chrono = { workspace = true }
|
chrono = { workspace = true }
|
||||||
regex = { workspace = true }
|
regex = { workspace = true }
|
||||||
|
|||||||
@@ -1,209 +0,0 @@
|
|||||||
use crate::{
|
|
||||||
Addressbook, AddressbookStore, Calendar, CalendarStore, Error, calendar::CalendarMetadata,
|
|
||||||
combined_calendar_store::PrefixedCalendarStore,
|
|
||||||
};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use derive_more::derive::Constructor;
|
|
||||||
use rustical_ical::{AddressObject, CalendarObject, CalendarObjectType};
|
|
||||||
use sha2::{Digest, Sha256};
|
|
||||||
use std::{collections::HashMap, sync::Arc};
|
|
||||||
|
|
||||||
pub const BIRTHDAYS_PREFIX: &str = "_birthdays_";
|
|
||||||
|
|
||||||
#[derive(Constructor, Clone)]
|
|
||||||
pub struct ContactBirthdayStore<AS: AddressbookStore>(Arc<AS>);
|
|
||||||
|
|
||||||
impl<AS: AddressbookStore> PrefixedCalendarStore for ContactBirthdayStore<AS> {
|
|
||||||
const PREFIX: &'static str = BIRTHDAYS_PREFIX;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn birthday_calendar(addressbook: Addressbook) -> Calendar {
|
|
||||||
Calendar {
|
|
||||||
principal: addressbook.principal,
|
|
||||||
id: format!("{}{}", BIRTHDAYS_PREFIX, addressbook.id),
|
|
||||||
meta: CalendarMetadata {
|
|
||||||
displayname: addressbook
|
|
||||||
.displayname
|
|
||||||
.map(|name| format!("{name} birthdays")),
|
|
||||||
order: 0,
|
|
||||||
description: None,
|
|
||||||
color: None,
|
|
||||||
},
|
|
||||||
timezone_id: None,
|
|
||||||
deleted_at: addressbook.deleted_at,
|
|
||||||
synctoken: addressbook.synctoken,
|
|
||||||
subscription_url: None,
|
|
||||||
push_topic: {
|
|
||||||
let mut hasher = Sha256::new();
|
|
||||||
hasher.update("birthdays");
|
|
||||||
hasher.update(addressbook.push_topic);
|
|
||||||
format!("{:x}", hasher.finalize())
|
|
||||||
},
|
|
||||||
components: vec![CalendarObjectType::Event],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Objects are all prefixed with `BIRTHDAYS_PREFIX`
|
|
||||||
#[async_trait]
|
|
||||||
impl<AS: AddressbookStore> CalendarStore for ContactBirthdayStore<AS> {
|
|
||||||
async fn get_calendar(
|
|
||||||
&self,
|
|
||||||
principal: &str,
|
|
||||||
id: &str,
|
|
||||||
show_deleted: bool,
|
|
||||||
) -> Result<Calendar, Error> {
|
|
||||||
let id = id.strip_prefix(BIRTHDAYS_PREFIX).ok_or(Error::NotFound)?;
|
|
||||||
let addressbook = self.0.get_addressbook(principal, id, show_deleted).await?;
|
|
||||||
Ok(birthday_calendar(addressbook))
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_calendars(&self, principal: &str) -> Result<Vec<Calendar>, Error> {
|
|
||||||
let addressbooks = self.0.get_addressbooks(principal).await?;
|
|
||||||
Ok(addressbooks.into_iter().map(birthday_calendar).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_deleted_calendars(&self, principal: &str) -> Result<Vec<Calendar>, Error> {
|
|
||||||
let addressbooks = self.0.get_deleted_addressbooks(principal).await?;
|
|
||||||
Ok(addressbooks.into_iter().map(birthday_calendar).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn update_calendar(
|
|
||||||
&self,
|
|
||||||
_principal: String,
|
|
||||||
_id: String,
|
|
||||||
_calendar: Calendar,
|
|
||||||
) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn insert_calendar(&self, _calendar: Calendar) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
async fn delete_calendar(
|
|
||||||
&self,
|
|
||||||
_principal: &str,
|
|
||||||
_name: &str,
|
|
||||||
_use_trashbin: bool,
|
|
||||||
) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn restore_calendar(&self, _principal: &str, _name: &str) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn import_calendar(
|
|
||||||
&self,
|
|
||||||
_calendar: Calendar,
|
|
||||||
_objects: Vec<CalendarObject>,
|
|
||||||
_merge_existing: bool,
|
|
||||||
) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn sync_changes(
|
|
||||||
&self,
|
|
||||||
principal: &str,
|
|
||||||
cal_id: &str,
|
|
||||||
synctoken: i64,
|
|
||||||
) -> Result<(Vec<CalendarObject>, Vec<String>, i64), Error> {
|
|
||||||
let cal_id = cal_id
|
|
||||||
.strip_prefix(BIRTHDAYS_PREFIX)
|
|
||||||
.ok_or(Error::NotFound)?;
|
|
||||||
let (objects, deleted_objects, new_synctoken) =
|
|
||||||
self.0.sync_changes(principal, cal_id, synctoken).await?;
|
|
||||||
let objects: Result<Vec<Option<CalendarObject>>, rustical_ical::Error> = objects
|
|
||||||
.iter()
|
|
||||||
.map(AddressObject::get_birthday_object)
|
|
||||||
.collect();
|
|
||||||
let objects = objects?.into_iter().flatten().collect();
|
|
||||||
|
|
||||||
Ok((objects, deleted_objects, new_synctoken))
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn calendar_metadata(
|
|
||||||
&self,
|
|
||||||
principal: &str,
|
|
||||||
cal_id: &str,
|
|
||||||
) -> Result<crate::CollectionMetadata, Error> {
|
|
||||||
let cal_id = cal_id
|
|
||||||
.strip_prefix(BIRTHDAYS_PREFIX)
|
|
||||||
.ok_or(Error::NotFound)?;
|
|
||||||
self.0.addressbook_metadata(principal, cal_id).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_objects(
|
|
||||||
&self,
|
|
||||||
principal: &str,
|
|
||||||
cal_id: &str,
|
|
||||||
) -> Result<Vec<CalendarObject>, Error> {
|
|
||||||
let cal_id = cal_id
|
|
||||||
.strip_prefix(BIRTHDAYS_PREFIX)
|
|
||||||
.ok_or(Error::NotFound)?;
|
|
||||||
let objects: Result<Vec<HashMap<&'static str, CalendarObject>>, rustical_ical::Error> =
|
|
||||||
self.0
|
|
||||||
.get_objects(principal, cal_id)
|
|
||||||
.await?
|
|
||||||
.iter()
|
|
||||||
.map(AddressObject::get_significant_dates)
|
|
||||||
.collect();
|
|
||||||
let objects = objects?
|
|
||||||
.into_iter()
|
|
||||||
.flat_map(HashMap::into_values)
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
Ok(objects)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_object(
|
|
||||||
&self,
|
|
||||||
principal: &str,
|
|
||||||
cal_id: &str,
|
|
||||||
object_id: &str,
|
|
||||||
show_deleted: bool,
|
|
||||||
) -> Result<CalendarObject, Error> {
|
|
||||||
let cal_id = cal_id
|
|
||||||
.strip_prefix(BIRTHDAYS_PREFIX)
|
|
||||||
.ok_or(Error::NotFound)?;
|
|
||||||
let (addressobject_id, date_type) = object_id.rsplit_once('-').ok_or(Error::NotFound)?;
|
|
||||||
self.0
|
|
||||||
.get_object(principal, cal_id, addressobject_id, show_deleted)
|
|
||||||
.await?
|
|
||||||
.get_significant_dates()?
|
|
||||||
.remove(date_type)
|
|
||||||
.ok_or(Error::NotFound)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn put_object(
|
|
||||||
&self,
|
|
||||||
_principal: String,
|
|
||||||
_cal_id: String,
|
|
||||||
_object: CalendarObject,
|
|
||||||
_overwrite: bool,
|
|
||||||
) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_object(
|
|
||||||
&self,
|
|
||||||
_principal: &str,
|
|
||||||
_cal_id: &str,
|
|
||||||
_object_id: &str,
|
|
||||||
_use_trashbin: bool,
|
|
||||||
) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn restore_object(
|
|
||||||
&self,
|
|
||||||
_principal: &str,
|
|
||||||
_cal_id: &str,
|
|
||||||
_object_id: &str,
|
|
||||||
) -> Result<(), Error> {
|
|
||||||
Err(Error::ReadOnly)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_read_only(&self, _cal_id: &str) -> bool {
|
|
||||||
true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -7,7 +7,6 @@ pub use error::Error;
|
|||||||
pub mod auth;
|
pub mod auth;
|
||||||
mod calendar;
|
mod calendar;
|
||||||
mod combined_calendar_store;
|
mod combined_calendar_store;
|
||||||
mod contact_birthday_store;
|
|
||||||
mod secret;
|
mod secret;
|
||||||
mod subscription_store;
|
mod subscription_store;
|
||||||
pub mod synctoken;
|
pub mod synctoken;
|
||||||
@@ -17,8 +16,7 @@ pub mod tests;
|
|||||||
|
|
||||||
pub use addressbook_store::AddressbookStore;
|
pub use addressbook_store::AddressbookStore;
|
||||||
pub use calendar_store::CalendarStore;
|
pub use calendar_store::CalendarStore;
|
||||||
pub use combined_calendar_store::CombinedCalendarStore;
|
pub use combined_calendar_store::{CombinedCalendarStore, PrefixedCalendarStore};
|
||||||
pub use contact_birthday_store::ContactBirthdayStore;
|
|
||||||
pub use secret::Secret;
|
pub use secret::Secret;
|
||||||
pub use subscription_store::*;
|
pub use subscription_store::*;
|
||||||
|
|
||||||
|
|||||||
@@ -29,3 +29,4 @@ uuid.workspace = true
|
|||||||
pbkdf2.workspace = true
|
pbkdf2.workspace = true
|
||||||
rustical_ical.workspace = true
|
rustical_ical.workspace = true
|
||||||
rstest = { workspace = true, optional = true }
|
rstest = { workspace = true, optional = true }
|
||||||
|
sha2.workspace = true
|
||||||
|
|||||||
353
crates/store_sqlite/src/addressbook_store/birthday_calendar.rs
Normal file
353
crates/store_sqlite/src/addressbook_store/birthday_calendar.rs
Normal file
@@ -0,0 +1,353 @@
|
|||||||
|
use crate::addressbook_store::SqliteAddressbookStore;
|
||||||
|
use async_trait::async_trait;
|
||||||
|
use chrono::NaiveDateTime;
|
||||||
|
use rustical_ical::{AddressObject, CalendarObject, CalendarObjectType};
|
||||||
|
use rustical_store::{
|
||||||
|
Addressbook, AddressbookStore, Calendar, CalendarMetadata, CalendarStore, CollectionMetadata,
|
||||||
|
Error, PrefixedCalendarStore,
|
||||||
|
};
|
||||||
|
use sha2::{Digest, Sha256};
|
||||||
|
use sqlx::{Executor, Sqlite};
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use tracing::instrument;
|
||||||
|
|
||||||
|
pub const BIRTHDAYS_PREFIX: &str = "_birthdays_";
|
||||||
|
|
||||||
|
struct BirthdayCalendarJoinRow {
|
||||||
|
principal: String,
|
||||||
|
id: String,
|
||||||
|
displayname: Option<String>,
|
||||||
|
description: Option<String>,
|
||||||
|
order: i64,
|
||||||
|
color: Option<String>,
|
||||||
|
timezone_id: Option<String>,
|
||||||
|
deleted_at: Option<NaiveDateTime>,
|
||||||
|
push_topic: String,
|
||||||
|
|
||||||
|
addr_synctoken: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<BirthdayCalendarJoinRow> for Calendar {
|
||||||
|
fn from(value: BirthdayCalendarJoinRow) -> Self {
|
||||||
|
Self {
|
||||||
|
principal: value.principal,
|
||||||
|
id: format!("{}{}", BIRTHDAYS_PREFIX, value.id),
|
||||||
|
meta: CalendarMetadata {
|
||||||
|
displayname: value.displayname,
|
||||||
|
order: value.order,
|
||||||
|
description: value.description,
|
||||||
|
color: value.color,
|
||||||
|
},
|
||||||
|
deleted_at: value.deleted_at,
|
||||||
|
components: vec![CalendarObjectType::Event],
|
||||||
|
timezone_id: value.timezone_id,
|
||||||
|
synctoken: value.addr_synctoken,
|
||||||
|
subscription_url: None,
|
||||||
|
push_topic: value.push_topic,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PrefixedCalendarStore for SqliteAddressbookStore {
|
||||||
|
const PREFIX: &'static str = BIRTHDAYS_PREFIX;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SqliteAddressbookStore {
|
||||||
|
#[instrument]
|
||||||
|
pub async fn _get_birthday_calendar<'e, E: Executor<'e, Database = Sqlite>>(
|
||||||
|
executor: E,
|
||||||
|
principal: &str,
|
||||||
|
id: &str,
|
||||||
|
show_deleted: bool,
|
||||||
|
) -> Result<Calendar, Error> {
|
||||||
|
let cal = sqlx::query_as!(
|
||||||
|
BirthdayCalendarJoinRow,
|
||||||
|
r#"SELECT principal, id, displayname, description, "order", color, timezone_id, deleted_at, addr_synctoken, push_topic
|
||||||
|
FROM birthday_calendars
|
||||||
|
INNER JOIN (
|
||||||
|
SELECT principal AS addr_principal,
|
||||||
|
id AS addr_id,
|
||||||
|
synctoken AS addr_synctoken
|
||||||
|
FROM addressbooks
|
||||||
|
) ON (principal, id) = (addr_principal, addr_id)
|
||||||
|
WHERE (principal, id) = (?, ?)
|
||||||
|
AND ((deleted_at IS NULL) OR ?)
|
||||||
|
"#,
|
||||||
|
principal,
|
||||||
|
id,
|
||||||
|
show_deleted
|
||||||
|
)
|
||||||
|
.fetch_one(executor)
|
||||||
|
.await
|
||||||
|
.map_err(crate::Error::from)?;
|
||||||
|
Ok(cal.into())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
pub async fn _get_birthday_calendars<'e, E: Executor<'e, Database = Sqlite>>(
|
||||||
|
executor: E,
|
||||||
|
principal: &str,
|
||||||
|
deleted: bool,
|
||||||
|
) -> Result<Vec<Calendar>, Error> {
|
||||||
|
Ok(
|
||||||
|
sqlx::query_as!(
|
||||||
|
BirthdayCalendarJoinRow,
|
||||||
|
r#"SELECT principal, id, displayname, description, "order", color, timezone_id, deleted_at, addr_synctoken, push_topic
|
||||||
|
FROM birthday_calendars
|
||||||
|
INNER JOIN (
|
||||||
|
SELECT principal AS addr_principal,
|
||||||
|
id AS addr_id,
|
||||||
|
synctoken AS addr_synctoken
|
||||||
|
FROM addressbooks
|
||||||
|
) ON (principal, id) = (addr_principal, addr_id)
|
||||||
|
WHERE principal = ?
|
||||||
|
AND (
|
||||||
|
(deleted_at IS NULL AND NOT ?) -- not deleted, want not deleted
|
||||||
|
OR (deleted_at IS NOT NULL AND ?) -- deleted, want deleted
|
||||||
|
)
|
||||||
|
"#,
|
||||||
|
principal,
|
||||||
|
deleted,
|
||||||
|
deleted
|
||||||
|
)
|
||||||
|
.fetch_all(executor)
|
||||||
|
.await
|
||||||
|
.map_err(crate::Error::from).map(|cals| cals.into_iter().map(BirthdayCalendarJoinRow::into).collect())?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
pub async fn _insert_birthday_calendar<'e, E: Executor<'e, Database = Sqlite>>(
|
||||||
|
executor: E,
|
||||||
|
addressbook: Addressbook,
|
||||||
|
) -> Result<(), rustical_store::Error> {
|
||||||
|
let birthday_name = addressbook
|
||||||
|
.displayname
|
||||||
|
.map(|name| format!("{name} birthdays"));
|
||||||
|
let birthday_push_topic = {
|
||||||
|
let mut hasher = Sha256::new();
|
||||||
|
hasher.update("birthdays");
|
||||||
|
hasher.update(addressbook.push_topic);
|
||||||
|
format!("{:x}", hasher.finalize())
|
||||||
|
};
|
||||||
|
|
||||||
|
sqlx::query!(
|
||||||
|
r#"INSERT INTO birthday_calendars (principal, id, displayname, push_topic)
|
||||||
|
VALUES (?, ?, ?, ?)"#,
|
||||||
|
addressbook.principal,
|
||||||
|
addressbook.id,
|
||||||
|
birthday_name,
|
||||||
|
birthday_push_topic,
|
||||||
|
)
|
||||||
|
.execute(executor)
|
||||||
|
.await
|
||||||
|
.map_err(crate::Error::from)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn _update_birthday_calendar<'e, E: Executor<'e, Database = Sqlite>>(
|
||||||
|
executor: E,
|
||||||
|
principal: &str,
|
||||||
|
calendar: &Calendar,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
let result = sqlx::query!(
|
||||||
|
r#"UPDATE birthday_calendars SET principal = ?, id = ?, displayname = ?, description = ?, "order" = ?, color = ?, timezone_id = ?, push_topic = ?
|
||||||
|
WHERE (principal, id) = (?, ?)"#,
|
||||||
|
calendar.principal,
|
||||||
|
calendar.id,
|
||||||
|
calendar.meta.displayname,
|
||||||
|
calendar.meta.description,
|
||||||
|
calendar.meta.order,
|
||||||
|
calendar.meta.color,
|
||||||
|
calendar.timezone_id,
|
||||||
|
calendar.push_topic,
|
||||||
|
principal,
|
||||||
|
calendar.id,
|
||||||
|
).execute(executor).await.map_err(crate::Error::from)?;
|
||||||
|
if result.rows_affected() == 0 {
|
||||||
|
return Err(rustical_store::Error::NotFound);
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl CalendarStore for SqliteAddressbookStore {
|
||||||
|
#[instrument]
|
||||||
|
async fn get_calendar(
|
||||||
|
&self,
|
||||||
|
principal: &str,
|
||||||
|
id: &str,
|
||||||
|
show_deleted: bool,
|
||||||
|
) -> Result<Calendar, Error> {
|
||||||
|
let id = id.strip_prefix(BIRTHDAYS_PREFIX).ok_or(Error::NotFound)?;
|
||||||
|
Self::_get_birthday_calendar(&self.db, principal, id, show_deleted).await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn get_calendars(&self, principal: &str) -> Result<Vec<Calendar>, Error> {
|
||||||
|
Self::_get_birthday_calendars(&self.db, principal, false).await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn get_deleted_calendars(&self, principal: &str) -> Result<Vec<Calendar>, Error> {
|
||||||
|
Self::_get_birthday_calendars(&self.db, principal, true).await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn update_calendar(
|
||||||
|
&self,
|
||||||
|
principal: String,
|
||||||
|
id: String,
|
||||||
|
mut calendar: Calendar,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
assert_eq!(id, calendar.id);
|
||||||
|
calendar.id = calendar
|
||||||
|
.id
|
||||||
|
.strip_prefix(BIRTHDAYS_PREFIX)
|
||||||
|
.ok_or(Error::NotFound)?
|
||||||
|
.to_string();
|
||||||
|
Self::_update_birthday_calendar(&self.db, &principal, &calendar).await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn insert_calendar(&self, _calendar: Calendar) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn delete_calendar(
|
||||||
|
&self,
|
||||||
|
_principal: &str,
|
||||||
|
_name: &str,
|
||||||
|
_use_trashbin: bool,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn restore_calendar(&self, _principal: &str, _name: &str) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn import_calendar(
|
||||||
|
&self,
|
||||||
|
_calendar: Calendar,
|
||||||
|
_objects: Vec<CalendarObject>,
|
||||||
|
_merge_existing: bool,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn sync_changes(
|
||||||
|
&self,
|
||||||
|
principal: &str,
|
||||||
|
cal_id: &str,
|
||||||
|
synctoken: i64,
|
||||||
|
) -> Result<(Vec<CalendarObject>, Vec<String>, i64), Error> {
|
||||||
|
let cal_id = cal_id
|
||||||
|
.strip_prefix(BIRTHDAYS_PREFIX)
|
||||||
|
.ok_or(Error::NotFound)?;
|
||||||
|
let (objects, deleted_objects, new_synctoken) =
|
||||||
|
AddressbookStore::sync_changes(self, principal, cal_id, synctoken).await?;
|
||||||
|
let objects: Result<Vec<Option<CalendarObject>>, rustical_ical::Error> = objects
|
||||||
|
.iter()
|
||||||
|
.map(AddressObject::get_birthday_object)
|
||||||
|
.collect();
|
||||||
|
let objects = objects?.into_iter().flatten().collect();
|
||||||
|
|
||||||
|
Ok((objects, deleted_objects, new_synctoken))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn calendar_metadata(
|
||||||
|
&self,
|
||||||
|
principal: &str,
|
||||||
|
cal_id: &str,
|
||||||
|
) -> Result<CollectionMetadata, Error> {
|
||||||
|
let cal_id = cal_id
|
||||||
|
.strip_prefix(BIRTHDAYS_PREFIX)
|
||||||
|
.ok_or(Error::NotFound)?;
|
||||||
|
self.addressbook_metadata(principal, cal_id).await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn get_objects(
|
||||||
|
&self,
|
||||||
|
principal: &str,
|
||||||
|
cal_id: &str,
|
||||||
|
) -> Result<Vec<CalendarObject>, Error> {
|
||||||
|
let cal_id = cal_id
|
||||||
|
.strip_prefix(BIRTHDAYS_PREFIX)
|
||||||
|
.ok_or(Error::NotFound)?;
|
||||||
|
let objects: Result<Vec<HashMap<&'static str, CalendarObject>>, rustical_ical::Error> =
|
||||||
|
AddressbookStore::get_objects(self, principal, cal_id)
|
||||||
|
.await?
|
||||||
|
.iter()
|
||||||
|
.map(AddressObject::get_significant_dates)
|
||||||
|
.collect();
|
||||||
|
let objects = objects?
|
||||||
|
.into_iter()
|
||||||
|
.flat_map(HashMap::into_values)
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
Ok(objects)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn get_object(
|
||||||
|
&self,
|
||||||
|
principal: &str,
|
||||||
|
cal_id: &str,
|
||||||
|
object_id: &str,
|
||||||
|
show_deleted: bool,
|
||||||
|
) -> Result<CalendarObject, Error> {
|
||||||
|
let cal_id = cal_id
|
||||||
|
.strip_prefix(BIRTHDAYS_PREFIX)
|
||||||
|
.ok_or(Error::NotFound)?;
|
||||||
|
let (addressobject_id, date_type) = object_id.rsplit_once('-').ok_or(Error::NotFound)?;
|
||||||
|
AddressbookStore::get_object(self, principal, cal_id, addressobject_id, show_deleted)
|
||||||
|
.await?
|
||||||
|
.get_significant_dates()?
|
||||||
|
.remove(date_type)
|
||||||
|
.ok_or(Error::NotFound)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn put_object(
|
||||||
|
&self,
|
||||||
|
_principal: String,
|
||||||
|
_cal_id: String,
|
||||||
|
_object: CalendarObject,
|
||||||
|
_overwrite: bool,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn delete_object(
|
||||||
|
&self,
|
||||||
|
_principal: &str,
|
||||||
|
_cal_id: &str,
|
||||||
|
_object_id: &str,
|
||||||
|
_use_trashbin: bool,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
async fn restore_object(
|
||||||
|
&self,
|
||||||
|
_principal: &str,
|
||||||
|
_cal_id: &str,
|
||||||
|
_object_id: &str,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
Err(Error::ReadOnly)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn is_read_only(&self, _cal_id: &str) -> bool {
|
||||||
|
true
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -11,6 +11,8 @@ use sqlx::{Acquire, Executor, Sqlite, SqlitePool, Transaction};
|
|||||||
use tokio::sync::mpsc::Sender;
|
use tokio::sync::mpsc::Sender;
|
||||||
use tracing::{error, instrument};
|
use tracing::{error, instrument};
|
||||||
|
|
||||||
|
pub mod birthday_calendar;
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
struct AddressObjectRow {
|
struct AddressObjectRow {
|
||||||
id: String,
|
id: String,
|
||||||
@@ -116,7 +118,7 @@ impl SqliteAddressbookStore {
|
|||||||
|
|
||||||
async fn _insert_addressbook<'e, E: Executor<'e, Database = Sqlite>>(
|
async fn _insert_addressbook<'e, E: Executor<'e, Database = Sqlite>>(
|
||||||
executor: E,
|
executor: E,
|
||||||
addressbook: Addressbook,
|
addressbook: &Addressbook,
|
||||||
) -> Result<(), rustical_store::Error> {
|
) -> Result<(), rustical_store::Error> {
|
||||||
sqlx::query!(
|
sqlx::query!(
|
||||||
r#"INSERT INTO addressbooks (principal, id, displayname, description, push_topic)
|
r#"INSERT INTO addressbooks (principal, id, displayname, description, push_topic)
|
||||||
@@ -283,9 +285,9 @@ impl SqliteAddressbookStore {
|
|||||||
|
|
||||||
async fn _put_object<'e, E: Executor<'e, Database = Sqlite>>(
|
async fn _put_object<'e, E: Executor<'e, Database = Sqlite>>(
|
||||||
executor: E,
|
executor: E,
|
||||||
principal: String,
|
principal: &str,
|
||||||
addressbook_id: String,
|
addressbook_id: &str,
|
||||||
object: AddressObject,
|
object: &AddressObject,
|
||||||
overwrite: bool,
|
overwrite: bool,
|
||||||
) -> Result<(), rustical_store::Error> {
|
) -> Result<(), rustical_store::Error> {
|
||||||
let (object_id, vcf) = (object.get_id(), object.get_vcf());
|
let (object_id, vcf) = (object.get_id(), object.get_vcf());
|
||||||
@@ -405,7 +407,15 @@ impl AddressbookStore for SqliteAddressbookStore {
|
|||||||
&self,
|
&self,
|
||||||
addressbook: Addressbook,
|
addressbook: Addressbook,
|
||||||
) -> Result<(), rustical_store::Error> {
|
) -> Result<(), rustical_store::Error> {
|
||||||
Self::_insert_addressbook(&self.db, addressbook).await
|
let mut tx = self
|
||||||
|
.db
|
||||||
|
.begin_with(BEGIN_IMMEDIATE)
|
||||||
|
.await
|
||||||
|
.map_err(crate::Error::from)?;
|
||||||
|
Self::_insert_addressbook(&mut *tx, &addressbook).await?;
|
||||||
|
Self::_insert_birthday_calendar(&mut *tx, addressbook).await?;
|
||||||
|
tx.commit().await.map_err(crate::Error::from)?;
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument]
|
#[instrument]
|
||||||
@@ -521,14 +531,7 @@ impl AddressbookStore for SqliteAddressbookStore {
|
|||||||
|
|
||||||
let object_id = object.get_id().to_owned();
|
let object_id = object.get_id().to_owned();
|
||||||
|
|
||||||
Self::_put_object(
|
Self::_put_object(&mut *tx, &principal, &addressbook_id, &object, overwrite).await?;
|
||||||
&mut *tx,
|
|
||||||
principal.clone(),
|
|
||||||
addressbook_id.clone(),
|
|
||||||
object,
|
|
||||||
overwrite,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let sync_token = log_object_operation(
|
let sync_token = log_object_operation(
|
||||||
&mut tx,
|
&mut tx,
|
||||||
@@ -659,15 +662,15 @@ impl AddressbookStore for SqliteAddressbookStore {
|
|||||||
return Err(Error::AlreadyExists);
|
return Err(Error::AlreadyExists);
|
||||||
}
|
}
|
||||||
if existing.is_none() {
|
if existing.is_none() {
|
||||||
Self::_insert_addressbook(&mut *tx, addressbook.clone()).await?;
|
Self::_insert_addressbook(&mut *tx, &addressbook).await?;
|
||||||
}
|
}
|
||||||
|
|
||||||
for object in objects {
|
for object in objects {
|
||||||
Self::_put_object(
|
Self::_put_object(
|
||||||
&mut *tx,
|
&mut *tx,
|
||||||
addressbook.principal.clone(),
|
&addressbook.principal,
|
||||||
addressbook.id.clone(),
|
&addressbook.id,
|
||||||
object,
|
&object,
|
||||||
false,
|
false,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
11
src/app.rs
11
src/app.rs
@@ -16,7 +16,8 @@ use rustical_frontend::{FrontendConfig, frontend_router};
|
|||||||
use rustical_oidc::OidcConfig;
|
use rustical_oidc::OidcConfig;
|
||||||
use rustical_store::auth::AuthenticationProvider;
|
use rustical_store::auth::AuthenticationProvider;
|
||||||
use rustical_store::{
|
use rustical_store::{
|
||||||
AddressbookStore, CalendarStore, CombinedCalendarStore, ContactBirthdayStore, SubscriptionStore,
|
AddressbookStore, CalendarStore, CombinedCalendarStore, PrefixedCalendarStore,
|
||||||
|
SubscriptionStore,
|
||||||
};
|
};
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@@ -33,7 +34,11 @@ use tracing::field::display;
|
|||||||
clippy::too_many_lines,
|
clippy::too_many_lines,
|
||||||
clippy::cognitive_complexity
|
clippy::cognitive_complexity
|
||||||
)]
|
)]
|
||||||
pub fn make_app<AS: AddressbookStore, CS: CalendarStore, S: SubscriptionStore>(
|
pub fn make_app<
|
||||||
|
AS: AddressbookStore + PrefixedCalendarStore,
|
||||||
|
CS: CalendarStore,
|
||||||
|
S: SubscriptionStore,
|
||||||
|
>(
|
||||||
addr_store: Arc<AS>,
|
addr_store: Arc<AS>,
|
||||||
cal_store: Arc<CS>,
|
cal_store: Arc<CS>,
|
||||||
subscription_store: Arc<S>,
|
subscription_store: Arc<S>,
|
||||||
@@ -45,7 +50,7 @@ pub fn make_app<AS: AddressbookStore, CS: CalendarStore, S: SubscriptionStore>(
|
|||||||
session_cookie_samesite_strict: bool,
|
session_cookie_samesite_strict: bool,
|
||||||
payload_limit_mb: usize,
|
payload_limit_mb: usize,
|
||||||
) -> Router<()> {
|
) -> Router<()> {
|
||||||
let birthday_store = Arc::new(ContactBirthdayStore::new(addr_store.clone()));
|
let birthday_store = addr_store.clone();
|
||||||
let combined_cal_store =
|
let combined_cal_store =
|
||||||
Arc::new(CombinedCalendarStore::new(cal_store).with_store(birthday_store));
|
Arc::new(CombinedCalendarStore::new(cal_store).with_store(birthday_store));
|
||||||
|
|
||||||
|
|||||||
@@ -13,7 +13,9 @@ use figment::Figment;
|
|||||||
use figment::providers::{Env, Format, Toml};
|
use figment::providers::{Env, Format, Toml};
|
||||||
use rustical_dav_push::DavPushController;
|
use rustical_dav_push::DavPushController;
|
||||||
use rustical_store::auth::AuthenticationProvider;
|
use rustical_store::auth::AuthenticationProvider;
|
||||||
use rustical_store::{AddressbookStore, CalendarStore, CollectionOperation, SubscriptionStore};
|
use rustical_store::{
|
||||||
|
AddressbookStore, CalendarStore, CollectionOperation, PrefixedCalendarStore, SubscriptionStore,
|
||||||
|
};
|
||||||
use rustical_store_sqlite::addressbook_store::SqliteAddressbookStore;
|
use rustical_store_sqlite::addressbook_store::SqliteAddressbookStore;
|
||||||
use rustical_store_sqlite::calendar_store::SqliteCalendarStore;
|
use rustical_store_sqlite::calendar_store::SqliteCalendarStore;
|
||||||
use rustical_store_sqlite::principal_store::SqlitePrincipalStore;
|
use rustical_store_sqlite::principal_store::SqlitePrincipalStore;
|
||||||
@@ -56,7 +58,7 @@ async fn get_data_stores(
|
|||||||
migrate: bool,
|
migrate: bool,
|
||||||
config: &DataStoreConfig,
|
config: &DataStoreConfig,
|
||||||
) -> Result<(
|
) -> Result<(
|
||||||
Arc<impl AddressbookStore>,
|
Arc<impl AddressbookStore + PrefixedCalendarStore>,
|
||||||
Arc<impl CalendarStore>,
|
Arc<impl CalendarStore>,
|
||||||
Arc<impl SubscriptionStore>,
|
Arc<impl SubscriptionStore>,
|
||||||
Arc<impl AuthenticationProvider>,
|
Arc<impl AuthenticationProvider>,
|
||||||
|
|||||||
Reference in New Issue
Block a user