diff --git a/.forgejo/workflows/test.yaml b/.forgejo/workflows/test.yaml index 29a3846..1fca279 100644 --- a/.forgejo/workflows/test.yaml +++ b/.forgejo/workflows/test.yaml @@ -1,4 +1,4 @@ -on: [push, workflow_dispatch] +on: [workflow_dispatch] jobs: integration-test--firefox: runs-on: playwright-latest diff --git a/Cargo.toml b/Cargo.toml index 9f7d62b..d246956 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -3,6 +3,10 @@ name = "mascarpone" version = "0.1.0" edition = "2024" +[profile.release] +strip = true +lto = true + [dependencies] anyhow = "1.0.100" axum = { version = "0.8.6", features = ["macros", "form"] } @@ -30,10 +34,10 @@ sqlx = { version = "0.8", features = ["macros", "runtim thiserror = "2.0.17" time = "0.3.44" tokio = { version = "1.47.1", features = ["macros", "rt-multi-thread", "signal"] } -tower-http = { version = "0.6.6", features = ["fs"] } +tower-http = { version = "0.6.6", features = ["fs", "trace"] } tower-sessions = { version = "0.14.0", features = ["signed"] } tower-sessions-sqlx-store = { version = "0.15.0", features = ["sqlite"] } -tracing = "0.1.41" +tracing = { version = "0.1.41", features = ["attributes"] } tracing-subscriber = { version = "0.3.20", features = ["env-filter"] } vcard = "0.4.13" diff --git a/Taskfile b/Taskfile index 5ddbe28..211d1e8 100755 --- a/Taskfile +++ b/Taskfile @@ -40,7 +40,7 @@ deploy_to_server() { } dev() { - _cargo run -- serve + find src static migrations | entr -ccdr ./Taskfile _cargo run -- serve } "$@" diff --git a/e2e/Taskfile b/e2e/Taskfile index 3d0be95..8cb16ea 100755 --- a/e2e/Taskfile +++ b/e2e/Taskfile @@ -22,7 +22,6 @@ playwright:ui() { --env DISPLAY="$DISPLAY" \ --volume /tmp/.X11-unix:/tmp/.X11-unix \ --volume "$SCRIPT_DIR":/e2e:rw --env BASE_URL="$BASE_URL" \ - --env ASTRO_TELEMETRY_DISABLED=1 \ "mcr.microsoft.com/playwright:$(_playwright_version)" \ /bin/bash -c "cd /e2e && ./Taskfile _test --ui $*" } @@ -31,7 +30,6 @@ playwright:ci() { exec docker run \ --interactive --tty --rm --ipc=host --net=host \ --volume "$SCRIPT_DIR":/e2e:rw --env BASE_URL="$BASE_URL" \ - --env ASTRO_TELEMETRY_DISABLED=1 \ "mcr.microsoft.com/playwright:$(_playwright_version)" \ bash -c "cd /e2e && ./Taskfile _test $*" } diff --git a/e2e/package.json b/e2e/package.json index 7baf433..c5c74c7 100644 --- a/e2e/package.json +++ b/e2e/package.json @@ -4,7 +4,7 @@ "description": "", "main": "index.js", "scripts": { - "test": "playwright test --project=firefox && playwright test" + "test": "echo use Taskfile instead" }, "keywords": [], "author": "", diff --git a/e2e/pages/util.ts b/e2e/pages/util.ts index eeb5fca..f73c74a 100644 --- a/e2e/pages/util.ts +++ b/e2e/pages/util.ts @@ -16,6 +16,9 @@ type UserFields = { export const verifyCreateUser = async (page: Page, fields: UserFields) => { await page.getByRole('button', { name: /add contact/i }).click(); + // TODO this is stupid but playwright kept filling while alpine was initializing + await page.waitForTimeout(200); + const { names, ...simple } = fields; for (const name of (names ?? [])) { await page.getByRole('textbox', { name: 'New name' }).fill(name); diff --git a/migrations/demo.sql b/migrations/demo.sql index a81f905..17bda4c 100644 --- a/migrations/demo.sql +++ b/migrations/demo.sql @@ -30,6 +30,21 @@ insert into names(contact_id, sort, name) values (3, 0, 'Eleanor Edgeworth'), (3, 1, 'Eleanor'); +insert into contacts(id, lives_with) values (4, '[[Henrietta]]'); +insert into names(contact_id, sort, name) values + (4, 0, 'Felicia Homeowner'); + +insert into contacts(id, lives_with) values (5, '[[Henrietta]]'); +insert into names(contact_id, sort, name) values + (5, 0, 'Gregory Homeowner'); + +insert into contacts(id) values (6); +insert into names(contact_id, sort, name) values + (6, 0, 'Henrietta Homeowner'), + (6, 1, 'Henrietta'); + +insert into addresses(contact_id, label, value) values + (6, null, '123 Main St., Realville, WI 99999'); insert into journal_entries(id, date, value) values (0, '2020-02-27', 'Lunch with [[Bazel Bagend]] and his wife'), @@ -39,13 +54,13 @@ insert into journal_entries(id, date, value) values (4, '2024-02-17', 'Friendship ended with [[Bazel]]. Now [[Eleanor Edgeworth]] is my best friend.'), (5, '2024-02-18', 'With [[Bazel]] gone, dissolved [[ABC]] Corp. Start discussions for a potential ACE, Inc. with [[Alexi]] and [[Eleanor]].'); -insert into journal_mentions values - (0, 'Bazel Bagend', 11, 27, '/contact/1'), - (1, 'Alexi', 12, 21, '/contact/0'), - (3, 'ABC', 24, 31, '/group/ABC'), - (4, 'Bazel', 22, 31, '/contact/1'), - (4, 'Eleanor Edgeworth', 37, 58, '/contact/3'), - (5, 'Eleanor', 108, 119, '/contact/3'), - (5, 'Alexi', 94, 103, '/contact/0'), - (5, 'Bazel', 5, 14, '/contact/1'), - (5, 'ABC', 31, 38, '/group/ABC'); +insert into mentions (entity_id, entity_type, input_text, byte_range_start, byte_range_end, url) values + (0, 0, 'Bazel Bagend', 11, 27, '/contact/1'), + (1, 0, 'Alexi', 12, 21, '/contact/0'), + (3, 0, 'ABC', 24, 31, '/group/ABC'), + (4, 0, 'Bazel', 22, 31, '/contact/1'), + (4, 0, 'Eleanor Edgeworth', 37, 58, '/contact/3'), + (5, 0, 'Eleanor', 108, 119, '/contact/3'), + (5, 0, 'Alexi', 94, 103, '/contact/0'), + (5, 0, 'Bazel', 5, 14, '/contact/1'), + (5, 0, 'ABC', 31, 38, '/group/ABC'); diff --git a/migrations/each_user/0009_contact-lives-with.sql b/migrations/each_user/0009_contact-lives-with.sql new file mode 100644 index 0000000..20f1edc --- /dev/null +++ b/migrations/each_user/0009_contact-lives-with.sql @@ -0,0 +1 @@ +alter table contacts add column lives_with text not null default ''; diff --git a/migrations/each_user/0010_more-mentions.sql b/migrations/each_user/0010_more-mentions.sql new file mode 100644 index 0000000..cf2d413 --- /dev/null +++ b/migrations/each_user/0010_more-mentions.sql @@ -0,0 +1,30 @@ +create table if not exists mentions ( + entity_id integer not null, + entity_type integer not null, + url text not null, + input_text text not null, + byte_range_start integer not null, + byte_range_end integer not null +); + +insert into mentions ( + entity_id, url, input_text, byte_range_start, byte_range_end, entity_type) + select entry_id, url, input_text, byte_range_start, byte_range_end, 'journal_entry' + from journal_mentions; + +drop table journal_mentions; + +-- entity types: +-- 0: journal_entry +-- 1: contact.text_body +-- 2: contact.lives_with +create trigger if not exists cascade_delete_journal_mentions + after delete on journal_entries for each row begin + delete from mentions where entity_type = 0 and entity_id = OLD.id; + end; + +create trigger if not exists cascade_delete_contact_text_body_mentions + after delete on contacts for each row begin + delete from mentions where entity_type = 1 and entity_id = OLD.id; + delete from mentions where entity_type = 2 and entity_id = OLD.id; + end; diff --git a/migrations/each_user/0011_phone_numbers.sql b/migrations/each_user/0011_phone_numbers.sql new file mode 100644 index 0000000..da8e95c --- /dev/null +++ b/migrations/each_user/0011_phone_numbers.sql @@ -0,0 +1,5 @@ +create table if not exists phone_numbers ( + contact_id integer not null references contacts(id) on delete cascade, + label text, + phone_number text not null +); diff --git a/migrations/each_user/0012_contact_fresh_type.sql b/migrations/each_user/0012_contact_fresh_type.sql new file mode 100644 index 0000000..a9685ef --- /dev/null +++ b/migrations/each_user/0012_contact_fresh_type.sql @@ -0,0 +1,19 @@ + +PRAGMA foreign_keys=OFF; +create table if not exists new_contacts ( + id integer primary key autoincrement, + birthday text, + manually_freshened_at text, + text_body text, + lives_with text not null default '' +); + +insert into new_contacts ( + id, birthday, manually_freshened_at, text_body, lives_with) + select id, birthday, manually_freshened_at, text_body, lives_with + from contacts; + +drop table contacts; +alter table new_contacts rename to contacts; +PRAGMA foreign_key_check; +PRAGMA foreign_keys=ON; diff --git a/src/db.rs b/src/db.rs index 359a3a3..4dd205e 100644 --- a/src/db.rs +++ b/src/db.rs @@ -23,12 +23,14 @@ impl Database { let pool = SqlitePoolOptions::new().connect_with(db_options).await?; + tracing::debug!("migrating..."); sqlx::migrate!("./migrations/each_user/").run(&pool).await?; if user.username == "demo" { sqlx::query_file!("./migrations/demo.sql") .execute(&pool) .await?; }; + tracing::debug!("...done."); Ok(Self { pool }) } diff --git a/src/main.rs b/src/main.rs index 39b8cd7..bafba17 100644 --- a/src/main.rs +++ b/src/main.rs @@ -17,7 +17,6 @@ use tower_sessions_sqlx_store::SqliteStore; use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt}; mod models; -use models::contact::MentionTrie; use models::user::{Backend, User}; mod db; @@ -26,10 +25,13 @@ use db::{Database, DbId}; mod web; use web::{auth, contact, group, home, ics, journal, settings}; +mod switchboard; +use switchboard::Switchboard; + #[derive(Clone)] struct AppStateEntry { database: Arc, - contact_search: Arc>, + switchboard: Arc>, } #[derive(Clone)] @@ -37,10 +39,6 @@ struct AppState { map: Arc>>, } -struct NameReference { - name: String, - contact_id: DbId, -} impl AppState { pub fn new() -> Self { AppState { @@ -49,39 +47,14 @@ impl AppState { } pub async fn init(&mut self, user: &User) -> Result, AppError> { let database = Database::for_user(&user).await?; - let mut trie = radix_trie::Trie::new(); - let mentionable_names = sqlx::query_as!( - NameReference, - "select name, contact_id from ( - select contact_id, name, count(name) as ct from names group by name - ) where ct = 1;", - ) - .fetch_all(&database.pool) - .await?; - - for row in mentionable_names { - trie.insert( - row.name, - format!("/contact/{}", DbId::try_from(row.contact_id)?), - ); - } - - let groups: Vec<(String, String)> = - sqlx::query_as("select distinct name, slug from groups") - .fetch_all(&database.pool) - .await?; - - for (group, slug) in groups { - // TODO urlencode - trie.insert(group, format!("/group/{}", slug)); - } + let switchboard = Switchboard::new(&database.pool).await?; let mut map = self.map.write().expect("rwlock poisoned"); Ok(map.insert( user.id(), crate::AppStateEntry { database: Arc::new(database), - contact_search: Arc::new(RwLock::new(trie)), + switchboard: Arc::new(RwLock::new(switchboard)), }, )) } @@ -93,9 +66,9 @@ impl AppState { let map = self.map.read().expect("rwlock poisoned"); map.get(&user.id()).unwrap().database.clone() } - pub fn contact_search(&self, user: &impl AuthUser) -> Arc> { + pub fn switchboard(&self, user: &impl AuthUser) -> Arc> { let map = self.map.read().expect("rwlock poisoned"); - map.get(&user.id()).unwrap().contact_search.clone() + map.get(&user.id()).unwrap().switchboard.clone() } } @@ -177,7 +150,7 @@ async fn serve(port: &u32) -> Result<(), anyhow::Error> { .with( tracing_subscriber::EnvFilter::try_from_default_env().unwrap_or_else(|_| { format!( - "{}=debug,tower_http=debug,axum=trace,sqlx=debug", + "{}=debug,tower_http=debug,axum=trace", env!("CARGO_CRATE_NAME") ) .into() @@ -197,6 +170,7 @@ async fn serve(port: &u32) -> Result<(), anyhow::Error> { .merge(ics::router()) .nest_service("/static", ServeDir::new("./hashed_static")) .layer(auth_layer) + .layer(tower_http::trace::TraceLayer::new_for_http()) .with_state(state); let listener = TcpListener::bind(format!("0.0.0.0:{}", port)).await?; diff --git a/src/models/contact.rs b/src/models/contact.rs index ca2bf98..bad0ff4 100644 --- a/src/models/contact.rs +++ b/src/models/contact.rs @@ -1,16 +1,50 @@ use chrono::{DateTime, NaiveDate, Utc}; -use sqlx::sqlite::SqliteRow; -use sqlx::{FromRow, Row}; +use sqlx::sqlite::SqlitePool; use std::str::FromStr; use super::Birthday; +use crate::AppError; use crate::db::DbId; +use crate::switchboard::MentionHostType; + +struct RawContact { + id: DbId, + birthday: Option, + manually_freshened_at: Option, + lives_with: String, +} #[derive(Clone, Debug)] pub struct Contact { pub id: DbId, pub birthday: Option, pub manually_freshened_at: Option>, + pub lives_with: String, +} + +impl Into for RawContact { + fn into(self) -> Contact { + Contact { + id: self.id, + birthday: self + .birthday + .and_then(|s| Birthday::from_str(s.as_ref()).ok()), + manually_freshened_at: self + .manually_freshened_at + .and_then(|str| DateTime::parse_from_str(str.as_ref(), "%+").ok()) + .map(|d| d.to_utc()), + lives_with: self.lives_with, + } + } +} + +struct RawHydratedContact { + id: DbId, + birthday: Option, + manually_freshened_at: Option, + lives_with: String, + last_mention_date: Option, + names: Option, } #[derive(Clone, Debug)] @@ -20,6 +54,28 @@ pub struct HydratedContact { pub names: Vec, } +impl Into for RawHydratedContact { + fn into(self) -> HydratedContact { + HydratedContact { + contact: Into::::into(RawContact { + id: self.id, + birthday: self.birthday, + manually_freshened_at: self.manually_freshened_at, + lives_with: self.lives_with, + }), + names: self + .names + .unwrap_or(String::new()) + .split('\x1c') + .map(|s| s.to_string()) + .collect::>(), + last_mention_date: self + .last_mention_date + .and_then(|str| NaiveDate::from_str(str.as_ref()).ok()), + } + } +} + impl std::ops::Deref for HydratedContact { type Target = Contact; fn deref(&self) -> &Self::Target { @@ -35,54 +91,60 @@ impl HydratedContact { "(unnamed)".to_string() } } -} -/* name/group, url */ -pub type MentionTrie = radix_trie::Trie; + pub async fn load(id: DbId, pool: &SqlitePool) -> Result { + // copy-paste the query from 'all', then add "where c.id = $2" to the last line + let raw = sqlx::query_as!( + RawHydratedContact, + r#"select id, birthday, lives_with, manually_freshened_at as "manually_freshened_at: String", ( + select string_agg(name,x'1c' order by sort) + from names where contact_id = c.id + ) as names, ( + select jes.date from journal_entries jes + join mentions ms on ms.entity_id = jes.id + where ms.entity_type = $1 + and ms.url = '/contact/'||c.id + or ms.url in ( + select '/group/'||slug from groups where + contact_id = c.id + ) + order by jes.date desc limit 1 + ) as last_mention_date from contacts c + where c.id = $2"#, + MentionHostType::JournalEntry as DbId, + id + ) + .fetch_one(pool) + .await?; -impl FromRow<'_, SqliteRow> for Contact { - fn from_row(row: &SqliteRow) -> sqlx::Result { - let id: DbId = row.try_get("id")?; + Ok(Into::::into(raw)) + } - let birthday = Birthday::from_row(row).ok(); + pub async fn all(pool: &SqlitePool) -> Result, AppError> { + let contacts = sqlx::query_as!( + RawHydratedContact, + r#"select id, birthday, lives_with, manually_freshened_at as "manually_freshened_at: String", ( + select string_agg(name,x'1c' order by sort) + from names where contact_id = c.id + ) as names, ( + select jes.date from journal_entries jes + join mentions ms on ms.entity_id = jes.id + where ms.entity_type = $1 + and ms.url = '/contact/'||c.id + or ms.url in ( + select '/group/'||slug from groups where + contact_id = c.id + ) + order by jes.date desc limit 1 + ) as last_mention_date from contacts c"#, + MentionHostType::JournalEntry as DbId + ) + .fetch_all(pool) + .await?; - let manually_freshened_at = row - .try_get::("manually_freshened_at") - .ok() - .and_then(|str| { - DateTime::parse_from_str(&str, "%+") - .ok() - .map(|d| d.to_utc()) - }); - - Ok(Self { - id, - birthday, - manually_freshened_at, - }) - } -} - -impl FromRow<'_, SqliteRow> for HydratedContact { - fn from_row(row: &SqliteRow) -> sqlx::Result { - let contact = Contact::from_row(row)?; - - let names_str: String = row.try_get("names").unwrap_or("".to_string()); - let names = if names_str.is_empty() { - vec![] - } else { - names_str.split('\x1c').map(|s| s.to_string()).collect() - }; - - let last_mention_date = row - .try_get::("last_mention_date") - .ok() - .and_then(|str| NaiveDate::from_str(&str).ok()); - - Ok(Self { - contact, - names, - last_mention_date, - }) + Ok(contacts + .into_iter() + .map(|raw| Into::::into(raw)) + .collect()) } } diff --git a/src/models/journal.rs b/src/models/journal.rs index 60e35d7..4a90d86 100644 --- a/src/models/journal.rs +++ b/src/models/journal.rs @@ -1,15 +1,12 @@ use chrono::NaiveDate; -use maud::{Markup, PreEscaped, html}; -use regex::Regex; +use maud::{Markup, html}; use serde_json::json; use sqlx::sqlite::{SqlitePool, SqliteRow}; use sqlx::{FromRow, Row}; -use std::collections::HashSet; -use std::sync::{Arc, RwLock}; -use super::contact::MentionTrie; use crate::AppError; use crate::db::DbId; +use crate::switchboard::{MentionHost, MentionHostType}; #[derive(Debug)] pub struct JournalEntry { @@ -18,92 +15,27 @@ pub struct JournalEntry { pub date: NaiveDate, } -#[derive(Debug, PartialEq, Eq, Hash, FromRow)] -pub struct Mention { - pub entry_id: DbId, - pub url: String, - pub input_text: String, - pub byte_range_start: u32, - pub byte_range_end: u32, +impl<'a> Into> for &'a JournalEntry { + fn into(self) -> MentionHost<'a> { + MentionHost { + entity_id: self.id, + entity_type: MentionHostType::JournalEntry as DbId, + input: &self.value, + } + } } impl JournalEntry { - pub fn extract_mentions(&self, trie: &MentionTrie) -> HashSet { - let name_re = Regex::new(r"\[\[(.+?)\]\]").unwrap(); - name_re - .captures_iter(&self.value) - .map(|caps| { - let range = caps.get_match().range(); - trie.get(&caps[1]).map(|url| Mention { - entry_id: self.id, - url: url.to_string(), - input_text: caps[1].to_string(), - byte_range_start: u32::try_from(range.start).unwrap(), - byte_range_end: u32::try_from(range.end).unwrap(), - }) - }) - .filter(|o| o.is_some()) - .map(|o| o.unwrap()) - .collect() - } - - pub async fn insert_mentions( - &self, - trie: Arc>, - pool: &SqlitePool, - ) -> Result, AppError> { - let mentions = { - let trie = trie.read().unwrap(); - self.extract_mentions(&trie) - }; - - for mention in &mentions { - sqlx::query!( - "insert into journal_mentions( - entry_id, url, input_text, - byte_range_start, byte_range_end - ) values ($1, $2, $3, $4, $5)", - mention.entry_id, - mention.url, - mention.input_text, - mention.byte_range_start, - mention.byte_range_end - ) - .execute(pool) - .await?; - } - - Ok(mentions) - } - pub async fn to_html(&self, pool: &SqlitePool) -> Result { - // important to sort desc so that changing contents early in the string - // doesn't break inserting mentions at byte offsets further in - let mentions: Vec = sqlx::query_as( - "select * from journal_mentions - where entry_id = $1 order by byte_range_start desc", - ) - .bind(self.id) - .fetch_all(pool) - .await?; - - let mut value = self.value.clone(); - for mention in mentions { - tracing::debug!("url ({})", mention.url); - value.replace_range( - (mention.byte_range_start as usize)..(mention.byte_range_end as usize), - &format!("[{}]({})", mention.input_text, mention.url), - ); - } - + let rendered = Into::::into(self).format_pool(pool).await?; let entry_url = format!("/journal_entry/{}", self.id); let date = self.date.to_string(); Ok(html! { - .entry { + .entry hx-target="this" { .view ":class"="{ hide: edit }" { .date { (date) } - .content { (PreEscaped(markdown::to_html(&value))) } + .content { (rendered) } } form .edit ":class"="{ hide: !edit }" x-data=(json!({ "date": date, "initial_date": date, "value": self.value, "initial_value": self.value })) { input name="date" x-model="date"; @@ -111,7 +43,6 @@ impl JournalEntry { textarea name="value" x-model="value" {} button title="Delete" hx-delete=(entry_url) - hx-target="closest .entry" hx-swap="delete" { svg .icon xmlns="http://www.w3.org/2000/svg" viewBox="0 0 640 640" { path d="M232.7 69.9C237.1 56.8 249.3 48 263.1 48L377 48C390.8 48 403 56.8 407.4 69.9L416 96L512 96C529.7 96 544 110.3 544 128C544 145.7 529.7 160 512 160L128 160C110.3 160 96 145.7 96 128C96 110.3 110.3 96 128 96L224 96L232.7 69.9zM128 208L512 208L512 512C512 547.3 483.3 576 448 576L192 576C156.7 576 128 547.3 128 512L128 208zM216 272C202.7 272 192 282.7 192 296L192 488C192 501.3 202.7 512 216 512C229.3 512 240 501.3 240 488L240 296C240 282.7 229.3 272 216 272zM320 272C306.7 272 296 282.7 296 296L296 488C296 501.3 306.7 512 320 512C333.3 512 344 501.3 344 488L344 296C344 282.7 333.3 272 320 272zM424 272C410.7 272 400 282.7 400 296L400 488C400 501.3 410.7 512 424 512C437.3 512 448 501.3 448 488L448 296C448 282.7 437.3 272 424 272z"; @@ -120,7 +51,6 @@ impl JournalEntry { button x-bind:disabled="(date === initial_date) && (value === initial_value)" x-on:click="initial_date = date; initial_value = value" hx-patch=(entry_url) - hx-target="closest .entry" hx-swap="outerHTML" title="Save" { "✓" } button x-bind:disabled="(date === initial_date) && (value === initial_value)" diff --git a/src/switchboard.rs b/src/switchboard.rs new file mode 100644 index 0000000..ede4111 --- /dev/null +++ b/src/switchboard.rs @@ -0,0 +1,150 @@ +use maud::{Markup, PreEscaped}; +use regex::Regex; +use sqlx::QueryBuilder; +use sqlx::sqlite::SqlitePool; +use std::collections::HashSet; + +use crate::AppError; +use crate::db::DbId; + +pub struct Switchboard { + trie: radix_trie::Trie, +} + +struct Mentionable { + text: String, + uri: String, +} + +#[derive(Debug, PartialEq, Eq, Hash)] +pub struct Mention { + pub entity_id: DbId, + pub entity_type: DbId, + pub url: String, + pub input_text: String, + pub byte_range_start: DbId, + pub byte_range_end: DbId, +} + +// must match the constants in trigger definitions in +// migrations/each_user/0010_more-mentions.sql (or future migrations) +#[derive(Copy, Clone)] +pub enum MentionHostType { + JournalEntry, + ContactTextBody, + ContactLivesWith, +} + +#[derive(Copy, Clone)] +pub struct MentionHost<'a> { + pub entity_id: DbId, + pub entity_type: DbId, + pub input: &'a String, +} + +impl MentionHost<'_> { + pub fn format<'a>( + self: &Self, + mentions: impl IntoIterator, + ) -> Result { + let mut out = self.input.clone(); + for mention in mentions.into_iter() { + out.replace_range( + (mention.byte_range_start as usize)..(mention.byte_range_end as usize), + &format!("[{}]({})", mention.input_text, mention.url), + ); + } + + Ok(PreEscaped(markdown::to_html(&out))) + } + pub async fn format_pool(self: &Self, pool: &SqlitePool) -> Result { + let mentions = sqlx::query_as!( + Mention, + "select * from mentions + where entity_id = $1 and entity_type = $2 + order by byte_range_start desc", + self.entity_id, + self.entity_type + ) + .fetch_all(pool) + .await?; + + self.format(&mentions) + } +} + +impl Switchboard { + pub async fn new(pool: &SqlitePool) -> Result { + let mut trie = radix_trie::Trie::new(); + + let mentionables = sqlx::query_as!( + Mentionable, + "select name as text, '/contact/'||contact_id as uri from ( + select contact_id, name, count(name) as ct from names group by name + ) where ct = 1 + union + select distinct name as text, '/group/'||slug as uri from groups", + ) + .fetch_all(pool) + .await?; + + for mentionable in mentionables { + trie.insert(mentionable.text, mentionable.uri); + } + + Ok(Switchboard { trie }) + } + + pub fn remove(self: &mut Self, text: &String) { + self.trie.remove(text); + } + + pub fn add_mentionable(self: &mut Self, text: String, uri: String) { + self.trie.insert(text, uri); + } + + pub fn extract_mentions<'a>(&self, host: impl Into>) -> HashSet { + let host: MentionHost = host.into(); + let name_re = Regex::new(r"\[\[(.+?)\]\]").unwrap(); + name_re + .captures_iter(host.input) + .map(|caps| { + let range = caps.get_match().range(); + self.trie.get(&caps[1]).map(|url| Mention { + entity_id: host.entity_id, + entity_type: host.entity_type, + url: url.to_string(), + input_text: caps[1].to_string(), + byte_range_start: DbId::try_from(range.start).unwrap(), + byte_range_end: DbId::try_from(range.end).unwrap(), + }) + }) + .filter(|o| o.is_some()) + .map(|o| o.unwrap()) + .collect() + } +} + +pub async fn insert_mentions<'a>( + mentions: impl IntoIterator, + pool: &SqlitePool, +) -> Result<(), AppError> { + let mut mentions = mentions.into_iter().peekable(); + if mentions.peek().is_some() { + let mut qb = QueryBuilder::::new( + "insert into mentions ( + entity_id, entity_type, url, input_text, + byte_range_start, byte_range_end) ", + ); + qb.push_values(mentions, |mut b, mention| { + b.push_bind(mention.entity_id) + .push_bind(mention.entity_type) + .push_bind(&mention.url) + .push_bind(&mention.input_text) + .push_bind(mention.byte_range_start) + .push_bind(mention.byte_range_end); + }); + qb.build().execute(pool).await?; + } + Ok(()) +} diff --git a/src/web/contact/fields.rs b/src/web/contact/fields.rs new file mode 100644 index 0000000..0e7bcf4 --- /dev/null +++ b/src/web/contact/fields.rs @@ -0,0 +1,131 @@ +use maud::{Markup, html}; +use serde_json::json; +use sqlx::sqlite::SqlitePool; + +use crate::AppError; +use crate::db::DbId; + +pub mod addresses { + use super::*; + + #[derive(serde::Serialize, Debug)] + pub struct Address { + pub id: DbId, + pub contact_id: DbId, + pub label: Option, + pub value: String, + } + + async fn all(pool: &SqlitePool, contact_id: DbId) -> Result, sqlx::Error> { + sqlx::query_as!( + Address, + "select * from addresses where contact_id = $1", + contact_id + ) + .fetch_all(pool) + .await + } + + pub async fn get(pool: &SqlitePool, contact_id: DbId) -> Result { + let addresses: Vec
= addresses::all(pool, contact_id).await?; + + Ok(html! { + @if addresses.len() == 1 { + label { "address" } + #addresses { + .label {} + .value { (addresses[0].value) } + } + } @else if addresses.len() > 0 { + label { "addresses" } + #addresses { + @for address in addresses { + @let lbl = address.label.unwrap_or(String::new()); + .label data-is-empty=(lbl.len() == 0) { + (lbl) + } + .value { (address.value) } + } + } + } + }) + } + + pub async fn edit(pool: &SqlitePool, contact_id: DbId) -> Result { + let addresses: Vec
= addresses::all(pool, contact_id).await?; + + Ok(html! { + label { "addresses" } + div x-data=(json!({ "addresses": addresses, "new_label": "", "new_address": "" })) { + template x-for="(address, index) in addresses" x-bind:key="index" { + .address-input { + input name="address_label" x-show="addresses.length" x-model="address.label" placeholder="label"; + .grow-wrap x-bind:data-replicated-value="address.value" { + textarea name="address_value" x-model="address.value" placeholder="address" {} + } + } + } + .address-input { + input x-show="addresses.length" name="address_label" x-model="new_label" placeholder="label"; + .grow-wrap x-bind:data-replicated-value="new_address" { + textarea name="address_value" x-model="new_address" placeholder="new address" {} + } + } + input type="button" value="Add" x-on:click="addresses.push({ label: new_label, value: new_address }); new_label = ''; new_address = ''"; + } + }) + } +} + +pub mod groups { + use super::*; + + #[derive(serde::Serialize, Debug)] + pub struct Group { + pub contact_id: DbId, + pub name: String, + pub slug: String, + } + + async fn all(pool: &SqlitePool, contact_id: DbId) -> Result, sqlx::Error> { + sqlx::query_as!( + Group, + "select * from groups where contact_id = $1", + contact_id + ) + .fetch_all(pool) + .await + } + + pub async fn get(pool: &SqlitePool, contact_id: DbId) -> Result { + let groups: Vec = groups::all(pool, contact_id).await?; + + Ok(html! { + @if groups.len() > 0 { + label { "in groups" } + #groups { + @for group in groups { + a .group href=(format!("/group/{}", group.slug)) { + (group.name) + } + } + } + } + }) + } + + pub async fn edit(pool: &SqlitePool, contact_id: DbId) -> Result { + let groups: Vec = groups::all(pool, contact_id).await?; + + Ok(html! { + label { "groups" } + #groups x-data=(json!({ "groups": groups, "new_group": "" })) { + template x-for="(group, index) in groups" x-bind:key="index" { + input name="group" x-model="group.name" placeholder="group name"; + } + input name="group" x-model="new_group" placeholder="group name"; + input type="button" value="Add" x-on:click="groups.push({ name: new_group }); new_group = ''"; + } + }) + } +} diff --git a/src/web/contact.rs b/src/web/contact/mod.rs similarity index 65% rename from src/web/contact.rs rename to src/web/contact/mod.rs index 1808025..8eea1b0 100644 --- a/src/web/contact.rs +++ b/src/web/contact/mod.rs @@ -8,7 +8,7 @@ use axum::{ use axum_extra::extract::Form; use cache_bust::asset; use chrono::DateTime; -use maud::{Markup, PreEscaped, html}; +use maud::{Markup, html}; use serde::Deserialize; use serde_json::json; use slug::slugify; @@ -19,21 +19,16 @@ use super::home::journal_section; use crate::db::DbId; use crate::models::user::AuthSession; use crate::models::{HydratedContact, JournalEntry}; +use crate::switchboard::{MentionHost, MentionHostType, insert_mentions}; use crate::{AppError, AppState}; -#[derive(serde::Serialize, Debug)] -pub struct Address { - pub id: DbId, - pub contact_id: DbId, - pub label: Option, - pub value: String, -} +pub mod fields; #[derive(serde::Serialize, Debug)] -pub struct Group { +pub struct PhoneNumber { pub contact_id: DbId, - pub name: String, - pub slug: String, + pub label: Option, + pub phone_number: String, } pub fn router() -> Router { @@ -69,51 +64,47 @@ mod get { pub async fn contact( auth_session: AuthSession, State(state): State, - Path(contact_id): Path, + Path(contact_id): Path, layout: Layout, ) -> Result { - let pool = &state.db(&auth_session.user.unwrap()).pool; - let contact: HydratedContact = sqlx::query_as( - "select id, birthday, manually_freshened_at, ( - select string_agg(name,'\x1c' order by sort) - from names where contact_id = c.id - ) as names - from contacts c - where c.id = $1", - ) - .bind(contact_id) - .fetch_one(pool) - .await?; + let user = auth_session.user.unwrap(); + let pool = &state.db(&user).pool; + + let contact = HydratedContact::load(contact_id, pool).await?; let entries: Vec = sqlx::query_as( "select distinct j.id, j.value, j.date from journal_entries j - join journal_mentions cm on j.id = cm.entry_id - where cm.url = '/contact/'||$1 or cm.url in ( + join mentions m on j.id = m.entity_id + where m.entity_type = $1 and (m.url = '/contact/'||$1 or m.url in ( select '/group/'||slug from groups - where contact_id = $1 - ) + where contact_id = $2 + )) order by j.date desc ", ) + .bind(MentionHostType::JournalEntry as DbId) .bind(contact_id) .fetch_all(pool) .await?; - let addresses: Vec
= sqlx::query_as!( - Address, - "select * from addresses where contact_id = $1", + let phone_numbers: Vec = sqlx::query_as!( + PhoneNumber, + "select * from phone_numbers where contact_id = $1", contact_id ) .fetch_all(pool) .await?; - let groups: Vec = sqlx::query_as!( - Group, - "select * from groups where contact_id = $1", - contact_id - ) - .fetch_all(pool) - .await?; + let lives_with = if contact.lives_with.len() > 1 { + let mention_host = MentionHost { + entity_id: contact_id, + entity_type: MentionHostType::ContactLivesWith as DbId, + input: &contact.lives_with, + }; + Some(mention_host.format_pool(pool).await?) + } else { + None + }; let text_body: Option = sqlx::query!("select text_body from contacts where id = $1", contact_id) @@ -156,41 +147,37 @@ mod get { "(never)" } } - @if addresses.len() == 1 { - label { "address" } - #addresses { - .label {} - .value { (addresses[0].value) } - } - } @else if addresses.len() > 0 { - label { "addresses" } - #addresses { - @for address in addresses { - @let lbl = address.label.unwrap_or(String::new()); - .label data-is-empty=(lbl.len() == 0) { - (lbl) + + @if phone_numbers.len() > 0 { + label { "phone" } + #phone_numbers { + @for phone_number in phone_numbers { + @let lbl = phone_number.label.unwrap_or(String::new()); + .label data-is-empty=(lbl.len() == 0) { (lbl) } + .phone_nunber { + a href=(format!("tel:{}", phone_number.phone_number)) { (phone_number.phone_number) } } - .value { (address.value) } } } } - @if groups.len() > 0 { - label { "in groups" } - #groups { - @for group in groups { - a .group href=(format!("/group/{}", group.slug)) { - (group.name) - } - } - } + @if let Some(lives_with) = lives_with { + label { "lives with" } + div { (lives_with) } } + + (fields::addresses::get(pool, contact_id).await?) + (fields::groups::get(pool, contact_id).await?) } @if let Some(text_body) = text_body { @if text_body.len() > 0 { - #text_body { (PreEscaped(markdown::to_html(&text_body))) } + #text_body { (MentionHost { + entity_id: contact_id, + entity_type: MentionHostType::ContactTextBody as DbId, + input: &text_body + }.format_pool(pool).await?) } } } @@ -202,29 +189,15 @@ mod get { pub async fn contact_edit( auth_session: AuthSession, State(state): State, - Path(contact_id): Path, + Path(contact_id): Path, layout: Layout, ) -> Result { let pool = &state.db(&auth_session.user.unwrap()).pool; - let contact: HydratedContact = sqlx::query_as( - "select id, birthday, manually_freshened_at, ( - select string_agg(name,'\x1c' order by sort) - from names where contact_id = c.id - ) as names, ( - select jes.date from journal_entries jes - join journal_mentions cms on cms.entry_id = jes.id - where cms.url = '/contact/'||c.id - order by jes.date desc limit 1 - ) as last_mention_date from contacts c - where c.id = $1", - ) - .bind(contact_id) - .fetch_one(pool) - .await?; + let contact = HydratedContact::load(contact_id, pool).await?; - let addresses: Vec
= sqlx::query_as!( - Address, - "select * from addresses where contact_id = $1", + let phone_numbers: Vec = sqlx::query_as!( + PhoneNumber, + "select * from phone_numbers where contact_id = $1", contact_id ) .fetch_all(pool) @@ -236,17 +209,6 @@ mod get { .clone() .map_or("".to_string(), |m| m.to_rfc3339()); - let groups: Vec = sqlx::query_as!( - Group, - "select * from groups where contact_id = $1", - contact_id - ) - .fetch_all(pool) - .await? - .into_iter() - .map(|group| group.name) - .collect(); - let text_body: String = sqlx::query!("select text_body from contacts where id = $1", contact_id) .fetch_one(pool) @@ -289,32 +251,26 @@ mod get { span x-text="date.length ? date.split('T')[0] : '(never)'" {} input type="button" value="Mark fresh now" x-on:click="date = new Date().toISOString()"; } - label { "addresses" } - div x-data=(json!({ "addresses": addresses, "new_label": "", "new_address": "" })) { - template x-for="(address, index) in addresses" x-bind:key="index" { - .address-input { - input name="address_label" x-show="addresses.length" x-model="address.label" placeholder="label"; - .grow-wrap x-bind:data-replicated-value="address.value" { - textarea name="address_value" x-model="address.value" placeholder="address" {} - } + label { "phone" } + #phone_numbers x-data=(json!({ "phones": phone_numbers, "new_label": "", "new_number": "" })) { + template x-for="(phone, index) in phones" x-bind:key="index" { + .phone_input { + input name="phone_label" x-model="phone.label" placeholder="home/work/mobile"; + input name="phone_number" x-model="phone.phone_number" placeholder="number"; } } - .address-input { - input x-show="addresses.length" name="address_label" x-model="new_label" placeholder="label"; - .grow-wrap x-bind:data-replicated-value="new_address" { - textarea name="address_value" x-model="new_address" placeholder="new address" {} - } + .phone_input { + input name="phone_label" x-model="new_label" placeholder="home/work/mobile"; + input name="phone_number" x-model="new_number" placeholder="number"; } - input type="button" value="Add" x-on:click="addresses.push({ label: new_label, value: new_address }); new_label = ''; new_address = ''"; + input type="button" value="Add" x-on:click="phones.push({ label: new_label, phone_number: new_number }); new_label=''; new_number = ''"; } - label { "groups" } - #groups x-data=(json!({ "groups": groups, "new_group": "" })) { - template x-for="(group, index) in groups" x-bind:key="index" { - input name="group" x-model="group" placeholder="group name"; - } - input name="group" x-model="new_group" placeholder="group name"; - input type="button" value="Add" x-on:click="groups.push(new_group); new_group = ''"; + label { "lives with" } + div { + input name="lives_with" value=(contact.lives_with); } + (fields::addresses::edit(pool, contact_id).await?) + (fields::groups::edit(pool, contact_id).await?) } div #text_body { div { "Free text (supports markdown)" } @@ -339,7 +295,7 @@ mod post { let user = auth_session.user.unwrap(); let pool = &state.db(&user).pool; - let contact_id: (u32,) = + let contact_id: (DbId,) = sqlx::query_as("insert into contacts (birthday) values (null) returning id") .fetch_one(pool) .await?; @@ -361,6 +317,9 @@ mod put { name: Option>, birthday: String, manually_freshened_at: String, + lives_with: String, + phone_label: Option>, + phone_number: Option>, address_label: Option>, address_value: Option>, group: Option>, @@ -375,6 +334,7 @@ mod put { ) -> Result { let user = auth_session.user.unwrap(); let pool = &state.db(&user).pool; + let sw_lock = state.switchboard(&user); let birthday = if payload.birthday.is_empty() { None @@ -399,19 +359,114 @@ mod put { Some(payload.text_body) }; + let old_contact = sqlx::query!("select * from contacts where id = $1", contact_id) + .fetch_one(pool) + .await?; + sqlx::query!( - "update contacts set (birthday, manually_freshened_at, text_body) = ($1, $2, $3) where id = $4", + "update contacts set + (birthday, manually_freshened_at, lives_with, text_body) = + ($1, $2, $3, $4) + where id = $5", birthday, manually_freshened_at, + payload.lives_with, text_body, contact_id ) .execute(pool) .await?; + if old_contact.lives_with != payload.lives_with { + sqlx::query!( + "delete from mentions where entity_id = $1 and entity_type = $2", + contact_id, + MentionHostType::ContactLivesWith as DbId + ) + .execute(pool) + .await?; + + let mention_host = MentionHost { + entity_id: contact_id, + entity_type: MentionHostType::ContactLivesWith as DbId, + input: &payload.lives_with, + }; + + let mentions = { + let switchboard = sw_lock.read().unwrap(); + switchboard.extract_mentions(mention_host) + }; + insert_mentions(&mentions, pool).await?; + } + + if old_contact.text_body != text_body { + sqlx::query!( + "delete from mentions where entity_id = $1 and entity_type = $2", + contact_id, + MentionHostType::ContactTextBody as DbId + ) + .execute(pool) + .await?; + + if text_body.is_some() { + let mention_host = MentionHost { + entity_id: contact_id, + entity_type: MentionHostType::ContactTextBody as DbId, + input: &text_body.unwrap(), + }; + + let mentions = { + let switchboard = sw_lock.read().unwrap(); + switchboard.extract_mentions(mention_host) + }; + insert_mentions(&mentions, pool).await?; + } + } + // these blocks are not in functions because payload gets progressively // partially moved as we handle each field and i don't want to deal with it + { + // update phone numbers + let new_numbers = payload.phone_number.clone().map_or(vec![], |numbers| { + let labels: Vec = payload.phone_label.clone().unwrap(); + + // TODO sanitize down to linkable on input + labels + .into_iter() + .zip(numbers) + .filter(|(_, val)| val.len() > 0) + .collect::>() + }); + + let old_numbers: Vec<(String, String)> = sqlx::query_as( + "select label, phone_number from phone_numbers where contact_id = $1", + ) + .bind(contact_id) + .fetch_all(pool) + .await?; + + if new_numbers != old_numbers { + sqlx::query!( + "delete from phone_numbers where contact_id = $1", + contact_id + ) + .execute(pool) + .await?; + + // trailing space in query intentional + QueryBuilder::new("insert into phone_numbers (contact_id, label, phone_number) ") + .push_values(new_numbers, |mut b, (label, phone_number)| { + b.push_bind(contact_id) + .push_bind(label) + .push_bind(phone_number); + }) + .build() + .execute(pool) + .await?; + } + } + { // update addresses let new_addresses = payload.address_value.clone().map(|values| { @@ -468,11 +523,11 @@ mod put { let old_names: Vec = old_names.into_iter().map(|(s,)| s).collect(); if old_names != new_names { - // delete and regen *all* journal mentions, not just the ones for the - // current user, since changing *this* user's names can change, *globally*, + // delete and regen *all* mentions, not just the ones for the current + // contact, since changing *this* contact's names can change, *globally*, // which names have n=1 and thus are eligible for mentioning sqlx::query!( - "delete from journal_mentions; delete from names where contact_id = $1", + "delete from mentions; delete from names where contact_id = $1", contact_id ) .execute(pool) @@ -514,14 +569,13 @@ mod put { .await?; { - let trie_mutex = state.contact_search(&user); - let mut trie = trie_mutex.write().unwrap(); + let mut switchboard = sw_lock.write().unwrap(); for name in &old_names { - trie.remove(name); + switchboard.remove(name); } for name in recalc_names { - trie.insert(name.0, format!("/contact/{}", name.1)); + switchboard.add_mentionable(name.0, format!("/contact/{}", name.1)); } } } @@ -541,35 +595,37 @@ mod put { if new_groups != old_groups { sqlx::query!( - "delete from journal_mentions; delete from groups where contact_id = $1", + "delete from mentions; delete from groups where contact_id = $1", contact_id ) .execute(pool) .await?; - QueryBuilder::new("insert into groups (contact_id, name, slug) ") - .push_values(&new_groups, |mut b, name| { - b.push_bind(contact_id) - .push_bind(name) - .push_bind(slugify(name)); - }) - .build() - .persistent(false) - .execute(pool) - .await?; + if new_groups.len() > 0 { + QueryBuilder::new("insert into groups (contact_id, name, slug) ") + .push_values(&new_groups, |mut b, name| { + b.push_bind(contact_id) + .push_bind(name) + .push_bind(slugify(name)); + }) + .build() + .persistent(false) + .execute(pool) + .await?; + } { - let trie_mutex = state.contact_search(&user); - let mut trie = trie_mutex.write().unwrap(); + let mut switchboard = sw_lock.write().unwrap(); for name in &old_groups { // TODO i think we care about group name vs contact name counts, // otherwise this will cause a problem (or we want to disallow // setting group names that are contact names or vice versa?) - trie.remove(name); + switchboard.remove(name); } for group in &new_groups { - trie.insert(group.clone(), format!("/group/{}", slugify(group))); + switchboard + .add_mentionable(group.clone(), format!("/group/{}", slugify(group))); } } } @@ -581,9 +637,11 @@ mod put { .await?; for entry in journal_entries { - entry - .insert_mentions(state.contact_search(&user), pool) - .await?; + let mentions = { + let switchboard = sw_lock.read().unwrap(); + switchboard.extract_mentions(&entry) + }; + insert_mentions(&mentions, pool).await?; } } } @@ -598,19 +656,14 @@ mod delete { pub async fn contact( auth_session: AuthSession, State(state): State, - Path(contact_id): Path, + Path(contact_id): Path, ) -> Result { let user = auth_session.user.unwrap(); let pool = &state.db(&user).pool; - sqlx::query( - "delete from journal_mentions where contact_id = $1; - delete from names where contact_id = $1; - delete from contacts where id = $1;", - ) - .bind(contact_id) - .execute(pool) - .await?; + sqlx::query!("delete from contacts where id = $1", contact_id) + .execute(pool) + .await?; let mut headers = HeaderMap::new(); headers.insert("HX-Redirect", "/".parse()?); diff --git a/src/web/home.rs b/src/web/home.rs index ce3ebae..0ed42ba 100644 --- a/src/web/home.rs +++ b/src/web/home.rs @@ -84,6 +84,7 @@ fn birthdays_section( }) } +#[tracing::instrument(level = "info")] pub async fn journal_section( pool: &SqlitePool, entries: &Vec, @@ -101,12 +102,14 @@ pub async fn journal_section( are now, or leave everything blank to default to 'today'. Entries will be added to the top of the list regardless of date; refresh the page to re-sort." } - form hx-post="/journal_entry" hx-target="next .entries" hx-swap="afterbegin" hx-on::after-request="if(event.detail.successful) this.reset()" { + form hx-post="/journal_entry" hx-target="next .entries" hx-target-error="#journal-error" hx-swap="afterbegin" hx-on::after-request="if(event.detail.successful) this.reset()" { input name="date" placeholder=(Local::now().date_naive().to_string()); textarea name="value" placeholder="New entry..." autofocus {} input type="submit" value="Add Entry"; } + #journal-error {} + .entries { @for entry in entries { (entry.to_html(pool).await?) @@ -124,21 +127,10 @@ pub mod get { State(state): State, layout: Layout, ) -> Result { - let pool = &state.db(&auth_session.user.unwrap()).pool; - let contacts: Vec = sqlx::query_as( - "select id, birthday, manually_freshened_at, ( - select string_agg(name,'\x1c' order by sort) - from names where contact_id = c.id - ) as names, ( - select jes.date from journal_entries jes - join journal_mentions cms on cms.entry_id = jes.id - where cms.url = '/contact/'||c.id - order by jes.date desc limit 1 - ) as last_mention_date from contacts c", - ) - .fetch_all(pool) - .await?; + let user = auth_session.user.unwrap(); + let pool = &state.db(&user).pool; + let contacts = HydratedContact::all(&pool).await?; let mut freshens: Vec = contacts .clone() .into_iter() diff --git a/src/web/ics.rs b/src/web/ics.rs index 107e270..b7780ec 100644 --- a/src/web/ics.rs +++ b/src/web/ics.rs @@ -55,15 +55,9 @@ mod get { let mut calendar = Calendar::new(); calendar.name(&calname); calendar.append_property(("PRODID", "Mascarpone CRM")); - let contacts: Vec = sqlx::query_as( - "select id, birthday, ( - select string_agg(name,'\x1c' order by sort) - from names where contact_id = c.id - ) as names - from contacts c", - ) - .fetch_all(&pool) - .await?; + + // TODO; this does some db work to pull in last_modified_date that we don't use + let contacts = HydratedContact::all(&pool).await?; for contact in &contacts { if let Some(Birthday::Date(yo_date)) = &contact.birthday { if let Some(date) = NaiveDate::from_ymd_opt( diff --git a/src/web/journal.rs b/src/web/journal.rs index 26ac93e..c12f15b 100644 --- a/src/web/journal.rs +++ b/src/web/journal.rs @@ -11,6 +11,7 @@ use serde::Deserialize; use crate::models::JournalEntry; use crate::models::user::AuthSession; +use crate::switchboard::{MentionHost, insert_mentions}; use crate::{AppError, AppState}; pub fn router() -> Router { @@ -36,6 +37,8 @@ mod post { ) -> Result { let user = auth_session.user.unwrap(); let pool = &state.db(&user).pool; + let sw_lock = state.switchboard(&user); + let now = Local::now().date_naive(); let date = if payload.date.is_empty() { @@ -73,9 +76,12 @@ mod post { .fetch_one(pool) .await?; - entry - .insert_mentions(state.contact_search(&user), pool) - .await?; + let mentions = { + let switchboard = sw_lock.read().unwrap(); + switchboard.extract_mentions(&entry) + }; + tracing::debug!("{:?}", mentions); + insert_mentions(&mentions, pool).await?; Ok(entry.to_html(pool).await?) } @@ -84,6 +90,7 @@ mod post { mod patch { use super::*; + #[axum::debug_handler] pub async fn entry( auth_session: AuthSession, State(state): State, @@ -92,8 +99,10 @@ mod patch { ) -> Result { let user = auth_session.user.unwrap(); let pool = &state.db(&user).pool; + let sw_lock = state.switchboard(&user); + // not a macro query, we want to use JournalEntry's custom FromRow - let entry: JournalEntry = sqlx::query_as("select * from journal_entries where id = $1") + let old_entry: JournalEntry = sqlx::query_as("select * from journal_entries where id = $1") .bind(entry_id) .fetch_one(pool) .await?; @@ -107,17 +116,24 @@ mod patch { .fetch_one(pool) .await?; - if entry.value != new_entry.value { - sqlx::query!("delete from journal_mentions where entry_id = $1", entry_id) - .execute(pool) - .await?; + if old_entry.value != new_entry.value { + sqlx::query!( + "delete from mentions where entity_id = $1 and entity_type = 'journal_entry'", + entry_id + ) + .execute(pool) + .await?; - new_entry - .insert_mentions(state.contact_search(&user), pool) - .await?; + let mentions = { + let switchboard = sw_lock.read().unwrap(); + switchboard.extract_mentions(&new_entry) + }; + insert_mentions(&mentions, pool).await?; } - Ok(new_entry.to_html(pool).await?) + Ok(Into::::into(&new_entry) + .format_pool(pool) + .await?) } } @@ -132,14 +148,11 @@ mod delete { let user = auth_session.user.unwrap(); let pool = &state.db(&user).pool; - sqlx::query( - "delete from journal_mentions where entry_id = $1; - delete from journal_entries where id = $2 returning id,date,value", - ) - .bind(entry_id) - .bind(entry_id) - .execute(pool) - .await?; + sqlx::query("delete from journal_entries where id = $2 returning id,date,value") + .bind(entry_id) + .bind(entry_id) + .execute(pool) + .await?; Ok(()) } diff --git a/src/web/mod.rs b/src/web/mod.rs index 97c0d1d..66ab59c 100644 --- a/src/web/mod.rs +++ b/src/web/mod.rs @@ -3,10 +3,10 @@ use axum::extract::FromRequestParts; use cache_bust::asset; use http::request::Parts; use maud::{DOCTYPE, Markup, html}; -use sqlx::FromRow; use super::models::user::{AuthSession, User}; use super::{AppError, AppState}; +use crate::db::DbId; pub mod auth; pub mod contact; @@ -16,11 +16,13 @@ pub mod ics; pub mod journal; pub mod settings; -#[derive(Debug, FromRow)] +#[derive(Debug)] struct ContactLink { name: String, - contact_id: u32, + contact_id: DbId, } + +#[derive(Debug)] pub struct Layout { contact_links: Vec, user: User, @@ -39,7 +41,8 @@ impl FromRequestParts for Layout { .map_err(|_| anyhow::Error::msg("could not get session"))?; let user = auth_session.user.unwrap(); - let contact_links: Vec = sqlx::query_as( + let contact_links = sqlx::query_as!( + ContactLink, "select c.id as contact_id, coalesce(n.name, '(unnamed)') as name from contacts c