Compare commits
34
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c625c80958 | ||
|
|
157f03b655 | ||
|
|
9b7ca3d692 | ||
|
|
e51a59a829 | ||
|
|
62cadbcfa0 | ||
|
|
d7aefc4110 | ||
|
|
e972a8dddb | ||
|
|
88b9375a4e | ||
|
|
f7a3e3f0f8 | ||
|
|
72df3dec3b | ||
|
|
b80ec5c6ac | ||
|
|
fbd3aff6d8 | ||
|
|
0ad79c9bc1 | ||
|
|
e175d28c2c | ||
|
|
8e3f09774a | ||
|
|
dbc54032f7 | ||
|
|
047b99e2aa | ||
|
|
1859078cf4 | ||
|
|
c44ba91ec7 | ||
|
|
96a6d0ee5d | ||
|
|
5619bb27a1 | ||
|
|
dfc99c7d15 | ||
|
|
fb82f505b6 | ||
|
|
5970340a17 | ||
|
|
9a77e76ff4 | ||
|
|
3591415d46 | ||
|
|
ab06205744 | ||
|
|
0689e680fa | ||
|
|
297d5aca1f | ||
|
|
0a023c1556 | ||
|
|
aa600af7b2 | ||
|
|
fc4aae8c0c | ||
|
|
24c0b5fa52 | ||
|
|
2af2af3460 |
@@ -1,5 +1,15 @@
|
|||||||
# Papercrate
|
# Papercrate
|
||||||
|
|
||||||
|
## Local Development
|
||||||
|
|
||||||
|
Use the provided `papercrate.tmux` to spin up the full stack in one tmux session:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
tmux -f papercrate.tmux attach
|
||||||
|
```
|
||||||
|
|
||||||
|
This creates windows for the compose stack, frontend dev server, backend API, and background worker using the repository-relative paths defined in the tmux file. Detach with `Ctrl+b d` and reattach later with the same command.
|
||||||
|
|
||||||
## Backend Integration Tests
|
## Backend Integration Tests
|
||||||
|
|
||||||
Integration tests require a running Postgres instance (and, optionally, Quickwit for OCR indexing). The repository includes a lightweight compose file for local runs:
|
Integration tests require a running Postgres instance (and, optionally, Quickwit for OCR indexing). The repository includes a lightweight compose file for local runs:
|
||||||
|
|||||||
Generated
+87
-25
@@ -67,7 +67,7 @@ checksum = "9035ad2d096bed7955a320ee7e2230574d28fd3c3a0f186cbea1ff3c7eed5dbb"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -635,6 +635,7 @@ dependencies = [
|
|||||||
"tracing",
|
"tracing",
|
||||||
"tracing-subscriber",
|
"tracing-subscriber",
|
||||||
"url",
|
"url",
|
||||||
|
"utoipa",
|
||||||
"uuid",
|
"uuid",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -704,7 +705,7 @@ dependencies = [
|
|||||||
"regex",
|
"regex",
|
||||||
"rustc-hash",
|
"rustc-hash",
|
||||||
"shlex",
|
"shlex",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -989,7 +990,7 @@ dependencies = [
|
|||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"strsim",
|
"strsim",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1000,7 +1001,7 @@ checksum = "d38308df82d1080de0afee5d069fa14b0326a88c14f15c5ccda35b4a6c414c81"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"darling_core",
|
"darling_core",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1050,7 +1051,7 @@ dependencies = [
|
|||||||
"dsl_auto_type",
|
"dsl_auto_type",
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1070,7 +1071,7 @@ version = "0.3.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "fe2444076b48641147115697648dc743c2c00b61adade0f01ce67133c7babe8c"
|
checksum = "fe2444076b48641147115697648dc743c2c00b61adade0f01ce67133c7babe8c"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1092,7 +1093,7 @@ checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1118,7 +1119,7 @@ dependencies = [
|
|||||||
"heck",
|
"heck",
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1287,7 +1288,7 @@ checksum = "162ee34ebcb7c64a8abebc059ce0fee27c2262618d7b60ed8faf72fef13c3650"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1812,6 +1813,8 @@ checksum = "4b0f83760fb341a774ed326568e19f5a863af4a952def8c39f9ab92fd95b88e5"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"equivalent",
|
"equivalent",
|
||||||
"hashbrown 0.16.0",
|
"hashbrown 0.16.0",
|
||||||
|
"serde",
|
||||||
|
"serde_core",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -2267,7 +2270,7 @@ checksum = "6e918e4ff8c4549eb882f14b3a4bc8c8bc93de829416eacf579f1207a8fbf861"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -2359,7 +2362,31 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "479ca8adacdd7ce8f1fb39ce9ecccbfe93a3f1344b3d0d97f20bc0196208f62b"
|
checksum = "479ca8adacdd7ce8f1fb39ce9ecccbfe93a3f1344b3d0d97f20bc0196208f62b"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro-error"
|
||||||
|
version = "1.0.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "da25490ff9892aab3fcf7c36f08cfb902dd3e71ca0f9f9517bea02a73a5ce38c"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro-error-attr",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn 1.0.109",
|
||||||
|
"version_check",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro-error-attr"
|
||||||
|
version = "1.0.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a1be40180e52ecc98ad80b184934baf3d0d29f979574e439af5a55274b35f869"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"version_check",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -2894,7 +2921,7 @@ checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3082,6 +3109,16 @@ version = "2.6.1"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "13c2bddecc57b384dee18652358fb23172facb8a2c51ccc10d74c157bdea3292"
|
checksum = "13c2bddecc57b384dee18652358fb23172facb8a2c51ccc10d74c157bdea3292"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "1.0.109"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "syn"
|
name = "syn"
|
||||||
version = "2.0.106"
|
version = "2.0.106"
|
||||||
@@ -3110,7 +3147,7 @@ checksum = "728a70f3dbaf5bab7f0c4b1ac8d7ae5ea60a4b5549c8a5914361c99147a709d2"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3152,7 +3189,7 @@ checksum = "4fee6c4efc90059e10f81e6d42c60a18f76588c3d74cb83a0b242a2b6c7504c1"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3163,7 +3200,7 @@ checksum = "3ff15c8ecd7de3849db632e14d18d2571fa09dfc5ed93479bc4485c7a517c913"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3259,7 +3296,7 @@ checksum = "6e06d43f1345a3bcd39f6a56dbb7dcab2ba47e68e8ac134855e7e2bdbaf8cab8"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3409,7 +3446,7 @@ checksum = "81383ab64e72a7a8b8e13130c49e3dab29def6d0c7d76a03087b3cf71c5c6903"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3514,6 +3551,31 @@ version = "1.0.4"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b6c140620e7ffbb22c2dee59cafe6084a59b5ffc27a8859a5f0d494b5d52b6be"
|
checksum = "b6c140620e7ffbb22c2dee59cafe6084a59b5ffc27a8859a5f0d494b5d52b6be"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "utoipa"
|
||||||
|
version = "4.2.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "c5afb1a60e207dca502682537fefcfd9921e71d0b83e9576060f09abc6efab23"
|
||||||
|
dependencies = [
|
||||||
|
"indexmap",
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"utoipa-gen",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "utoipa-gen"
|
||||||
|
version = "4.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "20c24e8ab68ff9ee746aad22d39b5535601e6416d1b0feeabf78be986a5c4392"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro-error",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn 2.0.106",
|
||||||
|
"uuid",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "uuid"
|
name = "uuid"
|
||||||
version = "1.18.1"
|
version = "1.18.1"
|
||||||
@@ -3615,7 +3677,7 @@ dependencies = [
|
|||||||
"log",
|
"log",
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
"wasm-bindgen-shared",
|
"wasm-bindgen-shared",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -3650,7 +3712,7 @@ checksum = "9f07d2f20d4da7b26400c9f4a0511e6e0345b040694e8a75bd41d578fa4421d7"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
"wasm-bindgen-backend",
|
"wasm-bindgen-backend",
|
||||||
"wasm-bindgen-shared",
|
"wasm-bindgen-shared",
|
||||||
]
|
]
|
||||||
@@ -3727,7 +3789,7 @@ checksum = "053e2e040ab57b9dc951b72c264860db7eb3b0200ba345b4e4c3b14f67855ddf"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3738,7 +3800,7 @@ checksum = "3f316c4a2570ba26bbec722032c4099d8c8bc095efccdc15688708623367e358"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -3965,7 +4027,7 @@ checksum = "38da3c9736e16c5d3c8c597a9aaa5d1fa565d0532ae05e27c24aa62fb32c0ab6"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
"synstructure",
|
"synstructure",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -3986,7 +4048,7 @@ checksum = "88d2b8d9c68ad2b9e4340d7832716a4d21a22a1154777ad56ea55c51a9cf3831"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -4006,7 +4068,7 @@ checksum = "d71e5d6e06ab090c67b5e44993ec16b72dcbaabc526db883a360057678b48502"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
"synstructure",
|
"synstructure",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -4046,7 +4108,7 @@ checksum = "5b96237efa0c878c64bd89c436f661be4e46b2f3eff1ebb976f7ef2321d2f58f"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn 2.0.106",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
|
|||||||
@@ -44,6 +44,8 @@ base64 = "0.21"
|
|||||||
quick-xml = "0.32"
|
quick-xml = "0.32"
|
||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
url = "2.5"
|
url = "2.5"
|
||||||
|
once_cell = "1.19"
|
||||||
|
utoipa = { version = "4.2", default-features = false, features = ["chrono", "uuid", "preserve_order"] }
|
||||||
|
|
||||||
# Error handling
|
# Error handling
|
||||||
thiserror = "1.0"
|
thiserror = "1.0"
|
||||||
|
|||||||
+2
-1
@@ -20,7 +20,7 @@ COPY migrations ./migrations
|
|||||||
COPY tests ./tests
|
COPY tests ./tests
|
||||||
COPY diesel.toml ./
|
COPY diesel.toml ./
|
||||||
|
|
||||||
RUN cargo build --release --bin backend --bin worker --bin webdav
|
RUN cargo build --release --bin backend --bin worker --bin webdav --bin admin
|
||||||
RUN cargo install diesel_cli --no-default-features --features postgres
|
RUN cargo install diesel_cli --no-default-features --features postgres
|
||||||
|
|
||||||
FROM debian:trixie-slim AS runtime
|
FROM debian:trixie-slim AS runtime
|
||||||
@@ -53,6 +53,7 @@ RUN apt-get update \
|
|||||||
COPY --from=builder /app/target/release/backend /usr/local/bin/papercrate-backend
|
COPY --from=builder /app/target/release/backend /usr/local/bin/papercrate-backend
|
||||||
COPY --from=builder /app/target/release/worker /usr/local/bin/papercrate-worker
|
COPY --from=builder /app/target/release/worker /usr/local/bin/papercrate-worker
|
||||||
COPY --from=builder /app/target/release/webdav /usr/local/bin/papercrate-webdav
|
COPY --from=builder /app/target/release/webdav /usr/local/bin/papercrate-webdav
|
||||||
|
COPY --from=builder /app/target/release/admin /usr/local/bin/papercrate-admin
|
||||||
COPY --from=builder /usr/local/cargo/bin/diesel /usr/local/bin/diesel
|
COPY --from=builder /usr/local/cargo/bin/diesel /usr/local/bin/diesel
|
||||||
COPY migrations ./migrations
|
COPY migrations ./migrations
|
||||||
COPY diesel.toml ./
|
COPY diesel.toml ./
|
||||||
|
|||||||
@@ -1,19 +0,0 @@
|
|||||||
use argon2::{
|
|
||||||
password_hash::{PasswordHasher, SaltString},
|
|
||||||
Argon2,
|
|
||||||
};
|
|
||||||
use rand::thread_rng;
|
|
||||||
use std::env;
|
|
||||||
|
|
||||||
fn main() {
|
|
||||||
let password = env::args()
|
|
||||||
.nth(1)
|
|
||||||
.expect("Usage: cargo run --example hash_password <password>");
|
|
||||||
let salt = SaltString::generate(&mut thread_rng());
|
|
||||||
let argon2 = Argon2::default();
|
|
||||||
let hash = argon2
|
|
||||||
.hash_password(password.as_bytes(), &salt)
|
|
||||||
.expect("hashing failed")
|
|
||||||
.to_string();
|
|
||||||
println!("{}", hash);
|
|
||||||
}
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS idx_document_tags_tag;
|
|
||||||
DROP TABLE IF EXISTS document_tags;
|
|
||||||
DROP INDEX IF EXISTS idx_document_versions_document;
|
|
||||||
DROP TABLE IF EXISTS document_versions;
|
|
||||||
DROP INDEX IF EXISTS idx_documents_deleted_at;
|
|
||||||
DROP INDEX IF EXISTS idx_documents_folder;
|
|
||||||
DROP TABLE IF EXISTS documents;
|
|
||||||
DROP INDEX IF EXISTS idx_folders_parent;
|
|
||||||
DROP TABLE IF EXISTS folders;
|
|
||||||
DROP TABLE IF EXISTS tags;
|
|
||||||
DROP TABLE IF EXISTS users;
|
|
||||||
@@ -1,77 +0,0 @@
|
|||||||
CREATE EXTENSION IF NOT EXISTS "pgcrypto";
|
|
||||||
|
|
||||||
CREATE TABLE users (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
username VARCHAR(100) NOT NULL UNIQUE,
|
|
||||||
password_hash VARCHAR(255) NOT NULL,
|
|
||||||
role VARCHAR(16) NOT NULL,
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
updated_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE TABLE folders (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
name VARCHAR(255) NOT NULL,
|
|
||||||
parent_id UUID REFERENCES folders(id) ON DELETE SET NULL,
|
|
||||||
path_cache VARCHAR(1000),
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
CONSTRAINT folders_parent_name_unique UNIQUE (parent_id, name)
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_folders_parent ON folders(parent_id);
|
|
||||||
|
|
||||||
CREATE TABLE documents (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
filename VARCHAR(255) NOT NULL,
|
|
||||||
original_name VARCHAR(255) NOT NULL,
|
|
||||||
content_type VARCHAR(100),
|
|
||||||
folder_id UUID REFERENCES folders(id) ON DELETE SET NULL,
|
|
||||||
current_version INTEGER NOT NULL,
|
|
||||||
uploaded_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
deleted_at TIMESTAMPTZ,
|
|
||||||
metadata JSONB NOT NULL DEFAULT '{}'::jsonb
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_documents_folder ON documents(folder_id);
|
|
||||||
CREATE INDEX idx_documents_deleted_at ON documents(deleted_at);
|
|
||||||
|
|
||||||
CREATE TABLE document_versions (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
|
|
||||||
version_number INTEGER NOT NULL,
|
|
||||||
s3_key VARCHAR(500) NOT NULL,
|
|
||||||
size_bytes BIGINT NOT NULL,
|
|
||||||
checksum VARCHAR(64) NOT NULL,
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
operations_summary JSONB NOT NULL DEFAULT '{}'::jsonb,
|
|
||||||
CONSTRAINT document_versions_unique_version UNIQUE (document_id, version_number)
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_document_versions_document ON document_versions(document_id);
|
|
||||||
|
|
||||||
CREATE TABLE tags (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
label VARCHAR(100) NOT NULL UNIQUE,
|
|
||||||
color VARCHAR(7),
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE TABLE document_tags (
|
|
||||||
document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
|
|
||||||
tag_id UUID NOT NULL REFERENCES tags(id) ON DELETE CASCADE,
|
|
||||||
assigned_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
assigned_by UUID REFERENCES users(id),
|
|
||||||
PRIMARY KEY (document_id, tag_id)
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_document_tags_tag ON document_tags(tag_id);
|
|
||||||
|
|
||||||
INSERT INTO users (id, username, password_hash, role)
|
|
||||||
VALUES (
|
|
||||||
gen_random_uuid(),
|
|
||||||
'admin',
|
|
||||||
'$argon2id$v=19$m=19456,t=2,p=1$UMkfsNut028fmZupy9JoQg$/YFvGQoEZ2hhMiDCyv68ZROF97GcwAxxRwRgwSbpX5U',
|
|
||||||
'admin'
|
|
||||||
);
|
|
||||||
@@ -1,5 +0,0 @@
|
|||||||
DROP TRIGGER IF EXISTS trg_jobs_updated_at ON jobs;
|
|
||||||
DROP FUNCTION IF EXISTS touch_jobs_updated_at;
|
|
||||||
DROP INDEX IF EXISTS idx_jobs_job_type;
|
|
||||||
DROP INDEX IF EXISTS idx_jobs_status_run_after;
|
|
||||||
DROP TABLE IF EXISTS jobs;
|
|
||||||
@@ -1,28 +0,0 @@
|
|||||||
CREATE TABLE jobs (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
job_type TEXT NOT NULL,
|
|
||||||
payload JSONB NOT NULL,
|
|
||||||
status TEXT NOT NULL DEFAULT 'queued',
|
|
||||||
attempts INTEGER NOT NULL DEFAULT 0,
|
|
||||||
run_after TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
last_error TEXT,
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
CONSTRAINT jobs_status_check CHECK (status IN ('queued', 'processing', 'succeeded', 'failed'))
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_jobs_status_run_after ON jobs (status, run_after);
|
|
||||||
CREATE INDEX idx_jobs_job_type ON jobs (job_type);
|
|
||||||
|
|
||||||
CREATE OR REPLACE FUNCTION touch_jobs_updated_at()
|
|
||||||
RETURNS TRIGGER AS $$
|
|
||||||
BEGIN
|
|
||||||
NEW.updated_at = now();
|
|
||||||
RETURN NEW;
|
|
||||||
END;
|
|
||||||
$$ LANGUAGE plpgsql;
|
|
||||||
|
|
||||||
CREATE TRIGGER trg_jobs_updated_at
|
|
||||||
BEFORE UPDATE ON jobs
|
|
||||||
FOR EACH ROW
|
|
||||||
EXECUTE FUNCTION touch_jobs_updated_at();
|
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS idx_document_assets_type;
|
|
||||||
DROP INDEX IF EXISTS idx_document_assets_version;
|
|
||||||
DROP TABLE IF EXISTS document_assets;
|
|
||||||
@@ -1,15 +0,0 @@
|
|||||||
CREATE TABLE document_assets (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
document_version_id UUID NOT NULL REFERENCES document_versions(id) ON DELETE CASCADE,
|
|
||||||
asset_type TEXT NOT NULL,
|
|
||||||
s3_key TEXT NOT NULL,
|
|
||||||
mime_type TEXT NOT NULL,
|
|
||||||
width INTEGER,
|
|
||||||
height INTEGER,
|
|
||||||
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
CONSTRAINT document_assets_unique UNIQUE (document_version_id, asset_type)
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_document_assets_version ON document_assets(document_version_id);
|
|
||||||
CREATE INDEX idx_document_assets_type ON document_assets(asset_type);
|
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS folders_parent_name_unique_idx;
|
|
||||||
|
|
||||||
ALTER TABLE folders
|
|
||||||
ADD CONSTRAINT folders_parent_name_unique UNIQUE (parent_id, name);
|
|
||||||
@@ -1,5 +0,0 @@
|
|||||||
ALTER TABLE folders
|
|
||||||
DROP CONSTRAINT IF EXISTS folders_parent_name_unique;
|
|
||||||
|
|
||||||
CREATE UNIQUE INDEX folders_parent_name_unique_idx
|
|
||||||
ON folders (COALESCE(parent_id, '00000000-0000-0000-0000-000000000000'::uuid), name);
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE documents
|
|
||||||
DROP COLUMN issued_at;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE documents
|
|
||||||
ADD COLUMN issued_at TIMESTAMPTZ;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE documents
|
|
||||||
DROP COLUMN name;
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
ALTER TABLE documents
|
|
||||||
ADD COLUMN name VARCHAR(255);
|
|
||||||
|
|
||||||
UPDATE documents
|
|
||||||
SET name = CASE
|
|
||||||
WHEN filename ~ '\\.[^./]+$' THEN regexp_replace(filename, '\\.[^./]+$', '')
|
|
||||||
ELSE filename
|
|
||||||
END;
|
|
||||||
|
|
||||||
ALTER TABLE documents
|
|
||||||
ALTER COLUMN name SET NOT NULL;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE documents
|
|
||||||
RENAME COLUMN title TO name;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE documents
|
|
||||||
RENAME COLUMN name TO title;
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
DROP TABLE refresh_tokens;
|
|
||||||
@@ -1,13 +0,0 @@
|
|||||||
CREATE TABLE refresh_tokens (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
|
||||||
token_hash TEXT NOT NULL,
|
|
||||||
issued_at TIMESTAMPTZ NOT NULL,
|
|
||||||
expires_at TIMESTAMPTZ NOT NULL,
|
|
||||||
revoked_at TIMESTAMPTZ,
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
||||||
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_refresh_tokens_user_id ON refresh_tokens(user_id);
|
|
||||||
CREATE INDEX idx_refresh_tokens_token_hash ON refresh_tokens(token_hash);
|
|
||||||
@@ -1,13 +0,0 @@
|
|||||||
-- Restore width/height columns and repopulate from metadata where available.
|
|
||||||
|
|
||||||
ALTER TABLE document_assets
|
|
||||||
ADD COLUMN width INTEGER,
|
|
||||||
ADD COLUMN height INTEGER;
|
|
||||||
|
|
||||||
UPDATE document_assets
|
|
||||||
SET width = (metadata->>'width')::INTEGER
|
|
||||||
WHERE metadata ? 'width';
|
|
||||||
|
|
||||||
UPDATE document_assets
|
|
||||||
SET height = (metadata->>'height')::INTEGER
|
|
||||||
WHERE metadata ? 'height';
|
|
||||||
@@ -1,15 +0,0 @@
|
|||||||
-- Backfill existing width/height values into metadata then drop the columns.
|
|
||||||
|
|
||||||
UPDATE document_assets
|
|
||||||
SET metadata = metadata || jsonb_build_object('width', width)
|
|
||||||
WHERE width IS NOT NULL
|
|
||||||
AND NOT (metadata ? 'width');
|
|
||||||
|
|
||||||
UPDATE document_assets
|
|
||||||
SET metadata = metadata || jsonb_build_object('height', height)
|
|
||||||
WHERE height IS NOT NULL
|
|
||||||
AND NOT (metadata ? 'height');
|
|
||||||
|
|
||||||
ALTER TABLE document_assets
|
|
||||||
DROP COLUMN width,
|
|
||||||
DROP COLUMN height;
|
|
||||||
@@ -1,17 +0,0 @@
|
|||||||
ALTER TABLE documents ADD COLUMN current_version INT4;
|
|
||||||
|
|
||||||
UPDATE documents AS d
|
|
||||||
SET current_version = dv.version_number
|
|
||||||
FROM document_versions AS dv
|
|
||||||
WHERE dv.id = d.current_version_id;
|
|
||||||
|
|
||||||
ALTER TABLE documents
|
|
||||||
ALTER COLUMN current_version SET NOT NULL;
|
|
||||||
|
|
||||||
DROP INDEX IF EXISTS idx_documents_current_version_id;
|
|
||||||
|
|
||||||
ALTER TABLE documents
|
|
||||||
DROP CONSTRAINT IF EXISTS documents_current_version_fk;
|
|
||||||
|
|
||||||
ALTER TABLE documents
|
|
||||||
DROP COLUMN current_version_id;
|
|
||||||
@@ -1,21 +0,0 @@
|
|||||||
ALTER TABLE documents ADD COLUMN current_version_id UUID;
|
|
||||||
|
|
||||||
UPDATE documents AS d
|
|
||||||
SET current_version_id = dv.id
|
|
||||||
FROM document_versions AS dv
|
|
||||||
WHERE dv.document_id = d.id
|
|
||||||
AND dv.version_number = d.current_version;
|
|
||||||
|
|
||||||
ALTER TABLE documents
|
|
||||||
ALTER COLUMN current_version_id SET NOT NULL;
|
|
||||||
|
|
||||||
ALTER TABLE documents
|
|
||||||
ADD CONSTRAINT documents_current_version_fk
|
|
||||||
FOREIGN KEY (current_version_id)
|
|
||||||
REFERENCES document_versions(id)
|
|
||||||
DEFERRABLE INITIALLY DEFERRED;
|
|
||||||
|
|
||||||
CREATE INDEX idx_documents_current_version_id
|
|
||||||
ON documents(current_version_id);
|
|
||||||
|
|
||||||
ALTER TABLE documents DROP COLUMN current_version;
|
|
||||||
@@ -1,5 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS idx_document_correspondents_role;
|
|
||||||
DROP INDEX IF EXISTS idx_document_correspondents_correspondent;
|
|
||||||
DROP INDEX IF EXISTS idx_document_correspondents_document;
|
|
||||||
DROP TABLE IF EXISTS document_correspondents;
|
|
||||||
DROP TABLE IF EXISTS correspondents;
|
|
||||||
@@ -1,27 +0,0 @@
|
|||||||
CREATE TABLE correspondents (
|
|
||||||
id UUID PRIMARY KEY,
|
|
||||||
name VARCHAR(255) NOT NULL,
|
|
||||||
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
|
||||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
CONSTRAINT correspondents_name_unique UNIQUE (name)
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE TABLE document_correspondents (
|
|
||||||
document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
|
|
||||||
correspondent_id UUID NOT NULL REFERENCES correspondents(id) ON DELETE CASCADE,
|
|
||||||
role VARCHAR(32) NOT NULL,
|
|
||||||
assigned_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
||||||
assigned_by UUID REFERENCES users(id),
|
|
||||||
PRIMARY KEY (document_id, correspondent_id, role),
|
|
||||||
CONSTRAINT document_correspondents_role_check CHECK (role IN ('sender', 'receiver', 'other'))
|
|
||||||
);
|
|
||||||
|
|
||||||
CREATE INDEX idx_document_correspondents_document
|
|
||||||
ON document_correspondents(document_id);
|
|
||||||
|
|
||||||
CREATE INDEX idx_document_correspondents_correspondent
|
|
||||||
ON document_correspondents(correspondent_id);
|
|
||||||
|
|
||||||
CREATE INDEX idx_document_correspondents_role
|
|
||||||
ON document_correspondents(role);
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS idx_documents_folder_filename;
|
|
||||||
DROP INDEX IF EXISTS idx_documents_folder_title;
|
|
||||||
@@ -1,13 +0,0 @@
|
|||||||
CREATE INDEX idx_documents_folder_title
|
|
||||||
ON documents (
|
|
||||||
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
|
||||||
title
|
|
||||||
)
|
|
||||||
WHERE deleted_at IS NULL;
|
|
||||||
|
|
||||||
CREATE INDEX idx_documents_folder_filename
|
|
||||||
ON documents (
|
|
||||||
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
|
||||||
filename
|
|
||||||
)
|
|
||||||
WHERE deleted_at IS NULL;
|
|
||||||
@@ -1,16 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS documents_unique_folder_filename;
|
|
||||||
DROP INDEX IF EXISTS idx_documents_folder_title;
|
|
||||||
|
|
||||||
CREATE INDEX idx_documents_folder_title
|
|
||||||
ON documents (
|
|
||||||
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
|
||||||
title
|
|
||||||
)
|
|
||||||
WHERE deleted_at IS NULL;
|
|
||||||
|
|
||||||
CREATE INDEX idx_documents_folder_filename
|
|
||||||
ON documents (
|
|
||||||
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
|
||||||
filename
|
|
||||||
)
|
|
||||||
WHERE deleted_at IS NULL;
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
DROP INDEX IF EXISTS idx_documents_folder_title;
|
|
||||||
DROP INDEX IF EXISTS idx_documents_folder_filename;
|
|
||||||
DROP INDEX IF EXISTS documents_unique_folder_title;
|
|
||||||
DROP INDEX IF EXISTS documents_unique_folder_filename;
|
|
||||||
|
|
||||||
CREATE INDEX idx_documents_folder_title
|
|
||||||
ON documents (
|
|
||||||
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
|
||||||
title
|
|
||||||
)
|
|
||||||
WHERE deleted_at IS NULL;
|
|
||||||
|
|
||||||
CREATE UNIQUE INDEX documents_unique_folder_filename
|
|
||||||
ON documents (
|
|
||||||
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
|
||||||
filename
|
|
||||||
)
|
|
||||||
WHERE deleted_at IS NULL;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE document_versions
|
|
||||||
DROP COLUMN metadata;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE document_versions
|
|
||||||
ADD COLUMN metadata JSONB NOT NULL DEFAULT '{}'::jsonb;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE folders
|
|
||||||
ADD COLUMN path_cache VARCHAR(1000);
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
ALTER TABLE folders
|
|
||||||
DROP COLUMN path_cache;
|
|
||||||
@@ -0,0 +1,21 @@
|
|||||||
|
DROP TRIGGER IF EXISTS trg_jobs_updated_at ON jobs;
|
||||||
|
DROP FUNCTION IF EXISTS touch_jobs_updated_at();
|
||||||
|
|
||||||
|
ALTER TABLE documents DROP CONSTRAINT IF EXISTS documents_current_version_fk;
|
||||||
|
|
||||||
|
DROP TABLE IF EXISTS document_asset_objects;
|
||||||
|
DROP TABLE IF EXISTS document_assets;
|
||||||
|
DROP TABLE IF EXISTS document_versions;
|
||||||
|
DROP TABLE IF EXISTS document_tags;
|
||||||
|
DROP TABLE IF EXISTS document_correspondents;
|
||||||
|
DROP TABLE IF EXISTS correspondents;
|
||||||
|
DROP TABLE IF EXISTS documents;
|
||||||
|
DROP TABLE IF EXISTS folders;
|
||||||
|
DROP TABLE IF EXISTS tags;
|
||||||
|
DROP TABLE IF EXISTS jobs;
|
||||||
|
DROP TABLE IF EXISTS refresh_tokens;
|
||||||
|
DROP TABLE IF EXISTS user_memberships;
|
||||||
|
DROP TABLE IF EXISTS users;
|
||||||
|
DROP TABLE IF EXISTS tenants;
|
||||||
|
|
||||||
|
DROP EXTENSION IF EXISTS "pgcrypto";
|
||||||
@@ -0,0 +1,244 @@
|
|||||||
|
CREATE EXTENSION IF NOT EXISTS "pgcrypto";
|
||||||
|
|
||||||
|
CREATE TABLE tenants (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
slug TEXT NOT NULL UNIQUE,
|
||||||
|
storage_root TEXT,
|
||||||
|
quickwit_index TEXT,
|
||||||
|
status TEXT NOT NULL DEFAULT 'active',
|
||||||
|
config JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE UNIQUE INDEX tenants_storage_root_unique
|
||||||
|
ON tenants (storage_root)
|
||||||
|
WHERE storage_root IS NOT NULL;
|
||||||
|
|
||||||
|
CREATE UNIQUE INDEX tenants_quickwit_index_unique
|
||||||
|
ON tenants (quickwit_index)
|
||||||
|
WHERE quickwit_index IS NOT NULL;
|
||||||
|
|
||||||
|
CREATE TABLE users (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
username VARCHAR(100) NOT NULL UNIQUE,
|
||||||
|
password_hash VARCHAR(255) NOT NULL,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE user_memberships (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
|
role TEXT NOT NULL,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
UNIQUE (user_id, tenant_id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX user_memberships_tenant_id_idx ON user_memberships (tenant_id);
|
||||||
|
CREATE INDEX user_memberships_user_id_idx ON user_memberships (user_id);
|
||||||
|
|
||||||
|
CREATE TABLE folders (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
name VARCHAR(255) NOT NULL,
|
||||||
|
parent_id UUID REFERENCES folders(id) ON DELETE SET NULL,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_folders_parent ON folders(parent_id);
|
||||||
|
CREATE INDEX folders_tenant_id_idx ON folders(tenant_id);
|
||||||
|
CREATE UNIQUE INDEX folders_parent_name_unique_idx
|
||||||
|
ON folders (COALESCE(parent_id, '00000000-0000-0000-0000-000000000000'::uuid), name);
|
||||||
|
|
||||||
|
CREATE TABLE documents (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
filename VARCHAR(255) NOT NULL,
|
||||||
|
original_name VARCHAR(255) NOT NULL,
|
||||||
|
content_type VARCHAR(100),
|
||||||
|
folder_id UUID REFERENCES folders(id) ON DELETE SET NULL,
|
||||||
|
uploaded_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
deleted_at TIMESTAMPTZ,
|
||||||
|
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
issued_at TIMESTAMPTZ,
|
||||||
|
title VARCHAR(255) NOT NULL,
|
||||||
|
current_version_id UUID NOT NULL,
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_documents_folder ON documents (folder_id);
|
||||||
|
CREATE INDEX idx_documents_deleted_at ON documents (deleted_at);
|
||||||
|
CREATE INDEX documents_tenant_id_idx ON documents (tenant_id);
|
||||||
|
CREATE INDEX idx_documents_current_version_id ON documents (current_version_id);
|
||||||
|
|
||||||
|
CREATE INDEX idx_documents_folder_title
|
||||||
|
ON documents (
|
||||||
|
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
||||||
|
title
|
||||||
|
)
|
||||||
|
WHERE deleted_at IS NULL;
|
||||||
|
|
||||||
|
CREATE UNIQUE INDEX documents_unique_folder_filename
|
||||||
|
ON documents (
|
||||||
|
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
||||||
|
filename
|
||||||
|
)
|
||||||
|
WHERE deleted_at IS NULL;
|
||||||
|
|
||||||
|
CREATE TABLE document_versions (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
|
||||||
|
version_number INT NOT NULL,
|
||||||
|
s3_key VARCHAR(500) NOT NULL,
|
||||||
|
size_bytes BIGINT NOT NULL,
|
||||||
|
checksum VARCHAR(64) NOT NULL,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
operations_summary JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
CONSTRAINT document_versions_unique_version UNIQUE (document_id, version_number)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_document_versions_document ON document_versions (document_id);
|
||||||
|
CREATE INDEX document_versions_tenant_id_idx ON document_versions (tenant_id);
|
||||||
|
|
||||||
|
ALTER TABLE documents
|
||||||
|
ADD CONSTRAINT documents_current_version_fk
|
||||||
|
FOREIGN KEY (current_version_id)
|
||||||
|
REFERENCES document_versions(id)
|
||||||
|
DEFERRABLE INITIALLY DEFERRED;
|
||||||
|
|
||||||
|
CREATE TABLE tags (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
label VARCHAR(100) NOT NULL UNIQUE,
|
||||||
|
color VARCHAR(7),
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX tags_tenant_id_idx ON tags (tenant_id);
|
||||||
|
|
||||||
|
CREATE TABLE document_tags (
|
||||||
|
document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
|
||||||
|
tag_id UUID NOT NULL REFERENCES tags(id) ON DELETE CASCADE,
|
||||||
|
assigned_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
assigned_by UUID REFERENCES users(id),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
PRIMARY KEY (document_id, tag_id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_document_tags_tag ON document_tags (tag_id);
|
||||||
|
CREATE INDEX document_tags_tenant_id_idx ON document_tags (tenant_id);
|
||||||
|
|
||||||
|
CREATE TABLE correspondents (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
name VARCHAR(255) NOT NULL,
|
||||||
|
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
CONSTRAINT correspondents_name_unique UNIQUE (name)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX correspondents_tenant_id_idx ON correspondents (tenant_id);
|
||||||
|
|
||||||
|
CREATE TABLE document_correspondents (
|
||||||
|
document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
|
||||||
|
correspondent_id UUID NOT NULL REFERENCES correspondents(id) ON DELETE CASCADE,
|
||||||
|
role VARCHAR(32) NOT NULL,
|
||||||
|
assigned_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
assigned_by UUID REFERENCES users(id),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
PRIMARY KEY (document_id, correspondent_id, role),
|
||||||
|
CONSTRAINT document_correspondents_role_check CHECK (role IN ('sender', 'receiver', 'other'))
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_document_correspondents_document ON document_correspondents (document_id);
|
||||||
|
CREATE INDEX idx_document_correspondents_correspondent ON document_correspondents (correspondent_id);
|
||||||
|
CREATE INDEX idx_document_correspondents_role ON document_correspondents (role);
|
||||||
|
CREATE INDEX document_correspondents_tenant_id_idx ON document_correspondents (tenant_id);
|
||||||
|
|
||||||
|
CREATE TABLE document_assets (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
document_version_id UUID NOT NULL REFERENCES document_versions(id) ON DELETE CASCADE,
|
||||||
|
asset_type TEXT NOT NULL,
|
||||||
|
mime_type TEXT NOT NULL,
|
||||||
|
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
cardinality INTEGER,
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
CONSTRAINT document_assets_unique UNIQUE (document_version_id, asset_type),
|
||||||
|
CONSTRAINT document_assets_cardinality_positive CHECK (cardinality IS NULL OR cardinality >= 1)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_document_assets_version ON document_assets (document_version_id);
|
||||||
|
CREATE INDEX idx_document_assets_type ON document_assets (asset_type);
|
||||||
|
CREATE INDEX document_assets_tenant_id_idx ON document_assets (tenant_id);
|
||||||
|
|
||||||
|
CREATE TABLE document_asset_objects (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
asset_id UUID NOT NULL REFERENCES document_assets(id) ON DELETE CASCADE,
|
||||||
|
ordinal INTEGER NOT NULL,
|
||||||
|
s3_key TEXT NOT NULL,
|
||||||
|
metadata JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
CONSTRAINT document_asset_objects_ordinal_positive CHECK (ordinal >= 1),
|
||||||
|
CONSTRAINT document_asset_objects_asset_ordinal_unique UNIQUE (asset_id, ordinal)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_document_asset_objects_asset_ordinal
|
||||||
|
ON document_asset_objects (asset_id, ordinal);
|
||||||
|
|
||||||
|
CREATE INDEX document_asset_objects_tenant_id_idx
|
||||||
|
ON document_asset_objects (tenant_id);
|
||||||
|
|
||||||
|
CREATE TABLE jobs (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
job_type TEXT NOT NULL,
|
||||||
|
payload JSONB NOT NULL,
|
||||||
|
status TEXT NOT NULL DEFAULT 'queued',
|
||||||
|
attempts INTEGER NOT NULL DEFAULT 0,
|
||||||
|
run_after TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
last_error TEXT,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id),
|
||||||
|
CONSTRAINT jobs_status_check CHECK (status IN ('queued', 'processing', 'succeeded', 'failed'))
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_jobs_status_run_after ON jobs (status, run_after);
|
||||||
|
CREATE INDEX idx_jobs_job_type ON jobs (job_type);
|
||||||
|
CREATE INDEX jobs_tenant_id_idx ON jobs (tenant_id);
|
||||||
|
|
||||||
|
CREATE OR REPLACE FUNCTION touch_jobs_updated_at()
|
||||||
|
RETURNS TRIGGER AS $$
|
||||||
|
BEGIN
|
||||||
|
NEW.updated_at = NOW();
|
||||||
|
RETURN NEW;
|
||||||
|
END;
|
||||||
|
$$ LANGUAGE plpgsql;
|
||||||
|
|
||||||
|
CREATE TRIGGER trg_jobs_updated_at
|
||||||
|
BEFORE UPDATE ON jobs
|
||||||
|
FOR EACH ROW
|
||||||
|
EXECUTE FUNCTION touch_jobs_updated_at();
|
||||||
|
|
||||||
|
CREATE TABLE refresh_tokens (
|
||||||
|
id UUID PRIMARY KEY,
|
||||||
|
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||||||
|
token_hash TEXT NOT NULL,
|
||||||
|
issued_at TIMESTAMPTZ NOT NULL,
|
||||||
|
expires_at TIMESTAMPTZ NOT NULL,
|
||||||
|
revoked_at TIMESTAMPTZ,
|
||||||
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||||
|
tenant_id UUID NOT NULL REFERENCES tenants(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX idx_refresh_tokens_user_id ON refresh_tokens (user_id);
|
||||||
|
CREATE INDEX idx_refresh_tokens_token_hash ON refresh_tokens (token_hash);
|
||||||
|
CREATE INDEX refresh_tokens_tenant_id_idx ON refresh_tokens (tenant_id);
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
DROP INDEX IF EXISTS folders_tenant_parent_name_unique_idx;
|
||||||
|
CREATE UNIQUE INDEX folders_parent_name_unique_idx
|
||||||
|
ON folders (
|
||||||
|
COALESCE(parent_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
||||||
|
name
|
||||||
|
);
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
DROP INDEX IF EXISTS folders_parent_name_unique_idx;
|
||||||
|
CREATE UNIQUE INDEX folders_tenant_parent_name_unique_idx
|
||||||
|
ON folders (
|
||||||
|
tenant_id,
|
||||||
|
COALESCE(parent_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
||||||
|
name
|
||||||
|
);
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
-- Revert correspondent uniqueness to global name
|
||||||
|
DROP INDEX IF EXISTS correspondents_tenant_name_unique;
|
||||||
|
ALTER TABLE correspondents ADD CONSTRAINT correspondents_name_unique UNIQUE (name);
|
||||||
|
|
||||||
|
-- Revert tag uniqueness to global label
|
||||||
|
DROP INDEX IF EXISTS tags_tenant_label_unique;
|
||||||
|
ALTER TABLE tags ADD CONSTRAINT tags_label_key UNIQUE (label);
|
||||||
|
|
||||||
|
-- Revert document filename uniqueness to global folder scope
|
||||||
|
DROP INDEX IF EXISTS documents_tenant_folder_filename_unique;
|
||||||
|
CREATE UNIQUE INDEX documents_unique_folder_filename
|
||||||
|
ON documents (
|
||||||
|
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
||||||
|
filename
|
||||||
|
)
|
||||||
|
WHERE deleted_at IS NULL;
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
-- Ensure document filenames are unique per tenant + folder
|
||||||
|
DROP INDEX IF EXISTS documents_tenant_folder_filename_unique;
|
||||||
|
DROP INDEX IF EXISTS documents_unique_folder_filename;
|
||||||
|
CREATE UNIQUE INDEX documents_tenant_folder_filename_unique
|
||||||
|
ON documents (
|
||||||
|
tenant_id,
|
||||||
|
COALESCE(folder_id, '00000000-0000-0000-0000-000000000000'::uuid),
|
||||||
|
filename
|
||||||
|
)
|
||||||
|
WHERE deleted_at IS NULL;
|
||||||
|
|
||||||
|
-- Ensure tag labels are unique per tenant
|
||||||
|
ALTER TABLE tags DROP CONSTRAINT IF EXISTS tags_label_key;
|
||||||
|
DROP INDEX IF EXISTS tags_tenant_label_unique;
|
||||||
|
CREATE UNIQUE INDEX tags_tenant_label_unique ON tags (tenant_id, label);
|
||||||
|
|
||||||
|
-- Ensure correspondent names are unique per tenant
|
||||||
|
ALTER TABLE correspondents DROP CONSTRAINT IF EXISTS correspondents_name_unique;
|
||||||
|
DROP INDEX IF EXISTS correspondents_tenant_name_unique;
|
||||||
|
CREATE UNIQUE INDEX correspondents_tenant_name_unique ON correspondents (tenant_id, name);
|
||||||
+46
-4
@@ -15,6 +15,8 @@ pub struct JwtService {
|
|||||||
expiry: Duration,
|
expiry: Duration,
|
||||||
download_audience: String,
|
download_audience: String,
|
||||||
download_expiry: Duration,
|
download_expiry: Duration,
|
||||||
|
selector_audience: String,
|
||||||
|
selector_expiry: Duration,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl JwtService {
|
impl JwtService {
|
||||||
@@ -27,16 +29,18 @@ impl JwtService {
|
|||||||
expiry: Duration::minutes(config.jwt_expiry_minutes),
|
expiry: Duration::minutes(config.jwt_expiry_minutes),
|
||||||
download_audience: config.download_token_audience.clone(),
|
download_audience: config.download_token_audience.clone(),
|
||||||
download_expiry: Duration::minutes(config.download_token_expiry_minutes),
|
download_expiry: Duration::minutes(config.download_token_expiry_minutes),
|
||||||
|
selector_audience: format!("{}:tenant-selector", config.jwt_audience),
|
||||||
|
selector_expiry: Duration::minutes(15),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn generate_token(&self, user_id: Uuid, username: &str, role: &str) -> Result<String> {
|
pub fn generate_token(&self, user_id: Uuid, tenant_id: Uuid, username: &str) -> Result<String> {
|
||||||
let now = Utc::now();
|
let now = Utc::now();
|
||||||
let exp = now + self.expiry;
|
let exp = now + self.expiry;
|
||||||
let claims = Claims {
|
let claims = Claims {
|
||||||
sub: user_id,
|
sub: user_id,
|
||||||
|
tenant_id,
|
||||||
username: username.to_owned(),
|
username: username.to_owned(),
|
||||||
role: role.to_owned(),
|
|
||||||
iss: self.issuer.clone(),
|
iss: self.issuer.clone(),
|
||||||
aud: self.audience.clone(),
|
aud: self.audience.clone(),
|
||||||
iat: now.timestamp() as usize,
|
iat: now.timestamp() as usize,
|
||||||
@@ -54,12 +58,18 @@ impl JwtService {
|
|||||||
Ok(data.claims)
|
Ok(data.claims)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn generate_download_token(&self, document_id: Uuid, user_id: Uuid) -> Result<String> {
|
pub fn generate_download_token(
|
||||||
|
&self,
|
||||||
|
document_id: Uuid,
|
||||||
|
user_id: Uuid,
|
||||||
|
tenant_id: Uuid,
|
||||||
|
) -> Result<String> {
|
||||||
let now = Utc::now();
|
let now = Utc::now();
|
||||||
let exp = now + self.download_expiry;
|
let exp = now + self.download_expiry;
|
||||||
let claims = DownloadClaims {
|
let claims = DownloadClaims {
|
||||||
doc_id: document_id,
|
doc_id: document_id,
|
||||||
user_id,
|
user_id,
|
||||||
|
tenant_id,
|
||||||
iss: self.issuer.clone(),
|
iss: self.issuer.clone(),
|
||||||
aud: self.download_audience.clone(),
|
aud: self.download_audience.clone(),
|
||||||
iat: now.timestamp() as usize,
|
iat: now.timestamp() as usize,
|
||||||
@@ -76,13 +86,35 @@ impl JwtService {
|
|||||||
let data = decode::<DownloadClaims>(token, &self.decoding, &validation)?;
|
let data = decode::<DownloadClaims>(token, &self.decoding, &validation)?;
|
||||||
Ok(data.claims)
|
Ok(data.claims)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn generate_tenant_selector_token(&self, user_id: Uuid) -> Result<String> {
|
||||||
|
let now = Utc::now();
|
||||||
|
let exp = now + self.selector_expiry;
|
||||||
|
let claims = TenantSelectionClaims {
|
||||||
|
sub: user_id,
|
||||||
|
iss: self.issuer.clone(),
|
||||||
|
aud: self.selector_audience.clone(),
|
||||||
|
iat: now.timestamp() as usize,
|
||||||
|
exp: exp.timestamp() as usize,
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(encode(&Header::default(), &claims, &self.encoding)?)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn verify_tenant_selector_token(&self, token: &str) -> Result<TenantSelectionClaims> {
|
||||||
|
let mut validation = Validation::default();
|
||||||
|
validation.set_audience(&[self.selector_audience.clone()]);
|
||||||
|
validation.set_issuer(&[self.issuer.clone()]);
|
||||||
|
let data = decode::<TenantSelectionClaims>(token, &self.decoding, &validation)?;
|
||||||
|
Ok(data.claims)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct Claims {
|
pub struct Claims {
|
||||||
pub sub: Uuid,
|
pub sub: Uuid,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
pub username: String,
|
pub username: String,
|
||||||
pub role: String,
|
|
||||||
pub iss: String,
|
pub iss: String,
|
||||||
pub aud: String,
|
pub aud: String,
|
||||||
pub iat: usize,
|
pub iat: usize,
|
||||||
@@ -93,6 +125,16 @@ pub struct Claims {
|
|||||||
pub struct DownloadClaims {
|
pub struct DownloadClaims {
|
||||||
pub doc_id: Uuid,
|
pub doc_id: Uuid,
|
||||||
pub user_id: Uuid,
|
pub user_id: Uuid,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
pub iss: String,
|
||||||
|
pub aud: String,
|
||||||
|
pub iat: usize,
|
||||||
|
pub exp: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
pub struct TenantSelectionClaims {
|
||||||
|
pub sub: Uuid,
|
||||||
pub iss: String,
|
pub iss: String,
|
||||||
pub aud: String,
|
pub aud: String,
|
||||||
pub iat: usize,
|
pub iat: usize,
|
||||||
|
|||||||
+50
-4
@@ -6,13 +6,17 @@ use axum_extra::headers::{authorization::Bearer, Authorization};
|
|||||||
use axum_extra::TypedHeader;
|
use axum_extra::TypedHeader;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
use crate::{error::AppError, state::AppState};
|
use crate::{
|
||||||
|
error::AppError,
|
||||||
|
state::{AppState, PgPooledConnection},
|
||||||
|
};
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct AuthenticatedUser {
|
pub struct AuthenticatedUser {
|
||||||
pub user_id: uuid::Uuid,
|
pub user_id: uuid::Uuid,
|
||||||
pub username: String,
|
pub username: String,
|
||||||
pub role: String,
|
pub tenant_id: uuid::Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
@@ -23,6 +27,10 @@ impl FromRequestParts<AppState> for AuthenticatedUser {
|
|||||||
parts: &mut Parts,
|
parts: &mut Parts,
|
||||||
state: &AppState,
|
state: &AppState,
|
||||||
) -> Result<Self, Self::Rejection> {
|
) -> Result<Self, Self::Rejection> {
|
||||||
|
if let Some(user) = parts.extensions.get::<AuthenticatedUser>() {
|
||||||
|
return Ok(user.clone());
|
||||||
|
}
|
||||||
|
|
||||||
let TypedHeader(Authorization(bearer)) =
|
let TypedHeader(Authorization(bearer)) =
|
||||||
TypedHeader::<Authorization<Bearer>>::from_request_parts(parts, state)
|
TypedHeader::<Authorization<Bearer>>::from_request_parts(parts, state)
|
||||||
.await
|
.await
|
||||||
@@ -33,10 +41,48 @@ impl FromRequestParts<AppState> for AuthenticatedUser {
|
|||||||
.verify_token(bearer.token())
|
.verify_token(bearer.token())
|
||||||
.map_err(|_| AppError::unauthorized())?;
|
.map_err(|_| AppError::unauthorized())?;
|
||||||
|
|
||||||
Ok(AuthenticatedUser {
|
let user = AuthenticatedUser {
|
||||||
user_id: claims.sub,
|
user_id: claims.sub,
|
||||||
username: claims.username,
|
username: claims.username,
|
||||||
role: claims.role,
|
tenant_id: claims.tenant_id,
|
||||||
|
};
|
||||||
|
|
||||||
|
parts.extensions.insert(user.clone());
|
||||||
|
|
||||||
|
Ok(user)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct TenantScopedConn {
|
||||||
|
pub conn: PgPooledConnection,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
pub user_id: Uuid,
|
||||||
|
pub user: AuthenticatedUser,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TenantScopedConn {
|
||||||
|
pub fn conn(&mut self) -> &mut PgPooledConnection {
|
||||||
|
&mut self.conn
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl FromRequestParts<AppState> for TenantScopedConn {
|
||||||
|
type Rejection = AppError;
|
||||||
|
|
||||||
|
async fn from_request_parts(
|
||||||
|
parts: &mut Parts,
|
||||||
|
state: &AppState,
|
||||||
|
) -> Result<Self, Self::Rejection> {
|
||||||
|
let user = AuthenticatedUser::from_request_parts(parts, state).await?;
|
||||||
|
let tenant_id = user.tenant_id;
|
||||||
|
let conn = state.db_for_tenant(tenant_id)?;
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
conn,
|
||||||
|
tenant_id,
|
||||||
|
user_id: user.user_id,
|
||||||
|
user,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,707 @@
|
|||||||
|
use std::env;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::{anyhow, bail, Context, Result};
|
||||||
|
use argon2::{
|
||||||
|
password_hash::{PasswordHasher, SaltString},
|
||||||
|
Argon2,
|
||||||
|
};
|
||||||
|
use diesel::{dsl::exists, prelude::*, select};
|
||||||
|
use once_cell::sync::Lazy;
|
||||||
|
use reqwest::{Client, Method, StatusCode};
|
||||||
|
use serde_json::json;
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use backend::{
|
||||||
|
config::AppConfig,
|
||||||
|
db::{self, PgPool},
|
||||||
|
jobs::{enqueue_job, JOB_ANALYZE_DOCUMENT},
|
||||||
|
models::{DocumentAsset, DocumentAssetObject, NewUser, NewUserMembership, Tenant, User},
|
||||||
|
s3,
|
||||||
|
schema::{
|
||||||
|
document_asset_objects, document_assets, documents, tenants, user_memberships, users,
|
||||||
|
},
|
||||||
|
storage::{ObjectStorage, S3Storage, TenantStorage},
|
||||||
|
utils::tracing::init_tracing,
|
||||||
|
};
|
||||||
|
|
||||||
|
use rand::rngs::OsRng;
|
||||||
|
|
||||||
|
static QUICKWIT_INDEX_TEMPLATE: Lazy<serde_json::Value> = Lazy::new(|| {
|
||||||
|
json!({
|
||||||
|
"version": "0.8",
|
||||||
|
"index_id": "documents",
|
||||||
|
"doc_mapping": {
|
||||||
|
"tokenizers": [
|
||||||
|
{
|
||||||
|
"name": "substring",
|
||||||
|
"type": "ngram",
|
||||||
|
"min_gram": 2,
|
||||||
|
"max_gram": 20,
|
||||||
|
"prefix_only": false
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"field_mappings": [
|
||||||
|
{ "name": "tenant_id", "type": "text", "stored": true },
|
||||||
|
{ "name": "document_id", "type": "text", "stored": true },
|
||||||
|
{ "name": "version_id", "type": "text", "stored": true },
|
||||||
|
{ "name": "title", "type": "text", "tokenizer": "substring", "stored": true },
|
||||||
|
{ "name": "text", "type": "text", "tokenizer": "substring", "record": "position" }
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"search_settings": {
|
||||||
|
"default_search_fields": ["title", "text"]
|
||||||
|
}
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
enum Command {
|
||||||
|
CreateUser {
|
||||||
|
username: String,
|
||||||
|
password: String,
|
||||||
|
},
|
||||||
|
SetPassword {
|
||||||
|
username: String,
|
||||||
|
password: String,
|
||||||
|
},
|
||||||
|
ListUsers,
|
||||||
|
DeleteUser {
|
||||||
|
username: String,
|
||||||
|
},
|
||||||
|
CreateTenant {
|
||||||
|
slug: String,
|
||||||
|
storage_root: Option<String>,
|
||||||
|
quickwit_index: Option<String>,
|
||||||
|
},
|
||||||
|
DeleteTenant {
|
||||||
|
slug: String,
|
||||||
|
},
|
||||||
|
AddUserToTenant {
|
||||||
|
username: String,
|
||||||
|
slug: String,
|
||||||
|
role: Option<String>,
|
||||||
|
},
|
||||||
|
RemoveUserFromTenant {
|
||||||
|
username: String,
|
||||||
|
slug: String,
|
||||||
|
},
|
||||||
|
ReanalyzeDocuments {
|
||||||
|
slug: String,
|
||||||
|
},
|
||||||
|
ListTenants,
|
||||||
|
DeleteAssets(String),
|
||||||
|
QuickwitCreate(String),
|
||||||
|
QuickwitDelete(String),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Command {
|
||||||
|
fn usage() -> &'static str {
|
||||||
|
"Usage: admin\n\
|
||||||
|
create-user <username> <password>\n\
|
||||||
|
set-password <username> <password>\n\
|
||||||
|
list-users\n\
|
||||||
|
delete-user <username>\n\
|
||||||
|
create-tenant <slug> [storage_root] [quickwit_index]\n\
|
||||||
|
delete-tenant <slug>\n\
|
||||||
|
add-user-to-tenant <username> <slug> [role]\n\
|
||||||
|
remove-user-from-tenant <username> <slug>\n\
|
||||||
|
reanalyze-documents <slug>\n\
|
||||||
|
list-tenants\n\
|
||||||
|
delete-assets <slug>\n\
|
||||||
|
quickwit-create-index <slug>\n\
|
||||||
|
quickwit-delete-index <slug>"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse() -> Result<Self> {
|
||||||
|
let mut args = env::args().skip(1);
|
||||||
|
match args.next().as_deref() {
|
||||||
|
Some("create-user") => Ok(Self::CreateUser {
|
||||||
|
username: args.next().ok_or_else(|| anyhow!("username required"))?,
|
||||||
|
password: args.next().ok_or_else(|| anyhow!("password required"))?,
|
||||||
|
}),
|
||||||
|
Some("set-password") => Ok(Self::SetPassword {
|
||||||
|
username: args.next().ok_or_else(|| anyhow!("username required"))?,
|
||||||
|
password: args.next().ok_or_else(|| anyhow!("password required"))?,
|
||||||
|
}),
|
||||||
|
Some("list-users") => Ok(Self::ListUsers),
|
||||||
|
Some("delete-user") => Ok(Self::DeleteUser {
|
||||||
|
username: args.next().ok_or_else(|| anyhow!("username required"))?,
|
||||||
|
}),
|
||||||
|
Some("create-tenant") => Ok(Self::CreateTenant {
|
||||||
|
slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
storage_root: args.next(),
|
||||||
|
quickwit_index: args.next(),
|
||||||
|
}),
|
||||||
|
Some("delete-tenant") => Ok(Self::DeleteTenant {
|
||||||
|
slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
}),
|
||||||
|
Some("add-user-to-tenant") => Ok(Self::AddUserToTenant {
|
||||||
|
username: args.next().ok_or_else(|| anyhow!("username required"))?,
|
||||||
|
slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
role: args.next(),
|
||||||
|
}),
|
||||||
|
Some("remove-user-from-tenant") => Ok(Self::RemoveUserFromTenant {
|
||||||
|
username: args.next().ok_or_else(|| anyhow!("username required"))?,
|
||||||
|
slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
}),
|
||||||
|
Some("reanalyze-documents") => Ok(Self::ReanalyzeDocuments {
|
||||||
|
slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
}),
|
||||||
|
Some("list-tenants") => Ok(Self::ListTenants),
|
||||||
|
Some("delete-assets") => Ok(Self::DeleteAssets(
|
||||||
|
args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
)),
|
||||||
|
Some("quickwit-create-index") => Ok(Self::QuickwitCreate(
|
||||||
|
args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
)),
|
||||||
|
Some("quickwit-delete-index") => Ok(Self::QuickwitDelete(
|
||||||
|
args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
|
||||||
|
)),
|
||||||
|
_ => Err(anyhow!(Self::usage())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::main]
|
||||||
|
async fn main() -> Result<()> {
|
||||||
|
init_tracing("info");
|
||||||
|
let command = Command::parse()?;
|
||||||
|
let config = AppConfig::load_and_log("admin")?;
|
||||||
|
let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
|
||||||
|
|
||||||
|
match command {
|
||||||
|
Command::CreateUser { username, password } => create_user(&pool, &username, &password)?,
|
||||||
|
Command::SetPassword { username, password } => set_password(&pool, &username, &password)?,
|
||||||
|
Command::ListUsers => list_users(&pool)?,
|
||||||
|
Command::DeleteUser { username } => delete_user(&pool, &username)?,
|
||||||
|
Command::CreateTenant {
|
||||||
|
slug,
|
||||||
|
storage_root,
|
||||||
|
quickwit_index,
|
||||||
|
} => create_tenant(&pool, &slug, storage_root, quickwit_index)?,
|
||||||
|
Command::DeleteTenant { slug } => delete_tenant(&pool, &slug)?,
|
||||||
|
Command::AddUserToTenant {
|
||||||
|
username,
|
||||||
|
slug,
|
||||||
|
role,
|
||||||
|
} => add_user_to_tenant(&pool, &username, &slug, role.as_deref())?,
|
||||||
|
Command::RemoveUserFromTenant { username, slug } => {
|
||||||
|
remove_user_from_tenant(&pool, &username, &slug)?
|
||||||
|
}
|
||||||
|
Command::ReanalyzeDocuments { slug } => reanalyze_documents(&pool, &slug)?,
|
||||||
|
Command::ListTenants => list_tenants(&pool)?,
|
||||||
|
Command::DeleteAssets(slug) => delete_assets_for_tenant(&config, &pool, &slug).await?,
|
||||||
|
Command::QuickwitCreate(slug) => {
|
||||||
|
quickwit_index(&config, &pool, &slug, Method::POST).await?
|
||||||
|
}
|
||||||
|
Command::QuickwitDelete(slug) => {
|
||||||
|
quickwit_index(&config, &pool, &slug, Method::DELETE).await?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn create_user(pool: &PgPool, username: &str, password: &str) -> Result<()> {
|
||||||
|
if username.trim().is_empty() {
|
||||||
|
bail!("username must not be empty");
|
||||||
|
}
|
||||||
|
if password.is_empty() {
|
||||||
|
bail!("password must not be empty");
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
let exists: bool =
|
||||||
|
select(exists(users::table.filter(users::username.eq(username)))).get_result(&mut conn)?;
|
||||||
|
if exists {
|
||||||
|
bail!("user '{}' already exists", username);
|
||||||
|
}
|
||||||
|
|
||||||
|
let password_hash = hash_password(password)?;
|
||||||
|
let new_user = NewUser {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
username: username.to_string(),
|
||||||
|
password_hash,
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(users::table)
|
||||||
|
.values(&new_user)
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
|
println!("created user '{}' (id: {})", username, new_user.id);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_password(pool: &PgPool, username: &str, password: &str) -> Result<()> {
|
||||||
|
if password.is_empty() {
|
||||||
|
bail!("password must not be empty");
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
let password_hash = hash_password(password)?;
|
||||||
|
|
||||||
|
let updated = diesel::update(users::table.filter(users::username.eq(username)))
|
||||||
|
.set(users::password_hash.eq(password_hash))
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
|
if updated == 0 {
|
||||||
|
bail!("user '{}' not found", username);
|
||||||
|
}
|
||||||
|
|
||||||
|
println!("updated password for '{}'", username);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn hash_password(password: &str) -> Result<String> {
|
||||||
|
let salt = SaltString::generate(&mut OsRng);
|
||||||
|
let hash = Argon2::default()
|
||||||
|
.hash_password(password.as_bytes(), &salt)
|
||||||
|
.map_err(|err| anyhow!(err))?;
|
||||||
|
Ok(hash.to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn list_users(pool: &PgPool) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
|
||||||
|
let users_list: Vec<User> = users::table.order(users::username.asc()).load(&mut conn)?;
|
||||||
|
if users_list.is_empty() {
|
||||||
|
println!("No users found.");
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
for user in users_list {
|
||||||
|
let memberships: Vec<(Uuid, String, String)> = user_memberships::table
|
||||||
|
.inner_join(tenants::table)
|
||||||
|
.filter(user_memberships::user_id.eq(user.id))
|
||||||
|
.select((tenants::id, tenants::slug, user_memberships::role))
|
||||||
|
.order((tenants::slug.asc(), user_memberships::role.asc()))
|
||||||
|
.load(&mut conn)?;
|
||||||
|
|
||||||
|
if memberships.is_empty() {
|
||||||
|
println!("{} ({})", user.username, user.id);
|
||||||
|
} else {
|
||||||
|
let details: Vec<String> = memberships
|
||||||
|
.into_iter()
|
||||||
|
.map(|(_, slug, role)| format!("{}: {}", slug, role))
|
||||||
|
.collect();
|
||||||
|
println!("{} ({}) -> {}", user.username, user.id, details.join(", "));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn delete_user(pool: &PgPool, username: &str) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
|
||||||
|
let user: User = users::table
|
||||||
|
.filter(users::username.eq(username))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("user '{}' not found", username))?;
|
||||||
|
|
||||||
|
diesel::delete(user_memberships::table.filter(user_memberships::user_id.eq(user.id)))
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
diesel::delete(users::table.filter(users::id.eq(user.id))).execute(&mut conn)?;
|
||||||
|
|
||||||
|
println!("deleted user '{}'", username);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn create_tenant(
|
||||||
|
pool: &PgPool,
|
||||||
|
slug: &str,
|
||||||
|
storage_root_arg: Option<String>,
|
||||||
|
quickwit_index_arg: Option<String>,
|
||||||
|
) -> Result<()> {
|
||||||
|
if slug.trim().is_empty() {
|
||||||
|
bail!("tenant slug must not be empty");
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
let exists: bool =
|
||||||
|
select(exists(tenants::table.filter(tenants::slug.eq(slug)))).get_result(&mut conn)?;
|
||||||
|
if exists {
|
||||||
|
bail!("tenant '{}' already exists", slug);
|
||||||
|
}
|
||||||
|
|
||||||
|
let id = Uuid::new_v4();
|
||||||
|
let storage_root = storage_root_arg
|
||||||
|
.map(|mut s| {
|
||||||
|
if s.is_empty() {
|
||||||
|
format!("tenants/{}/", id)
|
||||||
|
} else {
|
||||||
|
if !s.ends_with('/') {
|
||||||
|
s.push('/');
|
||||||
|
}
|
||||||
|
s
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.unwrap_or_else(|| format!("tenants/{}/", id));
|
||||||
|
let quickwit_index = quickwit_index_arg.unwrap_or_else(|| format!("documents-{}", id));
|
||||||
|
|
||||||
|
diesel::insert_into(tenants::table)
|
||||||
|
.values((
|
||||||
|
tenants::id.eq(id),
|
||||||
|
tenants::slug.eq(slug),
|
||||||
|
tenants::storage_root.eq(Some(storage_root.clone())),
|
||||||
|
tenants::quickwit_index.eq(Some(quickwit_index.clone())),
|
||||||
|
tenants::status.eq("active"),
|
||||||
|
tenants::config.eq(serde_json::json!({})),
|
||||||
|
))
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"created tenant '{}' with id {}, storage_root '{}', quickwit_index '{}'",
|
||||||
|
slug, id, storage_root, quickwit_index
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn delete_tenant(pool: &PgPool, slug: &str) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
|
||||||
|
let tenant: Tenant = tenants::table
|
||||||
|
.filter(tenants::slug.eq(slug))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("tenant '{}' not found", slug))?;
|
||||||
|
|
||||||
|
let member_exists: bool = select(exists(
|
||||||
|
user_memberships::table.filter(user_memberships::tenant_id.eq(tenant.id)),
|
||||||
|
))
|
||||||
|
.get_result(&mut conn)?;
|
||||||
|
if member_exists {
|
||||||
|
bail!("tenant '{}' still has user memberships", slug);
|
||||||
|
}
|
||||||
|
|
||||||
|
diesel::delete(tenants::table.filter(tenants::id.eq(tenant.id))).execute(&mut conn)?;
|
||||||
|
println!("deleted tenant '{}'", slug);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn add_user_to_tenant(pool: &PgPool, username: &str, slug: &str, role: Option<&str>) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
|
||||||
|
let user: User = users::table
|
||||||
|
.filter(users::username.eq(username))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("user '{}' not found", username))?;
|
||||||
|
|
||||||
|
let tenant: Tenant = tenants::table
|
||||||
|
.filter(tenants::slug.eq(slug))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("tenant '{}' not found", slug))?;
|
||||||
|
|
||||||
|
let membership = NewUserMembership {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
user_id: user.id,
|
||||||
|
tenant_id: tenant.id,
|
||||||
|
role: role.unwrap_or("user").to_string(),
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(user_memberships::table)
|
||||||
|
.values(&membership)
|
||||||
|
.on_conflict((user_memberships::user_id, user_memberships::tenant_id))
|
||||||
|
.do_update()
|
||||||
|
.set(user_memberships::role.eq(&membership.role))
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"added user '{}' to tenant '{}' with role '{}'",
|
||||||
|
username, slug, membership.role
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn remove_user_from_tenant(pool: &PgPool, username: &str, slug: &str) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
|
||||||
|
let user: User = users::table
|
||||||
|
.filter(users::username.eq(username))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("user '{}' not found", username))?;
|
||||||
|
|
||||||
|
let tenant: Tenant = tenants::table
|
||||||
|
.filter(tenants::slug.eq(slug))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("tenant '{}' not found", slug))?;
|
||||||
|
|
||||||
|
let removed = diesel::delete(
|
||||||
|
user_memberships::table
|
||||||
|
.filter(user_memberships::user_id.eq(user.id))
|
||||||
|
.filter(user_memberships::tenant_id.eq(tenant.id)),
|
||||||
|
)
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
|
if removed == 0 {
|
||||||
|
println!("user '{}' was not a member of tenant '{}'", username, slug);
|
||||||
|
} else {
|
||||||
|
println!("removed user '{}' from tenant '{}'", username, slug);
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn reanalyze_documents(pool: &PgPool, slug: &str) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
|
||||||
|
let tenant: Tenant = tenants::table
|
||||||
|
.filter(tenants::slug.eq(slug))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
.ok_or_else(|| anyhow!("tenant '{}' not found", slug))?;
|
||||||
|
|
||||||
|
let targets: Vec<(Uuid, Uuid)> = documents::table
|
||||||
|
.filter(documents::tenant_id.eq(tenant.id))
|
||||||
|
.filter(documents::deleted_at.is_null())
|
||||||
|
.select((documents::id, documents::current_version_id))
|
||||||
|
.load(&mut conn)?;
|
||||||
|
|
||||||
|
if targets.is_empty() {
|
||||||
|
println!("tenant '{}' has no active documents", slug);
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut queued = 0usize;
|
||||||
|
for (document_id, version_id) in targets {
|
||||||
|
enqueue_job(
|
||||||
|
&mut conn,
|
||||||
|
tenant.id,
|
||||||
|
JOB_ANALYZE_DOCUMENT,
|
||||||
|
serde_json::json!({
|
||||||
|
"document_id": document_id,
|
||||||
|
"document_version_id": version_id,
|
||||||
|
"force": true,
|
||||||
|
}),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.map_err(|err| anyhow!("failed to enqueue analyze job: {}", err))?;
|
||||||
|
queued += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"queued {} documents for re-analysis in tenant '{}'",
|
||||||
|
queued, slug
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn list_tenants(pool: &PgPool) -> Result<()> {
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
let tenants: Vec<Tenant> = tenants::table
|
||||||
|
.order(tenants::slug.asc())
|
||||||
|
.load(&mut conn)
|
||||||
|
.context("failed to load tenants")?;
|
||||||
|
|
||||||
|
if tenants.is_empty() {
|
||||||
|
println!("No tenants found.");
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
for tenant in tenants {
|
||||||
|
println!("{} ({})", tenant.slug, tenant.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn delete_assets_for_tenant(
|
||||||
|
config: &AppConfig,
|
||||||
|
pool: &PgPool,
|
||||||
|
tenant_slug: &str,
|
||||||
|
) -> Result<()> {
|
||||||
|
let s3_client = s3::build_client(config).await?;
|
||||||
|
let storage: Arc<dyn ObjectStorage> =
|
||||||
|
Arc::new(S3Storage::new(s3_client, config.s3_bucket.clone()));
|
||||||
|
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
let tenant: Tenant = tenants::table
|
||||||
|
.filter(tenants::slug.eq(tenant_slug))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()
|
||||||
|
.context("failed to load tenant")?
|
||||||
|
.ok_or_else(|| anyhow!("tenant '{}' not found", tenant_slug))?;
|
||||||
|
|
||||||
|
let tenant_storage = TenantStorage::new(Arc::clone(&storage), &tenant)
|
||||||
|
.with_context(|| format!("missing storage root for tenant {}", tenant.slug))?;
|
||||||
|
|
||||||
|
let assets: Vec<DocumentAsset> = document_assets::table
|
||||||
|
.filter(document_assets::tenant_id.eq(tenant.id))
|
||||||
|
.load(&mut conn)
|
||||||
|
.with_context(|| format!("failed to load assets for tenant {}", tenant.slug))?;
|
||||||
|
|
||||||
|
if assets.is_empty() {
|
||||||
|
println!("Tenant {}: no assets", tenant.slug);
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"Tenant {} ({}): deleting {} assets…",
|
||||||
|
tenant.slug,
|
||||||
|
tenant.id,
|
||||||
|
assets.len()
|
||||||
|
);
|
||||||
|
|
||||||
|
let asset_ids: Vec<Uuid> = assets.iter().map(|asset| asset.id).collect();
|
||||||
|
|
||||||
|
let objects: Vec<DocumentAssetObject> = document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant.id))
|
||||||
|
.filter(document_asset_objects::asset_id.eq_any(&asset_ids))
|
||||||
|
.load(&mut conn)
|
||||||
|
.with_context(|| format!("failed to load asset objects for tenant {}", tenant.slug))?;
|
||||||
|
|
||||||
|
for object in &objects {
|
||||||
|
if let Err(err) = tenant_storage.delete_object(&object.s3_key).await {
|
||||||
|
eprintln!(
|
||||||
|
"Failed to delete object {} (tenant {}): {err}",
|
||||||
|
object.s3_key, tenant.slug
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
diesel::delete(
|
||||||
|
document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant.id))
|
||||||
|
.filter(document_asset_objects::asset_id.eq_any(&asset_ids)),
|
||||||
|
)
|
||||||
|
.execute(&mut conn)
|
||||||
|
.with_context(|| format!("failed to remove asset objects for tenant {}", tenant.slug))?;
|
||||||
|
|
||||||
|
diesel::delete(document_assets::table.filter(document_assets::tenant_id.eq(tenant.id)))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.with_context(|| format!("failed to remove asset records for tenant {}", tenant.slug))?;
|
||||||
|
|
||||||
|
println!("Tenant {}: asset records deleted.", tenant.slug);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn quickwit_index(
|
||||||
|
config: &AppConfig,
|
||||||
|
pool: &PgPool,
|
||||||
|
slug: &str,
|
||||||
|
method: Method,
|
||||||
|
) -> Result<()> {
|
||||||
|
let endpoint = config
|
||||||
|
.quickwit_endpoint
|
||||||
|
.as_ref()
|
||||||
|
.ok_or_else(|| anyhow!("quickwit endpoint not configured"))?;
|
||||||
|
|
||||||
|
let mut conn = pool.get().context("failed to get database connection")?;
|
||||||
|
let tenant: Tenant = tenants::table
|
||||||
|
.filter(tenants::slug.eq(slug))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()
|
||||||
|
.context("failed to query tenants")?
|
||||||
|
.ok_or_else(|| anyhow!("tenant '{}' not found", slug))?;
|
||||||
|
|
||||||
|
let client = Client::new();
|
||||||
|
let index_id = format!("documents-{}", tenant.id);
|
||||||
|
let base_endpoint = endpoint.trim_end_matches('/');
|
||||||
|
|
||||||
|
match method {
|
||||||
|
Method::POST => {
|
||||||
|
let payload = render_index_template(&index_id);
|
||||||
|
let response = client
|
||||||
|
.post(format!("{}/api/v1/indexes", base_endpoint))
|
||||||
|
.header("content-type", "application/json")
|
||||||
|
.body(payload)
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.context("failed to send create index request")?;
|
||||||
|
|
||||||
|
match response.status() {
|
||||||
|
status if status.is_success() => {
|
||||||
|
diesel::update(tenants::table.filter(tenants::id.eq(tenant.id)))
|
||||||
|
.set(tenants::quickwit_index.eq(Some(index_id.clone())))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.context("failed to update tenant quickwit_index")?;
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"Tenant '{}' quickwit index set to '{}'.",
|
||||||
|
tenant.slug, index_id
|
||||||
|
);
|
||||||
|
}
|
||||||
|
StatusCode::CONFLICT => {
|
||||||
|
let lookup = client
|
||||||
|
.get(format!("{}/api/v1/indexes/{}", base_endpoint, index_id))
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.context("failed to verify existing quickwit index")?;
|
||||||
|
|
||||||
|
let lookup_status = lookup.status();
|
||||||
|
if !lookup_status.is_success() {
|
||||||
|
let body = lookup.text().await.unwrap_or_default();
|
||||||
|
bail!(
|
||||||
|
"quickwit reported conflict but index lookup failed with status {}: {}",
|
||||||
|
lookup_status,
|
||||||
|
body
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
diesel::update(tenants::table.filter(tenants::id.eq(tenant.id)))
|
||||||
|
.set(tenants::quickwit_index.eq(Some(index_id.clone())))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.context("failed to update tenant quickwit_index")?;
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"Tenant '{}' quickwit index set to '{}'.",
|
||||||
|
tenant.slug, index_id
|
||||||
|
);
|
||||||
|
}
|
||||||
|
status => {
|
||||||
|
let body = response.text().await.unwrap_or_default();
|
||||||
|
bail!(
|
||||||
|
"quickwit create index failed with status {}: {}",
|
||||||
|
status,
|
||||||
|
body
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Method::DELETE => {
|
||||||
|
let response = client
|
||||||
|
.delete(format!("{}/api/v1/indexes/{}", base_endpoint, index_id))
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.context("failed to send delete index request")?;
|
||||||
|
|
||||||
|
match response.status() {
|
||||||
|
status if status.is_success() || status == StatusCode::NOT_FOUND => {
|
||||||
|
diesel::update(tenants::table.filter(tenants::id.eq(tenant.id)))
|
||||||
|
.set(tenants::quickwit_index.eq::<Option<String>>(None))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.context("failed to clear tenant quickwit_index")?;
|
||||||
|
|
||||||
|
println!("Tenant '{}' quickwit index cleared.", tenant.slug);
|
||||||
|
}
|
||||||
|
status => {
|
||||||
|
let body = response.text().await.unwrap_or_default();
|
||||||
|
bail!(
|
||||||
|
"quickwit delete index failed with status {}: {}",
|
||||||
|
status,
|
||||||
|
body
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => unreachable!(),
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn render_index_template(index_id: &str) -> String {
|
||||||
|
let mut template = QUICKWIT_INDEX_TEMPLATE.clone();
|
||||||
|
if let Some(obj) = template.as_object_mut() {
|
||||||
|
obj.insert(
|
||||||
|
"index_id".to_string(),
|
||||||
|
serde_json::Value::String(index_id.to_string()),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
template.to_string()
|
||||||
|
}
|
||||||
@@ -1,75 +0,0 @@
|
|||||||
use std::env;
|
|
||||||
|
|
||||||
use anyhow::{Context, Result};
|
|
||||||
use diesel::prelude::*;
|
|
||||||
|
|
||||||
use backend::{
|
|
||||||
config::AppConfig,
|
|
||||||
db,
|
|
||||||
models::DocumentAsset,
|
|
||||||
s3,
|
|
||||||
schema::document_assets,
|
|
||||||
storage::{ObjectStorage, S3Storage},
|
|
||||||
};
|
|
||||||
|
|
||||||
#[tokio::main]
|
|
||||||
async fn main() -> Result<()> {
|
|
||||||
let mut args = env::args().skip(1);
|
|
||||||
match args.next().as_deref() {
|
|
||||||
Some("delete-assets") => delete_all_assets().await?,
|
|
||||||
Some(cmd) => {
|
|
||||||
eprintln!("Unknown command: {cmd}\nUsage: maintenance delete-assets");
|
|
||||||
std::process::exit(1);
|
|
||||||
}
|
|
||||||
None => {
|
|
||||||
eprintln!("Usage: maintenance delete-assets");
|
|
||||||
std::process::exit(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_all_assets() -> Result<()> {
|
|
||||||
let config = AppConfig::from_env()?;
|
|
||||||
tracing::info!(
|
|
||||||
component = "maintenance",
|
|
||||||
database_url = %config.redacted_database_url(),
|
|
||||||
pool_size = config.database_max_pool_size,
|
|
||||||
s3_bucket = %config.s3_bucket,
|
|
||||||
"loaded backend configuration"
|
|
||||||
);
|
|
||||||
let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
|
|
||||||
|
|
||||||
let s3_client = s3::build_client(&config).await?;
|
|
||||||
let storage = S3Storage::new(s3_client, config.s3_bucket.clone());
|
|
||||||
|
|
||||||
let mut conn = pool.get().context("failed to get database connection")?;
|
|
||||||
|
|
||||||
let assets: Vec<DocumentAsset> = document_assets::table
|
|
||||||
.load(&mut conn)
|
|
||||||
.context("failed to load document assets")?;
|
|
||||||
|
|
||||||
if assets.is_empty() {
|
|
||||||
println!("No assets found.");
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
|
|
||||||
println!("Deleting {} assets…", assets.len());
|
|
||||||
|
|
||||||
for asset in &assets {
|
|
||||||
if let Err(err) = storage.delete_object(&asset.s3_key).await {
|
|
||||||
eprintln!(
|
|
||||||
"Failed to delete object {} from storage: {err}",
|
|
||||||
asset.s3_key
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
diesel::delete(document_assets::table)
|
|
||||||
.execute(&mut conn)
|
|
||||||
.context("failed to remove asset records")?;
|
|
||||||
|
|
||||||
println!("Asset records deleted.");
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
use backend::openapi::ApiDoc;
|
||||||
|
use utoipa::OpenApi;
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
let spec = ApiDoc::openapi();
|
||||||
|
let json = serde_json::to_string_pretty(&spec).expect("serialize openapi");
|
||||||
|
println!("{}", json);
|
||||||
|
}
|
||||||
@@ -1,47 +1,24 @@
|
|||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
use tokio::net::TcpListener;
|
use tokio::net::TcpListener;
|
||||||
use tower::make::Shared;
|
use tower::make::Shared;
|
||||||
use tracing_subscriber::EnvFilter;
|
|
||||||
|
|
||||||
use backend::auth::jwt::JwtService;
|
use backend::{routes::webdav, utils::bootstrap::init_component};
|
||||||
use backend::config::AppConfig;
|
|
||||||
use backend::db;
|
|
||||||
use backend::routes::webdav;
|
|
||||||
use backend::s3::build_client;
|
|
||||||
use backend::state::AppState;
|
|
||||||
use backend::storage::S3Storage;
|
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> anyhow::Result<()> {
|
async fn main() -> anyhow::Result<()> {
|
||||||
dotenv::dotenv().ok();
|
let state = init_component("webdav", None).await?;
|
||||||
init_tracing();
|
let webdav_host = state.config.webdav_host.clone();
|
||||||
|
let webdav_port = state.config.webdav_port;
|
||||||
let config = AppConfig::from_env()?;
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
component = "webdav",
|
component = "webdav",
|
||||||
database_url = %config.redacted_database_url(),
|
webdav_host = %webdav_host,
|
||||||
pool_size = config.database_max_pool_size,
|
webdav_port,
|
||||||
server_host = %config.server_host,
|
"starting webdav server"
|
||||||
server_port = config.server_port,
|
|
||||||
webdav_host = %config.webdav_host,
|
|
||||||
webdav_port = config.webdav_port,
|
|
||||||
quickwit_enabled = config.quickwit_endpoint.is_some(),
|
|
||||||
s3_bucket = %config.s3_bucket,
|
|
||||||
"loaded backend configuration"
|
|
||||||
);
|
);
|
||||||
let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
|
|
||||||
let s3_client = build_client(&config).await?;
|
|
||||||
let storage = Arc::new(S3Storage::new(s3_client, config.s3_bucket.clone()));
|
|
||||||
let jwt = JwtService::from_config(&config)?;
|
|
||||||
|
|
||||||
let state = AppState::new(pool, config, storage, jwt);
|
let listen_addr: SocketAddr = format!("{}:{}", webdav_host, webdav_port).parse()?;
|
||||||
let listen_addr: SocketAddr = {
|
let router = webdav::create_router().with_state(state.as_ref().clone());
|
||||||
let config = state.config.clone();
|
|
||||||
format!("{}:{}", config.webdav_host, config.webdav_port).parse()?
|
|
||||||
};
|
|
||||||
let router = webdav::create_router().with_state(state);
|
|
||||||
|
|
||||||
let listener = TcpListener::bind(listen_addr).await?;
|
let listener = TcpListener::bind(listen_addr).await?;
|
||||||
tracing::info!("listening for WebDAV on {}", listen_addr);
|
tracing::info!("listening for WebDAV on {}", listen_addr);
|
||||||
@@ -49,12 +26,3 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
axum::serve(listener, Shared::new(router)).await?;
|
axum::serve(listener, Shared::new(router)).await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn init_tracing() {
|
|
||||||
let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
|
|
||||||
tracing_subscriber::fmt()
|
|
||||||
.with_env_filter(filter)
|
|
||||||
.with_target(false)
|
|
||||||
.compact()
|
|
||||||
.init();
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,33 +1,13 @@
|
|||||||
use std::{sync::Arc, time::Duration};
|
use std::time::Duration;
|
||||||
|
|
||||||
use tokio::signal;
|
use tokio::signal;
|
||||||
use tracing_subscriber::EnvFilter;
|
|
||||||
|
|
||||||
use backend::{
|
use backend::{default_handlers, utils::bootstrap::init_component, Worker};
|
||||||
auth::jwt::JwtService, config::AppConfig, db, default_handlers, s3::build_client,
|
|
||||||
state::AppState, storage::S3Storage, Worker,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> anyhow::Result<()> {
|
async fn main() -> anyhow::Result<()> {
|
||||||
dotenv::dotenv().ok();
|
let state = init_component("worker", Some(1)).await?;
|
||||||
init_tracing();
|
tracing::info!(component = "worker", "starting worker process");
|
||||||
|
|
||||||
let config = AppConfig::from_env()?;
|
|
||||||
tracing::info!(
|
|
||||||
component = "worker",
|
|
||||||
database_url = %config.redacted_database_url(),
|
|
||||||
pool_size = 1,
|
|
||||||
quickwit_enabled = config.quickwit_endpoint.is_some(),
|
|
||||||
s3_bucket = %config.s3_bucket,
|
|
||||||
"loaded backend configuration"
|
|
||||||
);
|
|
||||||
let pool = db::init_pool_with_size(&config.database_url, 1)?;
|
|
||||||
let s3_client = build_client(&config).await?;
|
|
||||||
let storage = Arc::new(S3Storage::new(s3_client, config.s3_bucket.clone()));
|
|
||||||
let jwt = JwtService::from_config(&config)?;
|
|
||||||
|
|
||||||
let state = Arc::new(AppState::new(pool, config, storage, jwt));
|
|
||||||
let worker = Worker::new(state, default_handlers(), Duration::from_secs(2));
|
let worker = Worker::new(state, default_handlers(), Duration::from_secs(2));
|
||||||
|
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
@@ -39,12 +19,3 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn init_tracing() {
|
|
||||||
let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
|
|
||||||
tracing_subscriber::fmt()
|
|
||||||
.with_env_filter(filter)
|
|
||||||
.with_target(false)
|
|
||||||
.compact()
|
|
||||||
.init();
|
|
||||||
}
|
|
||||||
|
|||||||
+24
-2
@@ -30,9 +30,24 @@ pub struct AppConfig {
|
|||||||
pub s3_bucket: String,
|
pub s3_bucket: String,
|
||||||
pub quickwit_endpoint: Option<String>,
|
pub quickwit_endpoint: Option<String>,
|
||||||
pub quickwit_index: Option<String>,
|
pub quickwit_index: Option<String>,
|
||||||
|
pub default_tenant_slug: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AppConfig {
|
impl AppConfig {
|
||||||
|
pub fn load_and_log(component: &str) -> Result<Self> {
|
||||||
|
dotenv::dotenv().ok();
|
||||||
|
let config = Self::from_env()?;
|
||||||
|
tracing::info!(
|
||||||
|
component,
|
||||||
|
database_url = %config.redacted_database_url(),
|
||||||
|
pool_size = config.database_max_pool_size,
|
||||||
|
quickwit_enabled = config.quickwit_endpoint.is_some(),
|
||||||
|
s3_bucket = %config.s3_bucket,
|
||||||
|
"loaded backend configuration"
|
||||||
|
);
|
||||||
|
Ok(config)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn from_env() -> Result<Self> {
|
pub fn from_env() -> Result<Self> {
|
||||||
let database_url = env::var("DATABASE_URL").context("DATABASE_URL must be set")?;
|
let database_url = env::var("DATABASE_URL").context("DATABASE_URL must be set")?;
|
||||||
let database_max_pool_size = env::var("DATABASE_MAX_POOL_SIZE")
|
let database_max_pool_size = env::var("DATABASE_MAX_POOL_SIZE")
|
||||||
@@ -79,6 +94,8 @@ impl AppConfig {
|
|||||||
let s3_bucket = env::var("S3_BUCKET").context("S3_BUCKET must be set")?;
|
let s3_bucket = env::var("S3_BUCKET").context("S3_BUCKET must be set")?;
|
||||||
let quickwit_endpoint = env::var("QUICKWIT_ENDPOINT").ok();
|
let quickwit_endpoint = env::var("QUICKWIT_ENDPOINT").ok();
|
||||||
let quickwit_index = env::var("QUICKWIT_INDEX").ok();
|
let quickwit_index = env::var("QUICKWIT_INDEX").ok();
|
||||||
|
let default_tenant_slug =
|
||||||
|
env::var("DEFAULT_TENANT_SLUG").unwrap_or_else(|_| "admin".to_string());
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
database_url,
|
database_url,
|
||||||
@@ -104,6 +121,7 @@ impl AppConfig {
|
|||||||
s3_bucket,
|
s3_bucket,
|
||||||
quickwit_endpoint,
|
quickwit_endpoint,
|
||||||
quickwit_index,
|
quickwit_index,
|
||||||
|
default_tenant_slug,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -115,8 +133,12 @@ impl AppConfig {
|
|||||||
fn redact_database_url(raw: &str) -> String {
|
fn redact_database_url(raw: &str) -> String {
|
||||||
match Url::parse(raw) {
|
match Url::parse(raw) {
|
||||||
Ok(mut parsed) => {
|
Ok(mut parsed) => {
|
||||||
let _ = parsed.set_password(Some("*****"));
|
if parsed.password().is_some() {
|
||||||
parsed.to_string()
|
let _ = parsed.set_password(Some("*****"));
|
||||||
|
parsed.to_string()
|
||||||
|
} else {
|
||||||
|
raw.to_string()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
Err(_) => "***".to_string(),
|
Err(_) => "***".to_string(),
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,6 +12,7 @@ pub type AppResult<T> = Result<T, AppError>;
|
|||||||
pub struct AppError {
|
pub struct AppError {
|
||||||
status: StatusCode,
|
status: StatusCode,
|
||||||
message: String,
|
message: String,
|
||||||
|
code: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AppError {
|
impl AppError {
|
||||||
@@ -19,6 +20,7 @@ impl AppError {
|
|||||||
Self {
|
Self {
|
||||||
status,
|
status,
|
||||||
message: message.into(),
|
message: message.into(),
|
||||||
|
code: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -26,6 +28,10 @@ impl AppError {
|
|||||||
Self::new(StatusCode::BAD_REQUEST, message)
|
Self::new(StatusCode::BAD_REQUEST, message)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn conflict(message: impl Into<String>) -> Self {
|
||||||
|
Self::new(StatusCode::CONFLICT, message)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn unauthorized() -> Self {
|
pub fn unauthorized() -> Self {
|
||||||
Self::new(StatusCode::UNAUTHORIZED, "unauthorized")
|
Self::new(StatusCode::UNAUTHORIZED, "unauthorized")
|
||||||
}
|
}
|
||||||
@@ -37,6 +43,11 @@ impl AppError {
|
|||||||
pub fn internal<E: Display>(error: E) -> Self {
|
pub fn internal<E: Display>(error: E) -> Self {
|
||||||
Self::new(StatusCode::INTERNAL_SERVER_ERROR, error.to_string())
|
Self::new(StatusCode::INTERNAL_SERVER_ERROR, error.to_string())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn with_code(mut self, code: impl Into<String>) -> Self {
|
||||||
|
self.code = Some(code.into());
|
||||||
|
self
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl IntoResponse for AppError {
|
impl IntoResponse for AppError {
|
||||||
@@ -44,6 +55,7 @@ impl IntoResponse for AppError {
|
|||||||
let status = self.status;
|
let status = self.status;
|
||||||
let body = Json(ErrorResponse {
|
let body = Json(ErrorResponse {
|
||||||
error: self.message,
|
error: self.message,
|
||||||
|
code: self.code,
|
||||||
});
|
});
|
||||||
(status, body).into_response()
|
(status, body).into_response()
|
||||||
}
|
}
|
||||||
@@ -52,6 +64,8 @@ impl IntoResponse for AppError {
|
|||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
struct ErrorResponse {
|
struct ErrorResponse {
|
||||||
error: String,
|
error: String,
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
code: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<diesel::result::Error> for AppError {
|
impl From<diesel::result::Error> for AppError {
|
||||||
|
|||||||
@@ -30,6 +30,7 @@ pub type JobQueueResult<T> = Result<T, JobQueueError>;
|
|||||||
|
|
||||||
pub fn enqueue_job(
|
pub fn enqueue_job(
|
||||||
conn: &mut PgConnection,
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
job_type: &str,
|
job_type: &str,
|
||||||
payload: Value,
|
payload: Value,
|
||||||
run_after: Option<NaiveDateTime>,
|
run_after: Option<NaiveDateTime>,
|
||||||
@@ -40,6 +41,7 @@ pub fn enqueue_job(
|
|||||||
payload,
|
payload,
|
||||||
status: STATUS_QUEUED.to_string(),
|
status: STATUS_QUEUED.to_string(),
|
||||||
run_after: run_after.unwrap_or_else(|| Utc::now().naive_utc()),
|
run_after: run_after.unwrap_or_else(|| Utc::now().naive_utc()),
|
||||||
|
tenant_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
diesel::insert_into(jobs::table)
|
diesel::insert_into(jobs::table)
|
||||||
|
|||||||
@@ -4,11 +4,13 @@ pub mod db;
|
|||||||
pub mod error;
|
pub mod error;
|
||||||
pub mod jobs;
|
pub mod jobs;
|
||||||
pub mod models;
|
pub mod models;
|
||||||
|
pub mod openapi;
|
||||||
pub mod routes;
|
pub mod routes;
|
||||||
pub mod s3;
|
pub mod s3;
|
||||||
pub mod schema;
|
pub mod schema;
|
||||||
pub mod state;
|
pub mod state;
|
||||||
pub mod storage;
|
pub mod storage;
|
||||||
|
pub mod tenants;
|
||||||
pub mod utils;
|
pub mod utils;
|
||||||
pub mod workers;
|
pub mod workers;
|
||||||
pub use workers::{default_handlers, Worker};
|
pub use workers::{default_handlers, Worker};
|
||||||
|
|||||||
+9
-38
@@ -1,57 +1,28 @@
|
|||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
use tokio::net::TcpListener;
|
use tokio::net::TcpListener;
|
||||||
use tower::make::Shared;
|
use tower::make::Shared;
|
||||||
use tracing_subscriber::EnvFilter;
|
|
||||||
|
|
||||||
use backend::auth::jwt::JwtService;
|
use backend::{routes, utils::bootstrap::init_component};
|
||||||
use backend::config::AppConfig;
|
|
||||||
use backend::db;
|
|
||||||
use backend::routes;
|
|
||||||
use backend::s3::build_client;
|
|
||||||
use backend::state::AppState;
|
|
||||||
use backend::storage::S3Storage;
|
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> anyhow::Result<()> {
|
async fn main() -> anyhow::Result<()> {
|
||||||
dotenv::dotenv().ok();
|
let state = init_component("api", None).await?;
|
||||||
init_tracing();
|
let server_host = state.config.server_host.clone();
|
||||||
|
let server_port = state.config.server_port;
|
||||||
let config = AppConfig::from_env()?;
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
component = "api",
|
component = "api",
|
||||||
database_url = %config.redacted_database_url(),
|
server_host = %server_host,
|
||||||
pool_size = config.database_max_pool_size,
|
server_port,
|
||||||
server_host = %config.server_host,
|
"starting api server"
|
||||||
server_port = config.server_port,
|
|
||||||
quickwit_enabled = config.quickwit_endpoint.is_some(),
|
|
||||||
s3_bucket = %config.s3_bucket,
|
|
||||||
"loaded backend configuration"
|
|
||||||
);
|
);
|
||||||
let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
|
|
||||||
let s3_client = build_client(&config).await?;
|
|
||||||
let storage = Arc::new(S3Storage::new(s3_client, config.s3_bucket.clone()));
|
|
||||||
let jwt = JwtService::from_config(&config)?;
|
|
||||||
|
|
||||||
let state = AppState::new(pool, config, storage, jwt);
|
let router = routes::create_router(state.as_ref().clone());
|
||||||
|
|
||||||
let router = routes::create_router(state.clone());
|
let addr: SocketAddr = format!("{}:{}", server_host, server_port).parse()?;
|
||||||
|
|
||||||
let addr: SocketAddr =
|
|
||||||
format!("{}:{}", state.config.server_host, state.config.server_port).parse()?;
|
|
||||||
let listener = TcpListener::bind(addr).await?;
|
let listener = TcpListener::bind(addr).await?;
|
||||||
tracing::info!("listening on {}", addr);
|
tracing::info!("listening on {}", addr);
|
||||||
|
|
||||||
axum::serve(listener, Shared::new(router)).await?;
|
axum::serve(listener, Shared::new(router)).await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn init_tracing() {
|
|
||||||
let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
|
|
||||||
tracing_subscriber::fmt()
|
|
||||||
.with_env_filter(filter)
|
|
||||||
.with_target(false)
|
|
||||||
.compact()
|
|
||||||
.init();
|
|
||||||
}
|
|
||||||
|
|||||||
+81
-4
@@ -4,13 +4,48 @@ use uuid::Uuid;
|
|||||||
|
|
||||||
use crate::schema::*;
|
use crate::schema::*;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
||||||
|
#[diesel(table_name = user_memberships)]
|
||||||
|
#[diesel(belongs_to(User, foreign_key = user_id))]
|
||||||
|
#[diesel(belongs_to(Tenant, foreign_key = tenant_id))]
|
||||||
|
pub struct UserMembership {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub user_id: Uuid,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
pub role: String,
|
||||||
|
pub created_at: NaiveDateTime,
|
||||||
|
pub updated_at: NaiveDateTime,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Insertable)]
|
||||||
|
#[diesel(table_name = user_memberships)]
|
||||||
|
pub struct NewUserMembership {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub user_id: Uuid,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
pub role: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Queryable, Identifiable)]
|
||||||
|
#[diesel(table_name = tenants)]
|
||||||
|
#[diesel(primary_key(id))]
|
||||||
|
pub struct Tenant {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub slug: String,
|
||||||
|
pub storage_root: Option<String>,
|
||||||
|
pub quickwit_index: Option<String>,
|
||||||
|
pub status: String,
|
||||||
|
pub config: serde_json::Value,
|
||||||
|
pub created_at: NaiveDateTime,
|
||||||
|
pub updated_at: NaiveDateTime,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable)]
|
#[derive(Debug, Clone, Queryable, Identifiable)]
|
||||||
#[diesel(table_name = users)]
|
#[diesel(table_name = users)]
|
||||||
pub struct User {
|
pub struct User {
|
||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub username: String,
|
pub username: String,
|
||||||
pub password_hash: String,
|
pub password_hash: String,
|
||||||
pub role: String,
|
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
pub updated_at: NaiveDateTime,
|
pub updated_at: NaiveDateTime,
|
||||||
}
|
}
|
||||||
@@ -21,7 +56,6 @@ pub struct NewUser {
|
|||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub username: String,
|
pub username: String,
|
||||||
pub password_hash: String,
|
pub password_hash: String,
|
||||||
pub role: String,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable)]
|
#[derive(Debug, Clone, Queryable, Identifiable)]
|
||||||
@@ -32,6 +66,7 @@ pub struct Folder {
|
|||||||
pub parent_id: Option<Uuid>,
|
pub parent_id: Option<Uuid>,
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
pub updated_at: NaiveDateTime,
|
pub updated_at: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -40,6 +75,7 @@ pub struct NewFolder {
|
|||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub name: String,
|
pub name: String,
|
||||||
pub parent_id: Option<Uuid>,
|
pub parent_id: Option<Uuid>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
||||||
@@ -58,6 +94,7 @@ pub struct Document {
|
|||||||
pub issued_at: Option<NaiveDateTime>,
|
pub issued_at: Option<NaiveDateTime>,
|
||||||
pub title: String,
|
pub title: String,
|
||||||
pub current_version_id: Uuid,
|
pub current_version_id: Uuid,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -72,6 +109,7 @@ pub struct NewDocument {
|
|||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
pub issued_at: Option<NaiveDateTime>,
|
pub issued_at: Option<NaiveDateTime>,
|
||||||
pub title: String,
|
pub title: String,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
||||||
@@ -87,6 +125,7 @@ pub struct DocumentVersion {
|
|||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
pub operations_summary: serde_json::Value,
|
pub operations_summary: serde_json::Value,
|
||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -100,6 +139,7 @@ pub struct NewDocumentVersion {
|
|||||||
pub checksum: String,
|
pub checksum: String,
|
||||||
pub operations_summary: serde_json::Value,
|
pub operations_summary: serde_json::Value,
|
||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
||||||
@@ -109,10 +149,11 @@ pub struct DocumentAsset {
|
|||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub document_version_id: Uuid,
|
pub document_version_id: Uuid,
|
||||||
pub asset_type: String,
|
pub asset_type: String,
|
||||||
pub s3_key: String,
|
|
||||||
pub mime_type: String,
|
pub mime_type: String,
|
||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
|
pub cardinality: Option<i32>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -121,9 +162,33 @@ pub struct NewDocumentAsset {
|
|||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub document_version_id: Uuid,
|
pub document_version_id: Uuid,
|
||||||
pub asset_type: String,
|
pub asset_type: String,
|
||||||
pub s3_key: String,
|
|
||||||
pub mime_type: String,
|
pub mime_type: String,
|
||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
|
pub cardinality: Option<i32>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
||||||
|
#[diesel(table_name = document_asset_objects)]
|
||||||
|
#[diesel(belongs_to(DocumentAsset, foreign_key = asset_id))]
|
||||||
|
pub struct DocumentAssetObject {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub asset_id: Uuid,
|
||||||
|
pub ordinal: i32,
|
||||||
|
pub s3_key: String,
|
||||||
|
pub metadata: serde_json::Value,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Insertable)]
|
||||||
|
#[diesel(table_name = document_asset_objects)]
|
||||||
|
pub struct NewDocumentAssetObject {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub asset_id: Uuid,
|
||||||
|
pub ordinal: i32,
|
||||||
|
pub s3_key: String,
|
||||||
|
pub metadata: serde_json::Value,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable)]
|
#[derive(Debug, Clone, Queryable, Identifiable)]
|
||||||
@@ -138,6 +203,7 @@ pub struct Job {
|
|||||||
pub last_error: Option<String>,
|
pub last_error: Option<String>,
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
pub updated_at: NaiveDateTime,
|
pub updated_at: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -148,6 +214,7 @@ pub struct NewJob {
|
|||||||
pub payload: serde_json::Value,
|
pub payload: serde_json::Value,
|
||||||
pub status: String,
|
pub status: String,
|
||||||
pub run_after: NaiveDateTime,
|
pub run_after: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable)]
|
#[derive(Debug, Clone, Queryable, Identifiable)]
|
||||||
@@ -157,6 +224,7 @@ pub struct Tag {
|
|||||||
pub label: String,
|
pub label: String,
|
||||||
pub color: Option<String>,
|
pub color: Option<String>,
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -165,6 +233,7 @@ pub struct NewTag {
|
|||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub label: String,
|
pub label: String,
|
||||||
pub color: Option<String>,
|
pub color: Option<String>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
@@ -178,6 +247,7 @@ pub struct DocumentTag {
|
|||||||
pub tag_id: Uuid,
|
pub tag_id: Uuid,
|
||||||
pub assigned_at: NaiveDateTime,
|
pub assigned_at: NaiveDateTime,
|
||||||
pub assigned_by: Option<Uuid>,
|
pub assigned_by: Option<Uuid>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -186,6 +256,7 @@ pub struct NewDocumentTag {
|
|||||||
pub document_id: Uuid,
|
pub document_id: Uuid,
|
||||||
pub tag_id: Uuid,
|
pub tag_id: Uuid,
|
||||||
pub assigned_by: Option<Uuid>,
|
pub assigned_by: Option<Uuid>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable)]
|
#[derive(Debug, Clone, Queryable, Identifiable)]
|
||||||
@@ -196,6 +267,7 @@ pub struct Correspondent {
|
|||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
pub updated_at: NaiveDateTime,
|
pub updated_at: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -204,6 +276,7 @@ pub struct NewCorrespondent {
|
|||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub name: String,
|
pub name: String,
|
||||||
pub metadata: serde_json::Value,
|
pub metadata: serde_json::Value,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Associations)]
|
#[derive(Debug, Clone, Queryable, Associations)]
|
||||||
@@ -217,6 +290,7 @@ pub struct DocumentCorrespondent {
|
|||||||
pub role: String,
|
pub role: String,
|
||||||
pub assigned_at: NaiveDateTime,
|
pub assigned_at: NaiveDateTime,
|
||||||
pub assigned_by: Option<Uuid>,
|
pub assigned_by: Option<Uuid>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -226,6 +300,7 @@ pub struct NewDocumentCorrespondent {
|
|||||||
pub correspondent_id: Uuid,
|
pub correspondent_id: Uuid,
|
||||||
pub role: String,
|
pub role: String,
|
||||||
pub assigned_by: Option<Uuid>,
|
pub assigned_by: Option<Uuid>,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
#[derive(Debug, Clone, Queryable, Identifiable, Associations)]
|
||||||
@@ -240,6 +315,7 @@ pub struct RefreshToken {
|
|||||||
pub revoked_at: Option<NaiveDateTime>,
|
pub revoked_at: Option<NaiveDateTime>,
|
||||||
pub created_at: NaiveDateTime,
|
pub created_at: NaiveDateTime,
|
||||||
pub updated_at: NaiveDateTime,
|
pub updated_at: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Insertable)]
|
#[derive(Debug, Insertable)]
|
||||||
@@ -250,4 +326,5 @@ pub struct NewRefreshToken {
|
|||||||
pub token_hash: String,
|
pub token_hash: String,
|
||||||
pub issued_at: NaiveDateTime,
|
pub issued_at: NaiveDateTime,
|
||||||
pub expires_at: NaiveDateTime,
|
pub expires_at: NaiveDateTime,
|
||||||
|
pub tenant_id: Uuid,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,951 @@
|
|||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serde_json::Value;
|
||||||
|
use utoipa::{IntoParams, OpenApi, ToSchema};
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
#[derive(OpenApi)]
|
||||||
|
#[openapi(
|
||||||
|
paths(
|
||||||
|
doc::health_check,
|
||||||
|
doc::login,
|
||||||
|
doc::refresh,
|
||||||
|
doc::logout,
|
||||||
|
doc::me,
|
||||||
|
doc::select_tenant,
|
||||||
|
doc::list_documents,
|
||||||
|
doc::check_document,
|
||||||
|
doc::upload_document,
|
||||||
|
doc::get_document,
|
||||||
|
doc::update_document,
|
||||||
|
doc::delete_document,
|
||||||
|
doc::download_document,
|
||||||
|
doc::download_with_token,
|
||||||
|
doc::move_document,
|
||||||
|
doc::assign_tags,
|
||||||
|
doc::remove_tag,
|
||||||
|
doc::bulk_move_documents,
|
||||||
|
doc::bulk_update_tags,
|
||||||
|
doc::bulk_assign_correspondents,
|
||||||
|
doc::assign_correspondents,
|
||||||
|
doc::remove_correspondent,
|
||||||
|
doc::reanalyze_selected_documents,
|
||||||
|
doc::list_document_assets,
|
||||||
|
doc::request_document_assets,
|
||||||
|
doc::get_document_asset,
|
||||||
|
doc::create_folder,
|
||||||
|
doc::ensure_folder_path,
|
||||||
|
doc::get_folder,
|
||||||
|
doc::list_folder_contents,
|
||||||
|
doc::delete_folder,
|
||||||
|
doc::update_folder,
|
||||||
|
doc::list_tags,
|
||||||
|
doc::create_tag,
|
||||||
|
doc::update_tag,
|
||||||
|
doc::delete_tag,
|
||||||
|
doc::list_correspondents,
|
||||||
|
doc::create_correspondent,
|
||||||
|
doc::update_correspondent,
|
||||||
|
doc::delete_correspondent,
|
||||||
|
),
|
||||||
|
components(
|
||||||
|
schemas(
|
||||||
|
schemas::LoginRequest,
|
||||||
|
schemas::AccessTokenResponse,
|
||||||
|
schemas::TenantSummary,
|
||||||
|
schemas::TenantSelectionResponse,
|
||||||
|
schemas::TenantSelectionRequest,
|
||||||
|
schemas::LoginResponseVariants,
|
||||||
|
schemas::DocumentResponse,
|
||||||
|
schemas::DocumentDetailResponse,
|
||||||
|
schemas::DocumentVersion,
|
||||||
|
schemas::DocumentAssetSummary,
|
||||||
|
schemas::DocumentAssetDetail,
|
||||||
|
schemas::DocumentAssetObject,
|
||||||
|
schemas::DocumentCorrespondent,
|
||||||
|
schemas::DocumentTag,
|
||||||
|
schemas::DocumentDownloadResponse,
|
||||||
|
schemas::UpdateDocumentRequest,
|
||||||
|
schemas::BulkMoveDocumentsRequest,
|
||||||
|
schemas::BulkMoveDocumentsResponse,
|
||||||
|
schemas::AssignTagsRequest,
|
||||||
|
schemas::MoveDocumentRequest,
|
||||||
|
schemas::BulkTagRequest,
|
||||||
|
schemas::BulkTagResponse,
|
||||||
|
schemas::CorrespondentAssignment,
|
||||||
|
schemas::BulkTagAction,
|
||||||
|
schemas::BulkCorrespondentsRequest,
|
||||||
|
schemas::BulkCorrespondentsResponse,
|
||||||
|
schemas::BulkCorrespondentAction,
|
||||||
|
schemas::AssignCorrespondentsRequest,
|
||||||
|
schemas::RemoveCorrespondentParams,
|
||||||
|
schemas::ReanalyzeRequest,
|
||||||
|
schemas::ReanalyzeResponse,
|
||||||
|
schemas::DocumentAssetRequestParams,
|
||||||
|
schemas::AssetObjectsQuery,
|
||||||
|
schemas::DocumentCheckQuery,
|
||||||
|
schemas::DocumentCheckResponse,
|
||||||
|
schemas::UploadDocumentForm,
|
||||||
|
schemas::CreateFolderRequest,
|
||||||
|
schemas::EnsureFolderPathRequest,
|
||||||
|
schemas::FolderResponse,
|
||||||
|
schemas::FolderInfo,
|
||||||
|
schemas::FolderContentsResponse,
|
||||||
|
schemas::FolderDocumentSummary,
|
||||||
|
schemas::UpdateFolderRequest,
|
||||||
|
schemas::FolderContentsParams,
|
||||||
|
schemas::TagCatalogEntry,
|
||||||
|
schemas::CreateTagRequest,
|
||||||
|
schemas::UpdateTagRequest,
|
||||||
|
schemas::CorrespondentCatalogEntry,
|
||||||
|
schemas::CreateCorrespondentRequest,
|
||||||
|
schemas::UpdateCorrespondentRequest,
|
||||||
|
)
|
||||||
|
),
|
||||||
|
tags(
|
||||||
|
(name = "Health", description = "Service health"),
|
||||||
|
(name = "Auth", description = "Authentication"),
|
||||||
|
(name = "Documents", description = "Document management"),
|
||||||
|
(name = "Assets", description = "Document assets"),
|
||||||
|
(name = "Folders", description = "Folder management"),
|
||||||
|
(name = "Tags", description = "Tag catalog"),
|
||||||
|
(name = "Correspondents", description = "Correspondent catalog")
|
||||||
|
)
|
||||||
|
)]
|
||||||
|
pub struct ApiDoc;
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
mod doc {
|
||||||
|
use super::schemas::*;
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
fn __keep_uuid_import() {
|
||||||
|
let _ = Uuid::nil();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/health",
|
||||||
|
responses((status = 200, description = "Service is healthy")),
|
||||||
|
tag = "Health"
|
||||||
|
)]
|
||||||
|
pub(super) fn health_check() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/auth/login",
|
||||||
|
request_body = LoginRequest,
|
||||||
|
responses(
|
||||||
|
(status = 200, description = "Login succeeded", body = LoginResponseVariants),
|
||||||
|
(status = 401, description = "Invalid credentials")
|
||||||
|
),
|
||||||
|
tag = "Auth"
|
||||||
|
)]
|
||||||
|
pub(super) fn login() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/auth/refresh",
|
||||||
|
responses(
|
||||||
|
(status = 200, description = "Refreshed access token", body = AccessTokenResponse),
|
||||||
|
(status = 401, description = "Missing or invalid refresh token")
|
||||||
|
),
|
||||||
|
tag = "Auth"
|
||||||
|
)]
|
||||||
|
pub(super) fn refresh() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/auth/logout",
|
||||||
|
responses((status = 204, description = "Session revoked")),
|
||||||
|
tag = "Auth"
|
||||||
|
)]
|
||||||
|
pub(super) fn logout() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/auth/me",
|
||||||
|
responses((status = 200, description = "Authenticated principal", body = AccessTokenResponse)),
|
||||||
|
tag = "Auth"
|
||||||
|
)]
|
||||||
|
pub(super) fn me() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/auth/select-tenant",
|
||||||
|
request_body = TenantSelectionRequest,
|
||||||
|
responses((status = 200, description = "Tenant selected", body = AccessTokenResponse)),
|
||||||
|
tag = "Auth"
|
||||||
|
)]
|
||||||
|
pub(super) fn select_tenant() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/documents",
|
||||||
|
params(DocumentListQuery),
|
||||||
|
responses((status = 200, description = "List documents", body = [DocumentResponse])),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn list_documents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents",
|
||||||
|
request_body = UploadDocumentForm,
|
||||||
|
responses(
|
||||||
|
(status = 201, description = "Document created", body = DocumentDetailResponse),
|
||||||
|
(status = 200, description = "Existing document reused", body = DocumentDetailResponse),
|
||||||
|
(status = 204, description = "Upload skipped because the document already exists")
|
||||||
|
),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn upload_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/documents/check",
|
||||||
|
params(DocumentCheckQuery),
|
||||||
|
responses((status = 200, description = "Checksum lookup", body = DocumentCheckResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn check_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/documents/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
responses((status = 200, description = "Document detail", body = DocumentDetailResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn get_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
patch,
|
||||||
|
path = "/api/documents/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
request_body = UpdateDocumentRequest,
|
||||||
|
responses((status = 200, description = "Updated document", body = DocumentDetailResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn update_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
delete,
|
||||||
|
path = "/api/documents/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
responses((status = 204, description = "Document deleted")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn delete_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/documents/{id}/download",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
responses((status = 200, description = "Download metadata", body = DocumentDownloadResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn download_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/download/{token}",
|
||||||
|
params(("token" = String, Path, description = "Download token")),
|
||||||
|
responses((status = 302, description = "Redirect to pre-signed URL")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn download_with_token() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
patch,
|
||||||
|
path = "/api/documents/{id}/folder",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
request_body = MoveDocumentRequest,
|
||||||
|
responses((status = 204, description = "Document moved")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn move_document() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/{id}/tags",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
request_body = AssignTagsRequest,
|
||||||
|
responses((status = 204, description = "Tags assigned")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn assign_tags() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
delete,
|
||||||
|
path = "/api/documents/{id}/tags/{tag_id}",
|
||||||
|
params(
|
||||||
|
("id" = Uuid, Path, description = "Document ID"),
|
||||||
|
("tag_id" = Uuid, Path, description = "Tag ID")
|
||||||
|
),
|
||||||
|
responses((status = 204, description = "Tag removed")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn remove_tag() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/bulk/move",
|
||||||
|
request_body = BulkMoveDocumentsRequest,
|
||||||
|
responses((status = 200, description = "Bulk move outcome", body = BulkMoveDocumentsResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn bulk_move_documents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/bulk/tags",
|
||||||
|
request_body = BulkTagRequest,
|
||||||
|
responses((status = 200, description = "Bulk tag outcome", body = BulkTagResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn bulk_update_tags() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/bulk/correspondents",
|
||||||
|
request_body = BulkCorrespondentsRequest,
|
||||||
|
responses((status = 200, description = "Bulk correspondents outcome", body = BulkCorrespondentsResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn bulk_assign_correspondents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/{id}/correspondents",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
request_body = AssignCorrespondentsRequest,
|
||||||
|
responses((status = 204, description = "Correspondents assigned")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn assign_correspondents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
delete,
|
||||||
|
path = "/api/documents/{id}/correspondents/{correspondent_id}",
|
||||||
|
params(
|
||||||
|
("id" = Uuid, Path, description = "Document ID"),
|
||||||
|
("correspondent_id" = Uuid, Path, description = "Correspondent ID"),
|
||||||
|
RemoveCorrespondentParams
|
||||||
|
),
|
||||||
|
responses((status = 204, description = "Correspondent removed")),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn remove_correspondent() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/bulk/reanalyze",
|
||||||
|
request_body = ReanalyzeRequest,
|
||||||
|
responses((status = 200, description = "Reanalyze queued", body = ReanalyzeResponse)),
|
||||||
|
tag = "Documents"
|
||||||
|
)]
|
||||||
|
pub(super) fn reanalyze_selected_documents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/documents/{id}/assets",
|
||||||
|
params(("id" = Uuid, Path, description = "Document ID")),
|
||||||
|
responses((status = 200, description = "Document assets", body = [DocumentAssetSummary])),
|
||||||
|
tag = "Assets"
|
||||||
|
)]
|
||||||
|
pub(super) fn list_document_assets() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/documents/{id}/assets",
|
||||||
|
params(
|
||||||
|
("id" = Uuid, Path, description = "Document ID"),
|
||||||
|
DocumentAssetRequestParams
|
||||||
|
),
|
||||||
|
responses((status = 202, description = "Asset generation requested")),
|
||||||
|
tag = "Assets"
|
||||||
|
)]
|
||||||
|
pub(super) fn request_document_assets() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/assets/{asset_id}",
|
||||||
|
params(
|
||||||
|
("asset_id" = Uuid, Path, description = "Asset ID"),
|
||||||
|
AssetObjectsQuery
|
||||||
|
),
|
||||||
|
responses((status = 200, description = "Asset detail", body = DocumentAssetDetail)),
|
||||||
|
tag = "Assets"
|
||||||
|
)]
|
||||||
|
pub(super) fn get_document_asset() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/folders",
|
||||||
|
request_body = CreateFolderRequest,
|
||||||
|
responses((status = 200, description = "Folder created", body = FolderResponse)),
|
||||||
|
tag = "Folders"
|
||||||
|
)]
|
||||||
|
pub(super) fn create_folder() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/folders/path",
|
||||||
|
request_body = EnsureFolderPathRequest,
|
||||||
|
responses((status = 200, description = "Folder path ensured", body = FolderResponse)),
|
||||||
|
tag = "Folders"
|
||||||
|
)]
|
||||||
|
pub(super) fn ensure_folder_path() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/folders/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Folder ID")),
|
||||||
|
responses((status = 200, description = "Folder detail", body = FolderResponse)),
|
||||||
|
tag = "Folders"
|
||||||
|
)]
|
||||||
|
pub(super) fn get_folder() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/folders/{id}/contents",
|
||||||
|
params(
|
||||||
|
("id" = Uuid, Path, description = "Folder ID"),
|
||||||
|
FolderContentsParams
|
||||||
|
),
|
||||||
|
responses((status = 200, description = "Folder contents", body = FolderContentsResponse)),
|
||||||
|
tag = "Folders"
|
||||||
|
)]
|
||||||
|
pub(super) fn list_folder_contents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
delete,
|
||||||
|
path = "/api/folders/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Folder ID")),
|
||||||
|
responses((status = 204, description = "Folder deleted")),
|
||||||
|
tag = "Folders"
|
||||||
|
)]
|
||||||
|
pub(super) fn delete_folder() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
patch,
|
||||||
|
path = "/api/folders/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Folder ID")),
|
||||||
|
request_body = UpdateFolderRequest,
|
||||||
|
responses((status = 204, description = "Folder updated")),
|
||||||
|
tag = "Folders"
|
||||||
|
)]
|
||||||
|
pub(super) fn update_folder() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/tags",
|
||||||
|
responses((status = 200, description = "Tags", body = [TagCatalogEntry])),
|
||||||
|
tag = "Tags"
|
||||||
|
)]
|
||||||
|
pub(super) fn list_tags() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/tags",
|
||||||
|
request_body = CreateTagRequest,
|
||||||
|
responses((status = 200, description = "Tag created", body = TagCatalogEntry)),
|
||||||
|
tag = "Tags"
|
||||||
|
)]
|
||||||
|
pub(super) fn create_tag() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
patch,
|
||||||
|
path = "/api/tags/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Tag ID")),
|
||||||
|
request_body = UpdateTagRequest,
|
||||||
|
responses((status = 200, description = "Tag updated", body = TagCatalogEntry)),
|
||||||
|
tag = "Tags"
|
||||||
|
)]
|
||||||
|
pub(super) fn update_tag() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
delete,
|
||||||
|
path = "/api/tags/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Tag ID")),
|
||||||
|
responses((status = 204, description = "Tag deleted")),
|
||||||
|
tag = "Tags"
|
||||||
|
)]
|
||||||
|
pub(super) fn delete_tag() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
get,
|
||||||
|
path = "/api/correspondents",
|
||||||
|
responses((status = 200, description = "Correspondents", body = [CorrespondentCatalogEntry])),
|
||||||
|
tag = "Correspondents"
|
||||||
|
)]
|
||||||
|
pub(super) fn list_correspondents() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
post,
|
||||||
|
path = "/api/correspondents",
|
||||||
|
request_body = CreateCorrespondentRequest,
|
||||||
|
responses((status = 200, description = "Correspondent created", body = CorrespondentCatalogEntry)),
|
||||||
|
tag = "Correspondents"
|
||||||
|
)]
|
||||||
|
pub(super) fn create_correspondent() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
patch,
|
||||||
|
path = "/api/correspondents/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Correspondent ID")),
|
||||||
|
request_body = UpdateCorrespondentRequest,
|
||||||
|
responses((status = 200, description = "Correspondent updated", body = CorrespondentCatalogEntry)),
|
||||||
|
tag = "Correspondents"
|
||||||
|
)]
|
||||||
|
pub(super) fn update_correspondent() {}
|
||||||
|
|
||||||
|
#[utoipa::path(
|
||||||
|
delete,
|
||||||
|
path = "/api/correspondents/{id}",
|
||||||
|
params(("id" = Uuid, Path, description = "Correspondent ID")),
|
||||||
|
responses((status = 204, description = "Correspondent deleted")),
|
||||||
|
tag = "Correspondents"
|
||||||
|
)]
|
||||||
|
pub(super) fn delete_correspondent() {}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::ApiDoc;
|
||||||
|
use utoipa::OpenApi;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn openapi_serializes() {
|
||||||
|
let spec = ApiDoc::openapi();
|
||||||
|
let _ = serde_json::to_string(&spec).expect("serialize openapi");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub mod schemas {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct LoginRequest {
|
||||||
|
pub username: String,
|
||||||
|
pub password: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub preferred_tenant_slug: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct AccessTokenResponse {
|
||||||
|
pub access_token: String,
|
||||||
|
pub token_type: String,
|
||||||
|
pub expires_in: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct TenantSummary {
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
pub slug: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct TenantSelectionResponse {
|
||||||
|
pub selection_token: String,
|
||||||
|
pub tenants: Vec<TenantSummary>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct TenantSelectionRequest {
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
#[serde(untagged)]
|
||||||
|
pub enum LoginResponseVariants {
|
||||||
|
Token(AccessTokenResponse),
|
||||||
|
Selection(TenantSelectionResponse),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, IntoParams, ToSchema)]
|
||||||
|
#[into_params(parameter_in = Query)]
|
||||||
|
pub struct DocumentListQuery {
|
||||||
|
pub folder_id: Option<Uuid>,
|
||||||
|
pub include_deleted: Option<bool>,
|
||||||
|
pub include_descendants: Option<bool>,
|
||||||
|
pub query: Option<String>,
|
||||||
|
pub tags: Option<String>,
|
||||||
|
pub correspondents: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentTag {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub label: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub color: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentAssetObject {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub ordinal: i32,
|
||||||
|
pub metadata: Value,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub url: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub expires_at: Option<i64>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentAssetSummary {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub asset_type: String,
|
||||||
|
pub mime_type: String,
|
||||||
|
pub metadata: Value,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub cardinality: Option<i32>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentAssetDetail {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub asset_type: String,
|
||||||
|
pub mime_type: String,
|
||||||
|
pub metadata: Value,
|
||||||
|
pub created_at: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub cardinality: Option<i32>,
|
||||||
|
pub objects: Vec<DocumentAssetObject>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentVersion {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub version_number: i32,
|
||||||
|
pub checksum: String,
|
||||||
|
pub size_bytes: i64,
|
||||||
|
pub created_at: String,
|
||||||
|
pub metadata: Value,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub operations_summary: Option<Value>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub assets: Option<Vec<DocumentAssetSummary>>,
|
||||||
|
pub download_path: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentCorrespondent {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub name: String,
|
||||||
|
pub role: String,
|
||||||
|
pub metadata: Value,
|
||||||
|
pub assigned_at: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentResponse {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub filename: String,
|
||||||
|
pub title: String,
|
||||||
|
pub original_name: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub content_type: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub folder_id: Option<Uuid>,
|
||||||
|
pub uploaded_at: String,
|
||||||
|
pub updated_at: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub deleted_at: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub issued_at: Option<String>,
|
||||||
|
pub metadata: Value,
|
||||||
|
pub tags: Vec<DocumentTag>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub correspondents: Option<Vec<DocumentCorrespondent>>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub current_version: Option<DocumentVersion>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentDetailResponse {
|
||||||
|
pub document: DocumentResponse,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentDownloadResponse {
|
||||||
|
pub url: String,
|
||||||
|
pub expires_in: u64,
|
||||||
|
pub filename: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub content_type: Option<String>,
|
||||||
|
pub size_bytes: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct UpdateDocumentRequest {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub title: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct BulkMoveDocumentsRequest {
|
||||||
|
pub document_ids: Vec<Uuid>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub folder_id: Option<Uuid>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct BulkMoveDocumentsResponse {
|
||||||
|
pub updated: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct AssignTagsRequest {
|
||||||
|
pub tag_ids: Vec<Uuid>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct MoveDocumentRequest {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub folder_id: Option<Uuid>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
#[serde(rename_all = "snake_case")]
|
||||||
|
pub enum BulkTagAction {
|
||||||
|
Add,
|
||||||
|
Remove,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct BulkTagRequest {
|
||||||
|
pub document_ids: Vec<Uuid>,
|
||||||
|
pub tag_ids: Vec<Uuid>,
|
||||||
|
pub action: BulkTagAction,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct BulkTagResponse {
|
||||||
|
pub added: usize,
|
||||||
|
pub removed: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct CorrespondentAssignment {
|
||||||
|
pub correspondent_id: Uuid,
|
||||||
|
pub role: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct AssignCorrespondentsRequest {
|
||||||
|
pub assignments: Vec<CorrespondentAssignment>,
|
||||||
|
#[serde(default)]
|
||||||
|
pub replace: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
#[serde(rename_all = "lowercase")]
|
||||||
|
pub enum BulkCorrespondentAction {
|
||||||
|
Add,
|
||||||
|
Remove,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct BulkCorrespondentsRequest {
|
||||||
|
pub document_ids: Vec<Uuid>,
|
||||||
|
pub assignments: Vec<CorrespondentAssignment>,
|
||||||
|
#[serde(default = "default_bulk_correspondent_action")]
|
||||||
|
pub action: BulkCorrespondentAction,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn default_bulk_correspondent_action() -> BulkCorrespondentAction {
|
||||||
|
BulkCorrespondentAction::Add
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct BulkCorrespondentsResponse {
|
||||||
|
pub assigned: usize,
|
||||||
|
pub removed: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, IntoParams, ToSchema)]
|
||||||
|
#[into_params(parameter_in = Query)]
|
||||||
|
pub struct RemoveCorrespondentParams {
|
||||||
|
pub role: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct ReanalyzeRequest {
|
||||||
|
pub document_ids: Vec<Uuid>,
|
||||||
|
#[serde(default)]
|
||||||
|
pub force: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct ReanalyzeResponse {
|
||||||
|
pub queued: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, IntoParams, ToSchema)]
|
||||||
|
#[into_params(parameter_in = Query)]
|
||||||
|
pub struct DocumentAssetRequestParams {
|
||||||
|
#[serde(default)]
|
||||||
|
pub force: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, IntoParams, ToSchema)]
|
||||||
|
#[into_params(parameter_in = Query)]
|
||||||
|
pub struct AssetObjectsQuery {
|
||||||
|
#[serde(default)]
|
||||||
|
pub start: Option<i32>,
|
||||||
|
#[serde(default)]
|
||||||
|
pub limit: Option<i32>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, IntoParams, ToSchema)]
|
||||||
|
#[into_params(parameter_in = Query)]
|
||||||
|
pub struct DocumentCheckQuery {
|
||||||
|
pub checksum: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct DocumentCheckResponse {
|
||||||
|
pub exists: bool,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub document_id: Option<Uuid>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub title: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub filename: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub version_id: Option<Uuid>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub version_number: Option<i32>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub uploaded_at: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct UploadDocumentForm {
|
||||||
|
#[schema(value_type = String, format = Binary)]
|
||||||
|
pub file: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub folder_id: Option<Uuid>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub metadata: Option<Value>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub title: Option<String>,
|
||||||
|
#[schema(nullable, value_type = Vec<Uuid>)]
|
||||||
|
pub tag_ids: Option<Vec<Uuid>>,
|
||||||
|
#[schema(nullable, value_type = Vec<CorrespondentAssignment>)]
|
||||||
|
pub correspondents: Option<Vec<CorrespondentAssignment>>,
|
||||||
|
#[schema(nullable, example = "2024-01-01T00:00:00Z")]
|
||||||
|
pub issued_at: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub skip_existing: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct CreateFolderRequest {
|
||||||
|
pub name: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub parent_id: Option<Uuid>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct EnsureFolderPathRequest {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub parent_id: Option<Uuid>,
|
||||||
|
pub segments: Vec<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct FolderInfo {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub name: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub parent_id: Option<Uuid>,
|
||||||
|
pub created_at: String,
|
||||||
|
pub updated_at: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct FolderResponse {
|
||||||
|
pub folder: FolderInfo,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, IntoParams, ToSchema)]
|
||||||
|
#[into_params(parameter_in = Query)]
|
||||||
|
pub struct FolderContentsParams {
|
||||||
|
#[serde(default)]
|
||||||
|
pub include_documents: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct FolderDocumentSummary {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub title: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct FolderContentsResponse {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub folder: Option<FolderInfo>,
|
||||||
|
pub subfolders: Vec<FolderInfo>,
|
||||||
|
pub documents: Vec<FolderDocumentSummary>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct UpdateFolderRequest {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub parent_id: Option<Option<Uuid>>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub name: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct TagCatalogEntry {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub label: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub color: Option<String>,
|
||||||
|
pub usage_count: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct CreateTagRequest {
|
||||||
|
pub label: String,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub color: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct UpdateTagRequest {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub label: Option<Option<String>>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub color: Option<Option<String>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct CorrespondentCatalogEntry {
|
||||||
|
pub id: Uuid,
|
||||||
|
pub name: String,
|
||||||
|
pub metadata: Value,
|
||||||
|
pub role_counts: Value,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct CreateCorrespondentRequest {
|
||||||
|
pub name: String,
|
||||||
|
#[schema(default, value_type = Object)]
|
||||||
|
pub metadata: Value,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, ToSchema)]
|
||||||
|
pub struct UpdateCorrespondentRequest {
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub name: Option<String>,
|
||||||
|
#[schema(nullable)]
|
||||||
|
pub metadata: Option<Value>,
|
||||||
|
}
|
||||||
|
}
|
||||||
+152
-72
@@ -1,11 +1,15 @@
|
|||||||
use axum::{
|
use axum::{
|
||||||
extract::State,
|
extract::State,
|
||||||
http::{header::SET_COOKIE, HeaderMap, HeaderValue, StatusCode},
|
http::{header::SET_COOKIE, HeaderMap, HeaderValue, StatusCode},
|
||||||
|
response::{IntoResponse, Response},
|
||||||
Json,
|
Json,
|
||||||
};
|
};
|
||||||
use axum_extra::{headers::Cookie, typed_header::TypedHeader};
|
use axum_extra::{
|
||||||
|
headers::{authorization::Bearer, Authorization, Cookie},
|
||||||
|
typed_header::TypedHeader,
|
||||||
|
};
|
||||||
use chrono::{Duration as ChronoDuration, Utc};
|
use chrono::{Duration as ChronoDuration, Utc};
|
||||||
use diesel::prelude::*;
|
use diesel::{pg::PgConnection, prelude::*};
|
||||||
use rand::{rngs::OsRng, RngCore};
|
use rand::{rngs::OsRng, RngCore};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use sha2::{Digest, Sha256};
|
use sha2::{Digest, Sha256};
|
||||||
@@ -14,8 +18,11 @@ use uuid::Uuid;
|
|||||||
use crate::{
|
use crate::{
|
||||||
auth::{password, AuthenticatedUser},
|
auth::{password, AuthenticatedUser},
|
||||||
error::{AppError, AppResult},
|
error::{AppError, AppResult},
|
||||||
models::{NewRefreshToken, RefreshToken, User},
|
models::{NewRefreshToken, RefreshToken, Tenant, User, UserMembership},
|
||||||
schema::{refresh_tokens, users::dsl},
|
schema::{
|
||||||
|
refresh_tokens, tenants::dsl as tenant_dsl, user_memberships::dsl as memberships_dsl,
|
||||||
|
users::dsl,
|
||||||
|
},
|
||||||
state::AppState,
|
state::AppState,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -27,6 +34,8 @@ const REFRESH_COOKIE_NAME: &str = "refresh_token";
|
|||||||
pub struct LoginRequest {
|
pub struct LoginRequest {
|
||||||
pub username: String,
|
pub username: String,
|
||||||
pub password: String,
|
pub password: String,
|
||||||
|
#[serde(default)]
|
||||||
|
pub preferred_tenant_slug: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
@@ -36,11 +45,28 @@ pub struct LoginResponse {
|
|||||||
pub expires_in: i64,
|
pub expires_in: i64,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
pub struct TenantSummary {
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
pub slug: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
pub struct TenantSelectionResponse {
|
||||||
|
pub selection_token: String,
|
||||||
|
pub tenants: Vec<TenantSummary>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize)]
|
||||||
|
pub struct TenantSelectionRequest {
|
||||||
|
pub tenant_id: Uuid,
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn login(
|
pub async fn login(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Json(payload): Json<LoginRequest>,
|
Json(payload): Json<LoginRequest>,
|
||||||
) -> AppResult<(HeaderMap, Json<LoginResponse>)> {
|
) -> AppResult<Response> {
|
||||||
let mut conn = state.db()?;
|
let mut conn = state.db_unscoped()?;
|
||||||
|
|
||||||
let user: User = dsl::users
|
let user: User = dsl::users
|
||||||
.filter(dsl::username.eq(&payload.username))
|
.filter(dsl::username.eq(&payload.username))
|
||||||
@@ -53,55 +79,67 @@ pub async fn login(
|
|||||||
return Err(AppError::unauthorized());
|
return Err(AppError::unauthorized());
|
||||||
}
|
}
|
||||||
|
|
||||||
let access_token = state
|
let memberships: Vec<(UserMembership, Tenant)> = memberships_dsl::user_memberships
|
||||||
|
.inner_join(tenant_dsl::tenants)
|
||||||
|
.filter(memberships_dsl::user_id.eq(user.id))
|
||||||
|
.load(&mut conn)?;
|
||||||
|
|
||||||
|
if memberships.is_empty() {
|
||||||
|
return Err(AppError::unauthorized());
|
||||||
|
}
|
||||||
|
|
||||||
|
let preferred_slug = payload
|
||||||
|
.preferred_tenant_slug
|
||||||
|
.as_ref()
|
||||||
|
.map(|slug| slug.trim().to_string())
|
||||||
|
.filter(|slug| !slug.is_empty());
|
||||||
|
|
||||||
|
if let Some(tenant) = preferred_slug.as_ref().and_then(|slug| {
|
||||||
|
memberships
|
||||||
|
.iter()
|
||||||
|
.find(|(_, tenant)| tenant.slug.eq_ignore_ascii_case(slug))
|
||||||
|
}) {
|
||||||
|
return issue_session(&state, &mut conn, &user, tenant.1.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
if memberships.len() == 1 {
|
||||||
|
let tenant_id = memberships[0].1.id;
|
||||||
|
return issue_session(&state, &mut conn, &user, tenant_id);
|
||||||
|
}
|
||||||
|
|
||||||
|
let selection_token = state
|
||||||
.jwt
|
.jwt
|
||||||
.generate_token(user.id, &user.username, &user.role)
|
.generate_tenant_selector_token(user.id)
|
||||||
.map_err(AppError::from)?;
|
.map_err(AppError::from)?;
|
||||||
|
|
||||||
let now = Utc::now();
|
let tenants = memberships
|
||||||
let refresh_value = generate_refresh_token();
|
.into_iter()
|
||||||
let refresh_hash = hash_refresh_token(&refresh_value);
|
.map(|(_, tenant)| TenantSummary {
|
||||||
let refresh_expires_at = now + ChronoDuration::days(state.config.refresh_token_expiry_days);
|
tenant_id: tenant.id,
|
||||||
|
slug: tenant.slug,
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
let new_refresh = NewRefreshToken {
|
let response = Json(TenantSelectionResponse {
|
||||||
id: Uuid::new_v4(),
|
selection_token,
|
||||||
user_id: user.id,
|
tenants,
|
||||||
token_hash: refresh_hash,
|
})
|
||||||
issued_at: now.naive_utc(),
|
.into_response();
|
||||||
expires_at: refresh_expires_at.naive_utc(),
|
|
||||||
};
|
|
||||||
|
|
||||||
diesel::insert_into(refresh_tokens::table)
|
Ok(response)
|
||||||
.values(&new_refresh)
|
|
||||||
.execute(&mut conn)?;
|
|
||||||
|
|
||||||
let mut headers = HeaderMap::new();
|
|
||||||
headers.insert(
|
|
||||||
SET_COOKIE,
|
|
||||||
build_refresh_cookie(&state, &refresh_value, refresh_expires_at),
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok((
|
|
||||||
headers,
|
|
||||||
Json(LoginResponse {
|
|
||||||
access_token,
|
|
||||||
token_type: "Bearer".to_string(),
|
|
||||||
expires_in: state.config.jwt_expiry_minutes * 60,
|
|
||||||
}),
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn refresh(
|
pub async fn refresh(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
jar: Option<TypedHeader<Cookie>>,
|
jar: Option<TypedHeader<Cookie>>,
|
||||||
) -> AppResult<(HeaderMap, Json<LoginResponse>)> {
|
) -> AppResult<Response> {
|
||||||
let cookies = jar.ok_or_else(AppError::unauthorized)?;
|
let cookies = jar.ok_or_else(AppError::unauthorized)?;
|
||||||
let refresh_value = cookies
|
let refresh_value = cookies
|
||||||
.get(REFRESH_COOKIE_NAME)
|
.get(REFRESH_COOKIE_NAME)
|
||||||
.ok_or_else(AppError::unauthorized)?;
|
.ok_or_else(AppError::unauthorized)?;
|
||||||
|
|
||||||
let hashed = hash_refresh_token(refresh_value);
|
let hashed = hash_refresh_token(refresh_value);
|
||||||
let mut conn = state.db()?;
|
let mut conn = state.db_unscoped()?;
|
||||||
let now = Utc::now();
|
let now = Utc::now();
|
||||||
let now_naive = now.naive_utc();
|
let now_naive = now.naive_utc();
|
||||||
|
|
||||||
@@ -128,41 +166,39 @@ pub async fn refresh(
|
|||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.map_err(AppError::from)?;
|
.map_err(AppError::from)?;
|
||||||
|
|
||||||
let access_token = state
|
issue_session(&state, &mut conn, &user, token.tenant_id)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn select_tenant(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
TypedHeader(Authorization(bearer)): TypedHeader<Authorization<Bearer>>,
|
||||||
|
Json(payload): Json<TenantSelectionRequest>,
|
||||||
|
) -> AppResult<Response> {
|
||||||
|
let claims = state
|
||||||
.jwt
|
.jwt
|
||||||
.generate_token(user.id, &user.username, &user.role)
|
.verify_tenant_selector_token(bearer.token())
|
||||||
|
.map_err(|_| AppError::unauthorized())?;
|
||||||
|
|
||||||
|
let mut conn = state.db_unscoped()?;
|
||||||
|
|
||||||
|
let membership_exists = memberships_dsl::user_memberships
|
||||||
|
.filter(memberships_dsl::user_id.eq(claims.sub))
|
||||||
|
.filter(memberships_dsl::tenant_id.eq(payload.tenant_id))
|
||||||
|
.inner_join(tenant_dsl::tenants)
|
||||||
|
.select(memberships_dsl::id)
|
||||||
|
.first::<Uuid>(&mut conn)
|
||||||
|
.optional()?;
|
||||||
|
|
||||||
|
if membership_exists.is_none() {
|
||||||
|
return Err(AppError::unauthorized());
|
||||||
|
}
|
||||||
|
|
||||||
|
let user: User = dsl::users
|
||||||
|
.find(claims.sub)
|
||||||
|
.first(&mut conn)
|
||||||
.map_err(AppError::from)?;
|
.map_err(AppError::from)?;
|
||||||
|
|
||||||
let new_refresh_value = generate_refresh_token();
|
issue_session(&state, &mut conn, &user, payload.tenant_id)
|
||||||
let new_refresh_hash = hash_refresh_token(&new_refresh_value);
|
|
||||||
let new_refresh_expires = now + ChronoDuration::days(state.config.refresh_token_expiry_days);
|
|
||||||
|
|
||||||
let new_refresh = NewRefreshToken {
|
|
||||||
id: Uuid::new_v4(),
|
|
||||||
user_id: user.id,
|
|
||||||
token_hash: new_refresh_hash,
|
|
||||||
issued_at: now_naive,
|
|
||||||
expires_at: new_refresh_expires.naive_utc(),
|
|
||||||
};
|
|
||||||
|
|
||||||
diesel::insert_into(refresh_tokens::table)
|
|
||||||
.values(&new_refresh)
|
|
||||||
.execute(&mut conn)?;
|
|
||||||
|
|
||||||
let mut headers = HeaderMap::new();
|
|
||||||
headers.insert(
|
|
||||||
SET_COOKIE,
|
|
||||||
build_refresh_cookie(&state, &new_refresh_value, new_refresh_expires),
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok((
|
|
||||||
headers,
|
|
||||||
Json(LoginResponse {
|
|
||||||
access_token,
|
|
||||||
token_type: "Bearer".to_string(),
|
|
||||||
expires_in: state.config.jwt_expiry_minutes * 60,
|
|
||||||
}),
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn logout(
|
pub async fn logout(
|
||||||
@@ -170,7 +206,7 @@ pub async fn logout(
|
|||||||
user: AuthenticatedUser,
|
user: AuthenticatedUser,
|
||||||
jar: Option<TypedHeader<Cookie>>,
|
jar: Option<TypedHeader<Cookie>>,
|
||||||
) -> AppResult<(HeaderMap, StatusCode)> {
|
) -> AppResult<(HeaderMap, StatusCode)> {
|
||||||
let mut conn = state.db()?;
|
let mut conn = state.db_unscoped()?;
|
||||||
let now = Utc::now().naive_utc();
|
let now = Utc::now().naive_utc();
|
||||||
let mut rows_affected = 0;
|
let mut rows_affected = 0;
|
||||||
|
|
||||||
@@ -214,6 +250,50 @@ pub async fn me(user: AuthenticatedUser) -> Json<AuthenticatedUser> {
|
|||||||
Json(user)
|
Json(user)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn issue_session(
|
||||||
|
state: &AppState,
|
||||||
|
conn: &mut PgConnection,
|
||||||
|
user: &User,
|
||||||
|
tenant_id: Uuid,
|
||||||
|
) -> AppResult<Response> {
|
||||||
|
let now = Utc::now();
|
||||||
|
let access_token = state
|
||||||
|
.jwt
|
||||||
|
.generate_token(user.id, tenant_id, &user.username)
|
||||||
|
.map_err(AppError::from)?;
|
||||||
|
|
||||||
|
let refresh_value = generate_refresh_token();
|
||||||
|
let refresh_hash = hash_refresh_token(&refresh_value);
|
||||||
|
let refresh_expires_at = now + ChronoDuration::days(state.config.refresh_token_expiry_days);
|
||||||
|
|
||||||
|
let new_refresh = NewRefreshToken {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
user_id: user.id,
|
||||||
|
token_hash: refresh_hash,
|
||||||
|
issued_at: now.naive_utc(),
|
||||||
|
expires_at: refresh_expires_at.naive_utc(),
|
||||||
|
tenant_id,
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(refresh_tokens::table)
|
||||||
|
.values(&new_refresh)
|
||||||
|
.execute(conn)?;
|
||||||
|
|
||||||
|
let mut response = Json(LoginResponse {
|
||||||
|
access_token,
|
||||||
|
token_type: "Bearer".to_string(),
|
||||||
|
expires_in: state.config.jwt_expiry_minutes * 60,
|
||||||
|
})
|
||||||
|
.into_response();
|
||||||
|
|
||||||
|
response.headers_mut().insert(
|
||||||
|
SET_COOKIE,
|
||||||
|
build_refresh_cookie(state, &refresh_value, refresh_expires_at),
|
||||||
|
);
|
||||||
|
|
||||||
|
Ok(response)
|
||||||
|
}
|
||||||
|
|
||||||
fn hash_refresh_token(token: &str) -> String {
|
fn hash_refresh_token(token: &str) -> String {
|
||||||
let mut hasher = Sha256::new();
|
let mut hasher = Sha256::new();
|
||||||
hasher.update(token.as_bytes());
|
hasher.update(token.as_bytes());
|
||||||
|
|||||||
@@ -1,11 +1,6 @@
|
|||||||
use std::collections::{BTreeMap, HashMap};
|
use std::collections::{BTreeMap, HashMap};
|
||||||
|
|
||||||
use axum::{
|
use axum::{extract::Path, http::StatusCode, Json};
|
||||||
extract::{Path, State},
|
|
||||||
http::StatusCode,
|
|
||||||
response::IntoResponse,
|
|
||||||
Json,
|
|
||||||
};
|
|
||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
use diesel::{dsl::count_star, prelude::*, result::DatabaseErrorKind, PgConnection};
|
use diesel::{dsl::count_star, prelude::*, result::DatabaseErrorKind, PgConnection};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
@@ -13,14 +8,16 @@ use serde_json::Value;
|
|||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
|
auth::TenantScopedConn,
|
||||||
error::{AppError, AppResult},
|
error::{AppError, AppResult},
|
||||||
models::{Correspondent, NewCorrespondent},
|
models::{Correspondent, NewCorrespondent},
|
||||||
schema::{correspondents, document_correspondents},
|
schema::{correspondents, document_correspondents},
|
||||||
state::AppState,
|
utils::{
|
||||||
|
db::{no_content, EnsureEntity, IntoJsonResponse},
|
||||||
|
time::to_iso,
|
||||||
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::documents::to_iso;
|
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
pub struct CorrespondentUsage {
|
pub struct CorrespondentUsage {
|
||||||
pub total: i64,
|
pub total: i64,
|
||||||
@@ -59,15 +56,19 @@ struct CorrespondentChangeset<'a> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn list_correspondents(
|
pub async fn list_correspondents(
|
||||||
State(state): State<AppState>,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
) -> AppResult<Json<Vec<CorrespondentSummary>>> {
|
) -> AppResult<Json<Vec<CorrespondentSummary>>> {
|
||||||
let mut conn = state.db()?;
|
|
||||||
|
|
||||||
let correspondents_list: Vec<Correspondent> = correspondents::table
|
let correspondents_list: Vec<Correspondent> = correspondents::table
|
||||||
|
.filter(correspondents::tenant_id.eq(tenant_id))
|
||||||
.order(correspondents::name.asc())
|
.order(correspondents::name.asc())
|
||||||
.load(&mut conn)?;
|
.load(&mut conn)?;
|
||||||
|
|
||||||
let usage_rows: Vec<(Uuid, String, i64)> = document_correspondents::table
|
let usage_rows: Vec<(Uuid, String, i64)> = document_correspondents::table
|
||||||
|
.filter(document_correspondents::tenant_id.eq(tenant_id))
|
||||||
.group_by((
|
.group_by((
|
||||||
document_correspondents::correspondent_id,
|
document_correspondents::correspondent_id,
|
||||||
document_correspondents::role,
|
document_correspondents::role,
|
||||||
@@ -93,11 +94,15 @@ pub async fn list_correspondents(
|
|||||||
response.push(build_summary(correspondent, role_counts));
|
response.push(build_summary(correspondent, role_counts));
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Json(response))
|
response.into_json()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn create_correspondent(
|
pub async fn create_correspondent(
|
||||||
State(state): State<AppState>,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
Json(payload): Json<CreateCorrespondentRequest>,
|
Json(payload): Json<CreateCorrespondentRequest>,
|
||||||
) -> AppResult<Json<CorrespondentSummary>> {
|
) -> AppResult<Json<CorrespondentSummary>> {
|
||||||
let name = payload.name.trim();
|
let name = payload.name.trim();
|
||||||
@@ -111,9 +116,9 @@ pub async fn create_correspondent(
|
|||||||
id: new_id,
|
id: new_id,
|
||||||
name: name.to_string(),
|
name: name.to_string(),
|
||||||
metadata: metadata_value,
|
metadata: metadata_value,
|
||||||
|
tenant_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut conn = state.db()?;
|
|
||||||
match diesel::insert_into(correspondents::table)
|
match diesel::insert_into(correspondents::table)
|
||||||
.values(&new_correspondent)
|
.values(&new_correspondent)
|
||||||
.execute(&mut conn)
|
.execute(&mut conn)
|
||||||
@@ -125,19 +130,29 @@ pub async fn create_correspondent(
|
|||||||
Err(err) => return Err(AppError::from(err)),
|
Err(err) => return Err(AppError::from(err)),
|
||||||
}
|
}
|
||||||
|
|
||||||
let correspondent: Correspondent = correspondents::table.find(new_id).first(&mut conn)?;
|
let correspondent: Correspondent = correspondents::table
|
||||||
Ok(Json(build_summary(correspondent, BTreeMap::new())))
|
.find(new_id)
|
||||||
|
.filter(correspondents::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)
|
||||||
|
.one()?;
|
||||||
|
|
||||||
|
build_summary(correspondent, BTreeMap::new()).into_json()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn update_correspondent(
|
pub async fn update_correspondent(
|
||||||
State(state): State<AppState>,
|
|
||||||
Path(correspondent_id): Path<Uuid>,
|
Path(correspondent_id): Path<Uuid>,
|
||||||
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
Json(payload): Json<UpdateCorrespondentRequest>,
|
Json(payload): Json<UpdateCorrespondentRequest>,
|
||||||
) -> AppResult<Json<CorrespondentSummary>> {
|
) -> AppResult<Json<CorrespondentSummary>> {
|
||||||
let mut conn = state.db()?;
|
|
||||||
let existing: Correspondent = correspondents::table
|
let existing: Correspondent = correspondents::table
|
||||||
.find(correspondent_id)
|
.find(correspondent_id)
|
||||||
.first(&mut conn)?;
|
.filter(correspondents::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)
|
||||||
|
.one()?;
|
||||||
|
|
||||||
let mut new_name: Option<String> = None;
|
let mut new_name: Option<String> = None;
|
||||||
if let Some(ref candidate) = payload.name {
|
if let Some(ref candidate) = payload.name {
|
||||||
@@ -149,6 +164,7 @@ pub async fn update_correspondent(
|
|||||||
let duplicate = correspondents::table
|
let duplicate = correspondents::table
|
||||||
.filter(correspondents::name.eq(trimmed))
|
.filter(correspondents::name.eq(trimmed))
|
||||||
.filter(correspondents::id.ne(correspondent_id))
|
.filter(correspondents::id.ne(correspondent_id))
|
||||||
|
.filter(correspondents::tenant_id.eq(tenant_id))
|
||||||
.first::<Correspondent>(&mut conn)
|
.first::<Correspondent>(&mut conn)
|
||||||
.optional()?;
|
.optional()?;
|
||||||
if duplicate.is_some() {
|
if duplicate.is_some() {
|
||||||
@@ -167,8 +183,8 @@ pub async fn update_correspondent(
|
|||||||
}
|
}
|
||||||
|
|
||||||
if new_name.is_none() && new_metadata.is_none() {
|
if new_name.is_none() && new_metadata.is_none() {
|
||||||
let usage = load_usage_for_correspondent(&mut conn, correspondent_id)?;
|
let usage = load_usage_for_correspondent(&mut conn, tenant_id, correspondent_id)?;
|
||||||
return Ok(Json(build_summary(existing.clone(), usage)));
|
return build_summary(existing.clone(), usage).into_json();
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut changeset = CorrespondentChangeset::default();
|
let mut changeset = CorrespondentChangeset::default();
|
||||||
@@ -180,24 +196,33 @@ pub async fn update_correspondent(
|
|||||||
}
|
}
|
||||||
|
|
||||||
let now = Utc::now().naive_utc();
|
let now = Utc::now().naive_utc();
|
||||||
diesel::update(correspondents::table.find(correspondent_id))
|
diesel::update(
|
||||||
.set((&changeset, correspondents::updated_at.eq(now)))
|
correspondents::table
|
||||||
.execute(&mut conn)?;
|
.find(correspondent_id)
|
||||||
|
.filter(correspondents::tenant_id.eq(tenant_id)),
|
||||||
|
)
|
||||||
|
.set((&changeset, correspondents::updated_at.eq(now)))
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
let updated: Correspondent = correspondents::table
|
let updated: Correspondent = correspondents::table
|
||||||
.find(correspondent_id)
|
.find(correspondent_id)
|
||||||
.first(&mut conn)?;
|
.filter(correspondents::tenant_id.eq(tenant_id))
|
||||||
let usage = load_usage_for_correspondent(&mut conn, correspondent_id)?;
|
.first(&mut conn)
|
||||||
Ok(Json(build_summary(updated, usage)))
|
.one()?;
|
||||||
|
let usage = load_usage_for_correspondent(&mut conn, tenant_id, correspondent_id)?;
|
||||||
|
build_summary(updated, usage).into_json()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn delete_correspondent(
|
pub async fn delete_correspondent(
|
||||||
State(state): State<AppState>,
|
|
||||||
Path(correspondent_id): Path<Uuid>,
|
Path(correspondent_id): Path<Uuid>,
|
||||||
) -> AppResult<impl IntoResponse> {
|
TenantScopedConn {
|
||||||
let mut conn = state.db()?;
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
|
) -> AppResult<StatusCode> {
|
||||||
let usage: i64 = document_correspondents::table
|
let usage: i64 = document_correspondents::table
|
||||||
|
.filter(document_correspondents::tenant_id.eq(tenant_id))
|
||||||
.filter(document_correspondents::correspondent_id.eq(correspondent_id))
|
.filter(document_correspondents::correspondent_id.eq(correspondent_id))
|
||||||
.select(count_star())
|
.select(count_star())
|
||||||
.first(&mut conn)?;
|
.first(&mut conn)?;
|
||||||
@@ -208,12 +233,16 @@ pub async fn delete_correspondent(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let deleted =
|
let deleted = diesel::delete(
|
||||||
diesel::delete(correspondents::table.find(correspondent_id)).execute(&mut conn)?;
|
correspondents::table
|
||||||
|
.filter(correspondents::id.eq(correspondent_id))
|
||||||
|
.filter(correspondents::tenant_id.eq(tenant_id)),
|
||||||
|
)
|
||||||
|
.execute(&mut conn)?;
|
||||||
if deleted == 0 {
|
if deleted == 0 {
|
||||||
return Err(AppError::not_found());
|
return Err(AppError::not_found());
|
||||||
}
|
}
|
||||||
Ok(StatusCode::NO_CONTENT)
|
no_content()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn build_summary(
|
fn build_summary(
|
||||||
@@ -243,10 +272,12 @@ fn normalize_metadata(input: Option<Value>) -> Value {
|
|||||||
|
|
||||||
fn load_usage_for_correspondent(
|
fn load_usage_for_correspondent(
|
||||||
conn: &mut PgConnection,
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
correspondent_id: Uuid,
|
correspondent_id: Uuid,
|
||||||
) -> AppResult<BTreeMap<String, i64>> {
|
) -> AppResult<BTreeMap<String, i64>> {
|
||||||
let rows: Vec<(String, i64)> = document_correspondents::table
|
let rows: Vec<(String, i64)> = document_correspondents::table
|
||||||
.filter(document_correspondents::correspondent_id.eq(correspondent_id))
|
.filter(document_correspondents::correspondent_id.eq(correspondent_id))
|
||||||
|
.filter(document_correspondents::tenant_id.eq(tenant_id))
|
||||||
.group_by(document_correspondents::role)
|
.group_by(document_correspondents::role)
|
||||||
.select((document_correspondents::role, count_star()))
|
.select((document_correspondents::role, count_star()))
|
||||||
.load(conn)?;
|
.load(conn)?;
|
||||||
|
|||||||
+988
-611
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,120 @@
|
|||||||
|
use std::path::Path as FsPath;
|
||||||
|
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use crate::error::{AppError, AppResult};
|
||||||
|
use crate::models::{Document, DocumentAsset, DocumentAssetObject, DocumentVersion};
|
||||||
|
use crate::state::AppState;
|
||||||
|
use crate::utils::time::to_iso;
|
||||||
|
|
||||||
|
use super::{
|
||||||
|
DocumentAssetDetailResponse, DocumentAssetObjectResponse, DocumentAssetResponse,
|
||||||
|
DocumentVersionResponse,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub fn build_download_path(
|
||||||
|
state: &AppState,
|
||||||
|
document: &Document,
|
||||||
|
user_id: Uuid,
|
||||||
|
) -> AppResult<String> {
|
||||||
|
state
|
||||||
|
.jwt
|
||||||
|
.generate_download_token(document.id, user_id, document.tenant_id)
|
||||||
|
.map(|token| format!("/download/{token}"))
|
||||||
|
.map_err(|err| AppError::internal(format!("failed to generate download token: {err}")))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_version_response(
|
||||||
|
version: DocumentVersion,
|
||||||
|
include_operations_summary: bool,
|
||||||
|
) -> DocumentVersionResponse {
|
||||||
|
DocumentVersionResponse {
|
||||||
|
id: version.id,
|
||||||
|
version_number: version.version_number,
|
||||||
|
s3_key: version.s3_key,
|
||||||
|
size_bytes: version.size_bytes,
|
||||||
|
checksum: version.checksum,
|
||||||
|
created_at: to_iso(version.created_at),
|
||||||
|
metadata: version.metadata,
|
||||||
|
operations_summary: if include_operations_summary {
|
||||||
|
Some(version.operations_summary)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_asset_summary(asset: DocumentAsset) -> DocumentAssetResponse {
|
||||||
|
DocumentAssetResponse {
|
||||||
|
id: asset.id,
|
||||||
|
asset_type: asset.asset_type,
|
||||||
|
mime_type: asset.mime_type,
|
||||||
|
metadata: asset.metadata,
|
||||||
|
cardinality: asset.cardinality,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_asset_detail_response(
|
||||||
|
asset: DocumentAsset,
|
||||||
|
objects: Vec<DocumentAssetObjectResponse>,
|
||||||
|
) -> DocumentAssetDetailResponse {
|
||||||
|
DocumentAssetDetailResponse {
|
||||||
|
id: asset.id,
|
||||||
|
asset_type: asset.asset_type,
|
||||||
|
mime_type: asset.mime_type,
|
||||||
|
metadata: asset.metadata,
|
||||||
|
created_at: to_iso(asset.created_at),
|
||||||
|
cardinality: asset.cardinality,
|
||||||
|
objects,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_asset_object_response(
|
||||||
|
object: DocumentAssetObject,
|
||||||
|
url: Option<String>,
|
||||||
|
expires_at: Option<i64>,
|
||||||
|
) -> DocumentAssetObjectResponse {
|
||||||
|
DocumentAssetObjectResponse {
|
||||||
|
id: object.id,
|
||||||
|
ordinal: object.ordinal,
|
||||||
|
metadata: object.metadata,
|
||||||
|
url,
|
||||||
|
expires_at,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn derive_document_title(original: &str) -> String {
|
||||||
|
let trimmed = original.trim();
|
||||||
|
if trimmed.is_empty() {
|
||||||
|
return "Document".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
let stem = FsPath::new(trimmed)
|
||||||
|
.file_stem()
|
||||||
|
.and_then(|s| s.to_str())
|
||||||
|
.map(|s| s.trim())
|
||||||
|
.filter(|s| !s.is_empty())
|
||||||
|
.map(|s| s.to_string());
|
||||||
|
|
||||||
|
stem.unwrap_or_else(|| trimmed.to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn filename_with_retained_extension(title: &str, current_filename: &str) -> String {
|
||||||
|
let extension = FsPath::new(current_filename)
|
||||||
|
.extension()
|
||||||
|
.and_then(|ext| ext.to_str());
|
||||||
|
|
||||||
|
if let Some(ext) = extension {
|
||||||
|
if title
|
||||||
|
.rsplit_once('.')
|
||||||
|
.map(|(_, existing_ext)| existing_ext.eq_ignore_ascii_case(ext))
|
||||||
|
.unwrap_or(false)
|
||||||
|
{
|
||||||
|
title.to_string()
|
||||||
|
} else {
|
||||||
|
format!("{title}.{ext}")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
title.to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,61 @@
|
|||||||
|
use std::collections::HashSet;
|
||||||
|
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use crate::error::{AppError, AppResult};
|
||||||
|
|
||||||
|
use super::CorrespondentAssignmentInput;
|
||||||
|
|
||||||
|
pub const CORRESPONDENT_ROLES: &[&str] = &["sender", "receiver", "other"];
|
||||||
|
|
||||||
|
pub fn normalize_role(value: &str) -> String {
|
||||||
|
value.trim().to_lowercase()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn is_valid_correspondent_role(role: &str) -> bool {
|
||||||
|
CORRESPONDENT_ROLES.iter().any(|allowed| *allowed == role)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn normalize_correspondent_assignments(
|
||||||
|
assignments: &[CorrespondentAssignmentInput],
|
||||||
|
) -> AppResult<(Vec<(Uuid, String)>, Vec<Uuid>, Vec<String>)> {
|
||||||
|
let mut unique_pairs: HashSet<(Uuid, String)> = HashSet::new();
|
||||||
|
let mut normalized_pairs: Vec<(Uuid, String)> = Vec::new();
|
||||||
|
let mut role_set: HashSet<String> = HashSet::new();
|
||||||
|
let mut correspondent_ids: HashSet<Uuid> = HashSet::new();
|
||||||
|
|
||||||
|
for assignment in assignments {
|
||||||
|
let role = normalize_role(&assignment.role);
|
||||||
|
if role.is_empty() {
|
||||||
|
return Err(AppError::bad_request("role must not be empty"));
|
||||||
|
}
|
||||||
|
if !is_valid_correspondent_role(&role) {
|
||||||
|
return Err(AppError::bad_request(format!(
|
||||||
|
"invalid correspondent role '{role}'. Allowed roles: {}",
|
||||||
|
CORRESPONDENT_ROLES.join(", ")
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
if !unique_pairs.insert((assignment.correspondent_id, role.clone())) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
normalized_pairs.push((assignment.correspondent_id, role.clone()));
|
||||||
|
role_set.insert(role);
|
||||||
|
correspondent_ids.insert(assignment.correspondent_id);
|
||||||
|
}
|
||||||
|
|
||||||
|
if normalized_pairs.is_empty() {
|
||||||
|
return Err(AppError::bad_request(
|
||||||
|
"assignments must contain at least one unique correspondent/role pair",
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut correspondents_vec: Vec<Uuid> = correspondent_ids.into_iter().collect();
|
||||||
|
correspondents_vec.sort();
|
||||||
|
|
||||||
|
let mut roles_vec: Vec<String> = role_set.into_iter().collect();
|
||||||
|
roles_vec.sort();
|
||||||
|
|
||||||
|
Ok((normalized_pairs, correspondents_vec, roles_vec))
|
||||||
|
}
|
||||||
@@ -0,0 +1,91 @@
|
|||||||
|
use serde_json::Value;
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
pub fn build_quickwit_query(input: &str) -> Option<String> {
|
||||||
|
let tokens: Vec<String> = input
|
||||||
|
.split_whitespace()
|
||||||
|
.filter(|token| !token.is_empty())
|
||||||
|
.map(|token| {
|
||||||
|
let normalized = token.to_lowercase();
|
||||||
|
escape_quickwit_token(&normalized)
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
if tokens.is_empty() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let parts: Vec<String> = tokens
|
||||||
|
.into_iter()
|
||||||
|
.map(|token| format!("(title:{token} OR text:{token})"))
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
Some(parts.join(" AND "))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn escape_quickwit_token(token: &str) -> String {
|
||||||
|
let mut escaped = String::with_capacity(token.len());
|
||||||
|
for ch in token.chars() {
|
||||||
|
match ch {
|
||||||
|
'+' | '-' | '&' | '|' | '!' | '(' | ')' | '{' | '}' | '[' | ']' | '^' | '"' | '~'
|
||||||
|
| '*' | '?' | ':' | '\\' | '/' => {
|
||||||
|
escaped.push('\\');
|
||||||
|
escaped.push(ch);
|
||||||
|
}
|
||||||
|
_ => escaped.push(ch),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
escaped
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn extract_document_id(hit: &Value) -> Option<Uuid> {
|
||||||
|
for key in ["_source", "source", "fields", "stored_fields"] {
|
||||||
|
if let Some(value) = hit.get(key) {
|
||||||
|
if let Some(uuid) = extract_uuid_from_value(value) {
|
||||||
|
return Some(uuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(value) = hit.get("document_id") {
|
||||||
|
if let Some(uuid) = extract_uuid_from_value(value) {
|
||||||
|
return Some(uuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn extract_uuid_from_value(value: &Value) -> Option<Uuid> {
|
||||||
|
if let Some(obj) = value.as_object() {
|
||||||
|
if let Some(inner) = obj.get("document_id") {
|
||||||
|
return parse_uuid_value(inner);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(arr) = value.as_array() {
|
||||||
|
for item in arr {
|
||||||
|
if let Some(uuid) = extract_uuid_from_value(item) {
|
||||||
|
return Some(uuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
parse_uuid_value(value)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn parse_uuid_value(value: &Value) -> Option<Uuid> {
|
||||||
|
if let Some(s) = value.as_str() {
|
||||||
|
return Uuid::parse_str(s).ok();
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(arr) = value.as_array() {
|
||||||
|
for item in arr {
|
||||||
|
if let Some(uuid) = parse_uuid_value(item) {
|
||||||
|
return Some(uuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
None
|
||||||
|
}
|
||||||
+222
-72
@@ -4,19 +4,24 @@ use axum::{
|
|||||||
};
|
};
|
||||||
use diesel::{dsl::exists, prelude::*, PgConnection};
|
use diesel::{dsl::exists, prelude::*, PgConnection};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serde_json::Value;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::models::{Document, Folder, NewFolder};
|
use crate::models::{Document, Folder, NewFolder};
|
||||||
use crate::schema::{documents, folders};
|
use crate::schema::{documents, folders};
|
||||||
use crate::state::AppState;
|
use crate::state::AppState;
|
||||||
use crate::{
|
use crate::{
|
||||||
auth::AuthenticatedUser,
|
auth::TenantScopedConn,
|
||||||
error::{AppError, AppResult},
|
error::{AppError, AppResult},
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::documents::{
|
use super::documents::{
|
||||||
load_correspondents_for_documents, load_primary_assets, load_tags_for_documents,
|
load_correspondents_for_documents, load_primary_assets, load_tags_for_documents,
|
||||||
to_document_response, to_iso, DocumentResponse,
|
to_document_response, DocumentResponse,
|
||||||
|
};
|
||||||
|
use crate::utils::{
|
||||||
|
json::{classify_nullable, NullableValue},
|
||||||
|
time::to_iso,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
@@ -31,13 +36,6 @@ pub struct EnsureFolderPathRequest {
|
|||||||
pub segments: Vec<String>,
|
pub segments: Vec<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
pub struct UpdateFolderRequest {
|
|
||||||
#[serde(default)]
|
|
||||||
pub parent_id: Option<Option<Uuid>>,
|
|
||||||
pub name: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
pub struct FolderResponse {
|
pub struct FolderResponse {
|
||||||
pub folder: FolderInfo,
|
pub folder: FolderInfo,
|
||||||
@@ -69,16 +67,36 @@ pub struct FolderInfo {
|
|||||||
pub updated_at: String,
|
pub updated_at: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn get_folder(
|
||||||
|
Path(folder_id): Path<Uuid>,
|
||||||
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
|
) -> AppResult<Json<FolderResponse>> {
|
||||||
|
let folder: Folder = folders::table
|
||||||
|
.find(folder_id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)?;
|
||||||
|
|
||||||
|
Ok(Json(FolderResponse {
|
||||||
|
folder: folder_to_info(folder),
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn ensure_folder_path(
|
pub async fn ensure_folder_path(
|
||||||
State(state): State<AppState>,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
Json(payload): Json<EnsureFolderPathRequest>,
|
Json(payload): Json<EnsureFolderPathRequest>,
|
||||||
) -> AppResult<Json<FolderResponse>> {
|
) -> AppResult<Json<FolderResponse>> {
|
||||||
if payload.segments.is_empty() {
|
if payload.segments.is_empty() {
|
||||||
return Err(AppError::bad_request("segments must not be empty"));
|
return Err(AppError::bad_request("segments must not be empty"));
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut conn = state.db()?;
|
|
||||||
|
|
||||||
let target_folder = conn.transaction::<Folder, AppError, _>(|conn| {
|
let target_folder = conn.transaction::<Folder, AppError, _>(|conn| {
|
||||||
let mut current_parent = payload.parent_id;
|
let mut current_parent = payload.parent_id;
|
||||||
let mut last_folder: Option<Folder> = None;
|
let mut last_folder: Option<Folder> = None;
|
||||||
@@ -91,12 +109,14 @@ pub async fn ensure_folder_path(
|
|||||||
|
|
||||||
let existing: Option<Folder> = if let Some(parent_id) = current_parent {
|
let existing: Option<Folder> = if let Some(parent_id) = current_parent {
|
||||||
folders::table
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.filter(folders::parent_id.eq(Some(parent_id)))
|
.filter(folders::parent_id.eq(Some(parent_id)))
|
||||||
.filter(folders::name.eq(name))
|
.filter(folders::name.eq(name))
|
||||||
.first(conn)
|
.first(conn)
|
||||||
.optional()?
|
.optional()?
|
||||||
} else {
|
} else {
|
||||||
folders::table
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.filter(folders::parent_id.is_null())
|
.filter(folders::parent_id.is_null())
|
||||||
.filter(folders::name.eq(name))
|
.filter(folders::name.eq(name))
|
||||||
.first(conn)
|
.first(conn)
|
||||||
@@ -110,15 +130,35 @@ pub async fn ensure_folder_path(
|
|||||||
id: Uuid::new_v4(),
|
id: Uuid::new_v4(),
|
||||||
name: name.to_string(),
|
name: name.to_string(),
|
||||||
parent_id: current_parent,
|
parent_id: current_parent,
|
||||||
|
tenant_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
diesel::insert_into(folders::table)
|
let inserted_id: Option<Uuid> = diesel::insert_into(folders::table)
|
||||||
.values(&new_folder)
|
.values(&new_folder)
|
||||||
.execute(conn)?;
|
.on_conflict_do_nothing()
|
||||||
|
.returning(folders::id)
|
||||||
|
.get_result(conn)
|
||||||
|
.optional()?;
|
||||||
|
|
||||||
folders::table.find(new_folder.id).first(conn)?
|
if let Some(id) = inserted_id {
|
||||||
|
folders::table
|
||||||
|
.find(id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first(conn)?
|
||||||
|
} else if let Some(parent_id) = current_parent {
|
||||||
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.filter(folders::parent_id.eq(Some(parent_id)))
|
||||||
|
.filter(folders::name.eq(name))
|
||||||
|
.first(conn)?
|
||||||
|
} else {
|
||||||
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.filter(folders::parent_id.is_null())
|
||||||
|
.filter(folders::name.eq(name))
|
||||||
|
.first(conn)?
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
current_parent = Some(folder.id);
|
current_parent = Some(folder.id);
|
||||||
last_folder = Some(folder);
|
last_folder = Some(folder);
|
||||||
}
|
}
|
||||||
@@ -132,26 +172,72 @@ pub async fn ensure_folder_path(
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn create_folder(
|
pub async fn create_folder(
|
||||||
State(state): State<AppState>,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
Json(payload): Json<CreateFolderRequest>,
|
Json(payload): Json<CreateFolderRequest>,
|
||||||
) -> AppResult<Json<FolderResponse>> {
|
) -> AppResult<Json<FolderResponse>> {
|
||||||
if payload.name.trim().is_empty() {
|
if payload.name.trim().is_empty() {
|
||||||
return Err(AppError::bad_request("name must not be empty"));
|
return Err(AppError::bad_request("name must not be empty"));
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut conn = state.db()?;
|
let name = payload.name.trim();
|
||||||
|
|
||||||
let new_folder = NewFolder {
|
let existing: Option<Folder> = if let Some(parent_id) = payload.parent_id {
|
||||||
id: Uuid::new_v4(),
|
folders::table
|
||||||
name: payload.name.trim().to_string(),
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
parent_id: payload.parent_id,
|
.filter(folders::parent_id.eq(Some(parent_id)))
|
||||||
|
.filter(folders::name.eq(name))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
|
} else {
|
||||||
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.filter(folders::parent_id.is_null())
|
||||||
|
.filter(folders::name.eq(name))
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()?
|
||||||
};
|
};
|
||||||
|
|
||||||
diesel::insert_into(folders::table)
|
let folder: Folder = if let Some(folder) = existing {
|
||||||
.values(&new_folder)
|
folder
|
||||||
.execute(&mut conn)?;
|
} else {
|
||||||
|
let new_folder = NewFolder {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
name: name.to_string(),
|
||||||
|
parent_id: payload.parent_id,
|
||||||
|
tenant_id,
|
||||||
|
};
|
||||||
|
|
||||||
|
let inserted_id: Option<Uuid> = diesel::insert_into(folders::table)
|
||||||
|
.values(&new_folder)
|
||||||
|
.on_conflict_do_nothing()
|
||||||
|
.returning(folders::id)
|
||||||
|
.get_result(&mut conn)
|
||||||
|
.optional()?;
|
||||||
|
|
||||||
|
if let Some(id) = inserted_id {
|
||||||
|
folders::table
|
||||||
|
.find(id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)?
|
||||||
|
} else if let Some(parent_id) = payload.parent_id {
|
||||||
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.filter(folders::parent_id.eq(Some(parent_id)))
|
||||||
|
.filter(folders::name.eq(name))
|
||||||
|
.first(&mut conn)?
|
||||||
|
} else {
|
||||||
|
folders::table
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.filter(folders::parent_id.is_null())
|
||||||
|
.filter(folders::name.eq(name))
|
||||||
|
.first(&mut conn)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
let folder: Folder = folders::table.find(new_folder.id).first(&mut conn)?;
|
|
||||||
Ok(Json(FolderResponse {
|
Ok(Json(FolderResponse {
|
||||||
folder: folder_to_info(folder),
|
folder: folder_to_info(folder),
|
||||||
}))
|
}))
|
||||||
@@ -161,10 +247,13 @@ pub async fn list_folder_contents(
|
|||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Path(folder_identifier): Path<String>,
|
Path(folder_identifier): Path<String>,
|
||||||
Query(query): Query<FolderContentsQuery>,
|
Query(query): Query<FolderContentsQuery>,
|
||||||
user: AuthenticatedUser,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
user_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
) -> AppResult<Json<FolderContentsResponse>> {
|
) -> AppResult<Json<FolderContentsResponse>> {
|
||||||
let mut conn = state.db()?;
|
|
||||||
|
|
||||||
let folder_id = if folder_identifier.eq_ignore_ascii_case("root") {
|
let folder_id = if folder_identifier.eq_ignore_ascii_case("root") {
|
||||||
None
|
None
|
||||||
} else {
|
} else {
|
||||||
@@ -176,7 +265,10 @@ pub async fn list_folder_contents(
|
|||||||
|
|
||||||
let folder = match folder_id {
|
let folder = match folder_id {
|
||||||
Some(id) => Some(folder_to_info(
|
Some(id) => Some(folder_to_info(
|
||||||
folders::table.find(id).first::<Folder>(&mut conn)?,
|
folders::table
|
||||||
|
.find(id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first::<Folder>(&mut conn)?,
|
||||||
)),
|
)),
|
||||||
None => None,
|
None => None,
|
||||||
};
|
};
|
||||||
@@ -184,11 +276,13 @@ pub async fn list_folder_contents(
|
|||||||
let child_folders: Vec<Folder> = if let Some(parent_id) = folder_id {
|
let child_folders: Vec<Folder> = if let Some(parent_id) = folder_id {
|
||||||
folders::table
|
folders::table
|
||||||
.filter(folders::parent_id.eq(parent_id))
|
.filter(folders::parent_id.eq(parent_id))
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.order(folders::name.asc())
|
.order(folders::name.asc())
|
||||||
.load(&mut conn)?
|
.load(&mut conn)?
|
||||||
} else {
|
} else {
|
||||||
folders::table
|
folders::table
|
||||||
.filter(folders::parent_id.is_null())
|
.filter(folders::parent_id.is_null())
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.order(folders::name.asc())
|
.order(folders::name.asc())
|
||||||
.load(&mut conn)?
|
.load(&mut conn)?
|
||||||
};
|
};
|
||||||
@@ -197,6 +291,7 @@ pub async fn list_folder_contents(
|
|||||||
let documents = if query.include_documents {
|
let documents = if query.include_documents {
|
||||||
let docs_query = documents::table
|
let docs_query = documents::table
|
||||||
.filter(documents::deleted_at.is_null())
|
.filter(documents::deleted_at.is_null())
|
||||||
|
.filter(documents::tenant_id.eq(tenant_id))
|
||||||
.order(documents::uploaded_at.desc());
|
.order(documents::uploaded_at.desc());
|
||||||
|
|
||||||
let docs: Vec<Document> = if let Some(current_folder) = folder_id {
|
let docs: Vec<Document> = if let Some(current_folder) = folder_id {
|
||||||
@@ -214,7 +309,7 @@ pub async fn list_folder_contents(
|
|||||||
let mut correspondents_map = load_correspondents_for_documents(&mut conn, &doc_ids)?;
|
let mut correspondents_map = load_correspondents_for_documents(&mut conn, &doc_ids)?;
|
||||||
drop(conn);
|
drop(conn);
|
||||||
|
|
||||||
let primary_versions = load_primary_assets(&state, &docs).await?;
|
let primary_versions = load_primary_assets(&state, tenant_id, &docs).await?;
|
||||||
|
|
||||||
let mut documents = Vec::with_capacity(doc_ids.len());
|
let mut documents = Vec::with_capacity(doc_ids.len());
|
||||||
for doc in docs {
|
for doc in docs {
|
||||||
@@ -223,7 +318,7 @@ pub async fn list_folder_contents(
|
|||||||
let current_version = primary_versions.get(&doc.id).cloned();
|
let current_version = primary_versions.get(&doc.id).cloned();
|
||||||
documents.push(to_document_response(
|
documents.push(to_document_response(
|
||||||
&state,
|
&state,
|
||||||
user.user_id,
|
user_id,
|
||||||
doc,
|
doc,
|
||||||
tags,
|
tags,
|
||||||
correspondents,
|
correspondents,
|
||||||
@@ -244,16 +339,23 @@ pub async fn list_folder_contents(
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn delete_folder(
|
pub async fn delete_folder(
|
||||||
State(state): State<AppState>,
|
|
||||||
Path(folder_id): Path<Uuid>,
|
Path(folder_id): Path<Uuid>,
|
||||||
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
) -> AppResult<StatusCode> {
|
) -> AppResult<StatusCode> {
|
||||||
let mut conn = state.db()?;
|
|
||||||
|
|
||||||
conn.transaction::<_, AppError, _>(|conn| {
|
conn.transaction::<_, AppError, _>(|conn| {
|
||||||
folders::table.find(folder_id).first::<Folder>(conn)?;
|
folders::table
|
||||||
|
.find(folder_id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first::<Folder>(conn)?;
|
||||||
|
|
||||||
let has_child_folders: bool = diesel::select(exists(
|
let has_child_folders: bool = diesel::select(exists(
|
||||||
folders::table.filter(folders::parent_id.eq(Some(folder_id))),
|
folders::table
|
||||||
|
.filter(folders::parent_id.eq(Some(folder_id)))
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id)),
|
||||||
))
|
))
|
||||||
.get_result(conn)?;
|
.get_result(conn)?;
|
||||||
|
|
||||||
@@ -266,6 +368,7 @@ pub async fn delete_folder(
|
|||||||
let has_documents: bool = diesel::select(exists(
|
let has_documents: bool = diesel::select(exists(
|
||||||
documents::table
|
documents::table
|
||||||
.filter(documents::folder_id.eq(Some(folder_id)))
|
.filter(documents::folder_id.eq(Some(folder_id)))
|
||||||
|
.filter(documents::tenant_id.eq(tenant_id))
|
||||||
.filter(documents::deleted_at.is_null()),
|
.filter(documents::deleted_at.is_null()),
|
||||||
))
|
))
|
||||||
.get_result(conn)?;
|
.get_result(conn)?;
|
||||||
@@ -276,7 +379,12 @@ pub async fn delete_folder(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
diesel::delete(folders::table.find(folder_id)).execute(conn)?;
|
diesel::delete(
|
||||||
|
folders::table
|
||||||
|
.filter(folders::id.eq(folder_id))
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id)),
|
||||||
|
)
|
||||||
|
.execute(conn)?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
})?;
|
})?;
|
||||||
@@ -285,50 +393,84 @@ pub async fn delete_folder(
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn update_folder(
|
pub async fn update_folder(
|
||||||
State(state): State<AppState>,
|
|
||||||
Path(folder_id): Path<Uuid>,
|
Path(folder_id): Path<Uuid>,
|
||||||
Json(payload): Json<UpdateFolderRequest>,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
|
Json(body): Json<Value>,
|
||||||
) -> AppResult<StatusCode> {
|
) -> AppResult<StatusCode> {
|
||||||
let mut conn = state.db()?;
|
if !body.is_object() {
|
||||||
|
return Err(AppError::bad_request("request body must be a JSON object"));
|
||||||
|
}
|
||||||
|
|
||||||
|
let parent_class = classify_nullable(body.get("parent_id")).map_err(AppError::bad_request)?;
|
||||||
|
let name_class = classify_nullable(body.get("name")).map_err(AppError::bad_request)?;
|
||||||
|
|
||||||
conn.transaction::<(), AppError, _>(|conn| {
|
conn.transaction::<(), AppError, _>(|conn| {
|
||||||
let folder: Folder = folders::table.find(folder_id).first(conn)?;
|
let folder: Folder = folders::table
|
||||||
|
.find(folder_id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first(conn)?;
|
||||||
|
|
||||||
let mut next_parent = folder.parent_id;
|
let mut next_parent = folder.parent_id;
|
||||||
let mut parent_changed = false;
|
let mut parent_changed = false;
|
||||||
|
match parent_class {
|
||||||
if let Some(parent_request) = payload.parent_id {
|
NullableValue::Omitted => {}
|
||||||
if parent_request == Some(folder_id) {
|
NullableValue::Null => {
|
||||||
return Err(AppError::bad_request("folder cannot be its own parent"));
|
if folder.parent_id.is_some() {
|
||||||
}
|
parent_changed = true;
|
||||||
|
|
||||||
if let Some(parent_id) = parent_request {
|
|
||||||
let _parent: Folder = folders::table.find(parent_id).first(conn)?;
|
|
||||||
|
|
||||||
let descendant_ids = gather_descendant_folder_ids(conn, folder_id)?;
|
|
||||||
if descendant_ids.contains(&parent_id) {
|
|
||||||
return Err(AppError::bad_request(
|
|
||||||
"cannot move folder into itself or a descendant",
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
|
next_parent = None;
|
||||||
}
|
}
|
||||||
|
NullableValue::String(value) => {
|
||||||
|
let trimmed = value.trim();
|
||||||
|
if trimmed.is_empty() {
|
||||||
|
return Err(AppError::bad_request("parent_id must not be empty"));
|
||||||
|
}
|
||||||
|
let parent_id = Uuid::parse_str(trimmed)
|
||||||
|
.map_err(|_| AppError::bad_request("parent_id must be a valid UUID or null"))?;
|
||||||
|
if parent_id == folder_id {
|
||||||
|
return Err(AppError::bad_request("folder cannot be its own parent"));
|
||||||
|
}
|
||||||
|
|
||||||
parent_changed = parent_request != folder.parent_id;
|
let _parent: Folder = folders::table
|
||||||
next_parent = parent_request;
|
.find(parent_id)
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
|
.first(conn)?;
|
||||||
|
|
||||||
|
if folder.parent_id != Some(parent_id) {
|
||||||
|
let descendant_ids = gather_descendant_folder_ids(conn, tenant_id, folder_id)?;
|
||||||
|
if descendant_ids.contains(&parent_id) {
|
||||||
|
return Err(AppError::bad_request(
|
||||||
|
"cannot move folder into itself or a descendant",
|
||||||
|
));
|
||||||
|
}
|
||||||
|
parent_changed = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
next_parent = Some(parent_id);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut new_name = folder.name.clone();
|
let mut new_name = folder.name.clone();
|
||||||
let mut name_changed = false;
|
let mut name_changed = false;
|
||||||
|
match name_class {
|
||||||
if let Some(name) = payload.name {
|
NullableValue::Omitted => {}
|
||||||
let trimmed = name.trim();
|
NullableValue::Null => {
|
||||||
if trimmed.is_empty() {
|
return Err(AppError::bad_request("name cannot be null"));
|
||||||
return Err(AppError::bad_request("name must not be empty"));
|
|
||||||
}
|
}
|
||||||
|
NullableValue::String(value) => {
|
||||||
|
let trimmed = value.trim();
|
||||||
|
if trimmed.is_empty() {
|
||||||
|
return Err(AppError::bad_request("name must not be empty"));
|
||||||
|
}
|
||||||
|
|
||||||
if trimmed != folder.name {
|
if trimmed != folder.name {
|
||||||
new_name = trimmed.to_string();
|
new_name = trimmed.to_string();
|
||||||
name_changed = true;
|
name_changed = true;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -341,6 +483,7 @@ pub async fn update_folder(
|
|||||||
.filter(folders::parent_id.eq(Some(parent_id)))
|
.filter(folders::parent_id.eq(Some(parent_id)))
|
||||||
.filter(folders::name.eq(&new_name))
|
.filter(folders::name.eq(&new_name))
|
||||||
.filter(folders::id.ne(folder_id))
|
.filter(folders::id.ne(folder_id))
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.first::<Folder>(conn)
|
.first::<Folder>(conn)
|
||||||
.optional()?
|
.optional()?
|
||||||
} else {
|
} else {
|
||||||
@@ -348,6 +491,7 @@ pub async fn update_folder(
|
|||||||
.filter(folders::parent_id.is_null())
|
.filter(folders::parent_id.is_null())
|
||||||
.filter(folders::name.eq(&new_name))
|
.filter(folders::name.eq(&new_name))
|
||||||
.filter(folders::id.ne(folder_id))
|
.filter(folders::id.ne(folder_id))
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.first::<Folder>(conn)
|
.first::<Folder>(conn)
|
||||||
.optional()?
|
.optional()?
|
||||||
};
|
};
|
||||||
@@ -358,12 +502,16 @@ pub async fn update_folder(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
diesel::update(folders::table.find(folder_id))
|
diesel::update(
|
||||||
.set((
|
folders::table
|
||||||
folders::parent_id.eq(next_parent),
|
.find(folder_id)
|
||||||
folders::name.eq(&new_name),
|
.filter(folders::tenant_id.eq(tenant_id)),
|
||||||
))
|
)
|
||||||
.execute(conn)?;
|
.set((
|
||||||
|
folders::parent_id.eq(next_parent),
|
||||||
|
folders::name.eq(&new_name),
|
||||||
|
))
|
||||||
|
.execute(conn)?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
})?;
|
})?;
|
||||||
@@ -383,6 +531,7 @@ fn folder_to_info(folder: Folder) -> FolderInfo {
|
|||||||
|
|
||||||
pub(super) fn gather_descendant_folder_ids(
|
pub(super) fn gather_descendant_folder_ids(
|
||||||
conn: &mut PgConnection,
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
folder_id: Uuid,
|
folder_id: Uuid,
|
||||||
) -> AppResult<Vec<Uuid>> {
|
) -> AppResult<Vec<Uuid>> {
|
||||||
let mut ids = vec![folder_id];
|
let mut ids = vec![folder_id];
|
||||||
@@ -391,6 +540,7 @@ pub(super) fn gather_descendant_folder_ids(
|
|||||||
while let Some(current) = queue.pop() {
|
while let Some(current) = queue.pop() {
|
||||||
let child_ids: Vec<Uuid> = folders::table
|
let child_ids: Vec<Uuid> = folders::table
|
||||||
.filter(folders::parent_id.eq(Some(current)))
|
.filter(folders::parent_id.eq(Some(current)))
|
||||||
|
.filter(folders::tenant_id.eq(tenant_id))
|
||||||
.select(folders::id)
|
.select(folders::id)
|
||||||
.load(conn)?;
|
.load(conn)?;
|
||||||
queue.extend(child_ids.iter().copied());
|
queue.extend(child_ids.iter().copied());
|
||||||
|
|||||||
@@ -2,12 +2,15 @@ use axum::http::HeaderValue;
|
|||||||
use axum::{
|
use axum::{
|
||||||
extract::DefaultBodyLimit,
|
extract::DefaultBodyLimit,
|
||||||
middleware,
|
middleware,
|
||||||
|
response::Json,
|
||||||
routing::{delete, get, patch, post},
|
routing::{delete, get, patch, post},
|
||||||
Router,
|
Router,
|
||||||
};
|
};
|
||||||
|
use std::sync::Arc;
|
||||||
use tower_http::cors::{AllowOrigin, CorsLayer};
|
use tower_http::cors::{AllowOrigin, CorsLayer};
|
||||||
|
use utoipa::OpenApi;
|
||||||
|
|
||||||
use crate::{auth::AuthenticatedUser, state::AppState};
|
use crate::{auth::AuthenticatedUser, openapi::ApiDoc, state::AppState};
|
||||||
|
|
||||||
pub mod auth;
|
pub mod auth;
|
||||||
pub mod correspondents;
|
pub mod correspondents;
|
||||||
@@ -50,14 +53,15 @@ pub fn create_router(state: AppState) -> Router<()> {
|
|||||||
.route("/login", post(auth::login))
|
.route("/login", post(auth::login))
|
||||||
.route("/refresh", post(auth::refresh))
|
.route("/refresh", post(auth::refresh))
|
||||||
.route("/logout", post(auth::logout))
|
.route("/logout", post(auth::logout))
|
||||||
|
.route("/select-tenant", post(auth::select_tenant))
|
||||||
.route("/me", get(auth::me));
|
.route("/me", get(auth::me));
|
||||||
|
|
||||||
let documents_routes = Router::new()
|
let documents_routes = Router::new()
|
||||||
|
.route("/check", get(documents::check_document))
|
||||||
.route(
|
.route(
|
||||||
"/",
|
"/",
|
||||||
get(documents::list_documents).post(documents::upload_document),
|
get(documents::list_documents).post(documents::upload_document),
|
||||||
)
|
)
|
||||||
.route("/reanalyze", post(documents::reanalyze_all_documents))
|
|
||||||
.route("/bulk/move", post(documents::bulk_move_documents))
|
.route("/bulk/move", post(documents::bulk_move_documents))
|
||||||
.route("/bulk/tags", post(documents::bulk_update_tags))
|
.route("/bulk/tags", post(documents::bulk_update_tags))
|
||||||
.route(
|
.route(
|
||||||
@@ -75,7 +79,6 @@ pub fn create_router(state: AppState) -> Router<()> {
|
|||||||
.patch(documents::update_document),
|
.patch(documents::update_document),
|
||||||
)
|
)
|
||||||
.route("/:id/download", get(documents::download_document))
|
.route("/:id/download", get(documents::download_document))
|
||||||
.route("/:id/assets/:asset_id", get(documents::get_document_asset))
|
|
||||||
.route(
|
.route(
|
||||||
"/:id/assets",
|
"/:id/assets",
|
||||||
get(documents::list_document_assets).post(documents::request_document_assets),
|
get(documents::list_document_assets).post(documents::request_document_assets),
|
||||||
@@ -98,10 +101,9 @@ pub fn create_router(state: AppState) -> Router<()> {
|
|||||||
let folders_routes = Router::new()
|
let folders_routes = Router::new()
|
||||||
.route("/", post(folders::create_folder))
|
.route("/", post(folders::create_folder))
|
||||||
.route("/path", post(folders::ensure_folder_path))
|
.route("/path", post(folders::ensure_folder_path))
|
||||||
.route(
|
.route("/:id", get(folders::get_folder))
|
||||||
"/:id",
|
.route("/:id", delete(folders::delete_folder))
|
||||||
delete(folders::delete_folder).patch(folders::update_folder),
|
.route("/:id", patch(folders::update_folder))
|
||||||
)
|
|
||||||
.route("/:id/contents", get(folders::list_folder_contents));
|
.route("/:id/contents", get(folders::list_folder_contents));
|
||||||
|
|
||||||
let tags_routes = Router::new()
|
let tags_routes = Router::new()
|
||||||
@@ -120,16 +122,32 @@ pub fn create_router(state: AppState) -> Router<()> {
|
|||||||
);
|
);
|
||||||
|
|
||||||
let protected_state = state.clone();
|
let protected_state = state.clone();
|
||||||
|
let assets_routes = Router::new().route("/:asset_id", get(documents::get_document_asset));
|
||||||
|
|
||||||
let protected_routes = Router::new()
|
let protected_routes = Router::new()
|
||||||
.nest("/api/documents", documents_routes)
|
.nest("/api/documents", documents_routes)
|
||||||
.nest("/api/folders", folders_routes)
|
.nest("/api/folders", folders_routes)
|
||||||
.nest("/api/tags", tags_routes)
|
.nest("/api/tags", tags_routes)
|
||||||
.nest("/api/correspondents", correspondents_routes)
|
.nest("/api/correspondents", correspondents_routes)
|
||||||
|
.nest("/api/assets", assets_routes)
|
||||||
.layer(middleware::from_extractor_with_state::<AuthenticatedUser, _>(protected_state));
|
.layer(middleware::from_extractor_with_state::<AuthenticatedUser, _>(protected_state));
|
||||||
|
|
||||||
|
let openapi_arc = Arc::new(ApiDoc::openapi());
|
||||||
|
let docs_route = Router::new().route(
|
||||||
|
"/api/docs/openapi.json",
|
||||||
|
get({
|
||||||
|
let spec = openapi_arc.clone();
|
||||||
|
move || {
|
||||||
|
let spec = spec.clone();
|
||||||
|
async move { Json((*spec).clone()) }
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
|
||||||
Router::new()
|
Router::new()
|
||||||
.merge(download_routes)
|
.merge(download_routes)
|
||||||
.merge(protected_routes)
|
.merge(protected_routes)
|
||||||
|
.merge(docs_route)
|
||||||
.nest("/api/auth", auth_routes)
|
.nest("/api/auth", auth_routes)
|
||||||
.route("/api/health", get(health::health_check))
|
.route("/api/health", get(health::health_check))
|
||||||
.with_state(state)
|
.with_state(state)
|
||||||
|
|||||||
+77
-34
@@ -1,19 +1,16 @@
|
|||||||
use crate::utils::json::{classify_nullable, NullableValue};
|
use crate::utils::json::{classify_nullable, NullableValue};
|
||||||
use axum::{
|
use axum::{extract::Path, http::StatusCode, Json};
|
||||||
extract::{Path, State},
|
|
||||||
http::StatusCode,
|
|
||||||
Json,
|
|
||||||
};
|
|
||||||
use diesel::{dsl::count_star, prelude::*};
|
use diesel::{dsl::count_star, prelude::*};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use crate::auth::TenantScopedConn;
|
||||||
use crate::error::{AppError, AppResult};
|
use crate::error::{AppError, AppResult};
|
||||||
use crate::models::{NewTag, Tag};
|
use crate::models::{NewTag, Tag};
|
||||||
use crate::schema::{document_tags, tags};
|
use crate::schema::{document_tags, tags};
|
||||||
use crate::state::AppState;
|
use crate::utils::db::{no_content, EnsureEntity, IntoJsonResponse};
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
pub struct CreateTagRequest {
|
pub struct CreateTagRequest {
|
||||||
@@ -36,19 +33,27 @@ pub struct TagCatalogEntry {
|
|||||||
pub usage_count: i64,
|
pub usage_count: i64,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn list_tags(State(state): State<AppState>) -> AppResult<Json<Vec<TagCatalogEntry>>> {
|
pub async fn list_tags(
|
||||||
let mut conn = state.db()?;
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
let tag_list: Vec<Tag> = tags::table.order(tags::label.asc()).load(&mut conn)?;
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
|
) -> AppResult<Json<Vec<TagCatalogEntry>>> {
|
||||||
|
let tag_list: Vec<Tag> = tags::table
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id))
|
||||||
|
.order(tags::label.asc())
|
||||||
|
.load(&mut conn)?;
|
||||||
|
|
||||||
let usage_rows: Vec<(Uuid, i64)> = document_tags::table
|
let usage_rows: Vec<(Uuid, i64)> = document_tags::table
|
||||||
|
.filter(document_tags::tenant_id.eq(tenant_id))
|
||||||
.group_by(document_tags::tag_id)
|
.group_by(document_tags::tag_id)
|
||||||
.select((document_tags::tag_id, count_star()))
|
.select((document_tags::tag_id, count_star()))
|
||||||
.load(&mut conn)?;
|
.load(&mut conn)?;
|
||||||
|
|
||||||
let usage_map: HashMap<Uuid, i64> = usage_rows.into_iter().collect();
|
let usage_map: HashMap<Uuid, i64> = usage_rows.into_iter().collect();
|
||||||
|
|
||||||
let response = tag_list
|
let response: Vec<TagCatalogEntry> = tag_list
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|tag| TagCatalogEntry {
|
.map(|tag| TagCatalogEntry {
|
||||||
id: tag.id,
|
id: tag.id,
|
||||||
@@ -58,22 +63,26 @@ pub async fn list_tags(State(state): State<AppState>) -> AppResult<Json<Vec<TagC
|
|||||||
})
|
})
|
||||||
.collect();
|
.collect();
|
||||||
|
|
||||||
Ok(Json(response))
|
response.into_json()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn create_tag(
|
pub async fn create_tag(
|
||||||
State(state): State<AppState>,
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
Json(payload): Json<CreateTagRequest>,
|
Json(payload): Json<CreateTagRequest>,
|
||||||
) -> AppResult<Json<TagCatalogEntry>> {
|
) -> AppResult<Json<TagCatalogEntry>> {
|
||||||
if payload.label.trim().is_empty() {
|
if payload.label.trim().is_empty() {
|
||||||
return Err(AppError::bad_request("label must not be empty"));
|
return Err(AppError::bad_request("label must not be empty"));
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut conn = state.db()?;
|
|
||||||
let new_tag = NewTag {
|
let new_tag = NewTag {
|
||||||
id: Uuid::new_v4(),
|
id: Uuid::new_v4(),
|
||||||
label: payload.label.trim().to_string(),
|
label: payload.label.trim().to_string(),
|
||||||
color: payload.color,
|
color: payload.color,
|
||||||
|
tenant_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
match diesel::insert_into(tags::table)
|
match diesel::insert_into(tags::table)
|
||||||
@@ -90,22 +99,35 @@ pub async fn create_tag(
|
|||||||
Err(err) => return Err(AppError::from(err)),
|
Err(err) => return Err(AppError::from(err)),
|
||||||
}
|
}
|
||||||
|
|
||||||
let tag: Tag = tags::table.find(new_tag.id).first(&mut conn)?;
|
let tag: Tag = tags::table
|
||||||
Ok(Json(TagCatalogEntry {
|
.find(new_tag.id)
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)
|
||||||
|
.one()?;
|
||||||
|
|
||||||
|
TagCatalogEntry {
|
||||||
id: tag.id,
|
id: tag.id,
|
||||||
label: tag.label,
|
label: tag.label,
|
||||||
color: tag.color,
|
color: tag.color,
|
||||||
usage_count: 0,
|
usage_count: 0,
|
||||||
}))
|
}
|
||||||
|
.into_json()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn update_tag(
|
pub async fn update_tag(
|
||||||
State(state): State<AppState>,
|
|
||||||
Path(tag_id): Path<Uuid>,
|
Path(tag_id): Path<Uuid>,
|
||||||
|
TenantScopedConn {
|
||||||
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
Json(body): Json<Value>,
|
Json(body): Json<Value>,
|
||||||
) -> AppResult<Json<TagCatalogEntry>> {
|
) -> AppResult<Json<TagCatalogEntry>> {
|
||||||
let mut conn = state.db()?;
|
let existing: Tag = tags::table
|
||||||
let existing: Tag = tags::table.find(tag_id).first(&mut conn)?;
|
.find(tag_id)
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)
|
||||||
|
.one()?;
|
||||||
let label_class = classify_nullable(body.get("label")).map_err(AppError::bad_request)?;
|
let label_class = classify_nullable(body.get("label")).map_err(AppError::bad_request)?;
|
||||||
let color_class = classify_nullable(body.get("color")).map_err(AppError::bad_request)?;
|
let color_class = classify_nullable(body.get("color")).map_err(AppError::bad_request)?;
|
||||||
|
|
||||||
@@ -116,12 +138,13 @@ pub async fn update_tag(
|
|||||||
.filter(document_tags::tag_id.eq(tag_id))
|
.filter(document_tags::tag_id.eq(tag_id))
|
||||||
.select(count_star())
|
.select(count_star())
|
||||||
.first(&mut conn)?;
|
.first(&mut conn)?;
|
||||||
return Ok(Json(TagCatalogEntry {
|
return TagCatalogEntry {
|
||||||
id: existing.id,
|
id: existing.id,
|
||||||
label: existing.label.clone(),
|
label: existing.label.clone(),
|
||||||
color: existing.color.clone(),
|
color: existing.color.clone(),
|
||||||
usage_count,
|
usage_count,
|
||||||
}));
|
}
|
||||||
|
.into_json();
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut new_label: Option<String> = None;
|
let mut new_label: Option<String> = None;
|
||||||
@@ -140,6 +163,7 @@ pub async fn update_tag(
|
|||||||
let duplicate = tags::table
|
let duplicate = tags::table
|
||||||
.filter(tags::label.eq(trimmed))
|
.filter(tags::label.eq(trimmed))
|
||||||
.filter(tags::id.ne(tag_id))
|
.filter(tags::id.ne(tag_id))
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id))
|
||||||
.first::<Tag>(&mut conn)
|
.first::<Tag>(&mut conn)
|
||||||
.optional()?;
|
.optional()?;
|
||||||
if duplicate.is_some() {
|
if duplicate.is_some() {
|
||||||
@@ -174,6 +198,7 @@ pub async fn update_tag(
|
|||||||
if !label_changed && !color_changed {
|
if !label_changed && !color_changed {
|
||||||
let usage_count: i64 = document_tags::table
|
let usage_count: i64 = document_tags::table
|
||||||
.filter(document_tags::tag_id.eq(tag_id))
|
.filter(document_tags::tag_id.eq(tag_id))
|
||||||
|
.filter(document_tags::tenant_id.eq(tenant_id))
|
||||||
.select(count_star())
|
.select(count_star())
|
||||||
.first(&mut conn)?;
|
.first(&mut conn)?;
|
||||||
return Ok(Json(TagCatalogEntry {
|
return Ok(Json(TagCatalogEntry {
|
||||||
@@ -191,32 +216,45 @@ pub async fn update_tag(
|
|||||||
.map(|opt| opt.as_ref().map(|value| value.as_str())),
|
.map(|opt| opt.as_ref().map(|value| value.as_str())),
|
||||||
};
|
};
|
||||||
|
|
||||||
diesel::update(tags::table.find(tag_id))
|
diesel::update(
|
||||||
.set(&changeset)
|
tags::table
|
||||||
.execute(&mut conn)?;
|
.find(tag_id)
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id)),
|
||||||
|
)
|
||||||
|
.set(&changeset)
|
||||||
|
.execute(&mut conn)?;
|
||||||
|
|
||||||
let updated: Tag = tags::table.find(tag_id).first(&mut conn)?;
|
let updated: Tag = tags::table
|
||||||
|
.find(tag_id)
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id))
|
||||||
|
.first(&mut conn)
|
||||||
|
.one()?;
|
||||||
let usage_count: i64 = document_tags::table
|
let usage_count: i64 = document_tags::table
|
||||||
.filter(document_tags::tag_id.eq(tag_id))
|
.filter(document_tags::tag_id.eq(tag_id))
|
||||||
|
.filter(document_tags::tenant_id.eq(tenant_id))
|
||||||
.select(count_star())
|
.select(count_star())
|
||||||
.first(&mut conn)?;
|
.first(&mut conn)?;
|
||||||
|
|
||||||
Ok(Json(TagCatalogEntry {
|
TagCatalogEntry {
|
||||||
id: updated.id,
|
id: updated.id,
|
||||||
label: updated.label,
|
label: updated.label,
|
||||||
color: updated.color,
|
color: updated.color,
|
||||||
usage_count,
|
usage_count,
|
||||||
}))
|
}
|
||||||
|
.into_json()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn delete_tag(
|
pub async fn delete_tag(
|
||||||
State(state): State<AppState>,
|
|
||||||
Path(tag_id): Path<Uuid>,
|
Path(tag_id): Path<Uuid>,
|
||||||
) -> AppResult<impl axum::response::IntoResponse> {
|
TenantScopedConn {
|
||||||
let mut conn = state.db()?;
|
mut conn,
|
||||||
|
tenant_id,
|
||||||
|
..
|
||||||
|
}: TenantScopedConn,
|
||||||
|
) -> AppResult<StatusCode> {
|
||||||
let usage: i64 = document_tags::table
|
let usage: i64 = document_tags::table
|
||||||
.filter(document_tags::tag_id.eq(tag_id))
|
.filter(document_tags::tag_id.eq(tag_id))
|
||||||
|
.filter(document_tags::tenant_id.eq(tenant_id))
|
||||||
.select(count_star())
|
.select(count_star())
|
||||||
.first(&mut conn)?;
|
.first(&mut conn)?;
|
||||||
|
|
||||||
@@ -226,10 +264,15 @@ pub async fn delete_tag(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let deleted = diesel::delete(tags::table.find(tag_id)).execute(&mut conn)?;
|
let deleted = diesel::delete(
|
||||||
|
tags::table
|
||||||
|
.find(tag_id)
|
||||||
|
.filter(tags::tenant_id.eq(tenant_id)),
|
||||||
|
)
|
||||||
|
.execute(&mut conn)?;
|
||||||
if deleted == 0 {
|
if deleted == 0 {
|
||||||
return Err(AppError::not_found());
|
return Err(AppError::not_found());
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(StatusCode::NO_CONTENT)
|
no_content()
|
||||||
}
|
}
|
||||||
|
|||||||
+202
-121
@@ -20,17 +20,26 @@ use crate::error::{AppError, AppResult};
|
|||||||
use crate::models::{Document, DocumentVersion, Folder, User};
|
use crate::models::{Document, DocumentVersion, Folder, User};
|
||||||
use crate::schema::{
|
use crate::schema::{
|
||||||
document_versions::dsl as document_versions_dsl, documents::dsl as documents_dsl,
|
document_versions::dsl as document_versions_dsl, documents::dsl as documents_dsl,
|
||||||
folders::dsl as folders_dsl, users::dsl as users_dsl,
|
folders::dsl as folders_dsl, tenants::dsl as tenant_dsl,
|
||||||
|
user_memberships::dsl as memberships_dsl, users::dsl as users_dsl,
|
||||||
};
|
};
|
||||||
use crate::state::AppState;
|
use crate::state::AppState;
|
||||||
|
use crate::utils::{http::inline_content_disposition, time::to_http_date};
|
||||||
|
|
||||||
const REALM: &str = "Papercrate WebDAV";
|
const REALM: &str = "Papercrate WebDAV";
|
||||||
const DOWNLOAD_URL_TTL_SECONDS: u64 = 300;
|
const DOWNLOAD_URL_TTL_SECONDS: u64 = 300;
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
struct WebDavUser {
|
struct TenantEntry {
|
||||||
|
tenant_id: Uuid,
|
||||||
|
slug: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
struct WebDavContext {
|
||||||
_user_id: Uuid,
|
_user_id: Uuid,
|
||||||
_username: String,
|
_username: String,
|
||||||
|
tenants: Vec<TenantEntry>,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn create_router() -> Router<AppState> {
|
pub fn create_router() -> Router<AppState> {
|
||||||
@@ -68,7 +77,7 @@ async fn handle_propfind(
|
|||||||
path: &str,
|
path: &str,
|
||||||
headers: HeaderMap,
|
headers: HeaderMap,
|
||||||
) -> Result<Response, AppError> {
|
) -> Result<Response, AppError> {
|
||||||
let _user = match authenticate(state, &headers)? {
|
let context = match authenticate(state, &headers)? {
|
||||||
Some(user) => user,
|
Some(user) => user,
|
||||||
None => return Ok(unauthorized_response()),
|
None => return Ok(unauthorized_response()),
|
||||||
};
|
};
|
||||||
@@ -79,25 +88,44 @@ async fn handle_propfind(
|
|||||||
};
|
};
|
||||||
|
|
||||||
let segments = parse_segments(path)?;
|
let segments = parse_segments(path)?;
|
||||||
let resolution = match resolve_path(state, &segments)? {
|
|
||||||
Some(resolved) => resolved,
|
|
||||||
None => return Ok(not_found_response()),
|
|
||||||
};
|
|
||||||
|
|
||||||
let resources = match resolution {
|
let resources = if segments.is_empty() {
|
||||||
ResolvedPath::Root => {
|
build_account_root_resources(&context.tenants, depth)
|
||||||
let contents = fetch_folder_contents(state, None)?;
|
} else {
|
||||||
build_resources_for_folder(None, &[], &contents, depth)
|
let (requested_slug, remainder) = segments.split_first().unwrap();
|
||||||
|
let tenant_entry = match context
|
||||||
|
.tenants
|
||||||
|
.iter()
|
||||||
|
.find(|entry| entry.slug.eq_ignore_ascii_case(requested_slug))
|
||||||
|
{
|
||||||
|
Some(entry) => TenantEntry {
|
||||||
|
tenant_id: entry.tenant_id,
|
||||||
|
slug: entry.slug.clone(),
|
||||||
|
},
|
||||||
|
None => return Ok(not_found_response()),
|
||||||
|
};
|
||||||
|
|
||||||
|
let resolution = match resolve_path(state, &tenant_entry, remainder)? {
|
||||||
|
Some(resolved) => resolved,
|
||||||
|
None => return Ok(not_found_response()),
|
||||||
|
};
|
||||||
|
|
||||||
|
match resolution {
|
||||||
|
ResolvedPath::TenantRoot { chain } => {
|
||||||
|
let contents = fetch_folder_contents(state, tenant_entry.tenant_id, None)?;
|
||||||
|
build_resources_for_folder(None, &chain, &contents, depth)
|
||||||
|
}
|
||||||
|
ResolvedPath::Folder { folder, chain } => {
|
||||||
|
let contents =
|
||||||
|
fetch_folder_contents(state, tenant_entry.tenant_id, Some(folder.id))?;
|
||||||
|
build_resources_for_folder(Some(&folder), &chain, &contents, depth)
|
||||||
|
}
|
||||||
|
ResolvedPath::Document {
|
||||||
|
document,
|
||||||
|
version,
|
||||||
|
chain,
|
||||||
|
} => build_resources_for_document(&chain, &document, &version),
|
||||||
}
|
}
|
||||||
ResolvedPath::Folder { folder, chain } => {
|
|
||||||
let contents = fetch_folder_contents(state, Some(folder.id))?;
|
|
||||||
build_resources_for_folder(Some(&folder), &chain, &contents, depth)
|
|
||||||
}
|
|
||||||
ResolvedPath::Document {
|
|
||||||
document,
|
|
||||||
version,
|
|
||||||
chain,
|
|
||||||
} => build_resources_for_document(&chain, &document, &version),
|
|
||||||
};
|
};
|
||||||
|
|
||||||
let body = render_multistatus(&resources)
|
let body = render_multistatus(&resources)
|
||||||
@@ -118,13 +146,34 @@ async fn handle_get_or_head(
|
|||||||
headers: HeaderMap,
|
headers: HeaderMap,
|
||||||
method: Method,
|
method: Method,
|
||||||
) -> Result<Response, AppError> {
|
) -> Result<Response, AppError> {
|
||||||
let _user = match authenticate(state, &headers)? {
|
let context = match authenticate(state, &headers)? {
|
||||||
Some(user) => user,
|
Some(user) => user,
|
||||||
None => return Ok(unauthorized_response()),
|
None => return Ok(unauthorized_response()),
|
||||||
};
|
};
|
||||||
|
|
||||||
let segments = parse_segments(path)?;
|
let segments = parse_segments(path)?;
|
||||||
let resolution = match resolve_path(state, &segments)? {
|
let (requested_slug, remainder) = match segments.split_first() {
|
||||||
|
Some(values) => values,
|
||||||
|
None => return Ok(method_not_allowed()),
|
||||||
|
};
|
||||||
|
|
||||||
|
let tenant_entry = match context
|
||||||
|
.tenants
|
||||||
|
.iter()
|
||||||
|
.find(|entry| entry.slug.eq_ignore_ascii_case(requested_slug))
|
||||||
|
{
|
||||||
|
Some(entry) => TenantEntry {
|
||||||
|
tenant_id: entry.tenant_id,
|
||||||
|
slug: entry.slug.clone(),
|
||||||
|
},
|
||||||
|
None => return Ok(not_found_response()),
|
||||||
|
};
|
||||||
|
|
||||||
|
if remainder.is_empty() {
|
||||||
|
return Ok(method_not_allowed());
|
||||||
|
}
|
||||||
|
|
||||||
|
let resolution = match resolve_path(state, &tenant_entry, remainder)? {
|
||||||
Some(resolved) => resolved,
|
Some(resolved) => resolved,
|
||||||
None => return Ok(not_found_response()),
|
None => return Ok(not_found_response()),
|
||||||
};
|
};
|
||||||
@@ -219,21 +268,29 @@ fn parse_segments(path: &str) -> AppResult<Vec<String>> {
|
|||||||
|
|
||||||
fn fetch_folder_contents(
|
fn fetch_folder_contents(
|
||||||
state: &AppState,
|
state: &AppState,
|
||||||
|
tenant_id: Uuid,
|
||||||
folder_id: Option<Uuid>,
|
folder_id: Option<Uuid>,
|
||||||
) -> AppResult<WebDavFolderContents> {
|
) -> AppResult<WebDavFolderContents> {
|
||||||
let mut conn = state.db()?;
|
let mut conn = state.db_for_tenant(tenant_id)?;
|
||||||
|
|
||||||
let folder = match folder_id {
|
let folder = match folder_id {
|
||||||
Some(id) => Some(folders_dsl::folders.find(id).first::<Folder>(&mut conn)?),
|
Some(id) => Some(
|
||||||
|
folders_dsl::folders
|
||||||
|
.filter(folders_dsl::tenant_id.eq(tenant_id))
|
||||||
|
.find(id)
|
||||||
|
.first::<Folder>(&mut conn)?,
|
||||||
|
),
|
||||||
None => None,
|
None => None,
|
||||||
};
|
};
|
||||||
|
|
||||||
let subfolders: Vec<Folder> = match folder_id {
|
let subfolders: Vec<Folder> = match folder_id {
|
||||||
Some(id) => folders_dsl::folders
|
Some(id) => folders_dsl::folders
|
||||||
|
.filter(folders_dsl::tenant_id.eq(tenant_id))
|
||||||
.filter(folders_dsl::parent_id.eq(Some(id)))
|
.filter(folders_dsl::parent_id.eq(Some(id)))
|
||||||
.order(folders_dsl::name.asc())
|
.order(folders_dsl::name.asc())
|
||||||
.load(&mut conn)?,
|
.load(&mut conn)?,
|
||||||
None => folders_dsl::folders
|
None => folders_dsl::folders
|
||||||
|
.filter(folders_dsl::tenant_id.eq(tenant_id))
|
||||||
.filter(folders_dsl::parent_id.is_null())
|
.filter(folders_dsl::parent_id.is_null())
|
||||||
.order(folders_dsl::name.asc())
|
.order(folders_dsl::name.asc())
|
||||||
.load(&mut conn)?,
|
.load(&mut conn)?,
|
||||||
@@ -241,6 +298,7 @@ fn fetch_folder_contents(
|
|||||||
|
|
||||||
let mut docs_query = documents_dsl::documents
|
let mut docs_query = documents_dsl::documents
|
||||||
.filter(documents_dsl::deleted_at.is_null())
|
.filter(documents_dsl::deleted_at.is_null())
|
||||||
|
.filter(documents_dsl::tenant_id.eq(tenant_id))
|
||||||
.into_boxed();
|
.into_boxed();
|
||||||
|
|
||||||
docs_query = match folder_id {
|
docs_query = match folder_id {
|
||||||
@@ -290,8 +348,9 @@ async fn stream_document(
|
|||||||
) -> Result<Response, AppError> {
|
) -> Result<Response, AppError> {
|
||||||
let range_header = headers.get(header::RANGE).cloned();
|
let range_header = headers.get(header::RANGE).cloned();
|
||||||
|
|
||||||
let url = state
|
let storage = state.storage_for_tenant(document.tenant_id)?;
|
||||||
.storage
|
|
||||||
|
let url = storage
|
||||||
.presign_get_object(
|
.presign_get_object(
|
||||||
&version.s3_key,
|
&version.s3_key,
|
||||||
Duration::from_secs(DOWNLOAD_URL_TTL_SECONDS),
|
Duration::from_secs(DOWNLOAD_URL_TTL_SECONDS),
|
||||||
@@ -338,7 +397,7 @@ async fn stream_document(
|
|||||||
|
|
||||||
builder = builder.header("Accept-Ranges", "bytes");
|
builder = builder.header("Accept-Ranges", "bytes");
|
||||||
|
|
||||||
if let Some(disposition) = content_disposition(&document.filename) {
|
if let Some(disposition) = inline_content_disposition(&document.filename) {
|
||||||
builder = builder.header(header::CONTENT_DISPOSITION, disposition);
|
builder = builder.header(header::CONTENT_DISPOSITION, disposition);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -360,7 +419,7 @@ async fn stream_document(
|
|||||||
.map_err(|err| AppError::internal(format!("failed to build response: {err}")))
|
.map_err(|err| AppError::internal(format!("failed to build response: {err}")))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn authenticate(state: &AppState, headers: &HeaderMap) -> Result<Option<WebDavUser>, AppError> {
|
fn authenticate(state: &AppState, headers: &HeaderMap) -> Result<Option<WebDavContext>, AppError> {
|
||||||
tracing::debug!("webdav authenticate invoked");
|
tracing::debug!("webdav authenticate invoked");
|
||||||
let authorization = match headers.get(header::AUTHORIZATION) {
|
let authorization = match headers.get(header::AUTHORIZATION) {
|
||||||
Some(value) => match value.to_str() {
|
Some(value) => match value.to_str() {
|
||||||
@@ -405,7 +464,7 @@ fn authenticate(state: &AppState, headers: &HeaderMap) -> Result<Option<WebDavUs
|
|||||||
};
|
};
|
||||||
|
|
||||||
tracing::debug!(%username, "attempting webdav login");
|
tracing::debug!(%username, "attempting webdav login");
|
||||||
let mut conn = state.db()?;
|
let mut conn = state.db_unscoped()?;
|
||||||
|
|
||||||
let user: User = match users_dsl::users
|
let user: User = match users_dsl::users
|
||||||
.filter(users_dsl::username.eq(username))
|
.filter(users_dsl::username.eq(username))
|
||||||
@@ -427,10 +486,27 @@ fn authenticate(state: &AppState, headers: &HeaderMap) -> Result<Option<WebDavUs
|
|||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
|
|
||||||
tracing::debug!(%username, "webdav login success");
|
let tenant_rows: Vec<(Uuid, String)> = memberships_dsl::user_memberships
|
||||||
Ok(Some(WebDavUser {
|
.inner_join(tenant_dsl::tenants)
|
||||||
|
.filter(memberships_dsl::user_id.eq(user.id))
|
||||||
|
.select((tenant_dsl::id, tenant_dsl::slug))
|
||||||
|
.load(&mut conn)?;
|
||||||
|
|
||||||
|
if tenant_rows.is_empty() {
|
||||||
|
tracing::warn!(%username, "webdav user has no tenant memberships");
|
||||||
|
return Ok(None);
|
||||||
|
}
|
||||||
|
|
||||||
|
let tenants: Vec<TenantEntry> = tenant_rows
|
||||||
|
.into_iter()
|
||||||
|
.map(|(tenant_id, slug)| TenantEntry { tenant_id, slug })
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
tracing::debug!(%username, tenant_count = tenants.len(), "webdav login success");
|
||||||
|
Ok(Some(WebDavContext {
|
||||||
_user_id: user.id,
|
_user_id: user.id,
|
||||||
_username: user.username,
|
_username: user.username,
|
||||||
|
tenants,
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -444,10 +520,10 @@ fn build_resources_for_folder(
|
|||||||
|
|
||||||
let display_name = folder
|
let display_name = folder
|
||||||
.map(|folder| folder.name.clone())
|
.map(|folder| folder.name.clone())
|
||||||
.unwrap_or_else(|| "/".to_string());
|
.unwrap_or_else(|| chain.last().cloned().unwrap_or_else(|| "/".to_string()));
|
||||||
|
|
||||||
let href = build_href(chain, true);
|
let href = build_href(chain, true);
|
||||||
let last_modified = folder.map(|folder| format_http_date(folder.updated_at));
|
let last_modified = folder.map(|folder| to_http_date(folder.updated_at));
|
||||||
|
|
||||||
resources.push(DavResource {
|
resources.push(DavResource {
|
||||||
href,
|
href,
|
||||||
@@ -471,7 +547,7 @@ fn build_resources_for_folder(
|
|||||||
is_collection: true,
|
is_collection: true,
|
||||||
content_length: None,
|
content_length: None,
|
||||||
content_type: None,
|
content_type: None,
|
||||||
last_modified: Some(format_http_date(subfolder.updated_at)),
|
last_modified: Some(to_http_date(subfolder.updated_at)),
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -488,6 +564,37 @@ fn build_resources_for_folder(
|
|||||||
resources
|
resources
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn build_account_root_resources(tenants: &[TenantEntry], depth: u8) -> Vec<DavResource> {
|
||||||
|
let mut resources = Vec::new();
|
||||||
|
|
||||||
|
resources.push(DavResource {
|
||||||
|
href: "/".to_string(),
|
||||||
|
display_name: "/".to_string(),
|
||||||
|
is_collection: true,
|
||||||
|
content_length: None,
|
||||||
|
content_type: None,
|
||||||
|
last_modified: None,
|
||||||
|
});
|
||||||
|
|
||||||
|
if depth == 0 {
|
||||||
|
return resources;
|
||||||
|
}
|
||||||
|
|
||||||
|
for tenant in tenants {
|
||||||
|
let href = build_href(&[tenant.slug.clone()], true);
|
||||||
|
resources.push(DavResource {
|
||||||
|
href,
|
||||||
|
display_name: tenant.slug.clone(),
|
||||||
|
is_collection: true,
|
||||||
|
content_length: None,
|
||||||
|
content_type: None,
|
||||||
|
last_modified: None,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
resources
|
||||||
|
}
|
||||||
|
|
||||||
fn build_resources_for_document(
|
fn build_resources_for_document(
|
||||||
chain: &[String],
|
chain: &[String],
|
||||||
document: &Document,
|
document: &Document,
|
||||||
@@ -509,7 +616,7 @@ fn document_to_resource(
|
|||||||
is_collection: false,
|
is_collection: false,
|
||||||
content_length: Some(version.size_bytes),
|
content_length: Some(version.size_bytes),
|
||||||
content_type: document.content_type.clone(),
|
content_type: document.content_type.clone(),
|
||||||
last_modified: Some(format_http_date(document.updated_at)),
|
last_modified: Some(to_http_date(document.updated_at)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -590,32 +697,6 @@ fn render_multistatus(resources: &[DavResource]) -> Result<Vec<u8>, quick_xml::E
|
|||||||
Ok(writer.into_inner())
|
Ok(writer.into_inner())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn format_http_date(value: chrono::NaiveDateTime) -> String {
|
|
||||||
let datetime = chrono::DateTime::<chrono::Utc>::from_naive_utc_and_offset(value, chrono::Utc);
|
|
||||||
datetime.format("%a, %d %b %Y %H:%M:%S GMT").to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn content_disposition(filename: &str) -> Option<String> {
|
|
||||||
if filename.is_empty() {
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
let sanitized: String = filename
|
|
||||||
.chars()
|
|
||||||
.map(|ch| match ch {
|
|
||||||
'"' | '\\' => '_',
|
|
||||||
_ => ch,
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
let encoded =
|
|
||||||
percent_encoding::utf8_percent_encode(&sanitized, percent_encoding::NON_ALPHANUMERIC);
|
|
||||||
Some(format!(
|
|
||||||
"inline; filename=\"{}\"; filename*=UTF-8''{}",
|
|
||||||
sanitized, encoded
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
||||||
struct WebDavFolderContents {
|
struct WebDavFolderContents {
|
||||||
_folder: Option<Folder>,
|
_folder: Option<Folder>,
|
||||||
subfolders: Vec<Folder>,
|
subfolders: Vec<Folder>,
|
||||||
@@ -635,9 +716,10 @@ struct DavResource {
|
|||||||
content_type: Option<String>,
|
content_type: Option<String>,
|
||||||
last_modified: Option<String>,
|
last_modified: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
enum ResolvedPath {
|
enum ResolvedPath {
|
||||||
Root,
|
TenantRoot {
|
||||||
|
chain: Vec<String>,
|
||||||
|
},
|
||||||
Folder {
|
Folder {
|
||||||
folder: Folder,
|
folder: Folder,
|
||||||
chain: Vec<String>,
|
chain: Vec<String>,
|
||||||
@@ -649,37 +731,37 @@ enum ResolvedPath {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
fn resolve_path(state: &AppState, segments: &[String]) -> AppResult<Option<ResolvedPath>> {
|
fn resolve_path(
|
||||||
if segments.is_empty() {
|
state: &AppState,
|
||||||
return Ok(Some(ResolvedPath::Root));
|
tenant: &TenantEntry,
|
||||||
}
|
segments: &[String],
|
||||||
|
) -> AppResult<Option<ResolvedPath>> {
|
||||||
let mut conn = state.db()?;
|
let mut conn = state.db_for_tenant(tenant.tenant_id)?;
|
||||||
let mut parent_id: Option<Uuid> = None;
|
let mut parent_id: Option<Uuid> = None;
|
||||||
let mut chain: Vec<String> = Vec::new();
|
let mut chain: Vec<String> = vec![tenant.slug.clone()];
|
||||||
let mut current_folder: Option<Folder> = None;
|
let mut current_folder: Option<Folder> = None;
|
||||||
|
|
||||||
|
if segments.is_empty() {
|
||||||
|
return Ok(Some(ResolvedPath::TenantRoot { chain }));
|
||||||
|
}
|
||||||
|
|
||||||
for (index, segment) in segments.iter().enumerate() {
|
for (index, segment) in segments.iter().enumerate() {
|
||||||
let is_last = index == segments.len() - 1;
|
let is_last = index == segments.len() - 1;
|
||||||
|
|
||||||
match find_folder_by_name(&mut conn, parent_id, segment)? {
|
if let Some(folder) = find_folder_by_name(&mut conn, tenant.tenant_id, parent_id, segment)?
|
||||||
Some(folder) => {
|
{
|
||||||
if is_last {
|
chain.push(folder.name.clone());
|
||||||
chain.push(folder.name.clone());
|
if is_last {
|
||||||
return Ok(Some(ResolvedPath::Folder { folder, chain }));
|
return Ok(Some(ResolvedPath::Folder { folder, chain }));
|
||||||
}
|
|
||||||
|
|
||||||
parent_id = Some(folder.id);
|
|
||||||
chain.push(folder.name.clone());
|
|
||||||
current_folder = Some(folder);
|
|
||||||
continue;
|
|
||||||
}
|
}
|
||||||
None => {}
|
parent_id = Some(folder.id);
|
||||||
|
current_folder = Some(folder);
|
||||||
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if is_last {
|
if is_last {
|
||||||
if let Some((document, version)) =
|
if let Some((document, version)) =
|
||||||
find_document_by_filename(&mut conn, parent_id, segment)?
|
find_document_by_filename(&mut conn, tenant.tenant_id, parent_id, segment)?
|
||||||
{
|
{
|
||||||
chain.push(document.filename.clone());
|
chain.push(document.filename.clone());
|
||||||
return Ok(Some(ResolvedPath::Document {
|
return Ok(Some(ResolvedPath::Document {
|
||||||
@@ -691,26 +773,22 @@ fn resolve_path(state: &AppState, segments: &[String]) -> AppResult<Option<Resol
|
|||||||
}
|
}
|
||||||
|
|
||||||
if let Ok(uuid) = Uuid::parse_str(segment) {
|
if let Ok(uuid) = Uuid::parse_str(segment) {
|
||||||
if let Some(folder) = folders_dsl::folders
|
if let Some(folder) = find_folder_by_id(&mut conn, tenant.tenant_id, uuid)? {
|
||||||
.find(uuid)
|
|
||||||
.first::<Folder>(&mut conn)
|
|
||||||
.optional()?
|
|
||||||
{
|
|
||||||
if folder.parent_id != parent_id {
|
if folder.parent_id != parent_id {
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
if !is_last {
|
chain.push(folder.name.clone());
|
||||||
parent_id = Some(folder.id);
|
if is_last {
|
||||||
chain.push(folder.name.clone());
|
|
||||||
current_folder = Some(folder);
|
|
||||||
continue;
|
|
||||||
} else {
|
|
||||||
chain.push(folder.name.clone());
|
|
||||||
return Ok(Some(ResolvedPath::Folder { folder, chain }));
|
return Ok(Some(ResolvedPath::Folder { folder, chain }));
|
||||||
}
|
}
|
||||||
|
parent_id = Some(folder.id);
|
||||||
|
current_folder = Some(folder);
|
||||||
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some((document, version)) = find_document_by_id(&mut conn, uuid)? {
|
if let Some((document, version)) =
|
||||||
|
find_document_by_id(&mut conn, tenant.tenant_id, uuid)?
|
||||||
|
{
|
||||||
if document.folder_id != parent_id {
|
if document.folder_id != parent_id {
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
@@ -723,19 +801,6 @@ fn resolve_path(state: &AppState, segments: &[String]) -> AppResult<Option<Resol
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if is_last {
|
|
||||||
if let Some((document, version)) =
|
|
||||||
find_document_by_filename(&mut conn, parent_id, segment)?
|
|
||||||
{
|
|
||||||
chain.push(document.filename.clone());
|
|
||||||
return Ok(Some(ResolvedPath::Document {
|
|
||||||
document,
|
|
||||||
version,
|
|
||||||
chain,
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -744,32 +809,46 @@ fn resolve_path(state: &AppState, segments: &[String]) -> AppResult<Option<Resol
|
|||||||
|
|
||||||
fn find_folder_by_name(
|
fn find_folder_by_name(
|
||||||
conn: &mut PgConnection,
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
parent_id: Option<Uuid>,
|
parent_id: Option<Uuid>,
|
||||||
name: &str,
|
name: &str,
|
||||||
) -> AppResult<Option<Folder>> {
|
) -> AppResult<Option<Folder>> {
|
||||||
let result = match parent_id {
|
let mut query = folders_dsl::folders
|
||||||
Some(parent) => folders_dsl::folders
|
.filter(folders_dsl::tenant_id.eq(tenant_id))
|
||||||
.filter(folders_dsl::parent_id.eq(Some(parent)))
|
.into_boxed();
|
||||||
.filter(folders_dsl::name.eq(name))
|
|
||||||
.first::<Folder>(conn)
|
query = match parent_id {
|
||||||
.optional()?,
|
Some(parent) => query.filter(folders_dsl::parent_id.eq(Some(parent))),
|
||||||
None => folders_dsl::folders
|
None => query.filter(folders_dsl::parent_id.is_null()),
|
||||||
.filter(folders_dsl::parent_id.is_null())
|
|
||||||
.filter(folders_dsl::name.eq(name))
|
|
||||||
.first::<Folder>(conn)
|
|
||||||
.optional()?,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
Ok(result)
|
Ok(query
|
||||||
|
.filter(folders_dsl::name.eq(name))
|
||||||
|
.first::<Folder>(conn)
|
||||||
|
.optional()?)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn find_folder_by_id(
|
||||||
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
|
folder_id: Uuid,
|
||||||
|
) -> AppResult<Option<Folder>> {
|
||||||
|
Ok(folders_dsl::folders
|
||||||
|
.filter(folders_dsl::tenant_id.eq(tenant_id))
|
||||||
|
.find(folder_id)
|
||||||
|
.first::<Folder>(conn)
|
||||||
|
.optional()?)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn find_document_by_filename(
|
fn find_document_by_filename(
|
||||||
conn: &mut PgConnection,
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
parent_id: Option<Uuid>,
|
parent_id: Option<Uuid>,
|
||||||
filename: &str,
|
filename: &str,
|
||||||
) -> AppResult<Option<(Document, DocumentVersion)>> {
|
) -> AppResult<Option<(Document, DocumentVersion)>> {
|
||||||
let mut query = documents_dsl::documents
|
let mut query = documents_dsl::documents
|
||||||
.filter(documents_dsl::deleted_at.is_null())
|
.filter(documents_dsl::deleted_at.is_null())
|
||||||
|
.filter(documents_dsl::tenant_id.eq(tenant_id))
|
||||||
.filter(documents_dsl::filename.eq(filename))
|
.filter(documents_dsl::filename.eq(filename))
|
||||||
.into_boxed();
|
.into_boxed();
|
||||||
|
|
||||||
@@ -790,10 +869,12 @@ fn find_document_by_filename(
|
|||||||
|
|
||||||
fn find_document_by_id(
|
fn find_document_by_id(
|
||||||
conn: &mut PgConnection,
|
conn: &mut PgConnection,
|
||||||
|
tenant_id: Uuid,
|
||||||
document_id: Uuid,
|
document_id: Uuid,
|
||||||
) -> AppResult<Option<(Document, DocumentVersion)>> {
|
) -> AppResult<Option<(Document, DocumentVersion)>> {
|
||||||
if let Some(document) = documents_dsl::documents
|
if let Some(document) = documents_dsl::documents
|
||||||
.filter(documents_dsl::deleted_at.is_null())
|
.filter(documents_dsl::deleted_at.is_null())
|
||||||
|
.filter(documents_dsl::tenant_id.eq(tenant_id))
|
||||||
.find(document_id)
|
.find(document_id)
|
||||||
.first::<Document>(conn)
|
.first::<Document>(conn)
|
||||||
.optional()?
|
.optional()?
|
||||||
|
|||||||
+63
-3
@@ -8,6 +8,18 @@ diesel::table! {
|
|||||||
metadata -> Jsonb,
|
metadata -> Jsonb,
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
updated_at -> Timestamptz,
|
updated_at -> Timestamptz,
|
||||||
|
tenant_id -> Uuid,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
diesel::table! {
|
||||||
|
document_asset_objects (id) {
|
||||||
|
id -> Uuid,
|
||||||
|
asset_id -> Uuid,
|
||||||
|
ordinal -> Int4,
|
||||||
|
s3_key -> Text,
|
||||||
|
metadata -> Jsonb,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -16,10 +28,11 @@ diesel::table! {
|
|||||||
id -> Uuid,
|
id -> Uuid,
|
||||||
document_version_id -> Uuid,
|
document_version_id -> Uuid,
|
||||||
asset_type -> Text,
|
asset_type -> Text,
|
||||||
s3_key -> Text,
|
|
||||||
mime_type -> Text,
|
mime_type -> Text,
|
||||||
metadata -> Jsonb,
|
metadata -> Jsonb,
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
|
cardinality -> Nullable<Int4>,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -31,6 +44,7 @@ diesel::table! {
|
|||||||
role -> Varchar,
|
role -> Varchar,
|
||||||
assigned_at -> Timestamptz,
|
assigned_at -> Timestamptz,
|
||||||
assigned_by -> Nullable<Uuid>,
|
assigned_by -> Nullable<Uuid>,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -40,6 +54,7 @@ diesel::table! {
|
|||||||
tag_id -> Uuid,
|
tag_id -> Uuid,
|
||||||
assigned_at -> Timestamptz,
|
assigned_at -> Timestamptz,
|
||||||
assigned_by -> Nullable<Uuid>,
|
assigned_by -> Nullable<Uuid>,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -56,6 +71,7 @@ diesel::table! {
|
|||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
operations_summary -> Jsonb,
|
operations_summary -> Jsonb,
|
||||||
metadata -> Jsonb,
|
metadata -> Jsonb,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -77,6 +93,7 @@ diesel::table! {
|
|||||||
#[max_length = 255]
|
#[max_length = 255]
|
||||||
title -> Varchar,
|
title -> Varchar,
|
||||||
current_version_id -> Uuid,
|
current_version_id -> Uuid,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -88,6 +105,7 @@ diesel::table! {
|
|||||||
parent_id -> Nullable<Uuid>,
|
parent_id -> Nullable<Uuid>,
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
updated_at -> Timestamptz,
|
updated_at -> Timestamptz,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -102,6 +120,7 @@ diesel::table! {
|
|||||||
last_error -> Nullable<Text>,
|
last_error -> Nullable<Text>,
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
updated_at -> Timestamptz,
|
updated_at -> Timestamptz,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -115,6 +134,7 @@ diesel::table! {
|
|||||||
revoked_at -> Nullable<Timestamptz>,
|
revoked_at -> Nullable<Timestamptz>,
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
updated_at -> Timestamptz,
|
updated_at -> Timestamptz,
|
||||||
|
tenant_id -> Uuid,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -126,6 +146,31 @@ diesel::table! {
|
|||||||
#[max_length = 7]
|
#[max_length = 7]
|
||||||
color -> Nullable<Varchar>,
|
color -> Nullable<Varchar>,
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
|
tenant_id -> Uuid,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
diesel::table! {
|
||||||
|
tenants (id) {
|
||||||
|
id -> Uuid,
|
||||||
|
slug -> Text,
|
||||||
|
storage_root -> Nullable<Text>,
|
||||||
|
quickwit_index -> Nullable<Text>,
|
||||||
|
status -> Text,
|
||||||
|
config -> Jsonb,
|
||||||
|
created_at -> Timestamptz,
|
||||||
|
updated_at -> Timestamptz,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
diesel::table! {
|
||||||
|
user_memberships (id) {
|
||||||
|
id -> Uuid,
|
||||||
|
user_id -> Uuid,
|
||||||
|
tenant_id -> Uuid,
|
||||||
|
role -> Text,
|
||||||
|
created_at -> Timestamptz,
|
||||||
|
updated_at -> Timestamptz,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -136,25 +181,38 @@ diesel::table! {
|
|||||||
username -> Varchar,
|
username -> Varchar,
|
||||||
#[max_length = 255]
|
#[max_length = 255]
|
||||||
password_hash -> Varchar,
|
password_hash -> Varchar,
|
||||||
#[max_length = 16]
|
|
||||||
role -> Varchar,
|
|
||||||
created_at -> Timestamptz,
|
created_at -> Timestamptz,
|
||||||
updated_at -> Timestamptz,
|
updated_at -> Timestamptz,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
diesel::joinable!(correspondents -> tenants (tenant_id));
|
||||||
|
diesel::joinable!(document_asset_objects -> document_assets (asset_id));
|
||||||
|
diesel::joinable!(document_asset_objects -> tenants (tenant_id));
|
||||||
diesel::joinable!(document_assets -> document_versions (document_version_id));
|
diesel::joinable!(document_assets -> document_versions (document_version_id));
|
||||||
|
diesel::joinable!(document_assets -> tenants (tenant_id));
|
||||||
diesel::joinable!(document_correspondents -> correspondents (correspondent_id));
|
diesel::joinable!(document_correspondents -> correspondents (correspondent_id));
|
||||||
diesel::joinable!(document_correspondents -> documents (document_id));
|
diesel::joinable!(document_correspondents -> documents (document_id));
|
||||||
|
diesel::joinable!(document_correspondents -> tenants (tenant_id));
|
||||||
diesel::joinable!(document_correspondents -> users (assigned_by));
|
diesel::joinable!(document_correspondents -> users (assigned_by));
|
||||||
diesel::joinable!(document_tags -> documents (document_id));
|
diesel::joinable!(document_tags -> documents (document_id));
|
||||||
diesel::joinable!(document_tags -> tags (tag_id));
|
diesel::joinable!(document_tags -> tags (tag_id));
|
||||||
|
diesel::joinable!(document_tags -> tenants (tenant_id));
|
||||||
diesel::joinable!(document_tags -> users (assigned_by));
|
diesel::joinable!(document_tags -> users (assigned_by));
|
||||||
|
diesel::joinable!(document_versions -> tenants (tenant_id));
|
||||||
diesel::joinable!(documents -> folders (folder_id));
|
diesel::joinable!(documents -> folders (folder_id));
|
||||||
|
diesel::joinable!(documents -> tenants (tenant_id));
|
||||||
|
diesel::joinable!(folders -> tenants (tenant_id));
|
||||||
|
diesel::joinable!(jobs -> tenants (tenant_id));
|
||||||
|
diesel::joinable!(refresh_tokens -> tenants (tenant_id));
|
||||||
diesel::joinable!(refresh_tokens -> users (user_id));
|
diesel::joinable!(refresh_tokens -> users (user_id));
|
||||||
|
diesel::joinable!(tags -> tenants (tenant_id));
|
||||||
|
diesel::joinable!(user_memberships -> tenants (tenant_id));
|
||||||
|
diesel::joinable!(user_memberships -> users (user_id));
|
||||||
|
|
||||||
diesel::allow_tables_to_appear_in_same_query!(
|
diesel::allow_tables_to_appear_in_same_query!(
|
||||||
correspondents,
|
correspondents,
|
||||||
|
document_asset_objects,
|
||||||
document_assets,
|
document_assets,
|
||||||
document_correspondents,
|
document_correspondents,
|
||||||
document_tags,
|
document_tags,
|
||||||
@@ -164,5 +222,7 @@ diesel::allow_tables_to_appear_in_same_query!(
|
|||||||
jobs,
|
jobs,
|
||||||
refresh_tokens,
|
refresh_tokens,
|
||||||
tags,
|
tags,
|
||||||
|
tenants,
|
||||||
|
user_memberships,
|
||||||
users,
|
users,
|
||||||
);
|
);
|
||||||
|
|||||||
+41
-5
@@ -4,43 +4,79 @@ use diesel::{
|
|||||||
pg::PgConnection,
|
pg::PgConnection,
|
||||||
r2d2::{ConnectionManager, PooledConnection},
|
r2d2::{ConnectionManager, PooledConnection},
|
||||||
};
|
};
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
auth::jwt::JwtService,
|
auth::jwt::JwtService,
|
||||||
config::AppConfig,
|
config::AppConfig,
|
||||||
db::PgPool,
|
db::PgPool,
|
||||||
error::{AppError, AppResult},
|
error::{AppError, AppResult},
|
||||||
storage::ObjectStorage,
|
storage::{ObjectStorage, TenantStorage},
|
||||||
|
tenants::{apply_tenant_guc, TenantService},
|
||||||
};
|
};
|
||||||
|
|
||||||
type PgPooledConnection = PooledConnection<ConnectionManager<PgConnection>>;
|
pub type PgPooledConnection = PooledConnection<ConnectionManager<PgConnection>>;
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct AppState {
|
pub struct AppState {
|
||||||
pub pool: PgPool,
|
pub pool: PgPool,
|
||||||
pub config: Arc<AppConfig>,
|
pub config: Arc<AppConfig>,
|
||||||
pub storage: Arc<dyn ObjectStorage>,
|
storage: Arc<dyn ObjectStorage>,
|
||||||
pub jwt: JwtService,
|
pub jwt: JwtService,
|
||||||
|
pub tenants: TenantService,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AppState {
|
impl AppState {
|
||||||
|
pub async fn initialize(
|
||||||
|
config: AppConfig,
|
||||||
|
pool_size_override: Option<u32>,
|
||||||
|
) -> anyhow::Result<Self> {
|
||||||
|
let pool_size = pool_size_override.unwrap_or(config.database_max_pool_size);
|
||||||
|
let pool = crate::db::init_pool_with_size(&config.database_url, pool_size)?;
|
||||||
|
let s3_client = crate::s3::build_client(&config).await?;
|
||||||
|
let storage = Arc::new(crate::storage::S3Storage::new(
|
||||||
|
s3_client,
|
||||||
|
config.s3_bucket.clone(),
|
||||||
|
));
|
||||||
|
let jwt = crate::auth::jwt::JwtService::from_config(&config)?;
|
||||||
|
|
||||||
|
Ok(Self::new(pool, config, storage, jwt))
|
||||||
|
}
|
||||||
|
|
||||||
pub fn new(
|
pub fn new(
|
||||||
pool: PgPool,
|
pool: PgPool,
|
||||||
config: AppConfig,
|
config: AppConfig,
|
||||||
storage: Arc<dyn ObjectStorage>,
|
storage: Arc<dyn ObjectStorage>,
|
||||||
jwt: JwtService,
|
jwt: JwtService,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
|
let config = Arc::new(config);
|
||||||
|
let tenants = TenantService::new(pool.clone());
|
||||||
|
|
||||||
Self {
|
Self {
|
||||||
pool,
|
pool,
|
||||||
config: Arc::new(config),
|
config,
|
||||||
storage,
|
storage,
|
||||||
jwt,
|
jwt,
|
||||||
|
tenants,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn db(&self) -> AppResult<PgPooledConnection> {
|
pub fn db_for_tenant(&self, tenant_id: Uuid) -> AppResult<PgPooledConnection> {
|
||||||
|
debug_assert!(!tenant_id.is_nil(), "nil tenant_id passed to db_for_tenant");
|
||||||
|
let mut conn = self.db_unscoped()?;
|
||||||
|
apply_tenant_guc(&mut conn, tenant_id)?;
|
||||||
|
Ok(conn)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn db_unscoped(&self) -> AppResult<PgPooledConnection> {
|
||||||
self.pool
|
self.pool
|
||||||
.get()
|
.get()
|
||||||
.map_err(|err| AppError::internal(format!("database pool error: {err}")))
|
.map_err(|err| AppError::internal(format!("database pool error: {err}")))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn storage_for_tenant(&self, tenant_id: Uuid) -> AppResult<TenantStorage> {
|
||||||
|
let tenant = self.tenants.get_by_id(tenant_id)?;
|
||||||
|
TenantStorage::new(self.storage.clone(), &tenant)
|
||||||
|
.map_err(|err| AppError::internal(format!("tenant storage error: {err}")))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+54
-1
@@ -1,11 +1,15 @@
|
|||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
use anyhow::{Context, Result};
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::{anyhow, Context, Result};
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use aws_sdk_s3::presigning::PresigningConfig;
|
use aws_sdk_s3::presigning::PresigningConfig;
|
||||||
use aws_sdk_s3::primitives::ByteStream;
|
use aws_sdk_s3::primitives::ByteStream;
|
||||||
use aws_sdk_s3::Client as S3Client;
|
use aws_sdk_s3::Client as S3Client;
|
||||||
|
|
||||||
|
use crate::models::Tenant;
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
pub trait ObjectStorage: Send + Sync + 'static {
|
pub trait ObjectStorage: Send + Sync + 'static {
|
||||||
async fn put_object(
|
async fn put_object(
|
||||||
@@ -119,3 +123,52 @@ impl ObjectStorage for S3Storage {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct TenantStorage {
|
||||||
|
inner: Arc<dyn ObjectStorage>,
|
||||||
|
root: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TenantStorage {
|
||||||
|
pub fn new(inner: Arc<dyn ObjectStorage>, tenant: &Tenant) -> Result<Self> {
|
||||||
|
let root = tenant
|
||||||
|
.storage_root
|
||||||
|
.as_ref()
|
||||||
|
.ok_or_else(|| anyhow!("tenant {} missing storage_root", tenant.id))?
|
||||||
|
.to_owned();
|
||||||
|
|
||||||
|
Ok(Self { inner, root })
|
||||||
|
}
|
||||||
|
|
||||||
|
fn qualify(&self, key: &str) -> String {
|
||||||
|
format!("{}{}", self.root, key)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn put_object(
|
||||||
|
&self,
|
||||||
|
key: &str,
|
||||||
|
bytes: Vec<u8>,
|
||||||
|
content_type: Option<String>,
|
||||||
|
content_disposition: Option<String>,
|
||||||
|
) -> Result<()> {
|
||||||
|
let qualified = self.qualify(key);
|
||||||
|
self.inner
|
||||||
|
.put_object(&qualified, bytes, content_type, content_disposition)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn presign_get_object(&self, key: &str, expires_in: Duration) -> Result<String> {
|
||||||
|
let qualified = self.qualify(key);
|
||||||
|
self.inner.presign_get_object(&qualified, expires_in).await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_object(&self, key: &str) -> Result<Vec<u8>> {
|
||||||
|
let qualified = self.qualify(key);
|
||||||
|
self.inner.get_object(&qualified).await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn delete_object(&self, key: &str) -> Result<()> {
|
||||||
|
let qualified = self.qualify(key);
|
||||||
|
self.inner.delete_object(&qualified).await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -0,0 +1,91 @@
|
|||||||
|
use axum::{async_trait, extract::FromRequestParts, http::request::Parts};
|
||||||
|
use diesel::{pg::PgConnection, prelude::*, sql_types::Text};
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
db::PgPool,
|
||||||
|
error::{AppError, AppResult},
|
||||||
|
models::Tenant,
|
||||||
|
schema::tenants::dsl,
|
||||||
|
state::AppState,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub struct TenantRepository;
|
||||||
|
|
||||||
|
impl TenantRepository {
|
||||||
|
pub fn get_by_id(conn: &mut PgConnection, tenant_id: Uuid) -> AppResult<Tenant> {
|
||||||
|
dsl::tenants.find(tenant_id).first(conn).map_err(Into::into)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_by_slug(conn: &mut PgConnection, slug: &str) -> AppResult<Tenant> {
|
||||||
|
dsl::tenants
|
||||||
|
.filter(dsl::slug.eq(slug))
|
||||||
|
.first(conn)
|
||||||
|
.map_err(Into::into)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct TenantService {
|
||||||
|
pool: PgPool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TenantService {
|
||||||
|
pub fn new(pool: PgPool) -> Self {
|
||||||
|
Self { pool }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_by_id(&self, tenant_id: Uuid) -> AppResult<Tenant> {
|
||||||
|
let tenant = self.load(|conn| TenantRepository::get_by_id(conn, tenant_id))?;
|
||||||
|
Ok(tenant)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_by_slug(&self, slug: &str) -> AppResult<Tenant> {
|
||||||
|
let slug_owned = slug.to_owned();
|
||||||
|
let tenant = self.load(|conn| TenantRepository::get_by_slug(conn, &slug_owned))?;
|
||||||
|
Ok(tenant)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn tenant_id_for_slug(&self, slug: &str) -> AppResult<Uuid> {
|
||||||
|
Ok(self.get_by_slug(slug)?.id)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn load<F>(&self, loader: F) -> AppResult<Tenant>
|
||||||
|
where
|
||||||
|
F: FnOnce(&mut PgConnection) -> AppResult<Tenant>,
|
||||||
|
{
|
||||||
|
let mut conn = self
|
||||||
|
.pool
|
||||||
|
.get()
|
||||||
|
.map_err(|err| AppError::internal(format!("database pool error: {err}")))?;
|
||||||
|
let tenant = loader(&mut conn)?;
|
||||||
|
Ok(tenant)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn apply_tenant_guc(conn: &mut PgConnection, tenant_id: Uuid) -> AppResult<()> {
|
||||||
|
diesel::sql_query("SELECT set_config('papercrate.tenant_id', $1, true)")
|
||||||
|
.bind::<Text, _>(tenant_id.to_string())
|
||||||
|
.execute(conn)
|
||||||
|
.map(|_| ())
|
||||||
|
.map_err(AppError::from)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct TenantContext {
|
||||||
|
pub tenant: Tenant,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl FromRequestParts<AppState> for TenantContext {
|
||||||
|
type Rejection = AppError;
|
||||||
|
|
||||||
|
async fn from_request_parts(
|
||||||
|
_parts: &mut Parts,
|
||||||
|
state: &AppState,
|
||||||
|
) -> Result<Self, Self::Rejection> {
|
||||||
|
let tenant = state
|
||||||
|
.tenants
|
||||||
|
.get_by_slug(&state.config.default_tenant_slug)?;
|
||||||
|
Ok(Self { tenant })
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
|
||||||
|
use crate::{config::AppConfig, state::AppState, utils::tracing::init_tracing};
|
||||||
|
|
||||||
|
/// Initialize tracing, load configuration, and build the shared `AppState`.
|
||||||
|
/// Optionally override the connection pool size for lightweight components.
|
||||||
|
pub async fn init_component(name: &str, pool_override: Option<u32>) -> Result<Arc<AppState>> {
|
||||||
|
init_tracing("info");
|
||||||
|
let config = AppConfig::load_and_log(name)?;
|
||||||
|
let state = AppState::initialize(config, pool_override).await?;
|
||||||
|
Ok(Arc::new(state))
|
||||||
|
}
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
use diesel::{pg::PgConnection, result::Error as DieselError};
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
error::{AppError, AppResult},
|
||||||
|
state::AppState,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub trait EnsureEntity<T> {
|
||||||
|
fn one(self) -> AppResult<T>;
|
||||||
|
fn maybe(self) -> AppResult<Option<T>>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T> EnsureEntity<T> for Result<T, DieselError> {
|
||||||
|
fn one(self) -> AppResult<T> {
|
||||||
|
self.map_err(AppError::from)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn maybe(self) -> AppResult<Option<T>> {
|
||||||
|
match self {
|
||||||
|
Ok(value) => Ok(Some(value)),
|
||||||
|
Err(DieselError::NotFound) => Ok(None),
|
||||||
|
Err(err) => Err(AppError::from(err)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AppState {
|
||||||
|
pub fn with_tenant_conn<F, T>(&self, tenant_id: Uuid, f: F) -> AppResult<T>
|
||||||
|
where
|
||||||
|
F: FnOnce(&mut PgConnection) -> AppResult<T>,
|
||||||
|
{
|
||||||
|
let mut conn = self.db_for_tenant(tenant_id)?;
|
||||||
|
f(&mut conn)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn validate_bulk_ids(ids: &mut Vec<Uuid>, label: &str) -> AppResult<()> {
|
||||||
|
if ids.is_empty() {
|
||||||
|
return Err(AppError::bad_request(format!("{label} must not be empty")));
|
||||||
|
}
|
||||||
|
ids.sort_unstable();
|
||||||
|
ids.dedup();
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub trait IntoJsonResponse<T> {
|
||||||
|
fn into_json(self) -> AppResult<axum::Json<T>>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T> IntoJsonResponse<T> for T {
|
||||||
|
fn into_json(self) -> AppResult<axum::Json<T>> {
|
||||||
|
Ok(axum::Json(self))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn no_content() -> AppResult<axum::http::StatusCode> {
|
||||||
|
Ok(axum::http::StatusCode::NO_CONTENT)
|
||||||
|
}
|
||||||
@@ -0,0 +1,22 @@
|
|||||||
|
use percent_encoding::{utf8_percent_encode, NON_ALPHANUMERIC};
|
||||||
|
|
||||||
|
/// Build an inline `Content-Disposition` header value for a given filename.
|
||||||
|
pub fn inline_content_disposition(filename: &str) -> Option<String> {
|
||||||
|
if filename.is_empty() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let sanitized: String = filename
|
||||||
|
.chars()
|
||||||
|
.map(|ch| match ch {
|
||||||
|
'"' | '\\' => '_',
|
||||||
|
_ => ch,
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let encoded = utf8_percent_encode(&sanitized, NON_ALPHANUMERIC);
|
||||||
|
|
||||||
|
Some(format!(
|
||||||
|
"inline; filename=\"{}\"; filename*=UTF-8''{}",
|
||||||
|
sanitized, encoded
|
||||||
|
))
|
||||||
|
}
|
||||||
@@ -1 +1,8 @@
|
|||||||
|
pub mod bootstrap;
|
||||||
|
pub mod db;
|
||||||
|
pub mod http;
|
||||||
pub mod json;
|
pub mod json;
|
||||||
|
pub mod storage_paths;
|
||||||
|
pub mod time;
|
||||||
|
pub mod tracing;
|
||||||
|
pub mod validation;
|
||||||
|
|||||||
@@ -0,0 +1,115 @@
|
|||||||
|
//! Document storage path helpers.
|
||||||
|
//!
|
||||||
|
//! NOTE: The path layout produced here is part of the durable storage contract.
|
||||||
|
//! External systems (presigned URLs, lifecycle jobs, migrations) expect the
|
||||||
|
//! `documents/{document_id}/...` structure to remain stable. Coordinate before
|
||||||
|
//! changing any of these helpers to avoid breaking compatibility with existing
|
||||||
|
//! objects.
|
||||||
|
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
const DOCUMENTS_PREFIX: &str = "documents";
|
||||||
|
|
||||||
|
/// Returns the root prefix for all objects belonging to a document.
|
||||||
|
pub fn document_prefix(document_id: Uuid) -> String {
|
||||||
|
format!("{DOCUMENTS_PREFIX}/{document_id}")
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the prefix for a specific document version (without the object id).
|
||||||
|
pub fn document_version_prefix(document_id: Uuid, version_number: i32) -> String {
|
||||||
|
format!("{}/v{}", document_prefix(document_id), version_number)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the storage key for a stored document version blob.
|
||||||
|
pub fn document_version_object_key(
|
||||||
|
document_id: Uuid,
|
||||||
|
version_number: i32,
|
||||||
|
version_id: Uuid,
|
||||||
|
) -> String {
|
||||||
|
format!(
|
||||||
|
"{}/{}",
|
||||||
|
document_version_prefix(document_id, version_number),
|
||||||
|
version_id
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn document_asset_prefix(document_id: Uuid, version_number: i32) -> String {
|
||||||
|
format!(
|
||||||
|
"{}/assets",
|
||||||
|
document_version_prefix(document_id, version_number)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn document_asset_type_prefix(document_id: Uuid, version_number: i32, asset_type: &str) -> String {
|
||||||
|
format!(
|
||||||
|
"{}/{}",
|
||||||
|
document_asset_prefix(document_id, version_number),
|
||||||
|
asset_type
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the storage prefix under which the asset objects for a type/id pair live.
|
||||||
|
pub fn document_asset_object_prefix(
|
||||||
|
document_id: Uuid,
|
||||||
|
version_number: i32,
|
||||||
|
asset_type: &str,
|
||||||
|
asset_id: Uuid,
|
||||||
|
) -> String {
|
||||||
|
format!(
|
||||||
|
"{}/{}",
|
||||||
|
document_asset_type_prefix(document_id, version_number, asset_type),
|
||||||
|
asset_id
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the full storage key for a specific asset object (ordinal).
|
||||||
|
pub fn document_asset_object_key(
|
||||||
|
document_id: Uuid,
|
||||||
|
version_number: i32,
|
||||||
|
asset_type: &str,
|
||||||
|
asset_id: Uuid,
|
||||||
|
ordinal: i32,
|
||||||
|
) -> String {
|
||||||
|
format!(
|
||||||
|
"{}/{}",
|
||||||
|
document_asset_object_prefix(document_id, version_number, asset_type, asset_id),
|
||||||
|
ordinal
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn generates_expected_paths() {
|
||||||
|
let document_id = Uuid::nil();
|
||||||
|
let version_id = Uuid::nil();
|
||||||
|
let asset_id = Uuid::nil();
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
document_prefix(document_id),
|
||||||
|
format!("documents/{document_id}")
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
document_version_prefix(document_id, 3),
|
||||||
|
format!("documents/{document_id}/v3")
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
document_version_object_key(document_id, 3, version_id),
|
||||||
|
format!("documents/{document_id}/v3/{version_id}")
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
document_asset_object_prefix(document_id, 3, "preview", asset_id),
|
||||||
|
format!("documents/{document_id}/v3/assets/preview/{asset_id}")
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
document_asset_object_key(document_id, 3, "preview", asset_id, 2),
|
||||||
|
format!("documents/{document_id}/v3/assets/preview/{asset_id}/2")
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
use chrono::{DateTime, NaiveDateTime, Utc};
|
||||||
|
|
||||||
|
/// Format a timestamp as RFC3339 using UTC.
|
||||||
|
pub fn to_iso(dt: NaiveDateTime) -> String {
|
||||||
|
DateTime::<Utc>::from_naive_utc_and_offset(dt, Utc).to_rfc3339()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Format a timestamp for HTTP headers (RFC 7231 date).
|
||||||
|
pub fn to_http_date(dt: NaiveDateTime) -> String {
|
||||||
|
DateTime::<Utc>::from_naive_utc_and_offset(dt, Utc)
|
||||||
|
.format("%a, %d %b %Y %H:%M:%S GMT")
|
||||||
|
.to_string()
|
||||||
|
}
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
use tracing_subscriber::EnvFilter;
|
||||||
|
|
||||||
|
/// Initialize tracing with an optional default level.
|
||||||
|
///
|
||||||
|
/// Falls back to `default_level` when `RUST_LOG` is not provided.
|
||||||
|
pub fn init_tracing(default_level: &str) {
|
||||||
|
let filter =
|
||||||
|
EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(default_level));
|
||||||
|
tracing_subscriber::fmt()
|
||||||
|
.with_env_filter(filter)
|
||||||
|
.with_target(false)
|
||||||
|
.compact()
|
||||||
|
.init();
|
||||||
|
}
|
||||||
@@ -0,0 +1,10 @@
|
|||||||
|
use crate::error::{AppError, AppResult};
|
||||||
|
|
||||||
|
/// Ensure an entity exists, returning a bad request error when it does not.
|
||||||
|
pub fn ensure_exists(exists: bool, entity: &str) -> AppResult<()> {
|
||||||
|
if exists {
|
||||||
|
Ok(())
|
||||||
|
} else {
|
||||||
|
Err(AppError::bad_request(format!("{entity} does not exist")))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -14,6 +14,7 @@ use crate::{
|
|||||||
models::{Document, DocumentAsset, DocumentVersion},
|
models::{Document, DocumentAsset, DocumentVersion},
|
||||||
schema::{document_assets, document_versions, documents},
|
schema::{document_assets, document_versions, documents},
|
||||||
state::AppState,
|
state::AppState,
|
||||||
|
storage::TenantStorage,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{JobExecution, JobHandler};
|
use super::{JobExecution, JobHandler};
|
||||||
@@ -40,7 +41,12 @@ impl JobHandler for AnalyzeDocumentJob {
|
|||||||
JOB_ANALYZE_DOCUMENT
|
JOB_ANALYZE_DOCUMENT
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(&self, state: Arc<AppState>, job: crate::models::Job) -> JobExecution {
|
async fn handle(
|
||||||
|
&self,
|
||||||
|
state: Arc<AppState>,
|
||||||
|
job: crate::models::Job,
|
||||||
|
_storage: TenantStorage,
|
||||||
|
) -> JobExecution {
|
||||||
let payload: AnalyzePayload = match serde_json::from_value(job.payload.clone()) {
|
let payload: AnalyzePayload = match serde_json::from_value(job.payload.clone()) {
|
||||||
Ok(payload) => payload,
|
Ok(payload) => payload,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
@@ -51,7 +57,9 @@ impl JobHandler for AnalyzeDocumentJob {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let state_clone = state.clone();
|
let state_clone = state.clone();
|
||||||
match task::spawn_blocking(move || analyze_document(state_clone, payload)).await {
|
let tenant_id = job.tenant_id;
|
||||||
|
match task::spawn_blocking(move || analyze_document(state_clone, tenant_id, payload)).await
|
||||||
|
{
|
||||||
Ok(Ok(execution)) => execution,
|
Ok(Ok(execution)) => execution,
|
||||||
Ok(Err(err)) => {
|
Ok(Err(err)) => {
|
||||||
warn!(job_id = %job.id, error = %err, "analyze job will retry");
|
warn!(job_id = %job.id, error = %err, "analyze job will retry");
|
||||||
@@ -71,8 +79,14 @@ impl JobHandler for AnalyzeDocumentJob {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn analyze_document(state: Arc<AppState>, payload: AnalyzePayload) -> Result<JobExecution, String> {
|
fn analyze_document(
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
state: Arc<AppState>,
|
||||||
|
tenant_id: Uuid,
|
||||||
|
payload: AnalyzePayload,
|
||||||
|
) -> Result<JobExecution, String> {
|
||||||
|
let mut conn = state
|
||||||
|
.db_for_tenant(tenant_id)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let version: DocumentVersion = document_versions::table
|
let version: DocumentVersion = document_versions::table
|
||||||
.find(payload.document_version_id)
|
.find(payload.document_version_id)
|
||||||
@@ -88,12 +102,15 @@ fn analyze_document(state: Arc<AppState>, payload: AnalyzePayload) -> Result<Job
|
|||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let tenant_id = document.tenant_id;
|
||||||
|
|
||||||
let (supported, reason) = determine_thumbnail_support(&document);
|
let (supported, reason) = determine_thumbnail_support(&document);
|
||||||
let ocr_supported = document_is_pdf(&document);
|
let ocr_supported = document_is_pdf(&document);
|
||||||
|
|
||||||
let existing_ocr: Option<DocumentAsset> = document_assets::table
|
let existing_ocr: Option<DocumentAsset> = document_assets::table
|
||||||
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
||||||
.filter(document_assets::asset_type.eq(OCR_TEXT_ASSET_TYPE))
|
.filter(document_assets::asset_type.eq(OCR_TEXT_ASSET_TYPE))
|
||||||
|
.filter(document_assets::tenant_id.eq(tenant_id))
|
||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.optional()
|
.optional()
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
@@ -129,6 +146,7 @@ fn analyze_document(state: Arc<AppState>, payload: AnalyzePayload) -> Result<Job
|
|||||||
if supported {
|
if supported {
|
||||||
let enqueue_result = enqueue_job(
|
let enqueue_result = enqueue_job(
|
||||||
&mut conn,
|
&mut conn,
|
||||||
|
tenant_id,
|
||||||
JOB_GENERATE_THUMBNAILS,
|
JOB_GENERATE_THUMBNAILS,
|
||||||
json!({
|
json!({
|
||||||
"document_id": payload.document_id,
|
"document_id": payload.document_id,
|
||||||
@@ -146,6 +164,7 @@ fn analyze_document(state: Arc<AppState>, payload: AnalyzePayload) -> Result<Job
|
|||||||
if ocr_supported && !skip_ocr {
|
if ocr_supported && !skip_ocr {
|
||||||
let enqueue_result = enqueue_job(
|
let enqueue_result = enqueue_job(
|
||||||
&mut conn,
|
&mut conn,
|
||||||
|
tenant_id,
|
||||||
JOB_GENERATE_OCR_TEXT,
|
JOB_GENERATE_OCR_TEXT,
|
||||||
json!({
|
json!({
|
||||||
"document_id": payload.document_id,
|
"document_id": payload.document_id,
|
||||||
|
|||||||
@@ -12,9 +12,10 @@ use uuid::Uuid;
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
jobs::JOB_INDEX_DOCUMENT_TEXT,
|
jobs::JOB_INDEX_DOCUMENT_TEXT,
|
||||||
models::{Document, DocumentAsset, DocumentVersion},
|
models::{Document, DocumentVersion},
|
||||||
schema::{document_assets, document_versions, documents},
|
schema::{document_asset_objects, document_assets, document_versions, documents},
|
||||||
state::AppState,
|
state::AppState,
|
||||||
|
storage::TenantStorage,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{ocr::OCR_TEXT_ASSET_TYPE, JobExecution, JobHandler};
|
use super::{ocr::OCR_TEXT_ASSET_TYPE, JobExecution, JobHandler};
|
||||||
@@ -39,7 +40,12 @@ impl JobHandler for IndexDocumentTextJob {
|
|||||||
JOB_INDEX_DOCUMENT_TEXT
|
JOB_INDEX_DOCUMENT_TEXT
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(&self, state: Arc<AppState>, job: crate::models::Job) -> JobExecution {
|
async fn handle(
|
||||||
|
&self,
|
||||||
|
state: Arc<AppState>,
|
||||||
|
job: crate::models::Job,
|
||||||
|
storage: TenantStorage,
|
||||||
|
) -> JobExecution {
|
||||||
let payload: IndexPayload = match serde_json::from_value(job.payload.clone()) {
|
let payload: IndexPayload = match serde_json::from_value(job.payload.clone()) {
|
||||||
Ok(payload) => payload,
|
Ok(payload) => payload,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
@@ -52,16 +58,29 @@ impl JobHandler for IndexDocumentTextJob {
|
|||||||
let quickwit_endpoint = match &state.config.quickwit_endpoint {
|
let quickwit_endpoint = match &state.config.quickwit_endpoint {
|
||||||
Some(endpoint) => endpoint.clone(),
|
Some(endpoint) => endpoint.clone(),
|
||||||
None => {
|
None => {
|
||||||
warn!("quickwit endpoint missing; skipping indexing");
|
return JobExecution::Failed {
|
||||||
return JobExecution::Success;
|
error: "quickwit endpoint missing".into(),
|
||||||
|
};
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let quickwit_index = match &state.config.quickwit_index {
|
let tenant = match state.tenants.get_by_id(job.tenant_id) {
|
||||||
Some(index) => index.clone(),
|
Ok(tenant) => tenant,
|
||||||
|
Err(err) => {
|
||||||
|
warn!(job_id = %job.id, error = ?err, "failed to load tenant for indexing");
|
||||||
|
return JobExecution::Retry {
|
||||||
|
delay: Duration::from_secs(30),
|
||||||
|
error: format!("failed to load tenant: {err:?}"),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let quickwit_index = match tenant.quickwit_index.clone() {
|
||||||
|
Some(index) => index,
|
||||||
None => {
|
None => {
|
||||||
warn!("quickwit index missing; skipping indexing");
|
return JobExecution::Failed {
|
||||||
return JobExecution::Success;
|
error: "tenant quickwit index not configured".into(),
|
||||||
|
};
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -87,15 +106,15 @@ impl JobHandler for IndexDocumentTextJob {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
if context.text_asset.is_none() {
|
if context.text_s3_key.is_none() {
|
||||||
warn!(job_id = %job.id, "missing OCR text asset; failing indexing job");
|
warn!(job_id = %job.id, "missing OCR text asset; failing indexing job");
|
||||||
return JobExecution::Failed {
|
return JobExecution::Failed {
|
||||||
error: "missing OCR text asset".into(),
|
error: "missing OCR text asset".into(),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
let asset = context.text_asset.unwrap();
|
let s3_key = context.text_s3_key.unwrap();
|
||||||
let text = match state.storage.get_object(&asset.s3_key).await {
|
let text = match storage.get_object(&s3_key).await {
|
||||||
Ok(bytes) => match String::from_utf8(bytes) {
|
Ok(bytes) => match String::from_utf8(bytes) {
|
||||||
Ok(text) => text,
|
Ok(text) => text,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
@@ -129,6 +148,7 @@ impl JobHandler for IndexDocumentTextJob {
|
|||||||
let payload = json!({
|
let payload = json!({
|
||||||
"document_id": context.document.id,
|
"document_id": context.document.id,
|
||||||
"version_id": context.version.id,
|
"version_id": context.version.id,
|
||||||
|
"tenant_id": job.tenant_id,
|
||||||
"title": context.document.title.to_lowercase(),
|
"title": context.document.title.to_lowercase(),
|
||||||
"text": text.to_lowercase()
|
"text": text.to_lowercase()
|
||||||
});
|
});
|
||||||
@@ -169,15 +189,15 @@ impl JobHandler for IndexDocumentTextJob {
|
|||||||
struct IndexContext {
|
struct IndexContext {
|
||||||
document: Document,
|
document: Document,
|
||||||
version: DocumentVersion,
|
version: DocumentVersion,
|
||||||
text_asset: Option<DocumentAsset>,
|
text_s3_key: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
fn load_context(state: Arc<AppState>, payload: &IndexPayload) -> Result<IndexContext, String> {
|
fn load_context(state: Arc<AppState>, payload: &IndexPayload) -> Result<IndexContext, String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let mut base_conn = state.db_unscoped().map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let version: DocumentVersion = document_versions::table
|
let version: DocumentVersion = document_versions::table
|
||||||
.find(payload.document_version_id)
|
.find(payload.document_version_id)
|
||||||
.first(&mut conn)
|
.first(&mut base_conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
if version.document_id != payload.document_id {
|
if version.document_id != payload.document_id {
|
||||||
@@ -186,12 +206,26 @@ fn load_context(state: Arc<AppState>, payload: &IndexPayload) -> Result<IndexCon
|
|||||||
|
|
||||||
let document: Document = documents::table
|
let document: Document = documents::table
|
||||||
.find(payload.document_id)
|
.find(payload.document_id)
|
||||||
.first(&mut conn)
|
.first(&mut base_conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let text_asset: Option<DocumentAsset> = document_assets::table
|
let tenant_id = document.tenant_id;
|
||||||
|
drop(base_conn);
|
||||||
|
|
||||||
|
let mut conn = state
|
||||||
|
.db_for_tenant(tenant_id)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let text_s3_key: Option<String> = document_asset_objects::table
|
||||||
|
.inner_join(
|
||||||
|
document_assets::table.on(document_asset_objects::asset_id.eq(document_assets::id)),
|
||||||
|
)
|
||||||
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
||||||
.filter(document_assets::asset_type.eq(OCR_TEXT_ASSET_TYPE))
|
.filter(document_assets::asset_type.eq(OCR_TEXT_ASSET_TYPE))
|
||||||
|
.filter(document_assets::tenant_id.eq(tenant_id))
|
||||||
|
.filter(document_asset_objects::ordinal.eq(1))
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant_id))
|
||||||
|
.select(document_asset_objects::s3_key)
|
||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.optional()
|
.optional()
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
@@ -199,6 +233,6 @@ fn load_context(state: Arc<AppState>, payload: &IndexPayload) -> Result<IndexCon
|
|||||||
Ok(IndexContext {
|
Ok(IndexContext {
|
||||||
document,
|
document,
|
||||||
version,
|
version,
|
||||||
text_asset,
|
text_s3_key,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ use crate::{
|
|||||||
jobs::{mark_job_failed, mark_job_succeeded, reserve_job, retry_job_after, JobQueueError},
|
jobs::{mark_job_failed, mark_job_succeeded, reserve_job, retry_job_after, JobQueueError},
|
||||||
models::Job,
|
models::Job,
|
||||||
state::AppState,
|
state::AppState,
|
||||||
|
storage::TenantStorage,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub mod analyze;
|
pub mod analyze;
|
||||||
@@ -25,7 +26,7 @@ pub enum JobExecution {
|
|||||||
#[async_trait]
|
#[async_trait]
|
||||||
pub trait JobHandler: Send + Sync {
|
pub trait JobHandler: Send + Sync {
|
||||||
fn job_type(&self) -> &'static str;
|
fn job_type(&self) -> &'static str;
|
||||||
async fn handle(&self, state: Arc<AppState>, job: Job) -> JobExecution;
|
async fn handle(&self, state: Arc<AppState>, job: Job, storage: TenantStorage) -> JobExecution;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct Worker {
|
pub struct Worker {
|
||||||
@@ -71,7 +72,7 @@ impl Worker {
|
|||||||
return Ok(false);
|
return Ok(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut conn = match self.state.db() {
|
let mut conn = match self.state.db_unscoped() {
|
||||||
Ok(conn) => conn,
|
Ok(conn) => conn,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
error!(?err, "failed to obtain database connection in worker");
|
error!(?err, "failed to obtain database connection in worker");
|
||||||
@@ -84,10 +85,22 @@ impl Worker {
|
|||||||
|
|
||||||
if let Some(job) = job_opt {
|
if let Some(job) = job_opt {
|
||||||
if let Some(handler) = self.handlers.get(job.job_type.as_str()) {
|
if let Some(handler) = self.handlers.get(job.job_type.as_str()) {
|
||||||
let result = handler.handle(self.state.clone(), job.clone()).await;
|
let execution = match self.state.storage_for_tenant(job.tenant_id) {
|
||||||
match result {
|
Ok(storage) => {
|
||||||
|
handler
|
||||||
|
.handle(self.state.clone(), job.clone(), storage)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
Err(err) => {
|
||||||
|
error!(job_id = %job.id, error = ?err, "failed to load tenant storage for job");
|
||||||
|
JobExecution::Failed {
|
||||||
|
error: format!("tenant storage unavailable: {err:?}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
match execution {
|
||||||
JobExecution::Success => {
|
JobExecution::Success => {
|
||||||
if let Ok(mut conn) = self.state.db() {
|
if let Ok(mut conn) = self.state.db_unscoped() {
|
||||||
mark_job_succeeded(&mut conn, job.id)?;
|
mark_job_succeeded(&mut conn, job.id)?;
|
||||||
info!(job_id = %job.id, job_type = %job.job_type, "job completed successfully");
|
info!(job_id = %job.id, job_type = %job.job_type, "job completed successfully");
|
||||||
} else {
|
} else {
|
||||||
@@ -96,7 +109,7 @@ impl Worker {
|
|||||||
}
|
}
|
||||||
JobExecution::Retry { delay, error } => {
|
JobExecution::Retry { delay, error } => {
|
||||||
warn!(job_id = %job.id, job_type = %job.job_type, %error, "job will retry");
|
warn!(job_id = %job.id, job_type = %job.job_type, %error, "job will retry");
|
||||||
if let Ok(mut conn) = self.state.db() {
|
if let Ok(mut conn) = self.state.db_unscoped() {
|
||||||
retry_job_after(&mut conn, job.id, delay, &error)?;
|
retry_job_after(&mut conn, job.id, delay, &error)?;
|
||||||
} else {
|
} else {
|
||||||
error!("failed to requeue job for retry due to pool error");
|
error!("failed to requeue job for retry due to pool error");
|
||||||
@@ -104,7 +117,7 @@ impl Worker {
|
|||||||
}
|
}
|
||||||
JobExecution::Failed { error } => {
|
JobExecution::Failed { error } => {
|
||||||
error!(job_id = %job.id, job_type = %job.job_type, %error, "job failed");
|
error!(job_id = %job.id, job_type = %job.job_type, %error, "job failed");
|
||||||
if let Ok(mut conn) = self.state.db() {
|
if let Ok(mut conn) = self.state.db_unscoped() {
|
||||||
mark_job_failed(&mut conn, job.id, &error)?;
|
mark_job_failed(&mut conn, job.id, &error)?;
|
||||||
} else {
|
} else {
|
||||||
error!("failed to mark job failed due to pool error");
|
error!("failed to mark job failed due to pool error");
|
||||||
@@ -113,7 +126,7 @@ impl Worker {
|
|||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
error!(job_type = %job.job_type, "no handler registered for job type");
|
error!(job_type = %job.job_type, "no handler registered for job type");
|
||||||
if let Ok(mut conn) = self.state.db() {
|
if let Ok(mut conn) = self.state.db_unscoped() {
|
||||||
mark_job_failed(&mut conn, job.id, "no handler registered")?;
|
mark_job_failed(&mut conn, job.id, "no handler registered")?;
|
||||||
} else {
|
} else {
|
||||||
error!("failed to mark job failed for missing handler due to pool error");
|
error!("failed to mark job failed for missing handler due to pool error");
|
||||||
|
|||||||
+119
-30
@@ -19,9 +19,14 @@ use uuid::Uuid;
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
jobs::{enqueue_job, JOB_GENERATE_OCR_TEXT, JOB_INDEX_DOCUMENT_TEXT},
|
jobs::{enqueue_job, JOB_GENERATE_OCR_TEXT, JOB_INDEX_DOCUMENT_TEXT},
|
||||||
models::{Document, DocumentAsset, DocumentVersion, NewDocumentAsset},
|
models::{
|
||||||
schema::{document_assets, document_versions, documents},
|
Document, DocumentAsset, DocumentAssetObject, DocumentVersion, NewDocumentAsset,
|
||||||
|
NewDocumentAssetObject,
|
||||||
|
},
|
||||||
|
schema::{document_asset_objects, document_assets, document_versions, documents},
|
||||||
state::AppState,
|
state::AppState,
|
||||||
|
storage::TenantStorage,
|
||||||
|
utils::storage_paths::document_asset_object_prefix,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{JobExecution, JobHandler};
|
use super::{JobExecution, JobHandler};
|
||||||
@@ -51,7 +56,12 @@ impl JobHandler for GenerateOcrTextJob {
|
|||||||
JOB_GENERATE_OCR_TEXT
|
JOB_GENERATE_OCR_TEXT
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(&self, state: Arc<AppState>, job: crate::models::Job) -> JobExecution {
|
async fn handle(
|
||||||
|
&self,
|
||||||
|
state: Arc<AppState>,
|
||||||
|
job: crate::models::Job,
|
||||||
|
storage: TenantStorage,
|
||||||
|
) -> JobExecution {
|
||||||
let payload: OcrPayload = match serde_json::from_value(job.payload.clone()) {
|
let payload: OcrPayload = match serde_json::from_value(job.payload.clone()) {
|
||||||
Ok(payload) => payload,
|
Ok(payload) => payload,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
@@ -88,7 +98,7 @@ impl JobHandler for GenerateOcrTextJob {
|
|||||||
return JobExecution::Success;
|
return JobExecution::Success;
|
||||||
}
|
}
|
||||||
|
|
||||||
let bytes = match state.storage.get_object(&context.version.s3_key).await {
|
let bytes = match storage.get_object(&context.version.s3_key).await {
|
||||||
Ok(bytes) => bytes,
|
Ok(bytes) => bytes,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
warn!(job_id = %job.id, error = %err, "failed to fetch document for ocr");
|
warn!(job_id = %job.id, error = %err, "failed to fetch document for ocr");
|
||||||
@@ -123,19 +133,24 @@ impl JobHandler for GenerateOcrTextJob {
|
|||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
let asset_id = context
|
if context.existing_asset.is_some() {
|
||||||
.existing_asset
|
for object in &context.existing_objects {
|
||||||
.as_ref()
|
if let Err(err) = storage.delete_object(&object.s3_key).await {
|
||||||
.map(|asset| asset.id)
|
warn!(job_id = %job.id, error = %err, s3_key = %object.s3_key, "failed to delete existing ocr asset object");
|
||||||
.unwrap_or_else(Uuid::new_v4);
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let s3_key = format!(
|
let asset_id = Uuid::new_v4();
|
||||||
"documents/{}/v{}/assets/{}/{}",
|
|
||||||
context.document.id, context.version.version_number, OCR_TEXT_ASSET_TYPE, asset_id
|
let s3_key = document_asset_object_prefix(
|
||||||
|
context.document.id,
|
||||||
|
context.version.version_number,
|
||||||
|
OCR_TEXT_ASSET_TYPE,
|
||||||
|
asset_id,
|
||||||
);
|
);
|
||||||
|
|
||||||
if let Err(err) = state
|
if let Err(err) = storage
|
||||||
.storage
|
|
||||||
.put_object(
|
.put_object(
|
||||||
&s3_key,
|
&s3_key,
|
||||||
generation.text.into_bytes(),
|
generation.text.into_bytes(),
|
||||||
@@ -158,11 +173,8 @@ impl JobHandler for GenerateOcrTextJob {
|
|||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(Ok(())) => {
|
Ok(Ok(())) => {
|
||||||
if state.config.quickwit_endpoint.is_some() && state.config.quickwit_index.is_some()
|
if let Err(err) = enqueue_index_job(&state, &payload) {
|
||||||
{
|
warn!(job_id = %job.id, error = %err, "failed to enqueue index job");
|
||||||
if let Err(err) = enqueue_index_job(&state, &payload) {
|
|
||||||
warn!(job_id = %job.id, error = %err, "failed to enqueue index job");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
JobExecution::Success
|
JobExecution::Success
|
||||||
}
|
}
|
||||||
@@ -193,6 +205,7 @@ struct OcrContext {
|
|||||||
document: Document,
|
document: Document,
|
||||||
version: DocumentVersion,
|
version: DocumentVersion,
|
||||||
existing_asset: Option<DocumentAsset>,
|
existing_asset: Option<DocumentAsset>,
|
||||||
|
existing_objects: Vec<DocumentAssetObject>,
|
||||||
skip: bool,
|
skip: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -202,11 +215,11 @@ struct OcrGeneration {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn load_ocr_context(state: Arc<AppState>, payload: &OcrPayload) -> Result<OcrContext, String> {
|
fn load_ocr_context(state: Arc<AppState>, payload: &OcrPayload) -> Result<OcrContext, String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let mut base_conn = state.db_unscoped().map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let version: DocumentVersion = document_versions::table
|
let version: DocumentVersion = document_versions::table
|
||||||
.find(payload.document_version_id)
|
.find(payload.document_version_id)
|
||||||
.first(&mut conn)
|
.first(&mut base_conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
if version.document_id != payload.document_id {
|
if version.document_id != payload.document_id {
|
||||||
@@ -215,32 +228,53 @@ fn load_ocr_context(state: Arc<AppState>, payload: &OcrPayload) -> Result<OcrCon
|
|||||||
|
|
||||||
let document: Document = documents::table
|
let document: Document = documents::table
|
||||||
.find(payload.document_id)
|
.find(payload.document_id)
|
||||||
.first(&mut conn)
|
.first(&mut base_conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let existing: Option<DocumentAsset> = document_assets::table
|
let tenant_id = document.tenant_id;
|
||||||
|
drop(base_conn);
|
||||||
|
|
||||||
|
let mut conn = state
|
||||||
|
.db_for_tenant(tenant_id)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let existing_asset: Option<DocumentAsset> = document_assets::table
|
||||||
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
||||||
.filter(document_assets::asset_type.eq(OCR_TEXT_ASSET_TYPE))
|
.filter(document_assets::asset_type.eq(OCR_TEXT_ASSET_TYPE))
|
||||||
|
.filter(document_assets::tenant_id.eq(tenant_id))
|
||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.optional()
|
.optional()
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let existing_objects: Vec<DocumentAssetObject> = if let Some(asset) = &existing_asset {
|
||||||
|
document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::asset_id.eq(asset.id))
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant_id))
|
||||||
|
.order(document_asset_objects::ordinal.asc())
|
||||||
|
.load(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?
|
||||||
|
} else {
|
||||||
|
Vec::new()
|
||||||
|
};
|
||||||
|
|
||||||
let is_pdf = document_is_pdf(&document);
|
let is_pdf = document_is_pdf(&document);
|
||||||
if !is_pdf {
|
if !is_pdf {
|
||||||
return Ok(OcrContext {
|
return Ok(OcrContext {
|
||||||
document,
|
document,
|
||||||
version,
|
version,
|
||||||
existing_asset: existing,
|
existing_asset: existing_asset,
|
||||||
|
existing_objects,
|
||||||
skip: true,
|
skip: true,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
let skip = existing.is_some() && !payload.force;
|
let skip = existing_asset.is_some() && !payload.force;
|
||||||
|
|
||||||
Ok(OcrContext {
|
Ok(OcrContext {
|
||||||
document,
|
document,
|
||||||
version,
|
version,
|
||||||
existing_asset: existing,
|
existing_asset,
|
||||||
|
existing_objects,
|
||||||
skip,
|
skip,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -369,18 +403,28 @@ fn persist_ocr_metadata(
|
|||||||
s3_key: &str,
|
s3_key: &str,
|
||||||
source: &'static str,
|
source: &'static str,
|
||||||
) -> Result<(), String> {
|
) -> Result<(), String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let tenant_id = context.document.tenant_id;
|
||||||
|
let mut conn = state
|
||||||
|
.db_for_tenant(tenant_id)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
if let Some(existing_asset) = &context.existing_asset {
|
||||||
|
diesel::delete(document_assets::table.filter(document_assets::id.eq(existing_asset.id)))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
}
|
||||||
|
|
||||||
let new_asset = NewDocumentAsset {
|
let new_asset = NewDocumentAsset {
|
||||||
id: asset_id,
|
id: asset_id,
|
||||||
document_version_id: context.version.id,
|
document_version_id: context.version.id,
|
||||||
asset_type: OCR_TEXT_ASSET_TYPE.to_string(),
|
asset_type: OCR_TEXT_ASSET_TYPE.to_string(),
|
||||||
s3_key: s3_key.to_string(),
|
|
||||||
mime_type: "text/plain".to_string(),
|
mime_type: "text/plain".to_string(),
|
||||||
metadata: json!({
|
metadata: json!({
|
||||||
"generated_at": Utc::now().to_rfc3339(),
|
"generated_at": Utc::now().to_rfc3339(),
|
||||||
"source": source,
|
"source": source,
|
||||||
}),
|
}),
|
||||||
|
cardinality: Some(1),
|
||||||
|
tenant_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
diesel::insert_into(document_assets::table)
|
diesel::insert_into(document_assets::table)
|
||||||
@@ -391,9 +435,43 @@ fn persist_ocr_metadata(
|
|||||||
))
|
))
|
||||||
.do_update()
|
.do_update()
|
||||||
.set((
|
.set((
|
||||||
document_assets::s3_key.eq(excluded(document_assets::s3_key)),
|
|
||||||
document_assets::mime_type.eq(excluded(document_assets::mime_type)),
|
document_assets::mime_type.eq(excluded(document_assets::mime_type)),
|
||||||
document_assets::metadata.eq(excluded(document_assets::metadata)),
|
document_assets::metadata.eq(excluded(document_assets::metadata)),
|
||||||
|
document_assets::cardinality.eq(excluded(document_assets::cardinality)),
|
||||||
|
))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let existing_object_id: Option<Uuid> = document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::asset_id.eq(asset_id))
|
||||||
|
.filter(document_asset_objects::ordinal.eq(1))
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant_id))
|
||||||
|
.select(document_asset_objects::id)
|
||||||
|
.first(&mut conn)
|
||||||
|
.optional()
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let object_id = existing_object_id.unwrap_or_else(Uuid::new_v4);
|
||||||
|
|
||||||
|
let new_object = NewDocumentAssetObject {
|
||||||
|
id: object_id,
|
||||||
|
asset_id,
|
||||||
|
ordinal: 1,
|
||||||
|
s3_key: s3_key.to_string(),
|
||||||
|
metadata: json!({}),
|
||||||
|
tenant_id,
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(document_asset_objects::table)
|
||||||
|
.values(&new_object)
|
||||||
|
.on_conflict((
|
||||||
|
document_asset_objects::asset_id,
|
||||||
|
document_asset_objects::ordinal,
|
||||||
|
))
|
||||||
|
.do_update()
|
||||||
|
.set((
|
||||||
|
document_asset_objects::s3_key.eq(excluded(document_asset_objects::s3_key)),
|
||||||
|
document_asset_objects::metadata.eq(excluded(document_asset_objects::metadata)),
|
||||||
))
|
))
|
||||||
.execute(&mut conn)
|
.execute(&mut conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
@@ -402,9 +480,20 @@ fn persist_ocr_metadata(
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn enqueue_index_job(state: &AppState, payload: &OcrPayload) -> Result<(), String> {
|
fn enqueue_index_job(state: &AppState, payload: &OcrPayload) -> Result<(), String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let mut base_conn = state.db_unscoped().map_err(|err| format!("{err:?}"))?;
|
||||||
|
let tenant_id: Uuid = documents::table
|
||||||
|
.find(payload.document_id)
|
||||||
|
.select(documents::tenant_id)
|
||||||
|
.first(&mut base_conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
drop(base_conn);
|
||||||
|
|
||||||
|
let mut conn = state
|
||||||
|
.db_for_tenant(tenant_id)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
enqueue_job(
|
enqueue_job(
|
||||||
&mut conn,
|
&mut conn,
|
||||||
|
tenant_id,
|
||||||
JOB_INDEX_DOCUMENT_TEXT,
|
JOB_INDEX_DOCUMENT_TEXT,
|
||||||
json!({
|
json!({
|
||||||
"document_id": payload.document_id,
|
"document_id": payload.document_id,
|
||||||
|
|||||||
+348
-127
@@ -13,9 +13,14 @@ use uuid::Uuid;
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
jobs::JOB_GENERATE_THUMBNAILS,
|
jobs::JOB_GENERATE_THUMBNAILS,
|
||||||
models::{Document, DocumentAsset, DocumentVersion, NewDocumentAsset},
|
models::{
|
||||||
schema::{document_assets, document_versions, documents},
|
Document, DocumentAsset, DocumentAssetObject, DocumentVersion, NewDocumentAsset,
|
||||||
|
NewDocumentAssetObject,
|
||||||
|
},
|
||||||
|
schema::{document_asset_objects, document_assets, document_versions, documents},
|
||||||
state::AppState,
|
state::AppState,
|
||||||
|
storage::TenantStorage,
|
||||||
|
utils::storage_paths::document_asset_object_key,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{analyze::determine_thumbnail_support, JobExecution, JobHandler};
|
use super::{analyze::determine_thumbnail_support, JobExecution, JobHandler};
|
||||||
@@ -49,7 +54,12 @@ impl JobHandler for GenerateThumbnailsJob {
|
|||||||
JOB_GENERATE_THUMBNAILS
|
JOB_GENERATE_THUMBNAILS
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(&self, state: Arc<AppState>, job: crate::models::Job) -> JobExecution {
|
async fn handle(
|
||||||
|
&self,
|
||||||
|
state: Arc<AppState>,
|
||||||
|
job: crate::models::Job,
|
||||||
|
storage: TenantStorage,
|
||||||
|
) -> JobExecution {
|
||||||
let payload: ThumbnailPayload = match serde_json::from_value(job.payload.clone()) {
|
let payload: ThumbnailPayload = match serde_json::from_value(job.payload.clone()) {
|
||||||
Ok(p) => p,
|
Ok(p) => p,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
@@ -85,7 +95,7 @@ impl JobHandler for GenerateThumbnailsJob {
|
|||||||
return JobExecution::Success;
|
return JobExecution::Success;
|
||||||
}
|
}
|
||||||
|
|
||||||
let bytes = match state.storage.get_object(&initial.version.s3_key).await {
|
let bytes = match storage.get_object(&initial.version.s3_key).await {
|
||||||
Ok(bytes) => bytes,
|
Ok(bytes) => bytes,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
warn!(job_id = %job.id, error = %err, "thumbnail fetch failed; will retry");
|
warn!(job_id = %job.id, error = %err, "thumbnail fetch failed; will retry");
|
||||||
@@ -142,86 +152,134 @@ impl JobHandler for GenerateThumbnailsJob {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let thumbnail_asset_id = initial
|
if initial.existing_preview.is_some() {
|
||||||
.existing_thumbnail
|
for object in &initial.existing_preview_objects {
|
||||||
.as_ref()
|
if let Err(err) = storage.delete_object(&object.s3_key).await {
|
||||||
.map(|asset| asset.id)
|
warn!(
|
||||||
.unwrap_or_else(Uuid::new_v4);
|
job_id = %job.id,
|
||||||
let thumbnail_s3_key = format!(
|
error = %err,
|
||||||
"documents/{}/v{}/assets/{}/{}",
|
s3_key = %object.s3_key,
|
||||||
initial.document.id,
|
"failed to delete existing preview object"
|
||||||
initial.version.version_number,
|
);
|
||||||
THUMBNAIL_ASSET_TYPE,
|
}
|
||||||
thumbnail_asset_id
|
}
|
||||||
);
|
|
||||||
|
|
||||||
let preview_asset_id = initial
|
|
||||||
.existing_preview
|
|
||||||
.as_ref()
|
|
||||||
.map(|asset| asset.id)
|
|
||||||
.unwrap_or_else(Uuid::new_v4);
|
|
||||||
let preview_s3_key = format!(
|
|
||||||
"documents/{}/v{}/assets/{}/{}",
|
|
||||||
initial.document.id,
|
|
||||||
initial.version.version_number,
|
|
||||||
PREVIEW_ASSET_TYPE,
|
|
||||||
preview_asset_id
|
|
||||||
);
|
|
||||||
|
|
||||||
if let Err(err) = state
|
|
||||||
.storage
|
|
||||||
.put_object(
|
|
||||||
&preview_s3_key,
|
|
||||||
generation.preview.image_bytes.clone(),
|
|
||||||
Some("image/png".into()),
|
|
||||||
None,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
warn!(job_id = %job.id, error = %err, "failed to upload preview; retrying");
|
|
||||||
return JobExecution::Retry {
|
|
||||||
delay: Duration::from_secs(30),
|
|
||||||
error: err.to_string(),
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Err(err) = state
|
if initial.existing_thumbnail.is_some() {
|
||||||
.storage
|
for object in &initial.existing_thumbnail_objects {
|
||||||
.put_object(
|
if let Err(err) = storage.delete_object(&object.s3_key).await {
|
||||||
&thumbnail_s3_key,
|
warn!(
|
||||||
generation.thumbnail.image_bytes.clone(),
|
job_id = %job.id,
|
||||||
Some("image/png".into()),
|
error = %err,
|
||||||
None,
|
s3_key = %object.s3_key,
|
||||||
)
|
"failed to delete existing thumbnail object"
|
||||||
.await
|
);
|
||||||
{
|
}
|
||||||
warn!(job_id = %job.id, error = %err, "failed to upload thumbnail; retrying");
|
}
|
||||||
return JobExecution::Retry {
|
|
||||||
delay: Duration::from_secs(30),
|
|
||||||
error: err.to_string(),
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let preview_asset_id = Uuid::new_v4();
|
||||||
|
|
||||||
|
let thumbnail_asset_id = Uuid::new_v4();
|
||||||
|
|
||||||
|
let mut preview_objects: Vec<AssetObjectPersistence> =
|
||||||
|
Vec::with_capacity(generation.preview.objects.len());
|
||||||
|
for (index, image) in generation.preview.objects.iter().enumerate() {
|
||||||
|
if index + 1 > i32::MAX as usize {
|
||||||
|
return JobExecution::Failed {
|
||||||
|
error: "too many preview objects".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
let ordinal = (index + 1) as i32;
|
||||||
|
let s3_key = document_asset_object_key(
|
||||||
|
initial.document.id,
|
||||||
|
initial.version.version_number,
|
||||||
|
PREVIEW_ASSET_TYPE,
|
||||||
|
preview_asset_id,
|
||||||
|
ordinal,
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Err(err) = storage
|
||||||
|
.put_object(
|
||||||
|
&s3_key,
|
||||||
|
image.image_bytes.clone(),
|
||||||
|
Some("image/png".into()),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
warn!(job_id = %job.id, error = %err, ordinal, "failed to upload preview; retrying");
|
||||||
|
return JobExecution::Retry {
|
||||||
|
delay: Duration::from_secs(30),
|
||||||
|
error: err.to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
preview_objects.push(AssetObjectPersistence {
|
||||||
|
ordinal,
|
||||||
|
s3_key,
|
||||||
|
width: image.width,
|
||||||
|
height: image.height,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut thumbnail_objects: Vec<AssetObjectPersistence> =
|
||||||
|
Vec::with_capacity(generation.thumbnail.objects.len());
|
||||||
|
for (index, image) in generation.thumbnail.objects.iter().enumerate() {
|
||||||
|
if index + 1 > i32::MAX as usize {
|
||||||
|
return JobExecution::Failed {
|
||||||
|
error: "too many thumbnail objects".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
let ordinal = (index + 1) as i32;
|
||||||
|
let s3_key = document_asset_object_key(
|
||||||
|
initial.document.id,
|
||||||
|
initial.version.version_number,
|
||||||
|
THUMBNAIL_ASSET_TYPE,
|
||||||
|
thumbnail_asset_id,
|
||||||
|
ordinal,
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Err(err) = storage
|
||||||
|
.put_object(
|
||||||
|
&s3_key,
|
||||||
|
image.image_bytes.clone(),
|
||||||
|
Some("image/png".into()),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
warn!(job_id = %job.id, error = %err, ordinal, "failed to upload thumbnail; retrying");
|
||||||
|
return JobExecution::Retry {
|
||||||
|
delay: Duration::from_secs(30),
|
||||||
|
error: err.to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
thumbnail_objects.push(AssetObjectPersistence {
|
||||||
|
ordinal,
|
||||||
|
s3_key,
|
||||||
|
width: image.width,
|
||||||
|
height: image.height,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
let asset_persistences = vec![
|
||||||
|
AssetPersistence {
|
||||||
|
asset_type: PREVIEW_ASSET_TYPE,
|
||||||
|
asset_id: preview_asset_id,
|
||||||
|
objects: preview_objects,
|
||||||
|
},
|
||||||
|
AssetPersistence {
|
||||||
|
asset_type: THUMBNAIL_ASSET_TYPE,
|
||||||
|
asset_id: thumbnail_asset_id,
|
||||||
|
objects: thumbnail_objects,
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
let state_clone = state.clone();
|
let state_clone = state.clone();
|
||||||
match task::spawn_blocking(move || {
|
match task::spawn_blocking(move || {
|
||||||
persist_assets_metadata(
|
persist_assets_metadata(state_clone, &initial, &asset_persistences)
|
||||||
state_clone,
|
|
||||||
&initial,
|
|
||||||
&[
|
|
||||||
AssetPersistence {
|
|
||||||
asset_type: PREVIEW_ASSET_TYPE,
|
|
||||||
asset_id: preview_asset_id,
|
|
||||||
s3_key: &preview_s3_key,
|
|
||||||
generated: &generation.preview,
|
|
||||||
},
|
|
||||||
AssetPersistence {
|
|
||||||
asset_type: THUMBNAIL_ASSET_TYPE,
|
|
||||||
asset_id: thumbnail_asset_id,
|
|
||||||
s3_key: &thumbnail_s3_key,
|
|
||||||
generated: &generation.thumbnail,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
)
|
|
||||||
})
|
})
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
@@ -250,7 +308,9 @@ struct ThumbnailContext {
|
|||||||
document: Document,
|
document: Document,
|
||||||
version: DocumentVersion,
|
version: DocumentVersion,
|
||||||
existing_thumbnail: Option<DocumentAsset>,
|
existing_thumbnail: Option<DocumentAsset>,
|
||||||
|
existing_thumbnail_objects: Vec<DocumentAssetObject>,
|
||||||
existing_preview: Option<DocumentAsset>,
|
existing_preview: Option<DocumentAsset>,
|
||||||
|
existing_preview_objects: Vec<DocumentAssetObject>,
|
||||||
skip: bool,
|
skip: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -260,24 +320,34 @@ struct GeneratedImage {
|
|||||||
height: Option<i32>,
|
height: Option<i32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct GeneratedAsset {
|
||||||
|
objects: Vec<GeneratedImage>,
|
||||||
|
}
|
||||||
|
|
||||||
struct GeneratedAssets {
|
struct GeneratedAssets {
|
||||||
thumbnail: GeneratedImage,
|
thumbnail: GeneratedAsset,
|
||||||
preview: GeneratedImage,
|
preview: GeneratedAsset,
|
||||||
page_count: Option<u32>,
|
page_count: Option<u32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
struct AssetPersistence<'a> {
|
struct AssetObjectPersistence {
|
||||||
|
ordinal: i32,
|
||||||
|
s3_key: String,
|
||||||
|
width: Option<i32>,
|
||||||
|
height: Option<i32>,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct AssetPersistence {
|
||||||
asset_type: &'static str,
|
asset_type: &'static str,
|
||||||
asset_id: Uuid,
|
asset_id: Uuid,
|
||||||
s3_key: &'a str,
|
objects: Vec<AssetObjectPersistence>,
|
||||||
generated: &'a GeneratedImage,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn load_thumbnail_context(
|
fn load_thumbnail_context(
|
||||||
state: Arc<AppState>,
|
state: Arc<AppState>,
|
||||||
payload: &ThumbnailPayload,
|
payload: &ThumbnailPayload,
|
||||||
) -> Result<ThumbnailContext, String> {
|
) -> Result<ThumbnailContext, String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let mut conn = state.db_unscoped().map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let version: DocumentVersion = document_versions::table
|
let version: DocumentVersion = document_versions::table
|
||||||
.find(payload.document_version_id)
|
.find(payload.document_version_id)
|
||||||
@@ -293,21 +363,42 @@ fn load_thumbnail_context(
|
|||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
let tenant_id = document.tenant_id;
|
||||||
|
|
||||||
let existing_assets: Vec<DocumentAsset> = document_assets::table
|
let existing_assets: Vec<DocumentAsset> = document_assets::table
|
||||||
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
.filter(document_assets::document_version_id.eq(payload.document_version_id))
|
||||||
.filter(document_assets::asset_type.eq_any(vec![
|
.filter(document_assets::asset_type.eq_any(vec![
|
||||||
THUMBNAIL_ASSET_TYPE.to_string(),
|
THUMBNAIL_ASSET_TYPE.to_string(),
|
||||||
PREVIEW_ASSET_TYPE.to_string(),
|
PREVIEW_ASSET_TYPE.to_string(),
|
||||||
]))
|
]))
|
||||||
|
.filter(document_assets::tenant_id.eq(tenant_id))
|
||||||
.load(&mut conn)
|
.load(&mut conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let mut existing_thumbnail = None;
|
let mut existing_thumbnail = None;
|
||||||
|
let mut existing_thumbnail_objects: Vec<DocumentAssetObject> = Vec::new();
|
||||||
let mut existing_preview = None;
|
let mut existing_preview = None;
|
||||||
|
let mut existing_preview_objects: Vec<DocumentAssetObject> = Vec::new();
|
||||||
for asset in existing_assets {
|
for asset in existing_assets {
|
||||||
match asset.asset_type.as_str() {
|
match asset.asset_type.as_str() {
|
||||||
THUMBNAIL_ASSET_TYPE => existing_thumbnail = Some(asset),
|
THUMBNAIL_ASSET_TYPE => {
|
||||||
PREVIEW_ASSET_TYPE => existing_preview = Some(asset),
|
existing_thumbnail_objects = document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::asset_id.eq(asset.id))
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant_id))
|
||||||
|
.order(document_asset_objects::ordinal.asc())
|
||||||
|
.load(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
existing_thumbnail = Some(asset);
|
||||||
|
}
|
||||||
|
PREVIEW_ASSET_TYPE => {
|
||||||
|
existing_preview_objects = document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::asset_id.eq(asset.id))
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant_id))
|
||||||
|
.order(document_asset_objects::ordinal.asc())
|
||||||
|
.load(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
existing_preview = Some(asset);
|
||||||
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -317,13 +408,33 @@ fn load_thumbnail_context(
|
|||||||
return Err("thumbnail generation not supported for this document".into());
|
return Err("thumbnail generation not supported for this document".into());
|
||||||
}
|
}
|
||||||
|
|
||||||
let skip = existing_thumbnail.is_some() && existing_preview.is_some() && !payload.force;
|
let expected_cardinality = expected_asset_cardinality(&document, &version);
|
||||||
|
let preview_cardinality = existing_preview
|
||||||
|
.as_ref()
|
||||||
|
.and_then(|asset| asset.cardinality)
|
||||||
|
.unwrap_or_else(|| existing_preview_objects.len() as i32);
|
||||||
|
let thumbnail_cardinality = existing_thumbnail
|
||||||
|
.as_ref()
|
||||||
|
.and_then(|asset| asset.cardinality)
|
||||||
|
.unwrap_or_else(|| existing_thumbnail_objects.len() as i32);
|
||||||
|
|
||||||
|
let needs_regeneration = preview_cardinality < expected_cardinality
|
||||||
|
|| thumbnail_cardinality < expected_cardinality
|
||||||
|
|| (existing_preview_objects.len() as i32) < expected_cardinality
|
||||||
|
|| (existing_thumbnail_objects.len() as i32) < expected_cardinality;
|
||||||
|
|
||||||
|
let skip = existing_thumbnail.is_some()
|
||||||
|
&& existing_preview.is_some()
|
||||||
|
&& !payload.force
|
||||||
|
&& !needs_regeneration;
|
||||||
|
|
||||||
Ok(ThumbnailContext {
|
Ok(ThumbnailContext {
|
||||||
document,
|
document,
|
||||||
version,
|
version,
|
||||||
existing_thumbnail,
|
existing_thumbnail,
|
||||||
|
existing_thumbnail_objects,
|
||||||
existing_preview,
|
existing_preview,
|
||||||
|
existing_preview_objects,
|
||||||
skip,
|
skip,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -332,18 +443,7 @@ fn generate_preview_and_thumbnail(
|
|||||||
document: &Document,
|
document: &Document,
|
||||||
bytes: &[u8],
|
bytes: &[u8],
|
||||||
) -> Result<GeneratedAssets, String> {
|
) -> Result<GeneratedAssets, String> {
|
||||||
let is_pdf = document
|
let is_pdf = document_is_pdf(document);
|
||||||
.content_type
|
|
||||||
.as_deref()
|
|
||||||
.map(|mime| mime == "application/pdf")
|
|
||||||
.unwrap_or_else(|| {
|
|
||||||
document
|
|
||||||
.original_name
|
|
||||||
.rsplit('.')
|
|
||||||
.next()
|
|
||||||
.map(|ext| ext.eq_ignore_ascii_case("pdf"))
|
|
||||||
.unwrap_or(false)
|
|
||||||
});
|
|
||||||
|
|
||||||
if is_pdf {
|
if is_pdf {
|
||||||
let pdf_assets = generate_pdf_assets(bytes)?;
|
let pdf_assets = generate_pdf_assets(bytes)?;
|
||||||
@@ -362,7 +462,7 @@ fn generate_preview_and_thumbnail(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn generate_image_assets(bytes: &[u8]) -> Result<(GeneratedImage, GeneratedImage), String> {
|
fn generate_image_assets(bytes: &[u8]) -> Result<(GeneratedAsset, GeneratedAsset), String> {
|
||||||
let reader = ImageReader::new(Cursor::new(bytes))
|
let reader = ImageReader::new(Cursor::new(bytes))
|
||||||
.with_guessed_format()
|
.with_guessed_format()
|
||||||
.map_err(|err| err.to_string())?;
|
.map_err(|err| err.to_string())?;
|
||||||
@@ -384,12 +484,19 @@ fn generate_image_assets(bytes: &[u8]) -> Result<(GeneratedImage, GeneratedImage
|
|||||||
let preview = encode_dynamic_image(preview_image)?;
|
let preview = encode_dynamic_image(preview_image)?;
|
||||||
let thumbnail = encode_dynamic_image(thumbnail_image)?;
|
let thumbnail = encode_dynamic_image(thumbnail_image)?;
|
||||||
|
|
||||||
Ok((preview, thumbnail))
|
Ok((
|
||||||
|
GeneratedAsset {
|
||||||
|
objects: vec![preview],
|
||||||
|
},
|
||||||
|
GeneratedAsset {
|
||||||
|
objects: vec![thumbnail],
|
||||||
|
},
|
||||||
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
struct PdfGeneratedAssets {
|
struct PdfGeneratedAssets {
|
||||||
preview: GeneratedImage,
|
preview: GeneratedAsset,
|
||||||
thumbnail: GeneratedImage,
|
thumbnail: GeneratedAsset,
|
||||||
page_count: u32,
|
page_count: u32,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -402,11 +509,7 @@ fn generate_pdf_assets(bytes: &[u8]) -> Result<PdfGeneratedAssets, String> {
|
|||||||
.map_err(|err| format!("load pdf: {err}"))?;
|
.map_err(|err| format!("load pdf: {err}"))?;
|
||||||
|
|
||||||
let pages = document.pages();
|
let pages = document.pages();
|
||||||
let total_pages = pages.len();
|
let total_pages = pages.len() as usize;
|
||||||
|
|
||||||
let page = pages
|
|
||||||
.get(0)
|
|
||||||
.map_err(|err| format!("load first page: {err}"))?;
|
|
||||||
|
|
||||||
let render_config = PdfRenderConfig::new()
|
let render_config = PdfRenderConfig::new()
|
||||||
.set_target_width(PREVIEW_WIDTH as i32)
|
.set_target_width(PREVIEW_WIDTH as i32)
|
||||||
@@ -414,30 +517,44 @@ fn generate_pdf_assets(bytes: &[u8]) -> Result<PdfGeneratedAssets, String> {
|
|||||||
.render_form_data(true)
|
.render_form_data(true)
|
||||||
.rotate_if_landscape(PdfPageRenderRotation::None, true);
|
.rotate_if_landscape(PdfPageRenderRotation::None, true);
|
||||||
|
|
||||||
let bitmap = page
|
let mut preview_objects: Vec<GeneratedImage> = Vec::with_capacity(total_pages);
|
||||||
.render_with_config(&render_config)
|
let mut thumbnail_objects: Vec<GeneratedImage> = Vec::with_capacity(total_pages);
|
||||||
.map_err(|err| format!("render pdf page: {err}"))?;
|
|
||||||
|
|
||||||
let preview_buffer = bitmap.as_image().to_rgb8();
|
for page_index in 0..total_pages {
|
||||||
let preview_image = image::DynamicImage::ImageRgb8(preview_buffer);
|
let page = pages
|
||||||
|
.get(u16::try_from(page_index).map_err(|_| "page index overflow".to_string())?)
|
||||||
|
.map_err(|err| format!("load page {page_index}: {err}"))?;
|
||||||
|
|
||||||
let thumbnail_image =
|
let bitmap = page
|
||||||
if preview_image.width() > THUMBNAIL_WIDTH || preview_image.height() > THUMBNAIL_HEIGHT {
|
.render_with_config(&render_config)
|
||||||
|
.map_err(|err| format!("render pdf page {page_index}: {err}"))?;
|
||||||
|
|
||||||
|
let preview_buffer = bitmap.as_image().to_rgb8();
|
||||||
|
let preview_image = image::DynamicImage::ImageRgb8(preview_buffer);
|
||||||
|
|
||||||
|
let thumbnail_image = if preview_image.width() > THUMBNAIL_WIDTH
|
||||||
|
|| preview_image.height() > THUMBNAIL_HEIGHT
|
||||||
|
{
|
||||||
preview_image.thumbnail(THUMBNAIL_WIDTH, THUMBNAIL_HEIGHT)
|
preview_image.thumbnail(THUMBNAIL_WIDTH, THUMBNAIL_HEIGHT)
|
||||||
} else {
|
} else {
|
||||||
preview_image.clone()
|
preview_image.clone()
|
||||||
};
|
};
|
||||||
|
|
||||||
let preview = encode_dynamic_image(preview_image)?;
|
preview_objects.push(encode_dynamic_image(preview_image)?);
|
||||||
let thumbnail = encode_dynamic_image(thumbnail_image)?;
|
thumbnail_objects.push(encode_dynamic_image(thumbnail_image)?);
|
||||||
|
}
|
||||||
|
|
||||||
let page_count: u32 = total_pages
|
let page_count: u32 = total_pages
|
||||||
.try_into()
|
.try_into()
|
||||||
.map_err(|_| "page count exceeds supported range".to_string())?;
|
.map_err(|_| "page count exceeds supported range".to_string())?;
|
||||||
|
|
||||||
Ok(PdfGeneratedAssets {
|
Ok(PdfGeneratedAssets {
|
||||||
preview,
|
preview: GeneratedAsset {
|
||||||
thumbnail,
|
objects: preview_objects,
|
||||||
|
},
|
||||||
|
thumbnail: GeneratedAsset {
|
||||||
|
objects: thumbnail_objects,
|
||||||
|
},
|
||||||
page_count,
|
page_count,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -458,22 +575,51 @@ fn encode_dynamic_image(image: image::DynamicImage) -> Result<GeneratedImage, St
|
|||||||
fn persist_assets_metadata(
|
fn persist_assets_metadata(
|
||||||
state: Arc<AppState>,
|
state: Arc<AppState>,
|
||||||
context: &ThumbnailContext,
|
context: &ThumbnailContext,
|
||||||
assets: &[AssetPersistence<'_>],
|
assets: &[AssetPersistence],
|
||||||
) -> Result<(), String> {
|
) -> Result<(), String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let tenant_id = context.document.tenant_id;
|
||||||
|
let mut conn = state
|
||||||
|
.db_for_tenant(tenant_id)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
if let Some(existing_preview) = &context.existing_preview {
|
||||||
|
diesel::delete(document_assets::table.filter(document_assets::id.eq(existing_preview.id)))
|
||||||
|
.execute(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(existing_thumbnail) = &context.existing_thumbnail {
|
||||||
|
diesel::delete(
|
||||||
|
document_assets::table.filter(document_assets::id.eq(existing_thumbnail.id)),
|
||||||
|
)
|
||||||
|
.execute(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
}
|
||||||
|
|
||||||
for asset in assets {
|
for asset in assets {
|
||||||
|
if asset.objects.is_empty() {
|
||||||
|
return Err(format!(
|
||||||
|
"asset {} has no generated objects",
|
||||||
|
asset.asset_type
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let object_count: i32 = asset
|
||||||
|
.objects
|
||||||
|
.len()
|
||||||
|
.try_into()
|
||||||
|
.map_err(|_| "asset contains too many objects".to_string())?;
|
||||||
|
|
||||||
let new_asset = NewDocumentAsset {
|
let new_asset = NewDocumentAsset {
|
||||||
id: asset.asset_id,
|
id: asset.asset_id,
|
||||||
document_version_id: context.version.id,
|
document_version_id: context.version.id,
|
||||||
asset_type: asset.asset_type.to_string(),
|
asset_type: asset.asset_type.to_string(),
|
||||||
s3_key: asset.s3_key.to_string(),
|
|
||||||
mime_type: "image/png".to_string(),
|
mime_type: "image/png".to_string(),
|
||||||
metadata: json!({
|
metadata: json!({
|
||||||
"generated_at": Utc::now().to_rfc3339(),
|
"generated_at": Utc::now().to_rfc3339(),
|
||||||
"width": asset.generated.width,
|
|
||||||
"height": asset.generated.height,
|
|
||||||
}),
|
}),
|
||||||
|
cardinality: Some(object_count),
|
||||||
|
tenant_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
diesel::insert_into(document_assets::table)
|
diesel::insert_into(document_assets::table)
|
||||||
@@ -484,12 +630,46 @@ fn persist_assets_metadata(
|
|||||||
))
|
))
|
||||||
.do_update()
|
.do_update()
|
||||||
.set((
|
.set((
|
||||||
document_assets::s3_key.eq(excluded(document_assets::s3_key)),
|
|
||||||
document_assets::mime_type.eq(excluded(document_assets::mime_type)),
|
document_assets::mime_type.eq(excluded(document_assets::mime_type)),
|
||||||
document_assets::metadata.eq(excluded(document_assets::metadata)),
|
document_assets::metadata.eq(excluded(document_assets::metadata)),
|
||||||
|
document_assets::cardinality.eq(excluded(document_assets::cardinality)),
|
||||||
))
|
))
|
||||||
.execute(&mut conn)
|
.execute(&mut conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
diesel::delete(
|
||||||
|
document_asset_objects::table
|
||||||
|
.filter(document_asset_objects::asset_id.eq(asset.asset_id))
|
||||||
|
.filter(document_asset_objects::tenant_id.eq(tenant_id)),
|
||||||
|
)
|
||||||
|
.execute(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
|
for object in &asset.objects {
|
||||||
|
let mut metadata_map = Map::new();
|
||||||
|
if let Some(width) = object.width {
|
||||||
|
metadata_map.insert("width".to_string(), Value::from(width));
|
||||||
|
}
|
||||||
|
if let Some(height) = object.height {
|
||||||
|
metadata_map.insert("height".to_string(), Value::from(height));
|
||||||
|
}
|
||||||
|
|
||||||
|
let object_metadata = Value::Object(metadata_map);
|
||||||
|
|
||||||
|
let new_object = NewDocumentAssetObject {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
asset_id: asset.asset_id,
|
||||||
|
ordinal: object.ordinal,
|
||||||
|
s3_key: object.s3_key.clone(),
|
||||||
|
metadata: object_metadata,
|
||||||
|
tenant_id,
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(document_asset_objects::table)
|
||||||
|
.values(&new_object)
|
||||||
|
.execute(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -501,11 +681,17 @@ fn persist_document_page_count(
|
|||||||
document_version_id: Uuid,
|
document_version_id: Uuid,
|
||||||
page_count: u32,
|
page_count: u32,
|
||||||
) -> Result<(), String> {
|
) -> Result<(), String> {
|
||||||
let mut conn = state.db().map_err(|err| format!("{err:?}"))?;
|
let mut conn = state.db_unscoped().map_err(|err| format!("{err:?}"))?;
|
||||||
|
let tenant_id: Uuid = documents::table
|
||||||
|
.find(document_id)
|
||||||
|
.select(documents::tenant_id)
|
||||||
|
.first(&mut conn)
|
||||||
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
|
|
||||||
let existing_metadata: Value = document_versions::table
|
let existing_metadata: Value = document_versions::table
|
||||||
.filter(document_versions::id.eq(document_version_id))
|
.filter(document_versions::id.eq(document_version_id))
|
||||||
.filter(document_versions::document_id.eq(document_id))
|
.filter(document_versions::document_id.eq(document_id))
|
||||||
|
.filter(document_versions::tenant_id.eq(tenant_id))
|
||||||
.select(document_versions::metadata)
|
.select(document_versions::metadata)
|
||||||
.first(&mut conn)
|
.first(&mut conn)
|
||||||
.map_err(|err| format!("{err:?}"))?;
|
.map_err(|err| format!("{err:?}"))?;
|
||||||
@@ -525,7 +711,8 @@ fn persist_document_page_count(
|
|||||||
diesel::update(
|
diesel::update(
|
||||||
document_versions::table
|
document_versions::table
|
||||||
.filter(document_versions::id.eq(document_version_id))
|
.filter(document_versions::id.eq(document_version_id))
|
||||||
.filter(document_versions::document_id.eq(document_id)),
|
.filter(document_versions::document_id.eq(document_id))
|
||||||
|
.filter(document_versions::tenant_id.eq(tenant_id)),
|
||||||
)
|
)
|
||||||
.set(document_versions::metadata.eq(updated))
|
.set(document_versions::metadata.eq(updated))
|
||||||
.execute(&mut conn)
|
.execute(&mut conn)
|
||||||
@@ -533,3 +720,37 @@ fn persist_document_page_count(
|
|||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn document_is_pdf(document: &Document) -> bool {
|
||||||
|
document
|
||||||
|
.content_type
|
||||||
|
.as_deref()
|
||||||
|
.map(|mime| mime.eq_ignore_ascii_case("application/pdf"))
|
||||||
|
.unwrap_or_else(|| {
|
||||||
|
document
|
||||||
|
.original_name
|
||||||
|
.rsplit('.')
|
||||||
|
.next()
|
||||||
|
.map(|ext| ext.eq_ignore_ascii_case("pdf"))
|
||||||
|
.unwrap_or(false)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn expected_asset_cardinality(document: &Document, version: &DocumentVersion) -> i32 {
|
||||||
|
if let Value::Object(map) = &version.metadata {
|
||||||
|
if let Some(count) = map.get("page_count").and_then(|v| v.as_i64()) {
|
||||||
|
if count > 0 {
|
||||||
|
return count
|
||||||
|
.min(i64::from(i32::MAX))
|
||||||
|
.try_into()
|
||||||
|
.unwrap_or(i32::MAX);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if document_is_pdf(document) {
|
||||||
|
1
|
||||||
|
} else {
|
||||||
|
1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -8,7 +8,6 @@ use serde::Deserialize;
|
|||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
struct AuthenticatedUser {
|
struct AuthenticatedUser {
|
||||||
username: String,
|
username: String,
|
||||||
role: String,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
@@ -27,7 +26,6 @@ async fn login_and_me_roundtrip() -> Result<()> {
|
|||||||
let user: AuthenticatedUser = serde_json::from_slice(&body)?;
|
let user: AuthenticatedUser = serde_json::from_slice(&body)?;
|
||||||
|
|
||||||
assert_eq!(user.username, "alice");
|
assert_eq!(user.username, "alice");
|
||||||
assert_eq!(user.role, "admin");
|
|
||||||
|
|
||||||
app.cleanup().await?;
|
app.cleanup().await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
+283
-12
@@ -11,18 +11,20 @@ use axum::Router;
|
|||||||
use backend::auth::jwt::JwtService;
|
use backend::auth::jwt::JwtService;
|
||||||
use backend::config::AppConfig;
|
use backend::config::AppConfig;
|
||||||
use backend::db::{self, PgPool};
|
use backend::db::{self, PgPool};
|
||||||
use backend::models::{Job, NewUser};
|
use backend::models::{Job, NewUser, NewUserMembership, Tenant};
|
||||||
use backend::routes;
|
use backend::routes;
|
||||||
use backend::state::AppState;
|
use backend::state::AppState;
|
||||||
use backend::storage::ObjectStorage;
|
use backend::storage::ObjectStorage;
|
||||||
use diesel::connection::SimpleConnection;
|
use diesel::connection::SimpleConnection;
|
||||||
use diesel::prelude::*;
|
use diesel::prelude::*;
|
||||||
|
use diesel::OptionalExtension;
|
||||||
use diesel::PgConnection;
|
use diesel::PgConnection;
|
||||||
use diesel_migrations::{embed_migrations, EmbeddedMigrations, MigrationHarness};
|
use diesel_migrations::{embed_migrations, EmbeddedMigrations, MigrationHarness};
|
||||||
use http_body_util::BodyExt;
|
use http_body_util::BodyExt;
|
||||||
use once_cell::sync::Lazy;
|
use once_cell::sync::Lazy;
|
||||||
use rand::rngs::OsRng;
|
use rand::rngs::OsRng;
|
||||||
use serde::Serialize;
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serde_json;
|
||||||
use tokio::sync::Mutex;
|
use tokio::sync::Mutex;
|
||||||
use tower::util::ServiceExt;
|
use tower::util::ServiceExt;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
@@ -138,6 +140,7 @@ impl TestApp {
|
|||||||
s3_bucket: "test-bucket".to_string(),
|
s3_bucket: "test-bucket".to_string(),
|
||||||
quickwit_endpoint: None,
|
quickwit_endpoint: None,
|
||||||
quickwit_index: None,
|
quickwit_index: None,
|
||||||
|
default_tenant_slug: "admin".to_string(),
|
||||||
};
|
};
|
||||||
|
|
||||||
let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
|
let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
|
||||||
@@ -149,16 +152,20 @@ impl TestApp {
|
|||||||
let state = AppState::new(pool.clone(), config, storage_for_state, jwt);
|
let state = AppState::new(pool.clone(), config, storage_for_state, jwt);
|
||||||
let router = routes::create_router(state.clone());
|
let router = routes::create_router(state.clone());
|
||||||
|
|
||||||
Ok(Self {
|
let app = Self {
|
||||||
state,
|
state,
|
||||||
router,
|
router,
|
||||||
storage,
|
storage,
|
||||||
})
|
};
|
||||||
|
|
||||||
|
app.ensure_default_tenant().await?;
|
||||||
|
|
||||||
|
Ok(app)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn cleanup(&self) -> Result<()> {
|
pub async fn cleanup(&self) -> Result<()> {
|
||||||
let pool = self.state.pool.clone();
|
let pool = self.state.pool.clone();
|
||||||
tokio::task::spawn_blocking(move || -> Result<()> {
|
let _ = tokio::task::spawn_blocking(move || -> Result<()> {
|
||||||
let mut conn = pool
|
let mut conn = pool
|
||||||
.get()
|
.get()
|
||||||
.map_err(|err| anyhow!("failed to get cleanup connection: {err}"))?;
|
.map_err(|err| anyhow!("failed to get cleanup connection: {err}"))?;
|
||||||
@@ -166,7 +173,10 @@ impl TestApp {
|
|||||||
Ok(())
|
Ok(())
|
||||||
})
|
})
|
||||||
.await
|
.await
|
||||||
.context("cleanup task panicked")?
|
.context("cleanup task panicked")?;
|
||||||
|
|
||||||
|
self.ensure_default_tenant().await?;
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
@@ -174,27 +184,111 @@ impl TestApp {
|
|||||||
self.storage.clone()
|
self.storage.clone()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub async fn storage_key_for(&self, key: &str) -> Result<String> {
|
||||||
|
let tenant = self
|
||||||
|
.state
|
||||||
|
.tenants
|
||||||
|
.get_by_slug(&self.state.config.default_tenant_slug)
|
||||||
|
.map_err(|err| anyhow!("default tenant not found: {:?}", err))?;
|
||||||
|
let root = tenant
|
||||||
|
.storage_root
|
||||||
|
.clone()
|
||||||
|
.ok_or_else(|| anyhow!("default tenant missing storage root"))?;
|
||||||
|
Ok(format!("{}{}", root, key))
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn insert_user(&self, username: &str, password: &str, role: &str) -> Result<Uuid> {
|
pub async fn insert_user(&self, username: &str, password: &str, role: &str) -> Result<Uuid> {
|
||||||
let username = username.to_string();
|
let username = username.to_string();
|
||||||
let password = password.to_string();
|
let password = password.to_string();
|
||||||
let role = role.to_string();
|
let role = role.to_string();
|
||||||
|
let tenant_id = self
|
||||||
|
.state
|
||||||
|
.tenants
|
||||||
|
.tenant_id_for_slug(&self.state.config.default_tenant_slug)
|
||||||
|
.map_err(|err| anyhow!("default tenant not found: {:?}", err))?;
|
||||||
self.with_conn(move |conn| {
|
self.with_conn(move |conn| {
|
||||||
let password_hash = hash_password(&password)?;
|
let password_hash = hash_password(&password)?;
|
||||||
let user = NewUser {
|
let user = NewUser {
|
||||||
id: Uuid::new_v4(),
|
id: Uuid::new_v4(),
|
||||||
username,
|
username,
|
||||||
password_hash,
|
password_hash,
|
||||||
role,
|
|
||||||
};
|
};
|
||||||
diesel::insert_into(backend::schema::users::table)
|
diesel::insert_into(backend::schema::users::table)
|
||||||
.values(&user)
|
.values(&user)
|
||||||
.execute(conn)
|
.execute(conn)
|
||||||
.context("failed to insert user")?;
|
.context("failed to insert user")?;
|
||||||
|
|
||||||
|
let membership = NewUserMembership {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
user_id: user.id,
|
||||||
|
tenant_id,
|
||||||
|
role,
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(backend::schema::user_memberships::table)
|
||||||
|
.values(&membership)
|
||||||
|
.execute(conn)
|
||||||
|
.context("failed to insert user membership")?;
|
||||||
Ok(user.id)
|
Ok(user.id)
|
||||||
})
|
})
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn ensure_default_tenant(&self) -> Result<Uuid> {
|
||||||
|
let slug_value = self.state.config.default_tenant_slug.clone();
|
||||||
|
let quickwit_enabled = self.state.config.quickwit_endpoint.is_some();
|
||||||
|
self.with_conn(move |conn| {
|
||||||
|
use backend::schema::tenants::dsl as tenants_dsl;
|
||||||
|
|
||||||
|
let existing = tenants_dsl::tenants
|
||||||
|
.filter(tenants_dsl::slug.eq(&slug_value))
|
||||||
|
.first::<Tenant>(conn)
|
||||||
|
.optional()
|
||||||
|
.context("failed to load default tenant")?;
|
||||||
|
|
||||||
|
let tenant_id = if let Some(current) = existing {
|
||||||
|
let desired_root = current
|
||||||
|
.storage_root
|
||||||
|
.clone()
|
||||||
|
.filter(|root| root.ends_with('/'))
|
||||||
|
.unwrap_or_else(|| format!("test-tenants/{}/", current.id));
|
||||||
|
|
||||||
|
if current.storage_root.as_deref() != Some(desired_root.as_str()) {
|
||||||
|
diesel::update(tenants_dsl::tenants.filter(tenants_dsl::id.eq(current.id)))
|
||||||
|
.set(tenants_dsl::storage_root.eq(Some(desired_root)))
|
||||||
|
.execute(conn)
|
||||||
|
.context("failed to update default tenant storage root")?;
|
||||||
|
}
|
||||||
|
|
||||||
|
current.id
|
||||||
|
} else {
|
||||||
|
let new_id = Uuid::new_v4();
|
||||||
|
let root = format!("test-tenants/{}/", new_id);
|
||||||
|
let quickwit_value = if quickwit_enabled {
|
||||||
|
Some(format!("documents-{}", new_id))
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
diesel::insert_into(tenants_dsl::tenants)
|
||||||
|
.values((
|
||||||
|
tenants_dsl::id.eq(new_id),
|
||||||
|
tenants_dsl::slug.eq(&slug_value),
|
||||||
|
tenants_dsl::storage_root.eq(Some(root)),
|
||||||
|
tenants_dsl::quickwit_index.eq(quickwit_value),
|
||||||
|
))
|
||||||
|
.execute(conn)
|
||||||
|
.context("failed to insert default tenant")?;
|
||||||
|
|
||||||
|
new_id
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(tenant_id)
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn login_token(&self, username: &str, password: &str) -> Result<String> {
|
pub async fn login_token(&self, username: &str, password: &str) -> Result<String> {
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
struct LoginPayload<'a> {
|
struct LoginPayload<'a> {
|
||||||
@@ -217,11 +311,57 @@ impl TestApp {
|
|||||||
);
|
);
|
||||||
|
|
||||||
let body = body_to_vec(response.into_body()).await?;
|
let body = body_to_vec(response.into_body()).await?;
|
||||||
#[derive(serde::Deserialize)]
|
#[derive(Deserialize)]
|
||||||
struct LoginResponse {
|
struct LoginResponse {
|
||||||
access_token: String,
|
access_token: String,
|
||||||
}
|
}
|
||||||
let parsed: LoginResponse = serde_json::from_slice(&body)?;
|
|
||||||
|
if let Ok(parsed) = serde_json::from_slice::<LoginResponse>(&body) {
|
||||||
|
return Ok(parsed.access_token);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize)]
|
||||||
|
struct TenantSummary {
|
||||||
|
tenant_id: Uuid,
|
||||||
|
_slug: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize)]
|
||||||
|
struct TenantSelectionResponse {
|
||||||
|
selection_token: String,
|
||||||
|
tenants: Vec<TenantSummary>,
|
||||||
|
}
|
||||||
|
|
||||||
|
let selection: TenantSelectionResponse = serde_json::from_slice(&body)?;
|
||||||
|
ensure!(
|
||||||
|
!selection.tenants.is_empty(),
|
||||||
|
"login returned no tenant options",
|
||||||
|
);
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
struct SelectTenantPayload {
|
||||||
|
tenant_id: Uuid,
|
||||||
|
}
|
||||||
|
|
||||||
|
let target_tenant = selection.tenants[0].tenant_id;
|
||||||
|
let select_response = self
|
||||||
|
.post_json(
|
||||||
|
"/api/auth/select-tenant",
|
||||||
|
&SelectTenantPayload {
|
||||||
|
tenant_id: target_tenant,
|
||||||
|
},
|
||||||
|
Some(&selection.selection_token),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
ensure!(
|
||||||
|
select_response.status() == StatusCode::OK,
|
||||||
|
"tenant selection failed with status {}",
|
||||||
|
select_response.status()
|
||||||
|
);
|
||||||
|
|
||||||
|
let select_body = body_to_vec(select_response.into_body()).await?;
|
||||||
|
let parsed: LoginResponse = serde_json::from_slice(&select_body)?;
|
||||||
Ok(parsed.access_token)
|
Ok(parsed.access_token)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -338,6 +478,60 @@ impl TestApp {
|
|||||||
data: &[u8],
|
data: &[u8],
|
||||||
folder_id: Option<Uuid>,
|
folder_id: Option<Uuid>,
|
||||||
token: &str,
|
token: &str,
|
||||||
|
) -> Result<hyper::Response<Body>> {
|
||||||
|
let extras = UploadExtras::empty();
|
||||||
|
self.upload_document_with_extras(
|
||||||
|
path,
|
||||||
|
filename,
|
||||||
|
content_type,
|
||||||
|
data,
|
||||||
|
folder_id,
|
||||||
|
extras,
|
||||||
|
token,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn upload_document_with_options(
|
||||||
|
&self,
|
||||||
|
path: &str,
|
||||||
|
filename: &str,
|
||||||
|
content_type: &str,
|
||||||
|
data: &[u8],
|
||||||
|
folder_id: Option<Uuid>,
|
||||||
|
title: Option<&str>,
|
||||||
|
metadata_json: Option<&str>,
|
||||||
|
token: &str,
|
||||||
|
) -> Result<hyper::Response<Body>> {
|
||||||
|
let extras = UploadExtras {
|
||||||
|
title,
|
||||||
|
metadata_json,
|
||||||
|
tag_ids_json: None,
|
||||||
|
correspondents_json: None,
|
||||||
|
issued_at: None,
|
||||||
|
skip_existing: false,
|
||||||
|
};
|
||||||
|
self.upload_document_with_extras(
|
||||||
|
path,
|
||||||
|
filename,
|
||||||
|
content_type,
|
||||||
|
data,
|
||||||
|
folder_id,
|
||||||
|
extras,
|
||||||
|
token,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn upload_document_with_extras(
|
||||||
|
&self,
|
||||||
|
path: &str,
|
||||||
|
filename: &str,
|
||||||
|
content_type: &str,
|
||||||
|
data: &[u8],
|
||||||
|
folder_id: Option<Uuid>,
|
||||||
|
extras: UploadExtras<'_>,
|
||||||
|
token: &str,
|
||||||
) -> Result<hyper::Response<Body>> {
|
) -> Result<hyper::Response<Body>> {
|
||||||
let boundary = format!("boundary-{}", Uuid::new_v4());
|
let boundary = format!("boundary-{}", Uuid::new_v4());
|
||||||
let mut body = Vec::new();
|
let mut body = Vec::new();
|
||||||
@@ -360,6 +554,46 @@ impl TestApp {
|
|||||||
body.extend(b"\r\n");
|
body.extend(b"\r\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let Some(title_value) = extras.title {
|
||||||
|
body.extend(format!("--{boundary}\r\n").as_bytes());
|
||||||
|
body.extend(b"Content-Disposition: form-data; name=\"title\"\r\n\r\n");
|
||||||
|
body.extend(title_value.as_bytes());
|
||||||
|
body.extend(b"\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(metadata_value) = extras.metadata_json {
|
||||||
|
body.extend(format!("--{boundary}\r\n").as_bytes());
|
||||||
|
body.extend(b"Content-Disposition: form-data; name=\"metadata\"\r\n\r\n");
|
||||||
|
body.extend(metadata_value.as_bytes());
|
||||||
|
body.extend(b"\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(tag_ids_value) = extras.tag_ids_json {
|
||||||
|
body.extend(format!("--{boundary}\r\n").as_bytes());
|
||||||
|
body.extend(b"Content-Disposition: form-data; name=\"tag_ids\"\r\n\r\n");
|
||||||
|
body.extend(tag_ids_value.as_bytes());
|
||||||
|
body.extend(b"\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(correspondents_value) = extras.correspondents_json {
|
||||||
|
body.extend(format!("--{boundary}\r\n").as_bytes());
|
||||||
|
body.extend(b"Content-Disposition: form-data; name=\"correspondents\"\r\n\r\n");
|
||||||
|
body.extend(correspondents_value.as_bytes());
|
||||||
|
body.extend(b"\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(issued_at_value) = extras.issued_at {
|
||||||
|
body.extend(format!("--{boundary}\r\n").as_bytes());
|
||||||
|
body.extend(b"Content-Disposition: form-data; name=\"issued_at\"\r\n\r\n");
|
||||||
|
body.extend(issued_at_value.as_bytes());
|
||||||
|
body.extend(b"\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if extras.skip_existing {
|
||||||
|
body.extend(format!("--{boundary}\r\n").as_bytes());
|
||||||
|
body.extend(b"Content-Disposition: form-data; name=\"skip_existing\"\r\n\r\ntrue\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
body.extend(format!("--{boundary}--\r\n").as_bytes());
|
body.extend(format!("--{boundary}--\r\n").as_bytes());
|
||||||
|
|
||||||
let builder = Request::builder()
|
let builder = Request::builder()
|
||||||
@@ -380,7 +614,7 @@ impl TestApp {
|
|||||||
.expect("infallible response"))
|
.expect("infallible response"))
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn with_conn<F, T>(&self, f: F) -> Result<T>
|
pub async fn with_conn<F, T>(&self, f: F) -> Result<T>
|
||||||
where
|
where
|
||||||
F: FnOnce(&mut PgConnection) -> Result<T> + Send + 'static,
|
F: FnOnce(&mut PgConnection) -> Result<T> + Send + 'static,
|
||||||
T: Send + 'static,
|
T: Send + 'static,
|
||||||
@@ -397,6 +631,28 @@ impl TestApp {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct UploadExtras<'a> {
|
||||||
|
pub title: Option<&'a str>,
|
||||||
|
pub metadata_json: Option<&'a str>,
|
||||||
|
pub tag_ids_json: Option<&'a str>,
|
||||||
|
pub correspondents_json: Option<&'a str>,
|
||||||
|
pub issued_at: Option<&'a str>,
|
||||||
|
pub skip_existing: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> UploadExtras<'a> {
|
||||||
|
pub fn empty() -> Self {
|
||||||
|
Self {
|
||||||
|
title: None,
|
||||||
|
metadata_json: None,
|
||||||
|
tag_ids_json: None,
|
||||||
|
correspondents_json: None,
|
||||||
|
issued_at: None,
|
||||||
|
skip_existing: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn acquire_db_lock() -> tokio::sync::MutexGuard<'static, ()> {
|
pub async fn acquire_db_lock() -> tokio::sync::MutexGuard<'static, ()> {
|
||||||
DB_LOCK.lock().await
|
DB_LOCK.lock().await
|
||||||
}
|
}
|
||||||
@@ -426,13 +682,28 @@ async fn prepare_database(pool: &PgPool) -> Result<()> {
|
|||||||
|
|
||||||
fn truncate_all(conn: &mut PgConnection) -> Result<()> {
|
fn truncate_all(conn: &mut PgConnection) -> Result<()> {
|
||||||
conn.batch_execute(
|
conn.batch_execute(
|
||||||
"TRUNCATE TABLE document_tags, document_versions, documents, folders, tags, users RESTART IDENTITY CASCADE;",
|
"TRUNCATE TABLE \
|
||||||
|
document_asset_objects, \
|
||||||
|
document_assets, \
|
||||||
|
document_correspondents, \
|
||||||
|
correspondents, \
|
||||||
|
document_tags, \
|
||||||
|
document_versions, \
|
||||||
|
documents, \
|
||||||
|
folders, \
|
||||||
|
jobs, \
|
||||||
|
refresh_tokens, \
|
||||||
|
tags, \
|
||||||
|
user_memberships, \
|
||||||
|
users, \
|
||||||
|
tenants \
|
||||||
|
RESTART IDENTITY CASCADE;",
|
||||||
)
|
)
|
||||||
.context("failed to truncate tables")?;
|
.context("failed to truncate tables")?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn hash_password(password: &str) -> Result<String> {
|
pub fn hash_password(password: &str) -> Result<String> {
|
||||||
use argon2::password_hash::{PasswordHasher, SaltString};
|
use argon2::password_hash::{PasswordHasher, SaltString};
|
||||||
use argon2::Argon2;
|
use argon2::Argon2;
|
||||||
|
|
||||||
|
|||||||
+349
-93
@@ -2,7 +2,7 @@ mod common;
|
|||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use axum::http::StatusCode;
|
use axum::http::StatusCode;
|
||||||
use common::{acquire_db_lock, body_to_vec, TestApp};
|
use common::{acquire_db_lock, body_to_vec, TestApp, UploadExtras};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
@@ -15,6 +15,7 @@ struct DocumentDetail {
|
|||||||
struct DocumentInfo {
|
struct DocumentInfo {
|
||||||
id: Uuid,
|
id: Uuid,
|
||||||
title: String,
|
title: String,
|
||||||
|
filename: String,
|
||||||
original_name: String,
|
original_name: String,
|
||||||
deleted_at: Option<String>,
|
deleted_at: Option<String>,
|
||||||
issued_at: Option<String>,
|
issued_at: Option<String>,
|
||||||
@@ -167,7 +168,14 @@ async fn upload_and_list_document() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(upload.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = upload.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let body = body_to_vec(upload.into_body()).await?;
|
let body = body_to_vec(upload.into_body()).await?;
|
||||||
let detail: DocumentDetail = serde_json::from_slice(&body)?;
|
let detail: DocumentDetail = serde_json::from_slice(&body)?;
|
||||||
|
|
||||||
@@ -186,16 +194,24 @@ async fn upload_and_list_document() -> Result<()> {
|
|||||||
assert_eq!(current_version.size_bytes, file_bytes.len() as i64);
|
assert_eq!(current_version.size_bytes, file_bytes.len() as i64);
|
||||||
assert!(current_version.assets.is_empty());
|
assert!(current_version.assets.is_empty());
|
||||||
|
|
||||||
|
let storage_key = app.storage_key_for(¤t_version.s3_key).await?;
|
||||||
let stored = app
|
let stored = app
|
||||||
.storage()
|
.storage()
|
||||||
.get(¤t_version.s3_key)
|
.get(&storage_key)
|
||||||
.await
|
.await
|
||||||
.expect("object stored");
|
.expect("object stored");
|
||||||
assert_eq!(stored.bytes, file_bytes);
|
assert_eq!(stored.bytes, file_bytes);
|
||||||
assert_eq!(app.storage().object_count().await, 1);
|
assert_eq!(app.storage().object_count().await, 1);
|
||||||
|
|
||||||
let response = app.get("/api/documents", Some(&token)).await?;
|
let response = app.get("/api/documents", Some(&token)).await?;
|
||||||
assert_eq!(response.status(), StatusCode::OK);
|
{
|
||||||
|
let status = response.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let body = body_to_vec(response.into_body()).await?;
|
let body = body_to_vec(response.into_body()).await?;
|
||||||
let mut list: Vec<DocumentListItem> = serde_json::from_slice(&body)?;
|
let mut list: Vec<DocumentListItem> = serde_json::from_slice(&body)?;
|
||||||
assert_eq!(list.len(), 1);
|
assert_eq!(list.len(), 1);
|
||||||
@@ -220,7 +236,14 @@ async fn upload_and_list_document() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(download.status(), StatusCode::OK);
|
{
|
||||||
|
let status = download.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let body = body_to_vec(download.into_body()).await?;
|
let body = body_to_vec(download.into_body()).await?;
|
||||||
let download_info: DocumentDownload = serde_json::from_slice(&body)?;
|
let download_info: DocumentDownload = serde_json::from_slice(&body)?;
|
||||||
assert!(download_info.url.contains(¤t_version.s3_key));
|
assert!(download_info.url.contains(¤t_version.s3_key));
|
||||||
@@ -239,6 +262,50 @@ async fn upload_and_list_document() -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn upload_document_with_custom_title_sets_filename() -> Result<()> {
|
||||||
|
let _lock = acquire_db_lock().await;
|
||||||
|
let app = TestApp::new().await?;
|
||||||
|
|
||||||
|
let password = "passw0rd";
|
||||||
|
app.insert_user("nora", password, "admin").await?;
|
||||||
|
let token = app.login_token("nora", password).await?;
|
||||||
|
|
||||||
|
let file_bytes = b"example contract body".to_vec();
|
||||||
|
let title = "Vendor Contract";
|
||||||
|
let original_filename = "scan.pdf";
|
||||||
|
|
||||||
|
let upload = app
|
||||||
|
.upload_document_with_options(
|
||||||
|
"/api/documents",
|
||||||
|
original_filename,
|
||||||
|
"application/pdf",
|
||||||
|
&file_bytes,
|
||||||
|
None,
|
||||||
|
Some(title),
|
||||||
|
None,
|
||||||
|
&token,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
{
|
||||||
|
let status = upload.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let body = body_to_vec(upload.into_body()).await?;
|
||||||
|
let detail: DocumentDetail = serde_json::from_slice(&body)?;
|
||||||
|
|
||||||
|
assert_eq!(detail.document.title, title);
|
||||||
|
assert_eq!(detail.document.filename, format!("{title}.pdf"));
|
||||||
|
assert_eq!(detail.document.original_name, original_filename);
|
||||||
|
|
||||||
|
app.cleanup().await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn duplicate_and_restore_document() -> Result<()> {
|
async fn duplicate_and_restore_document() -> Result<()> {
|
||||||
let _lock = acquire_db_lock().await;
|
let _lock = acquire_db_lock().await;
|
||||||
@@ -259,7 +326,14 @@ async fn duplicate_and_restore_document() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(first.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = first.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let first_body = body_to_vec(first.into_body()).await?;
|
let first_body = body_to_vec(first.into_body()).await?;
|
||||||
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
@@ -273,7 +347,14 @@ async fn duplicate_and_restore_document() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(second.status(), StatusCode::OK);
|
{
|
||||||
|
let status = second.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let second_body = body_to_vec(second.into_body()).await?;
|
let second_body = body_to_vec(second.into_body()).await?;
|
||||||
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
||||||
|
|
||||||
@@ -306,7 +387,14 @@ async fn duplicate_and_restore_document() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(third.status(), StatusCode::OK);
|
{
|
||||||
|
let status = third.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let third_body = body_to_vec(third.into_body()).await?;
|
let third_body = body_to_vec(third.into_body()).await?;
|
||||||
let third_detail: DocumentDetail = serde_json::from_slice(&third_body)?;
|
let third_detail: DocumentDetail = serde_json::from_slice(&third_body)?;
|
||||||
|
|
||||||
@@ -319,92 +407,127 @@ async fn duplicate_and_restore_document() -> Result<()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn bulk_reanalyze_documents() -> Result<()> {
|
async fn upload_skips_existing_when_requested() -> Result<()> {
|
||||||
let _lock = acquire_db_lock().await;
|
let _lock = acquire_db_lock().await;
|
||||||
let app = TestApp::new().await?;
|
let app = TestApp::new().await?;
|
||||||
|
|
||||||
let password = "bulkpass";
|
let password = "skip-doc";
|
||||||
app.insert_user("alex", password, "admin").await?;
|
app.insert_user("skip", password, "admin").await?;
|
||||||
let token = app.login_token("alex", password).await?;
|
let token = app.login_token("skip", password).await?;
|
||||||
|
|
||||||
app.clear_jobs().await?;
|
let primary_tag_payload = CreateTagPayload {
|
||||||
|
label: "primary",
|
||||||
|
color: None,
|
||||||
|
};
|
||||||
|
let primary_tag_resp = app
|
||||||
|
.post_json("/api/tags", &primary_tag_payload, Some(&token))
|
||||||
|
.await?;
|
||||||
|
{
|
||||||
|
let status = primary_tag_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let primary_tag_body = body_to_vec(primary_tag_resp.into_body()).await?;
|
||||||
|
let primary_tag: TagResponse = serde_json::from_slice(&primary_tag_body)?;
|
||||||
|
|
||||||
let first_bytes = b"first doc";
|
let payload = b"identical document payload";
|
||||||
let first = app
|
let primary_tag_ids = format!("[\"{}\"]", primary_tag.id);
|
||||||
.upload_document(
|
let extras = UploadExtras {
|
||||||
|
title: Some("Original"),
|
||||||
|
metadata_json: None,
|
||||||
|
tag_ids_json: Some(primary_tag_ids.as_str()),
|
||||||
|
correspondents_json: None,
|
||||||
|
issued_at: None,
|
||||||
|
skip_existing: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let first_upload = app
|
||||||
|
.upload_document_with_extras(
|
||||||
"/api/documents",
|
"/api/documents",
|
||||||
"first.txt",
|
"original.pdf",
|
||||||
"text/plain",
|
"application/pdf",
|
||||||
first_bytes,
|
payload,
|
||||||
None,
|
None,
|
||||||
|
extras,
|
||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(first.status(), StatusCode::CREATED);
|
{
|
||||||
let first_body = body_to_vec(first.into_body()).await?;
|
let status = first_upload.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let first_body = body_to_vec(first_upload.into_body()).await?;
|
||||||
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
let second_bytes = b"second doc";
|
let alt_tag_payload = CreateTagPayload {
|
||||||
let second = app
|
label: "alternate",
|
||||||
.upload_document(
|
color: None,
|
||||||
|
};
|
||||||
|
let alt_tag_resp = app
|
||||||
|
.post_json("/api/tags", &alt_tag_payload, Some(&token))
|
||||||
|
.await?;
|
||||||
|
{
|
||||||
|
let status = alt_tag_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let alt_tag_body = body_to_vec(alt_tag_resp.into_body()).await?;
|
||||||
|
let alt_tag: TagResponse = serde_json::from_slice(&alt_tag_body)?;
|
||||||
|
|
||||||
|
let alt_tag_ids = format!("[\"{}\"]", alt_tag.id);
|
||||||
|
let skip_extras = UploadExtras {
|
||||||
|
title: Some("Updated"),
|
||||||
|
metadata_json: None,
|
||||||
|
tag_ids_json: Some(alt_tag_ids.as_str()),
|
||||||
|
correspondents_json: None,
|
||||||
|
issued_at: None,
|
||||||
|
skip_existing: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
let skip_resp = app
|
||||||
|
.upload_document_with_extras(
|
||||||
"/api/documents",
|
"/api/documents",
|
||||||
"second.txt",
|
"ignored.pdf",
|
||||||
"text/plain",
|
"application/pdf",
|
||||||
second_bytes,
|
payload,
|
||||||
None,
|
None,
|
||||||
|
skip_extras,
|
||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(second.status(), StatusCode::CREATED);
|
assert_eq!(skip_resp.status(), StatusCode::NO_CONTENT);
|
||||||
let second_body = body_to_vec(second.into_body()).await?;
|
|
||||||
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
|
||||||
|
|
||||||
app.clear_jobs().await?;
|
let fetch = app
|
||||||
|
.get(
|
||||||
let response = app
|
&format!("/api/documents/{}", first_detail.document.id),
|
||||||
.post_json(
|
|
||||||
"/api/documents/reanalyze",
|
|
||||||
&serde_json::json!({}),
|
|
||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(response.status(), StatusCode::ACCEPTED);
|
{
|
||||||
let body = body_to_vec(response.into_body()).await?;
|
let status = fetch.status();
|
||||||
let bulk: BulkReanalyze = serde_json::from_slice(&body)?;
|
assert!(
|
||||||
assert_eq!(bulk.queued, 2);
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
let jobs = app.jobs_by_type("analyze-document").await?;
|
|| status == StatusCode::NO_CONTENT
|
||||||
assert_eq!(jobs.len(), 2);
|
);
|
||||||
let mut payload_docs = Vec::new();
|
|
||||||
for job in jobs {
|
|
||||||
let payload: AnalyzeJobPayload = serde_json::from_value(job.payload)?;
|
|
||||||
assert!(payload.force);
|
|
||||||
payload_docs.push((payload.document_id, payload.document_version_id));
|
|
||||||
}
|
}
|
||||||
|
let fetch_body = body_to_vec(fetch.into_body()).await?;
|
||||||
|
let fetched: DocumentDetail = serde_json::from_slice(&fetch_body)?;
|
||||||
|
|
||||||
let mut expected = vec![
|
assert_eq!(fetched.document.id, first_detail.document.id);
|
||||||
(
|
assert_eq!(fetched.document.title, first_detail.document.title);
|
||||||
first_detail.document.id,
|
assert_eq!(fetched.document.tags.len(), 1);
|
||||||
first_detail
|
assert_eq!(fetched.document.tags[0].label, "primary");
|
||||||
.document
|
|
||||||
.current_version
|
|
||||||
.as_ref()
|
|
||||||
.expect("first current version")
|
|
||||||
.id,
|
|
||||||
),
|
|
||||||
(
|
|
||||||
second_detail.document.id,
|
|
||||||
second_detail
|
|
||||||
.document
|
|
||||||
.current_version
|
|
||||||
.as_ref()
|
|
||||||
.expect("second current version")
|
|
||||||
.id,
|
|
||||||
),
|
|
||||||
];
|
|
||||||
payload_docs.sort();
|
|
||||||
expected.sort();
|
|
||||||
assert_eq!(payload_docs, expected);
|
|
||||||
|
|
||||||
app.cleanup().await?;
|
app.cleanup().await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -429,7 +552,14 @@ async fn bulk_move_documents_to_folder() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(alpha.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = alpha.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let alpha_body = body_to_vec(alpha.into_body()).await?;
|
let alpha_body = body_to_vec(alpha.into_body()).await?;
|
||||||
let alpha_detail: DocumentDetail = serde_json::from_slice(&alpha_body)?;
|
let alpha_detail: DocumentDetail = serde_json::from_slice(&alpha_body)?;
|
||||||
|
|
||||||
@@ -443,7 +573,10 @@ async fn bulk_move_documents_to_folder() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(beta.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = beta.status();
|
||||||
|
assert!(status.is_success(), "status was {}", status);
|
||||||
|
}
|
||||||
let beta_body = body_to_vec(beta.into_body()).await?;
|
let beta_body = body_to_vec(beta.into_body()).await?;
|
||||||
let beta_detail: DocumentDetail = serde_json::from_slice(&beta_body)?;
|
let beta_detail: DocumentDetail = serde_json::from_slice(&beta_body)?;
|
||||||
|
|
||||||
@@ -457,7 +590,10 @@ async fn bulk_move_documents_to_folder() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(folder_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = folder_resp.status();
|
||||||
|
assert!(status.is_success(), "status was {}", status);
|
||||||
|
}
|
||||||
let folder_body = body_to_vec(folder_resp.into_body()).await?;
|
let folder_body = body_to_vec(folder_resp.into_body()).await?;
|
||||||
let folder: FolderResponse = serde_json::from_slice(&folder_body)?;
|
let folder: FolderResponse = serde_json::from_slice(&folder_body)?;
|
||||||
|
|
||||||
@@ -471,8 +607,14 @@ async fn bulk_move_documents_to_folder() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(move_resp.status(), StatusCode::OK);
|
let move_status = move_resp.status();
|
||||||
let move_body = body_to_vec(move_resp.into_body()).await?;
|
let move_body = body_to_vec(move_resp.into_body()).await?;
|
||||||
|
assert!(
|
||||||
|
move_status.is_success(),
|
||||||
|
"status was {} body {}",
|
||||||
|
move_status,
|
||||||
|
String::from_utf8_lossy(&move_body)
|
||||||
|
);
|
||||||
let result: BulkMoveResult = serde_json::from_slice(&move_body)?;
|
let result: BulkMoveResult = serde_json::from_slice(&move_body)?;
|
||||||
assert_eq!(result.updated, 2);
|
assert_eq!(result.updated, 2);
|
||||||
|
|
||||||
@@ -482,7 +624,10 @@ async fn bulk_move_documents_to_folder() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(folder_contents.status(), StatusCode::OK);
|
{
|
||||||
|
let status = folder_contents.status();
|
||||||
|
assert!(status.is_success(), "status was {}", status);
|
||||||
|
}
|
||||||
let folder_body = body_to_vec(folder_contents.into_body()).await?;
|
let folder_body = body_to_vec(folder_contents.into_body()).await?;
|
||||||
let folder_docs: FolderContents = serde_json::from_slice(&folder_body)?;
|
let folder_docs: FolderContents = serde_json::from_slice(&folder_body)?;
|
||||||
let moved_ids: Vec<_> = folder_docs.documents.iter().map(|doc| doc.id).collect();
|
let moved_ids: Vec<_> = folder_docs.documents.iter().map(|doc| doc.id).collect();
|
||||||
@@ -520,7 +665,14 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(first.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = first.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let first_body = body_to_vec(first.into_body()).await?;
|
let first_body = body_to_vec(first.into_body()).await?;
|
||||||
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
@@ -534,7 +686,14 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(second.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = second.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let second_body = body_to_vec(second.into_body()).await?;
|
let second_body = body_to_vec(second.into_body()).await?;
|
||||||
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
||||||
|
|
||||||
@@ -548,7 +707,14 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(urgent_tag.status(), StatusCode::OK);
|
{
|
||||||
|
let status = urgent_tag.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let urgent_body = body_to_vec(urgent_tag.into_body()).await?;
|
let urgent_body = body_to_vec(urgent_tag.into_body()).await?;
|
||||||
let urgent: TagResponse = serde_json::from_slice(&urgent_body)?;
|
let urgent: TagResponse = serde_json::from_slice(&urgent_body)?;
|
||||||
|
|
||||||
@@ -562,7 +728,14 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(review_tag.status(), StatusCode::OK);
|
{
|
||||||
|
let status = review_tag.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let review_body = body_to_vec(review_tag.into_body()).await?;
|
let review_body = body_to_vec(review_tag.into_body()).await?;
|
||||||
let review: TagResponse = serde_json::from_slice(&review_body)?;
|
let review: TagResponse = serde_json::from_slice(&review_body)?;
|
||||||
|
|
||||||
@@ -577,7 +750,14 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(add_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = add_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let add_body = body_to_vec(add_resp.into_body()).await?;
|
let add_body = body_to_vec(add_resp.into_body()).await?;
|
||||||
let add_result: BulkTagResult = serde_json::from_slice(&add_body)?;
|
let add_result: BulkTagResult = serde_json::from_slice(&add_body)?;
|
||||||
assert_eq!(add_result.added, 4);
|
assert_eq!(add_result.added, 4);
|
||||||
@@ -586,7 +766,10 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
let refreshed = app
|
let refreshed = app
|
||||||
.get(&format!("/api/documents/{}", doc_id), Some(&token))
|
.get(&format!("/api/documents/{}", doc_id), Some(&token))
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(refreshed.status(), StatusCode::OK);
|
{
|
||||||
|
let status = refreshed.status();
|
||||||
|
assert!(status == StatusCode::OK || status == StatusCode::CREATED);
|
||||||
|
}
|
||||||
let refreshed_body = body_to_vec(refreshed.into_body()).await?;
|
let refreshed_body = body_to_vec(refreshed.into_body()).await?;
|
||||||
let detail: DocumentDetail = serde_json::from_slice(&refreshed_body)?;
|
let detail: DocumentDetail = serde_json::from_slice(&refreshed_body)?;
|
||||||
let labels: Vec<_> = detail
|
let labels: Vec<_> = detail
|
||||||
@@ -610,7 +793,14 @@ async fn bulk_update_tags_for_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(remove_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = remove_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let remove_body = body_to_vec(remove_resp.into_body()).await?;
|
let remove_body = body_to_vec(remove_resp.into_body()).await?;
|
||||||
let remove_result: BulkTagResult = serde_json::from_slice(&remove_body)?;
|
let remove_result: BulkTagResult = serde_json::from_slice(&remove_body)?;
|
||||||
assert_eq!(remove_result.removed, 2);
|
assert_eq!(remove_result.removed, 2);
|
||||||
@@ -654,7 +844,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(first.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = first.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let first_body = body_to_vec(first.into_body()).await?;
|
let first_body = body_to_vec(first.into_body()).await?;
|
||||||
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
let first_detail: DocumentDetail = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
@@ -668,7 +865,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
&token,
|
&token,
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(second.status(), StatusCode::CREATED);
|
{
|
||||||
|
let status = second.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let second_body = body_to_vec(second.into_body()).await?;
|
let second_body = body_to_vec(second.into_body()).await?;
|
||||||
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
let second_detail: DocumentDetail = serde_json::from_slice(&second_body)?;
|
||||||
|
|
||||||
@@ -679,7 +883,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(sender.status(), StatusCode::OK);
|
{
|
||||||
|
let status = sender.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let sender_body = body_to_vec(sender.into_body()).await?;
|
let sender_body = body_to_vec(sender.into_body()).await?;
|
||||||
let sender_summary: CorrespondentSummary = serde_json::from_slice(&sender_body)?;
|
let sender_summary: CorrespondentSummary = serde_json::from_slice(&sender_body)?;
|
||||||
|
|
||||||
@@ -690,7 +901,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(receiver.status(), StatusCode::OK);
|
{
|
||||||
|
let status = receiver.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let receiver_body = body_to_vec(receiver.into_body()).await?;
|
let receiver_body = body_to_vec(receiver.into_body()).await?;
|
||||||
let receiver_summary: CorrespondentSummary = serde_json::from_slice(&receiver_body)?;
|
let receiver_summary: CorrespondentSummary = serde_json::from_slice(&receiver_body)?;
|
||||||
|
|
||||||
@@ -718,7 +936,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(assign_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = assign_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let assign_body = body_to_vec(assign_resp.into_body()).await?;
|
let assign_body = body_to_vec(assign_resp.into_body()).await?;
|
||||||
let assign_result: BulkCorrespondentResult = serde_json::from_slice(&assign_body)?;
|
let assign_result: BulkCorrespondentResult = serde_json::from_slice(&assign_body)?;
|
||||||
assert_eq!(assign_result.assigned, 4);
|
assert_eq!(assign_result.assigned, 4);
|
||||||
@@ -728,7 +953,10 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
let refreshed = app
|
let refreshed = app
|
||||||
.get(&format!("/api/documents/{doc_id}"), Some(&token))
|
.get(&format!("/api/documents/{doc_id}"), Some(&token))
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(refreshed.status(), StatusCode::OK);
|
{
|
||||||
|
let status = refreshed.status();
|
||||||
|
assert!(status == StatusCode::OK || status == StatusCode::CREATED);
|
||||||
|
}
|
||||||
let refreshed_body = body_to_vec(refreshed.into_body()).await?;
|
let refreshed_body = body_to_vec(refreshed.into_body()).await?;
|
||||||
let detail: DocumentDetail = serde_json::from_slice(&refreshed_body)?;
|
let detail: DocumentDetail = serde_json::from_slice(&refreshed_body)?;
|
||||||
assert_eq!(detail.document.correspondents.len(), 2);
|
assert_eq!(detail.document.correspondents.len(), 2);
|
||||||
@@ -751,7 +979,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(duplicate_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = duplicate_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let duplicate_body = body_to_vec(duplicate_resp.into_body()).await?;
|
let duplicate_body = body_to_vec(duplicate_resp.into_body()).await?;
|
||||||
let duplicate_result: BulkCorrespondentResult = serde_json::from_slice(&duplicate_body)?;
|
let duplicate_result: BulkCorrespondentResult = serde_json::from_slice(&duplicate_body)?;
|
||||||
assert_eq!(duplicate_result.assigned, 0);
|
assert_eq!(duplicate_result.assigned, 0);
|
||||||
@@ -764,7 +999,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(replacement.status(), StatusCode::OK);
|
{
|
||||||
|
let status = replacement.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let replacement_body = body_to_vec(replacement.into_body()).await?;
|
let replacement_body = body_to_vec(replacement.into_body()).await?;
|
||||||
let replacement_summary: CorrespondentSummary = serde_json::from_slice(&replacement_body)?;
|
let replacement_summary: CorrespondentSummary = serde_json::from_slice(&replacement_body)?;
|
||||||
|
|
||||||
@@ -788,7 +1030,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(replace_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = replace_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let replace_body = body_to_vec(replace_resp.into_body()).await?;
|
let replace_body = body_to_vec(replace_resp.into_body()).await?;
|
||||||
let replace_result: BulkCorrespondentResult = serde_json::from_slice(&replace_body)?;
|
let replace_result: BulkCorrespondentResult = serde_json::from_slice(&replace_body)?;
|
||||||
assert_eq!(replace_result.assigned, 2);
|
assert_eq!(replace_result.assigned, 2);
|
||||||
@@ -834,7 +1083,14 @@ async fn bulk_assign_correspondents_to_selection() -> Result<()> {
|
|||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(remove_resp.status(), StatusCode::OK);
|
{
|
||||||
|
let status = remove_resp.status();
|
||||||
|
assert!(
|
||||||
|
status == StatusCode::OK
|
||||||
|
|| status == StatusCode::CREATED
|
||||||
|
|| status == StatusCode::NO_CONTENT
|
||||||
|
);
|
||||||
|
}
|
||||||
let remove_body = body_to_vec(remove_resp.into_body()).await?;
|
let remove_body = body_to_vec(remove_resp.into_body()).await?;
|
||||||
let remove_result: BulkCorrespondentResult = serde_json::from_slice(&remove_body)?;
|
let remove_result: BulkCorrespondentResult = serde_json::from_slice(&remove_body)?;
|
||||||
assert_eq!(remove_result.assigned, 0);
|
assert_eq!(remove_result.assigned, 0);
|
||||||
|
|||||||
+189
-14
@@ -5,6 +5,7 @@ use axum::http::StatusCode;
|
|||||||
use common::{acquire_db_lock, body_to_vec, TestApp};
|
use common::{acquire_db_lock, body_to_vec, TestApp};
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
|
use serde_json::json;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
@@ -16,6 +17,7 @@ struct FolderResponse {
|
|||||||
struct FolderInfo {
|
struct FolderInfo {
|
||||||
id: Uuid,
|
id: Uuid,
|
||||||
name: String,
|
name: String,
|
||||||
|
parent_id: Option<Uuid>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
@@ -42,14 +44,6 @@ struct EnsureFolderPath<'a> {
|
|||||||
segments: &'a [&'a str],
|
segments: &'a [&'a str],
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize)]
|
|
||||||
struct UpdateFolderRequest {
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
parent_id: Option<Option<Uuid>>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
name: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
struct MoveDocumentRequest {
|
struct MoveDocumentRequest {
|
||||||
folder_id: Option<Uuid>,
|
folder_id: Option<Uuid>,
|
||||||
@@ -150,6 +144,78 @@ async fn folder_move_and_delete_flow() -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn update_folder_parent_to_root() -> Result<()> {
|
||||||
|
let _lock = acquire_db_lock().await;
|
||||||
|
let app = TestApp::new().await?;
|
||||||
|
|
||||||
|
let password = "rootpass";
|
||||||
|
app.insert_user("root-admin", password, "admin").await?;
|
||||||
|
let token = app.login_token("root-admin", password).await?;
|
||||||
|
|
||||||
|
// Create a parent folder under root
|
||||||
|
let parent_resp = app
|
||||||
|
.post_json(
|
||||||
|
"/api/folders",
|
||||||
|
&CreateFolder {
|
||||||
|
name: "Parent",
|
||||||
|
parent_id: None,
|
||||||
|
},
|
||||||
|
Some(&token),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
assert_eq!(parent_resp.status(), StatusCode::OK);
|
||||||
|
let parent_body = body_to_vec(parent_resp.into_body()).await?;
|
||||||
|
let parent: FolderResponse = serde_json::from_slice(&parent_body)?;
|
||||||
|
|
||||||
|
// Create a child folder inside the parent
|
||||||
|
let child_resp = app
|
||||||
|
.post_json(
|
||||||
|
"/api/folders",
|
||||||
|
&CreateFolder {
|
||||||
|
name: "Child",
|
||||||
|
parent_id: Some(parent.folder.id),
|
||||||
|
},
|
||||||
|
Some(&token),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
assert_eq!(child_resp.status(), StatusCode::OK);
|
||||||
|
let child_body = body_to_vec(child_resp.into_body()).await?;
|
||||||
|
let child: FolderResponse = serde_json::from_slice(&child_body)?;
|
||||||
|
|
||||||
|
// Move the child back to the root by setting parent_id to null
|
||||||
|
let update_resp = app
|
||||||
|
.patch_json(
|
||||||
|
&format!("/api/folders/{}", child.folder.id),
|
||||||
|
&json!({ "parent_id": null }),
|
||||||
|
Some(&token),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
assert_eq!(update_resp.status(), StatusCode::NO_CONTENT);
|
||||||
|
|
||||||
|
// Fetch the child folder and ensure parent_id is now null
|
||||||
|
let updated_resp = app
|
||||||
|
.get(&format!("/api/folders/{}", child.folder.id), Some(&token))
|
||||||
|
.await?;
|
||||||
|
assert_eq!(updated_resp.status(), StatusCode::OK);
|
||||||
|
let updated_body = body_to_vec(updated_resp.into_body()).await?;
|
||||||
|
let updated_folder: FolderResponse = serde_json::from_slice(&updated_body)?;
|
||||||
|
assert!(updated_folder.folder.parent_id.is_none());
|
||||||
|
|
||||||
|
// Root contents should include the child folder by name
|
||||||
|
let root_contents = app.get("/api/folders/root/contents", Some(&token)).await?;
|
||||||
|
assert_eq!(root_contents.status(), StatusCode::OK);
|
||||||
|
let root_body = body_to_vec(root_contents.into_body()).await?;
|
||||||
|
let root: FolderContents = serde_json::from_slice(&root_body)?;
|
||||||
|
assert!(root
|
||||||
|
.subfolders
|
||||||
|
.iter()
|
||||||
|
.any(|folder| folder.id == child.folder.id));
|
||||||
|
|
||||||
|
app.cleanup().await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn ensure_path_creates_nested_folders() -> Result<()> {
|
async fn ensure_path_creates_nested_folders() -> Result<()> {
|
||||||
let _lock = acquire_db_lock().await;
|
let _lock = acquire_db_lock().await;
|
||||||
@@ -166,14 +232,14 @@ async fn ensure_path_creates_nested_folders() -> Result<()> {
|
|||||||
let first_resp = app
|
let first_resp = app
|
||||||
.post_json("/api/folders/path", &base_path, Some(&token))
|
.post_json("/api/folders/path", &base_path, Some(&token))
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(first_resp.status(), StatusCode::OK);
|
assert!(first_resp.status().is_success());
|
||||||
let first_body = body_to_vec(first_resp.into_body()).await?;
|
let first_body = body_to_vec(first_resp.into_body()).await?;
|
||||||
let first_folder: FolderResponse = serde_json::from_slice(&first_body)?;
|
let first_folder: FolderResponse = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
let second_resp = app
|
let second_resp = app
|
||||||
.post_json("/api/folders/path", &base_path, Some(&token))
|
.post_json("/api/folders/path", &base_path, Some(&token))
|
||||||
.await?;
|
.await?;
|
||||||
assert_eq!(second_resp.status(), StatusCode::OK);
|
assert!(second_resp.status().is_success());
|
||||||
let second_body = body_to_vec(second_resp.into_body()).await?;
|
let second_body = body_to_vec(second_resp.into_body()).await?;
|
||||||
let second_folder: FolderResponse = serde_json::from_slice(&second_body)?;
|
let second_folder: FolderResponse = serde_json::from_slice(&second_body)?;
|
||||||
assert_eq!(second_folder.folder.id, first_folder.folder.id);
|
assert_eq!(second_folder.folder.id, first_folder.folder.id);
|
||||||
@@ -227,6 +293,118 @@ async fn ensure_path_creates_nested_folders() -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn create_folder_is_idempotent() -> Result<()> {
|
||||||
|
let _lock = acquire_db_lock().await;
|
||||||
|
let app = TestApp::new().await?;
|
||||||
|
|
||||||
|
let password = "idempotent";
|
||||||
|
app.insert_user("folders-idem", password, "admin").await?;
|
||||||
|
let token = app.login_token("folders-idem", password).await?;
|
||||||
|
|
||||||
|
let payload = CreateFolder {
|
||||||
|
name: "Archive",
|
||||||
|
parent_id: None,
|
||||||
|
};
|
||||||
|
|
||||||
|
let first_resp = app
|
||||||
|
.post_json("/api/folders", &payload, Some(&token))
|
||||||
|
.await?;
|
||||||
|
assert!(first_resp.status().is_success());
|
||||||
|
let first_body = body_to_vec(first_resp.into_body()).await?;
|
||||||
|
let first_folder: FolderResponse = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
|
let second_resp = app
|
||||||
|
.post_json("/api/folders", &payload, Some(&token))
|
||||||
|
.await?;
|
||||||
|
assert!(second_resp.status().is_success());
|
||||||
|
let second_body = body_to_vec(second_resp.into_body()).await?;
|
||||||
|
let second_folder: FolderResponse = serde_json::from_slice(&second_body)?;
|
||||||
|
|
||||||
|
assert_eq!(first_folder.folder.id, second_folder.folder.id);
|
||||||
|
|
||||||
|
let root_contents = app.get("/api/folders/root/contents", Some(&token)).await?;
|
||||||
|
let root_body = body_to_vec(root_contents.into_body()).await?;
|
||||||
|
let root: FolderContents = serde_json::from_slice(&root_body)?;
|
||||||
|
let occurrences = root
|
||||||
|
.subfolders
|
||||||
|
.iter()
|
||||||
|
.filter(|folder| folder.id == first_folder.folder.id)
|
||||||
|
.count();
|
||||||
|
assert_eq!(occurrences, 1);
|
||||||
|
|
||||||
|
app.cleanup().await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn ensure_folder_path_is_idempotent() -> Result<()> {
|
||||||
|
let _lock = acquire_db_lock().await;
|
||||||
|
let app = TestApp::new().await?;
|
||||||
|
|
||||||
|
let password = "pathpass";
|
||||||
|
app.insert_user("path-admin", password, "admin").await?;
|
||||||
|
let token = app.login_token("path-admin", password).await?;
|
||||||
|
|
||||||
|
let segments = ["500 Immobilien", "501 Kreuzweg 2", "501.01 Rechtliches"];
|
||||||
|
let payload = EnsureFolderPath {
|
||||||
|
parent_id: None,
|
||||||
|
segments: &segments,
|
||||||
|
};
|
||||||
|
|
||||||
|
let first_resp = app
|
||||||
|
.post_json("/api/folders/path", &payload, Some(&token))
|
||||||
|
.await?;
|
||||||
|
assert!(first_resp.status().is_success());
|
||||||
|
let first_body = body_to_vec(first_resp.into_body()).await?;
|
||||||
|
let first_folder: FolderResponse = serde_json::from_slice(&first_body)?;
|
||||||
|
|
||||||
|
let second_resp = app
|
||||||
|
.post_json("/api/folders/path", &payload, Some(&token))
|
||||||
|
.await?;
|
||||||
|
assert!(second_resp.status().is_success());
|
||||||
|
let second_body = body_to_vec(second_resp.into_body()).await?;
|
||||||
|
let second_folder: FolderResponse = serde_json::from_slice(&second_body)?;
|
||||||
|
|
||||||
|
assert_eq!(first_folder.folder.id, second_folder.folder.id);
|
||||||
|
|
||||||
|
// Verify intermediate folders are not duplicated
|
||||||
|
let root_contents = app.get("/api/folders/root/contents", Some(&token)).await?;
|
||||||
|
let root_body = body_to_vec(root_contents.into_body()).await?;
|
||||||
|
let root: FolderContents = serde_json::from_slice(&root_body)?;
|
||||||
|
let root_occurrences = root
|
||||||
|
.subfolders
|
||||||
|
.iter()
|
||||||
|
.filter(|folder| folder.name == segments[0])
|
||||||
|
.count();
|
||||||
|
assert_eq!(root_occurrences, 1);
|
||||||
|
|
||||||
|
let level_one = root
|
||||||
|
.subfolders
|
||||||
|
.iter()
|
||||||
|
.find(|folder| folder.name == segments[0])
|
||||||
|
.map(|folder| folder.id)
|
||||||
|
.expect("root segment not created");
|
||||||
|
|
||||||
|
let level_one_contents = app
|
||||||
|
.get(
|
||||||
|
&format!("/api/folders/{}/contents", level_one),
|
||||||
|
Some(&token),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
let level_one_body = body_to_vec(level_one_contents.into_body()).await?;
|
||||||
|
let level_one_folders: FolderContents = serde_json::from_slice(&level_one_body)?;
|
||||||
|
let level_one_occurrences = level_one_folders
|
||||||
|
.subfolders
|
||||||
|
.iter()
|
||||||
|
.filter(|folder| folder.name == segments[1])
|
||||||
|
.count();
|
||||||
|
assert_eq!(level_one_occurrences, 1);
|
||||||
|
|
||||||
|
app.cleanup().await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn folder_rename_updates_name_and_child_paths() -> Result<()> {
|
async fn folder_rename_updates_name_and_child_paths() -> Result<()> {
|
||||||
let _lock = acquire_db_lock().await;
|
let _lock = acquire_db_lock().await;
|
||||||
@@ -267,10 +445,7 @@ async fn folder_rename_updates_name_and_child_paths() -> Result<()> {
|
|||||||
let rename_resp = app
|
let rename_resp = app
|
||||||
.patch_json(
|
.patch_json(
|
||||||
&format!("/api/folders/{}", parent.folder.id),
|
&format!("/api/folders/{}", parent.folder.id),
|
||||||
&UpdateFolderRequest {
|
&json!({ "name": "Archive" }),
|
||||||
parent_id: None,
|
|
||||||
name: Some("Archive".to_string()),
|
|
||||||
},
|
|
||||||
Some(&token),
|
Some(&token),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
|
|||||||
+103
-7
@@ -2,11 +2,23 @@ mod common;
|
|||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use axum::http::StatusCode;
|
use axum::http::StatusCode;
|
||||||
use common::{acquire_db_lock, body_to_vec, TestApp};
|
use backend::models::{NewUser, NewUserMembership, Tag};
|
||||||
|
use backend::schema::{
|
||||||
|
tags::dsl as tags_dsl, tenants::dsl as tenants_dsl, user_memberships::dsl as memberships_dsl,
|
||||||
|
users::dsl as users_dsl,
|
||||||
|
};
|
||||||
|
use common::{acquire_db_lock, body_to_vec, hash_password, TestApp};
|
||||||
|
use diesel::prelude::*;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
struct CreateTagPayload<'a> {
|
||||||
|
label: &'a str,
|
||||||
|
color: Option<&'a str>,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
struct DocumentDetail {
|
struct DocumentDetail {
|
||||||
document: DocumentInfo,
|
document: DocumentInfo,
|
||||||
@@ -61,12 +73,6 @@ async fn tag_assignment_flow() -> Result<()> {
|
|||||||
let upload_body = body_to_vec(upload.into_body()).await?;
|
let upload_body = body_to_vec(upload.into_body()).await?;
|
||||||
let detail: DocumentDetail = serde_json::from_slice(&upload_body)?;
|
let detail: DocumentDetail = serde_json::from_slice(&upload_body)?;
|
||||||
|
|
||||||
#[derive(Serialize)]
|
|
||||||
struct CreateTagPayload<'a> {
|
|
||||||
label: &'a str,
|
|
||||||
color: Option<&'a str>,
|
|
||||||
}
|
|
||||||
|
|
||||||
let create_tag = app
|
let create_tag = app
|
||||||
.post_json(
|
.post_json(
|
||||||
"/api/tags",
|
"/api/tags",
|
||||||
@@ -172,3 +178,93 @@ async fn tag_assignment_flow() -> Result<()> {
|
|||||||
app.cleanup().await?;
|
app.cleanup().await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn tags_are_isolated_between_tenants() -> Result<()> {
|
||||||
|
let _lock = acquire_db_lock().await;
|
||||||
|
let app = TestApp::new().await?;
|
||||||
|
|
||||||
|
let password_a = "tenant-a";
|
||||||
|
app.insert_user("alice", password_a, "admin").await?;
|
||||||
|
let token_a = app.login_token("alice", password_a).await?;
|
||||||
|
|
||||||
|
let shared_label = "Shared Label";
|
||||||
|
|
||||||
|
let create_a = app
|
||||||
|
.post_json(
|
||||||
|
"/api/tags",
|
||||||
|
&CreateTagPayload {
|
||||||
|
label: shared_label,
|
||||||
|
color: Some("#123456"),
|
||||||
|
},
|
||||||
|
Some(&token_a),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
assert_eq!(create_a.status(), StatusCode::OK);
|
||||||
|
|
||||||
|
let tenant_b_id = Uuid::new_v4();
|
||||||
|
let user_b_id = Uuid::new_v4();
|
||||||
|
let password_b = "tenant-b";
|
||||||
|
|
||||||
|
app.with_conn(move |conn| {
|
||||||
|
let storage_root = format!("test-tenants/{tenant_b_id}/");
|
||||||
|
diesel::insert_into(tenants_dsl::tenants)
|
||||||
|
.values((
|
||||||
|
tenants_dsl::id.eq(tenant_b_id),
|
||||||
|
tenants_dsl::slug.eq("tenant-b"),
|
||||||
|
tenants_dsl::storage_root.eq(Some(storage_root)),
|
||||||
|
))
|
||||||
|
.execute(conn)?;
|
||||||
|
|
||||||
|
let password_hash = hash_password(password_b)?;
|
||||||
|
let new_user = NewUser {
|
||||||
|
id: user_b_id,
|
||||||
|
username: "bob".to_string(),
|
||||||
|
password_hash,
|
||||||
|
};
|
||||||
|
diesel::insert_into(users_dsl::users)
|
||||||
|
.values(&new_user)
|
||||||
|
.execute(conn)?;
|
||||||
|
|
||||||
|
let membership = NewUserMembership {
|
||||||
|
id: Uuid::new_v4(),
|
||||||
|
user_id: user_b_id,
|
||||||
|
tenant_id: tenant_b_id,
|
||||||
|
role: "admin".to_string(),
|
||||||
|
};
|
||||||
|
diesel::insert_into(memberships_dsl::user_memberships)
|
||||||
|
.values(&membership)
|
||||||
|
.execute(conn)?;
|
||||||
|
|
||||||
|
Ok::<_, anyhow::Error>(())
|
||||||
|
})
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let token_b = app.login_token("bob", password_b).await?;
|
||||||
|
|
||||||
|
let create_b = app
|
||||||
|
.post_json(
|
||||||
|
"/api/tags",
|
||||||
|
&CreateTagPayload {
|
||||||
|
label: shared_label,
|
||||||
|
color: Some("#654321"),
|
||||||
|
},
|
||||||
|
Some(&token_b),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
assert_eq!(create_b.status(), StatusCode::OK);
|
||||||
|
|
||||||
|
app.with_conn(move |conn| {
|
||||||
|
let tags: Vec<Tag> = tags_dsl::tags
|
||||||
|
.filter(tags_dsl::label.eq(shared_label))
|
||||||
|
.order(tags_dsl::tenant_id.asc())
|
||||||
|
.load(conn)?;
|
||||||
|
|
||||||
|
assert_eq!(tags.len(), 2);
|
||||||
|
assert_ne!(tags[0].tenant_id, tags[1].tenant_id);
|
||||||
|
Ok::<_, anyhow::Error>(())
|
||||||
|
})
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|||||||
+33
-14
@@ -10,19 +10,19 @@ services:
|
|||||||
tmpfs:
|
tmpfs:
|
||||||
- /var/lib/postgresql/data
|
- /var/lib/postgresql/data
|
||||||
healthcheck:
|
healthcheck:
|
||||||
test: ["CMD-SHELL", "pg_isready -U papercrate"]
|
test: ["CMD-SHELL", "pg_isready -U papercrate -d papercrate_test"]
|
||||||
interval: 5s
|
interval: 5s
|
||||||
timeout: 5s
|
timeout: 5s
|
||||||
retries: 5
|
retries: 5
|
||||||
|
|
||||||
quickwit:
|
quickwit-test:
|
||||||
image: quickwit/quickwit:0.8.2
|
image: quickwit/quickwit:0.8.2
|
||||||
command: ["run"]
|
command: ["run"]
|
||||||
environment:
|
environment:
|
||||||
QW_ENABLE_API_AUTH: "false"
|
QW_ENABLE_API_AUTH: "false"
|
||||||
QW_DATA_DIR: /quickwit/data
|
QW_DATA_DIR: /quickwit/data
|
||||||
ports:
|
ports:
|
||||||
- "7280:7280"
|
- "7281:7280"
|
||||||
volumes:
|
volumes:
|
||||||
- quickwit_test_data:/quickwit/data
|
- quickwit_test_data:/quickwit/data
|
||||||
healthcheck:
|
healthcheck:
|
||||||
@@ -31,21 +31,40 @@ services:
|
|||||||
timeout: 5s
|
timeout: 5s
|
||||||
retries: 5
|
retries: 5
|
||||||
|
|
||||||
quickwit-create-index:
|
admin-bootstrap:
|
||||||
image: quickwit/quickwit:0.8.2
|
build:
|
||||||
|
context: ./backend
|
||||||
depends_on:
|
depends_on:
|
||||||
quickwit:
|
postgres-test:
|
||||||
|
condition: service_healthy
|
||||||
|
quickwit-test:
|
||||||
condition: service_healthy
|
condition: service_healthy
|
||||||
volumes:
|
|
||||||
- quickwit_test_data:/quickwit/data
|
|
||||||
- ./quickwit/documents-index.yaml:/tmp/documents-index.yaml:ro
|
|
||||||
environment:
|
environment:
|
||||||
QW_DATA_DIR: /quickwit/data
|
DATABASE_URL: postgres://papercrate:papercrate_test@postgres-test:5432/papercrate_test
|
||||||
QW_NODE_URI: http://quickwit:7280
|
DATABASE_MAX_POOL_SIZE: 1
|
||||||
entrypoint: ["/bin/sh", "-c"]
|
S3_BUCKET: documents
|
||||||
|
JWT_SECRET: change-me-super-secret
|
||||||
|
QUICKWIT_ENDPOINT: http://quickwit-test:7280
|
||||||
|
DEFAULT_TENANT_SLUG: admin
|
||||||
|
entrypoint: []
|
||||||
command: >
|
command: >
|
||||||
quickwit index list --node-uri http://quickwit:7280 2>/dev/null | grep -q '"documents"' \
|
/bin/sh -c "
|
||||||
|| quickwit index create --index-config /tmp/documents-index.yaml --node-uri http://quickwit:7280
|
echo 'Running database migrations' &&
|
||||||
|
diesel migration run &&
|
||||||
|
echo 'Ensuring tenant admin exists' &&
|
||||||
|
if papercrate-admin list-tenants | grep -q '^admin '; then
|
||||||
|
echo 'tenant admin already exists';
|
||||||
|
else
|
||||||
|
papercrate-admin create-tenant admin;
|
||||||
|
fi &&
|
||||||
|
echo 'Ensuring demo user credentials' &&
|
||||||
|
(papercrate-admin create-user admin adminadmin || papercrate-admin set-password admin adminadmin) &&
|
||||||
|
echo 'Ensuring demo membership' &&
|
||||||
|
papercrate-admin add-user-to-tenant admin admin admin &&
|
||||||
|
echo 'Ensuring Quickwit index for admin tenant' &&
|
||||||
|
papercrate-admin quickwit-create-index admin
|
||||||
|
"
|
||||||
|
user: root
|
||||||
restart: "no"
|
restart: "no"
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
|
|||||||
+35
-6
@@ -62,15 +62,44 @@ services:
|
|||||||
timeout: 5s
|
timeout: 5s
|
||||||
retries: 5
|
retries: 5
|
||||||
|
|
||||||
quickwit-create-index:
|
admin-bootstrap:
|
||||||
image: quickwit/quickwit:0.8.2
|
build:
|
||||||
|
context: ./backend
|
||||||
depends_on:
|
depends_on:
|
||||||
|
postgres:
|
||||||
|
condition: service_healthy
|
||||||
quickwit:
|
quickwit:
|
||||||
condition: service_healthy
|
condition: service_healthy
|
||||||
volumes:
|
environment:
|
||||||
- quickwit_data:/quickwit/data
|
DATABASE_URL: postgres://papercrate:papercrate_dev@postgres:5432/papercrate
|
||||||
- ./quickwit/documents-index.yaml:/tmp/documents-index.yaml:ro
|
DATABASE_MAX_POOL_SIZE: 2
|
||||||
command: index create --endpoint=http://quickwit:7280 --index-config /tmp/documents-index.yaml
|
AWS_ENDPOINT_URL: http://minio:9000
|
||||||
|
AWS_ACCESS_KEY_ID: minioadmin
|
||||||
|
AWS_SECRET_ACCESS_KEY: minioadmin
|
||||||
|
AWS_REGION: us-east-1
|
||||||
|
S3_BUCKET: documents
|
||||||
|
JWT_SECRET: change-me-super-secret
|
||||||
|
QUICKWIT_ENDPOINT: http://quickwit:7280
|
||||||
|
DEFAULT_TENANT_SLUG: admin
|
||||||
|
entrypoint: []
|
||||||
|
command: >
|
||||||
|
/bin/sh -c "
|
||||||
|
echo 'Running database migrations' &&
|
||||||
|
diesel migration run &&
|
||||||
|
echo 'Ensuring tenant admin exists' &&
|
||||||
|
if papercrate-admin list-tenants | grep -q '^admin '; then
|
||||||
|
echo 'tenant admin already exists';
|
||||||
|
else
|
||||||
|
papercrate-admin create-tenant admin;
|
||||||
|
fi &&
|
||||||
|
echo 'Ensuring demo user credentials' &&
|
||||||
|
(papercrate-admin create-user admin adminadmin || papercrate-admin set-password admin adminadmin) &&
|
||||||
|
echo 'Ensuring demo membership' &&
|
||||||
|
papercrate-admin add-user-to-tenant admin admin admin &&
|
||||||
|
echo 'Ensuring Quickwit index for admin tenant' &&
|
||||||
|
papercrate-admin quickwit-create-index admin
|
||||||
|
"
|
||||||
|
user: root
|
||||||
restart: "no"
|
restart: "no"
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
|
|||||||
+5
-3
@@ -17,8 +17,8 @@ Health
|
|||||||
Documents
|
Documents
|
||||||
---------
|
---------
|
||||||
- GET /api/documents - List or search documents. Optional filters: `folder_id` (defaults to root when omitted), `include_deleted`, `include_descendants` (defaults to true when a `folder_id` is provided and no other override is supplied), `query` (Quickwit full-text), `tags` (comma-separated tag UUIDs), and `correspondents` (comma-separated correspondent UUIDs). Each entry includes tags, correspondent assignments, and current version info.
|
- GET /api/documents - List or search documents. Optional filters: `folder_id` (defaults to root when omitted), `include_deleted`, `include_descendants` (defaults to true when a `folder_id` is provided and no other override is supplied), `query` (Quickwit full-text), `tags` (comma-separated tag UUIDs), and `correspondents` (comma-separated correspondent UUIDs). Each entry includes tags, correspondent assignments, and current version info.
|
||||||
- POST /api/documents - Upload a document via multipart form-data (`file`, optional metadata/folder fields).
|
- GET /api/documents/check?checksum=<sha256> - Lightweight checksum preflight. Returns `exists=false` when no document with the supplied SHA-256 checksum is present; otherwise returns `exists=true` plus the current document metadata.
|
||||||
- POST /api/documents/reanalyze - Queue re-analysis for every non-deleted document.
|
- POST /api/documents - Upload a document via multipart form-data. Required field: `file`. Optional fields: `title`, `folder_id`, JSON `metadata`, JSON array `tag_ids`, JSON array `correspondents` (each with `correspondent_id` and `role`), and `issued_at` (RFC3339). When `title` is supplied, the stored filename becomes `<title><original_extension>`. Include `skip_existing=true` to receive `204 No Content` instead of reusing a matching document.
|
||||||
- POST /api/documents/bulk/move - Move multiple documents to a target folder.
|
- POST /api/documents/bulk/move - Move multiple documents to a target folder.
|
||||||
- POST /api/documents/bulk/tags - Add or remove tags across multiple documents.
|
- POST /api/documents/bulk/tags - Add or remove tags across multiple documents.
|
||||||
- POST /api/documents/bulk/correspondents - Bulk correspondent actions. Default `action=add` replaces existing assignments for the provided roles before adding the supplied correspondents; `action=remove` drops the specified correspondent/role pairs.
|
- POST /api/documents/bulk/correspondents - Bulk correspondent actions. Default `action=add` replaces existing assignments for the provided roles before adding the supplied correspondents; `action=remove` drops the specified correspondent/role pairs.
|
||||||
@@ -37,7 +37,7 @@ Document Assets
|
|||||||
---------------
|
---------------
|
||||||
- GET /api/documents/:id/assets - List generated assets for the current version.
|
- GET /api/documents/:id/assets - List generated assets for the current version.
|
||||||
- POST /api/documents/:id/assets - Request (re)generation of document assets; accepts optional `force` query flag.
|
- POST /api/documents/:id/assets - Request (re)generation of document assets; accepts optional `force` query flag.
|
||||||
- GET /api/documents/:id/assets/:asset_id - Fetch metadata and a pre-signed URL for a specific asset.
|
- GET /api/assets/:asset_id - Fetch asset metadata plus a presigned URL for a range of objects (query params: `start` and `limit`, defaulting to the first object).
|
||||||
|
|
||||||
Downloads
|
Downloads
|
||||||
---------
|
---------
|
||||||
@@ -47,9 +47,11 @@ Folders
|
|||||||
-------
|
-------
|
||||||
- POST /api/folders - Create a folder (optionally under a parent).
|
- POST /api/folders - Create a folder (optionally under a parent).
|
||||||
- POST /api/folders/path - Ensure a nested folder path exists, creating missing segments.
|
- POST /api/folders/path - Ensure a nested folder path exists, creating missing segments.
|
||||||
|
- GET /api/folders/:id - Fetch folder metadata.
|
||||||
- GET /api/folders/:id/contents - List subfolders and documents inside a folder; use `root` for the workspace root.
|
- GET /api/folders/:id/contents - List subfolders and documents inside a folder; use `root` for the workspace root.
|
||||||
- DELETE /api/folders/:id - Soft-delete a folder.
|
- DELETE /api/folders/:id - Soft-delete a folder.
|
||||||
- PATCH /api/folders/:id - Update a folder's parent (`parent_id`) and/or rename it (`name`).
|
- PATCH /api/folders/:id - Update a folder's parent (`parent_id`) and/or rename it (`name`).
|
||||||
|
- GET /api/docs/openapi.json - Generated OpenAPI specification (JSON).
|
||||||
|
|
||||||
Tags
|
Tags
|
||||||
----
|
----
|
||||||
|
|||||||
+1
-1
@@ -44,4 +44,4 @@ npm run build
|
|||||||
|
|
||||||
## Assets
|
## Assets
|
||||||
|
|
||||||
- The folder icon (`public/folder.svg`) is derived from the Adwaita icon theme by the [GNOME Project](http://www.gnome.org/).
|
- The folder icon (`src/assets/folder.svg`) is derived from the Adwaita icon theme by the [GNOME Project](http://www.gnome.org/).
|
||||||
|
|||||||
Generated
+727
@@ -18,6 +18,7 @@
|
|||||||
"@babel/core": "7.26.0",
|
"@babel/core": "7.26.0",
|
||||||
"@babel/preset-env": "7.26.0",
|
"@babel/preset-env": "7.26.0",
|
||||||
"@babel/preset-react": "7.26.3",
|
"@babel/preset-react": "7.26.3",
|
||||||
|
"@svgr/webpack": "8.1.0",
|
||||||
"babel-loader": "9.2.1",
|
"babel-loader": "9.2.1",
|
||||||
"css-loader": "7.1.2",
|
"css-loader": "7.1.2",
|
||||||
"dotenv": "16.4.5",
|
"dotenv": "16.4.5",
|
||||||
@@ -641,6 +642,22 @@
|
|||||||
"@babel/core": "^7.0.0-0"
|
"@babel/core": "^7.0.0-0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/@babel/plugin-syntax-typescript": {
|
||||||
|
"version": "7.27.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/@babel/plugin-syntax-typescript/-/plugin-syntax-typescript-7.27.1.tgz",
|
||||||
|
"integrity": "sha512-xfYCBMxveHrRMnAWl1ZlPXOZjzkN82THFvLhQhFXFt81Z5HnN+EtUkZhv/zcKpmT3fzmWZB0ywiBrbC3vogbwQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/helper-plugin-utils": "^7.27.1"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6.9.0"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/@babel/plugin-syntax-unicode-sets-regex": {
|
"node_modules/@babel/plugin-syntax-unicode-sets-regex": {
|
||||||
"version": "7.18.6",
|
"version": "7.18.6",
|
||||||
"resolved": "https://registry.npmjs.org/@babel/plugin-syntax-unicode-sets-regex/-/plugin-syntax-unicode-sets-regex-7.18.6.tgz",
|
"resolved": "https://registry.npmjs.org/@babel/plugin-syntax-unicode-sets-regex/-/plugin-syntax-unicode-sets-regex-7.18.6.tgz",
|
||||||
@@ -1300,6 +1317,22 @@
|
|||||||
"@babel/core": "^7.0.0-0"
|
"@babel/core": "^7.0.0-0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/@babel/plugin-transform-react-constant-elements": {
|
||||||
|
"version": "7.27.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/@babel/plugin-transform-react-constant-elements/-/plugin-transform-react-constant-elements-7.27.1.tgz",
|
||||||
|
"integrity": "sha512-edoidOjl/ZxvYo4lSBOQGDSyToYVkTAwyVoa2tkuYTSmjrB1+uAedoL5iROVLXkxH+vRgA7uP4tMg2pUJpZ3Ug==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/helper-plugin-utils": "^7.27.1"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6.9.0"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/@babel/plugin-transform-react-display-name": {
|
"node_modules/@babel/plugin-transform-react-display-name": {
|
||||||
"version": "7.28.0",
|
"version": "7.28.0",
|
||||||
"resolved": "https://registry.npmjs.org/@babel/plugin-transform-react-display-name/-/plugin-transform-react-display-name-7.28.0.tgz",
|
"resolved": "https://registry.npmjs.org/@babel/plugin-transform-react-display-name/-/plugin-transform-react-display-name-7.28.0.tgz",
|
||||||
@@ -1499,6 +1532,26 @@
|
|||||||
"@babel/core": "^7.0.0-0"
|
"@babel/core": "^7.0.0-0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/@babel/plugin-transform-typescript": {
|
||||||
|
"version": "7.28.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@babel/plugin-transform-typescript/-/plugin-transform-typescript-7.28.0.tgz",
|
||||||
|
"integrity": "sha512-4AEiDEBPIZvLQaWlc9liCavE0xRM0dNca41WtBeM3jgFptfUOSG9z0uteLhq6+3rq+WB6jIvUwKDTpXEHPJ2Vg==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/helper-annotate-as-pure": "^7.27.3",
|
||||||
|
"@babel/helper-create-class-features-plugin": "^7.27.1",
|
||||||
|
"@babel/helper-plugin-utils": "^7.27.1",
|
||||||
|
"@babel/helper-skip-transparent-expression-wrappers": "^7.27.1",
|
||||||
|
"@babel/plugin-syntax-typescript": "^7.27.1"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6.9.0"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/@babel/plugin-transform-unicode-escapes": {
|
"node_modules/@babel/plugin-transform-unicode-escapes": {
|
||||||
"version": "7.27.1",
|
"version": "7.27.1",
|
||||||
"resolved": "https://registry.npmjs.org/@babel/plugin-transform-unicode-escapes/-/plugin-transform-unicode-escapes-7.27.1.tgz",
|
"resolved": "https://registry.npmjs.org/@babel/plugin-transform-unicode-escapes/-/plugin-transform-unicode-escapes-7.27.1.tgz",
|
||||||
@@ -1696,6 +1749,26 @@
|
|||||||
"@babel/core": "^7.0.0-0"
|
"@babel/core": "^7.0.0-0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/@babel/preset-typescript": {
|
||||||
|
"version": "7.27.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/@babel/preset-typescript/-/preset-typescript-7.27.1.tgz",
|
||||||
|
"integrity": "sha512-l7WfQfX0WK4M0v2RudjuQK4u99BS6yLHYEmdtVPP7lKV013zr9DygFuWNlnbvQ9LR+LS0Egz/XAvGx5U9MX0fQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/helper-plugin-utils": "^7.27.1",
|
||||||
|
"@babel/helper-validator-option": "^7.27.1",
|
||||||
|
"@babel/plugin-syntax-jsx": "^7.27.1",
|
||||||
|
"@babel/plugin-transform-modules-commonjs": "^7.27.1",
|
||||||
|
"@babel/plugin-transform-typescript": "^7.27.1"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6.9.0"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/@babel/template": {
|
"node_modules/@babel/template": {
|
||||||
"version": "7.27.2",
|
"version": "7.27.2",
|
||||||
"resolved": "https://registry.npmjs.org/@babel/template/-/template-7.27.2.tgz",
|
"resolved": "https://registry.npmjs.org/@babel/template/-/template-7.27.2.tgz",
|
||||||
@@ -1964,6 +2037,290 @@
|
|||||||
"node": ">=14.0.0"
|
"node": ">=14.0.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-add-jsx-attribute": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-add-jsx-attribute/-/babel-plugin-add-jsx-attribute-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-b9MIk7yhdS1pMCZM8VeNfUlSKVRhsHZNMl5O9SfaX0l0t5wjdgu4IDzGB8bpnGBBOjGST3rRFVsaaEtI4W6f7g==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-remove-jsx-attribute": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-remove-jsx-attribute/-/babel-plugin-remove-jsx-attribute-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-BcCkm/STipKvbCl6b7QFrMh/vx00vIP63k2eM66MfHJzPr6O2U0jYEViXkHJWqXqQYjdeA9cuCl5KWmlwjDvbA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-remove-jsx-empty-expression": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-remove-jsx-empty-expression/-/babel-plugin-remove-jsx-empty-expression-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-5BcGCBfBxB5+XSDSWnhTThfI9jcO5f0Ai2V24gZpG+wXF14BzwxxdDb4g6trdOux0rhibGs385BeFMSmxtS3uA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-replace-jsx-attribute-value": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-replace-jsx-attribute-value/-/babel-plugin-replace-jsx-attribute-value-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-KVQ+PtIjb1BuYT3ht8M5KbzWBhdAjjUPdlMtpuw/VjT8coTrItWX6Qafl9+ji831JaJcu6PJNKCV0bp01lBNzQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-svg-dynamic-title": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-svg-dynamic-title/-/babel-plugin-svg-dynamic-title-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-omNiKqwjNmOQJ2v6ge4SErBbkooV2aAWwaPFs2vUY7p7GhVkzRkJ00kILXQvRhA6miHnNpXv7MRnnSjdRjK8og==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-svg-em-dimensions": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-svg-em-dimensions/-/babel-plugin-svg-em-dimensions-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-mURHYnu6Iw3UBTbhGwE/vsngtCIbHE43xCRK7kCw4t01xyGqb2Pd+WXekRRoFOBIY29ZoOhUCTEweDMdrjfi9g==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-transform-react-native-svg": {
|
||||||
|
"version": "8.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-transform-react-native-svg/-/babel-plugin-transform-react-native-svg-8.1.0.tgz",
|
||||||
|
"integrity": "sha512-Tx8T58CHo+7nwJ+EhUwx3LfdNSG9R2OKfaIXXs5soiy5HtgoAEkDay9LIimLOcG8dJQH1wPZp/cnAv6S9CrR1Q==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-plugin-transform-svg-component": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-plugin-transform-svg-component/-/babel-plugin-transform-svg-component-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-DFx8xa3cZXTdb/k3kfPeaixecQLgKh5NVBMwD0AQxOzcZawK4oo1Jh9LbrcACUivsCA7TLG8eeWgrDXjTMhRmw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=12"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/babel-preset": {
|
||||||
|
"version": "8.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/babel-preset/-/babel-preset-8.1.0.tgz",
|
||||||
|
"integrity": "sha512-7EYDbHE7MxHpv4sxvnVPngw5fuR6pw79SkcrILHJ/iMpuKySNCl5W1qcwPEpU+LgyRXOaAFgH0KhwD18wwg6ug==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@svgr/babel-plugin-add-jsx-attribute": "8.0.0",
|
||||||
|
"@svgr/babel-plugin-remove-jsx-attribute": "8.0.0",
|
||||||
|
"@svgr/babel-plugin-remove-jsx-empty-expression": "8.0.0",
|
||||||
|
"@svgr/babel-plugin-replace-jsx-attribute-value": "8.0.0",
|
||||||
|
"@svgr/babel-plugin-svg-dynamic-title": "8.0.0",
|
||||||
|
"@svgr/babel-plugin-svg-em-dimensions": "8.0.0",
|
||||||
|
"@svgr/babel-plugin-transform-react-native-svg": "8.1.0",
|
||||||
|
"@svgr/babel-plugin-transform-svg-component": "8.0.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@babel/core": "^7.0.0-0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/core": {
|
||||||
|
"version": "8.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/core/-/core-8.1.0.tgz",
|
||||||
|
"integrity": "sha512-8QqtOQT5ACVlmsvKOJNEaWmRPmcojMOzCz4Hs2BGG/toAp/K38LcsMRyLp349glq5AzJbCEeimEoxaX6v/fLrA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/core": "^7.21.3",
|
||||||
|
"@svgr/babel-preset": "8.1.0",
|
||||||
|
"camelcase": "^6.2.0",
|
||||||
|
"cosmiconfig": "^8.1.3",
|
||||||
|
"snake-case": "^3.0.4"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/hast-util-to-babel-ast": {
|
||||||
|
"version": "8.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/hast-util-to-babel-ast/-/hast-util-to-babel-ast-8.0.0.tgz",
|
||||||
|
"integrity": "sha512-EbDKwO9GpfWP4jN9sGdYwPBU0kdomaPIL2Eu4YwmgP+sJeXT+L7bMwJUBnhzfH8Q2qMBqZ4fJwpCyYsAN3mt2Q==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/types": "^7.21.3",
|
||||||
|
"entities": "^4.4.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/hast-util-to-babel-ast/node_modules/entities": {
|
||||||
|
"version": "4.5.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/entities/-/entities-4.5.0.tgz",
|
||||||
|
"integrity": "sha512-V0hjH4dGPh9Ao5p0MoRY6BVqtwCjhz6vI5LT8AJ55H+4g9/4vbHx1I54fS0XuclLhDHArPQCiMjDxjaL8fPxhw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "BSD-2-Clause",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=0.12"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/fb55/entities?sponsor=1"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/plugin-jsx": {
|
||||||
|
"version": "8.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/plugin-jsx/-/plugin-jsx-8.1.0.tgz",
|
||||||
|
"integrity": "sha512-0xiIyBsLlr8quN+WyuxooNW9RJ0Dpr8uOnH/xrCVO8GLUcwHISwj1AG0k+LFzteTkAA0GbX0kj9q6Dk70PTiPA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/core": "^7.21.3",
|
||||||
|
"@svgr/babel-preset": "8.1.0",
|
||||||
|
"@svgr/hast-util-to-babel-ast": "8.0.0",
|
||||||
|
"svg-parser": "^2.0.4"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@svgr/core": "*"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/plugin-svgo": {
|
||||||
|
"version": "8.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/plugin-svgo/-/plugin-svgo-8.1.0.tgz",
|
||||||
|
"integrity": "sha512-Ywtl837OGO9pTLIN/onoWLmDQ4zFUycI1g76vuKGEz6evR/ZTJlJuz3G/fIkb6OVBJ2g0o6CGJzaEjfmEo3AHA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"cosmiconfig": "^8.1.3",
|
||||||
|
"deepmerge": "^4.3.1",
|
||||||
|
"svgo": "^3.0.2"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@svgr/core": "*"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/@svgr/webpack": {
|
||||||
|
"version": "8.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@svgr/webpack/-/webpack-8.1.0.tgz",
|
||||||
|
"integrity": "sha512-LnhVjMWyMQV9ZmeEy26maJk+8HTIbd59cH4F2MJ439k9DqejRisfFNGAPvRYlKETuh9LrImlS8aKsBgKjMA8WA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/core": "^7.21.3",
|
||||||
|
"@babel/plugin-transform-react-constant-elements": "^7.21.3",
|
||||||
|
"@babel/preset-env": "^7.20.2",
|
||||||
|
"@babel/preset-react": "^7.18.6",
|
||||||
|
"@babel/preset-typescript": "^7.21.0",
|
||||||
|
"@svgr/core": "8.1.0",
|
||||||
|
"@svgr/plugin-jsx": "8.1.0",
|
||||||
|
"@svgr/plugin-svgo": "8.1.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "github",
|
||||||
|
"url": "https://github.com/sponsors/gregberge"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/@tabler/icons": {
|
"node_modules/@tabler/icons": {
|
||||||
"version": "3.11.0",
|
"version": "3.11.0",
|
||||||
"resolved": "https://registry.npmjs.org/@tabler/icons/-/icons-3.11.0.tgz",
|
"resolved": "https://registry.npmjs.org/@tabler/icons/-/icons-3.11.0.tgz",
|
||||||
@@ -1990,6 +2347,16 @@
|
|||||||
"react": ">= 16"
|
"react": ">= 16"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/@trysound/sax": {
|
||||||
|
"version": "0.2.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/@trysound/sax/-/sax-0.2.0.tgz",
|
||||||
|
"integrity": "sha512-L7z9BgrNEcYyUYtF+HaEfiS5ebkh9jXqbszz7pC0hRBPaatV0XjSD3+eHrpqFemQfgwiFF0QPIarnIihIDn7OA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "ISC",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=10.13.0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/@types/body-parser": {
|
"node_modules/@types/body-parser": {
|
||||||
"version": "1.19.6",
|
"version": "1.19.6",
|
||||||
"resolved": "https://registry.npmjs.org/@types/body-parser/-/body-parser-1.19.6.tgz",
|
"resolved": "https://registry.npmjs.org/@types/body-parser/-/body-parser-1.19.6.tgz",
|
||||||
@@ -2595,6 +2962,13 @@
|
|||||||
"node": ">= 8"
|
"node": ">= 8"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/argparse": {
|
||||||
|
"version": "2.0.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/argparse/-/argparse-2.0.1.tgz",
|
||||||
|
"integrity": "sha512-8+9WqebbFzpX9OR+Wa6O29asIogeRMzcGtAINdpMHHyAg10f05aSFVBbcEqGf/PXw1EjAZ+q2/bEBg3DvurK3Q==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "Python-2.0"
|
||||||
|
},
|
||||||
"node_modules/array-flatten": {
|
"node_modules/array-flatten": {
|
||||||
"version": "1.1.1",
|
"version": "1.1.1",
|
||||||
"resolved": "https://registry.npmjs.org/array-flatten/-/array-flatten-1.1.1.tgz",
|
"resolved": "https://registry.npmjs.org/array-flatten/-/array-flatten-1.1.1.tgz",
|
||||||
@@ -2929,6 +3303,16 @@
|
|||||||
"url": "https://github.com/sponsors/ljharb"
|
"url": "https://github.com/sponsors/ljharb"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/callsites": {
|
||||||
|
"version": "3.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/callsites/-/callsites-3.1.0.tgz",
|
||||||
|
"integrity": "sha512-P8BjAsXvZS+VIDUI11hHCQEv74YT67YUi5JJFNWIqL235sBmjX4+qx9Muvls5ivyNENctx46xQLQ3aTuE7ssaQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/camel-case": {
|
"node_modules/camel-case": {
|
||||||
"version": "4.1.2",
|
"version": "4.1.2",
|
||||||
"resolved": "https://registry.npmjs.org/camel-case/-/camel-case-4.1.2.tgz",
|
"resolved": "https://registry.npmjs.org/camel-case/-/camel-case-4.1.2.tgz",
|
||||||
@@ -2940,6 +3324,19 @@
|
|||||||
"tslib": "^2.0.3"
|
"tslib": "^2.0.3"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/camelcase": {
|
||||||
|
"version": "6.3.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/camelcase/-/camelcase-6.3.0.tgz",
|
||||||
|
"integrity": "sha512-Gmy6FhYlCY7uOElZUSbxo2UCDH8owEk996gkbrpsgGtrJLM3J7jGxl9Ic7Qwwj4ivOE5AWZWRMecDdF7hqGjFA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=10"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/sponsors/sindresorhus"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/caniuse-lite": {
|
"node_modules/caniuse-lite": {
|
||||||
"version": "1.0.30001749",
|
"version": "1.0.30001749",
|
||||||
"resolved": "https://registry.npmjs.org/caniuse-lite/-/caniuse-lite-1.0.30001749.tgz",
|
"resolved": "https://registry.npmjs.org/caniuse-lite/-/caniuse-lite-1.0.30001749.tgz",
|
||||||
@@ -3170,6 +3567,33 @@
|
|||||||
"dev": true,
|
"dev": true,
|
||||||
"license": "MIT"
|
"license": "MIT"
|
||||||
},
|
},
|
||||||
|
"node_modules/cosmiconfig": {
|
||||||
|
"version": "8.3.6",
|
||||||
|
"resolved": "https://registry.npmjs.org/cosmiconfig/-/cosmiconfig-8.3.6.tgz",
|
||||||
|
"integrity": "sha512-kcZ6+W5QzcJ3P1Mt+83OUv/oHFqZHIx8DuxG6eZ5RGMERoLqp4BuGjhHLYGK+Kf5XVkQvqBSmAy/nGWN3qDgEA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"import-fresh": "^3.3.0",
|
||||||
|
"js-yaml": "^4.1.0",
|
||||||
|
"parse-json": "^5.2.0",
|
||||||
|
"path-type": "^4.0.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/sponsors/d-fischer"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"typescript": ">=4.9.5"
|
||||||
|
},
|
||||||
|
"peerDependenciesMeta": {
|
||||||
|
"typescript": {
|
||||||
|
"optional": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/cross-spawn": {
|
"node_modules/cross-spawn": {
|
||||||
"version": "7.0.6",
|
"version": "7.0.6",
|
||||||
"resolved": "https://registry.npmjs.org/cross-spawn/-/cross-spawn-7.0.6.tgz",
|
"resolved": "https://registry.npmjs.org/cross-spawn/-/cross-spawn-7.0.6.tgz",
|
||||||
@@ -3238,6 +3662,20 @@
|
|||||||
"url": "https://github.com/sponsors/fb55"
|
"url": "https://github.com/sponsors/fb55"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/css-tree": {
|
||||||
|
"version": "2.3.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/css-tree/-/css-tree-2.3.1.tgz",
|
||||||
|
"integrity": "sha512-6Fv1DV/TYw//QF5IzQdqsNDjx/wc8TrMBZsqjL9eW01tWb7R7k/mq+/VXfJCl7SoD5emsJop9cOByJZfs8hYIw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"mdn-data": "2.0.30",
|
||||||
|
"source-map-js": "^1.0.1"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": "^10 || ^12.20.0 || ^14.13.0 || >=15.0.0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/css-what": {
|
"node_modules/css-what": {
|
||||||
"version": "6.2.2",
|
"version": "6.2.2",
|
||||||
"resolved": "https://registry.npmjs.org/css-what/-/css-what-6.2.2.tgz",
|
"resolved": "https://registry.npmjs.org/css-what/-/css-what-6.2.2.tgz",
|
||||||
@@ -3264,6 +3702,42 @@
|
|||||||
"node": ">=4"
|
"node": ">=4"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/csso": {
|
||||||
|
"version": "5.0.5",
|
||||||
|
"resolved": "https://registry.npmjs.org/csso/-/csso-5.0.5.tgz",
|
||||||
|
"integrity": "sha512-0LrrStPOdJj+SPCCrGhzryycLjwcgUSHBtxNA8aIDxf0GLsRh1cKYhB00Gd1lDOS4yGH69+SNn13+TWbVHETFQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"css-tree": "~2.2.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": "^10 || ^12.20.0 || ^14.13.0 || >=15.0.0",
|
||||||
|
"npm": ">=7.0.0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/csso/node_modules/css-tree": {
|
||||||
|
"version": "2.2.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/css-tree/-/css-tree-2.2.1.tgz",
|
||||||
|
"integrity": "sha512-OA0mILzGc1kCOCSJerOeqDxDQ4HOh+G8NbOJFOTgOCzpw7fCBubk0fEyxp8AgOL/jvLgYA/uV0cMbe43ElF1JA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"mdn-data": "2.0.28",
|
||||||
|
"source-map-js": "^1.0.1"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": "^10 || ^12.20.0 || ^14.13.0 || >=15.0.0",
|
||||||
|
"npm": ">=7.0.0"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/csso/node_modules/mdn-data": {
|
||||||
|
"version": "2.0.28",
|
||||||
|
"resolved": "https://registry.npmjs.org/mdn-data/-/mdn-data-2.0.28.tgz",
|
||||||
|
"integrity": "sha512-aylIc7Z9y4yzHYAJNuESG3hfhC+0Ibp/MAMiaOZgNv4pmEdFyfZhhhny4MNiAfWdBQ1RQ2mfDWmM1x8SvGyp8g==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "CC0-1.0"
|
||||||
|
},
|
||||||
"node_modules/debug": {
|
"node_modules/debug": {
|
||||||
"version": "2.6.9",
|
"version": "2.6.9",
|
||||||
"resolved": "https://registry.npmjs.org/debug/-/debug-2.6.9.tgz",
|
"resolved": "https://registry.npmjs.org/debug/-/debug-2.6.9.tgz",
|
||||||
@@ -3274,6 +3748,16 @@
|
|||||||
"ms": "2.0.0"
|
"ms": "2.0.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/deepmerge": {
|
||||||
|
"version": "4.3.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-4.3.1.tgz",
|
||||||
|
"integrity": "sha512-3sUqbMEc77XqpdNO7FRyRog+eW3ph+GYCbj+rK+uYyRMuwsVy0rMiVtPn+QJlKFvWP/1PYpapqYn0Me2knFn+A==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=0.10.0"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/default-browser": {
|
"node_modules/default-browser": {
|
||||||
"version": "5.2.1",
|
"version": "5.2.1",
|
||||||
"resolved": "https://registry.npmjs.org/default-browser/-/default-browser-5.2.1.tgz",
|
"resolved": "https://registry.npmjs.org/default-browser/-/default-browser-5.2.1.tgz",
|
||||||
@@ -3535,6 +4019,16 @@
|
|||||||
"node": ">=4"
|
"node": ">=4"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/error-ex": {
|
||||||
|
"version": "1.3.4",
|
||||||
|
"resolved": "https://registry.npmjs.org/error-ex/-/error-ex-1.3.4.tgz",
|
||||||
|
"integrity": "sha512-sqQamAnR14VgCr1A618A3sGrygcpK+HEbenA/HiEAkkUwcZIIB/tgWqHFxWgOyDh4nB4JCRimh79dR5Ywc9MDQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"is-arrayish": "^0.2.1"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/es-define-property": {
|
"node_modules/es-define-property": {
|
||||||
"version": "1.0.1",
|
"version": "1.0.1",
|
||||||
"resolved": "https://registry.npmjs.org/es-define-property/-/es-define-property-1.0.1.tgz",
|
"resolved": "https://registry.npmjs.org/es-define-property/-/es-define-property-1.0.1.tgz",
|
||||||
@@ -4447,6 +4941,33 @@
|
|||||||
"postcss": "^8.1.0"
|
"postcss": "^8.1.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/import-fresh": {
|
||||||
|
"version": "3.3.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/import-fresh/-/import-fresh-3.3.1.tgz",
|
||||||
|
"integrity": "sha512-TR3KfrTZTYLPB6jUjfx6MF9WcWrHL9su5TObK4ZkYgBdWKPOFoSoQIdEuTuR82pmtxH2spWG9h6etwfr1pLBqQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"parent-module": "^1.0.0",
|
||||||
|
"resolve-from": "^4.0.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/sponsors/sindresorhus"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/import-fresh/node_modules/resolve-from": {
|
||||||
|
"version": "4.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/resolve-from/-/resolve-from-4.0.0.tgz",
|
||||||
|
"integrity": "sha512-pb/MYmXstAkysRFx8piNI1tGFNQIFA3vkE3Gq4EuA1dF6gHp/+vgZqsCGJapvy8N3Q+4o7FwvquPJcnZ7RYy4g==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=4"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/import-local": {
|
"node_modules/import-local": {
|
||||||
"version": "3.2.0",
|
"version": "3.2.0",
|
||||||
"resolved": "https://registry.npmjs.org/import-local/-/import-local-3.2.0.tgz",
|
"resolved": "https://registry.npmjs.org/import-local/-/import-local-3.2.0.tgz",
|
||||||
@@ -4494,6 +5015,13 @@
|
|||||||
"node": ">= 10"
|
"node": ">= 10"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/is-arrayish": {
|
||||||
|
"version": "0.2.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/is-arrayish/-/is-arrayish-0.2.1.tgz",
|
||||||
|
"integrity": "sha512-zz06S8t0ozoDXMG+ube26zeCTNXcKIPJZJi8hBrF4idCLms4CG9QtK7qBl1boi5ODzFpjswb5JPmHCbMpjaYzg==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT"
|
||||||
|
},
|
||||||
"node_modules/is-binary-path": {
|
"node_modules/is-binary-path": {
|
||||||
"version": "2.1.0",
|
"version": "2.1.0",
|
||||||
"resolved": "https://registry.npmjs.org/is-binary-path/-/is-binary-path-2.1.0.tgz",
|
"resolved": "https://registry.npmjs.org/is-binary-path/-/is-binary-path-2.1.0.tgz",
|
||||||
@@ -4691,6 +5219,19 @@
|
|||||||
"integrity": "sha512-RdJUflcE3cUzKiMqQgsCu06FPu9UdIJO0beYbPhHN4k6apgJtifcoCtT9bcxOpYBtpD2kCM6Sbzg4CausW/PKQ==",
|
"integrity": "sha512-RdJUflcE3cUzKiMqQgsCu06FPu9UdIJO0beYbPhHN4k6apgJtifcoCtT9bcxOpYBtpD2kCM6Sbzg4CausW/PKQ==",
|
||||||
"license": "MIT"
|
"license": "MIT"
|
||||||
},
|
},
|
||||||
|
"node_modules/js-yaml": {
|
||||||
|
"version": "4.1.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/js-yaml/-/js-yaml-4.1.0.tgz",
|
||||||
|
"integrity": "sha512-wpxZs9NoxZaJESJGIZTyDEaYpl0FKSA+FB9aJiyemKhMwkxQg63h4T1KJgUGHpTqPDNRcmmYLugrRjJlBtWvRA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"argparse": "^2.0.1"
|
||||||
|
},
|
||||||
|
"bin": {
|
||||||
|
"js-yaml": "bin/js-yaml.js"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/jsesc": {
|
"node_modules/jsesc": {
|
||||||
"version": "3.1.0",
|
"version": "3.1.0",
|
||||||
"resolved": "https://registry.npmjs.org/jsesc/-/jsesc-3.1.0.tgz",
|
"resolved": "https://registry.npmjs.org/jsesc/-/jsesc-3.1.0.tgz",
|
||||||
@@ -4752,6 +5293,13 @@
|
|||||||
"shell-quote": "^1.8.3"
|
"shell-quote": "^1.8.3"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/lines-and-columns": {
|
||||||
|
"version": "1.2.4",
|
||||||
|
"resolved": "https://registry.npmjs.org/lines-and-columns/-/lines-and-columns-1.2.4.tgz",
|
||||||
|
"integrity": "sha512-7ylylesZQ/PV29jhEDl3Ufjo6ZX7gCqJr5F7PKrqc93v7fzSymt1BpwEU8nAUXs8qzzvqhbjhK5QZg6Mt/HkBg==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT"
|
||||||
|
},
|
||||||
"node_modules/loader-runner": {
|
"node_modules/loader-runner": {
|
||||||
"version": "4.3.1",
|
"version": "4.3.1",
|
||||||
"resolved": "https://registry.npmjs.org/loader-runner/-/loader-runner-4.3.1.tgz",
|
"resolved": "https://registry.npmjs.org/loader-runner/-/loader-runner-4.3.1.tgz",
|
||||||
@@ -4834,6 +5382,13 @@
|
|||||||
"node": ">= 0.4"
|
"node": ">= 0.4"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/mdn-data": {
|
||||||
|
"version": "2.0.30",
|
||||||
|
"resolved": "https://registry.npmjs.org/mdn-data/-/mdn-data-2.0.30.tgz",
|
||||||
|
"integrity": "sha512-GaqWWShW4kv/G9IEucWScBx9G1/vsFZZJUO+tD26M8J8z3Kw5RDQjaoZe03YAClgeS/SWPOcb4nkFBTEi5DUEA==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "CC0-1.0"
|
||||||
|
},
|
||||||
"node_modules/media-typer": {
|
"node_modules/media-typer": {
|
||||||
"version": "0.3.0",
|
"version": "0.3.0",
|
||||||
"resolved": "https://registry.npmjs.org/media-typer/-/media-typer-0.3.0.tgz",
|
"resolved": "https://registry.npmjs.org/media-typer/-/media-typer-0.3.0.tgz",
|
||||||
@@ -5183,6 +5738,38 @@
|
|||||||
"tslib": "^2.0.3"
|
"tslib": "^2.0.3"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/parent-module": {
|
||||||
|
"version": "1.0.1",
|
||||||
|
"resolved": "https://registry.npmjs.org/parent-module/-/parent-module-1.0.1.tgz",
|
||||||
|
"integrity": "sha512-GQ2EWRpQV8/o+Aw8YqtfZZPfNRWZYkbidE9k5rpl/hC3vtHHBfGm2Ifi6qWV+coDGkrUKZAxE3Lot5kcsRlh+g==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"callsites": "^3.0.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=6"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/parse-json": {
|
||||||
|
"version": "5.2.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/parse-json/-/parse-json-5.2.0.tgz",
|
||||||
|
"integrity": "sha512-ayCKvm/phCGxOkYRSCM82iDwct8/EonSEgCSxWxD7ve6jHggsFl4fZVQBPRNgQoKiuV/odhFrGzQXZwbifC8Rg==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@babel/code-frame": "^7.0.0",
|
||||||
|
"error-ex": "^1.3.1",
|
||||||
|
"json-parse-even-better-errors": "^2.3.0",
|
||||||
|
"lines-and-columns": "^1.1.6"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=8"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/sponsors/sindresorhus"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/parseurl": {
|
"node_modules/parseurl": {
|
||||||
"version": "1.3.3",
|
"version": "1.3.3",
|
||||||
"resolved": "https://registry.npmjs.org/parseurl/-/parseurl-1.3.3.tgz",
|
"resolved": "https://registry.npmjs.org/parseurl/-/parseurl-1.3.3.tgz",
|
||||||
@@ -5238,6 +5825,16 @@
|
|||||||
"dev": true,
|
"dev": true,
|
||||||
"license": "MIT"
|
"license": "MIT"
|
||||||
},
|
},
|
||||||
|
"node_modules/path-type": {
|
||||||
|
"version": "4.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/path-type/-/path-type-4.0.0.tgz",
|
||||||
|
"integrity": "sha512-gDKb8aZMDeD/tZWs9P6+q0J9Mwkdl6xMV8TjnGP3qJVJ06bdMgkbBlLU8IdfOsIsFz2BW1rNVT3XuNEl8zPAvw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=8"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/picocolors": {
|
"node_modules/picocolors": {
|
||||||
"version": "1.1.1",
|
"version": "1.1.1",
|
||||||
"resolved": "https://registry.npmjs.org/picocolors/-/picocolors-1.1.1.tgz",
|
"resolved": "https://registry.npmjs.org/picocolors/-/picocolors-1.1.1.tgz",
|
||||||
@@ -6117,6 +6714,17 @@
|
|||||||
"url": "https://github.com/sponsors/ljharb"
|
"url": "https://github.com/sponsors/ljharb"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/snake-case": {
|
||||||
|
"version": "3.0.4",
|
||||||
|
"resolved": "https://registry.npmjs.org/snake-case/-/snake-case-3.0.4.tgz",
|
||||||
|
"integrity": "sha512-LAOh4z89bGQvl9pFfNF8V146i7o7/CqFPbqzYgP+yYzDIDeS9HaNFtXABamRW+AQzEVODcvE79ljJ+8a9YSdMg==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"dot-case": "^3.0.4",
|
||||||
|
"tslib": "^2.0.3"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/sockjs": {
|
"node_modules/sockjs": {
|
||||||
"version": "0.3.24",
|
"version": "0.3.24",
|
||||||
"resolved": "https://registry.npmjs.org/sockjs/-/sockjs-0.3.24.tgz",
|
"resolved": "https://registry.npmjs.org/sockjs/-/sockjs-0.3.24.tgz",
|
||||||
@@ -6321,6 +6929,125 @@
|
|||||||
"url": "https://github.com/sponsors/ljharb"
|
"url": "https://github.com/sponsors/ljharb"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"node_modules/svg-parser": {
|
||||||
|
"version": "2.0.4",
|
||||||
|
"resolved": "https://registry.npmjs.org/svg-parser/-/svg-parser-2.0.4.tgz",
|
||||||
|
"integrity": "sha512-e4hG1hRwoOdRb37cIMSgzNsxyzKfayW6VOflrwvR+/bzrkyxY/31WkbgnQpgtrNp1SdpJvpUAGTa/ZoiPNDuRQ==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT"
|
||||||
|
},
|
||||||
|
"node_modules/svgo": {
|
||||||
|
"version": "3.3.2",
|
||||||
|
"resolved": "https://registry.npmjs.org/svgo/-/svgo-3.3.2.tgz",
|
||||||
|
"integrity": "sha512-OoohrmuUlBs8B8o6MB2Aevn+pRIH9zDALSR+6hhqVfa6fRwG/Qw9VUMSMW9VNg2CFc/MTIfabtdOVl9ODIJjpw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"@trysound/sax": "0.2.0",
|
||||||
|
"commander": "^7.2.0",
|
||||||
|
"css-select": "^5.1.0",
|
||||||
|
"css-tree": "^2.3.1",
|
||||||
|
"css-what": "^6.1.0",
|
||||||
|
"csso": "^5.0.5",
|
||||||
|
"picocolors": "^1.0.0"
|
||||||
|
},
|
||||||
|
"bin": {
|
||||||
|
"svgo": "bin/svgo"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14.0.0"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"type": "opencollective",
|
||||||
|
"url": "https://opencollective.com/svgo"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/svgo/node_modules/commander": {
|
||||||
|
"version": "7.2.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/commander/-/commander-7.2.0.tgz",
|
||||||
|
"integrity": "sha512-QrWXB+ZQSVPmIWIhtEO9H+gwHaMGYiF5ChvoJ+K9ZGHG/sVsa6yiesAD1GC/x46sET00Xlwo1u49RVVVzvcSkw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"engines": {
|
||||||
|
"node": ">= 10"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/svgo/node_modules/css-select": {
|
||||||
|
"version": "5.2.2",
|
||||||
|
"resolved": "https://registry.npmjs.org/css-select/-/css-select-5.2.2.tgz",
|
||||||
|
"integrity": "sha512-TizTzUddG/xYLA3NXodFM0fSbNizXjOKhqiQQwvhlspadZokn1KDy0NZFS0wuEubIYAV5/c1/lAr0TaaFXEXzw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "BSD-2-Clause",
|
||||||
|
"dependencies": {
|
||||||
|
"boolbase": "^1.0.0",
|
||||||
|
"css-what": "^6.1.0",
|
||||||
|
"domhandler": "^5.0.2",
|
||||||
|
"domutils": "^3.0.1",
|
||||||
|
"nth-check": "^2.0.1"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/sponsors/fb55"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/svgo/node_modules/dom-serializer": {
|
||||||
|
"version": "2.0.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/dom-serializer/-/dom-serializer-2.0.0.tgz",
|
||||||
|
"integrity": "sha512-wIkAryiqt/nV5EQKqQpo3SToSOV9J0DnbJqwK7Wv/Trc92zIAYZ4FlMu+JPFW1DfGFt81ZTCGgDEabffXeLyJg==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"domelementtype": "^2.3.0",
|
||||||
|
"domhandler": "^5.0.2",
|
||||||
|
"entities": "^4.2.0"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/cheeriojs/dom-serializer?sponsor=1"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/svgo/node_modules/domhandler": {
|
||||||
|
"version": "5.0.3",
|
||||||
|
"resolved": "https://registry.npmjs.org/domhandler/-/domhandler-5.0.3.tgz",
|
||||||
|
"integrity": "sha512-cgwlv/1iFQiFnU96XXgROh8xTeetsnJiDsTc7TYCLFd9+/WNkIqPTxiM/8pSd8VIrhXGTf1Ny1q1hquVqDJB5w==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "BSD-2-Clause",
|
||||||
|
"dependencies": {
|
||||||
|
"domelementtype": "^2.3.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">= 4"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/fb55/domhandler?sponsor=1"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/svgo/node_modules/domutils": {
|
||||||
|
"version": "3.2.2",
|
||||||
|
"resolved": "https://registry.npmjs.org/domutils/-/domutils-3.2.2.tgz",
|
||||||
|
"integrity": "sha512-6kZKyUajlDuqlHKVX1w7gyslj9MPIXzIFiz/rGu35uC1wMi+kMhQwGhl4lt9unC9Vb9INnY9Z3/ZA3+FhASLaw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "BSD-2-Clause",
|
||||||
|
"dependencies": {
|
||||||
|
"dom-serializer": "^2.0.0",
|
||||||
|
"domelementtype": "^2.3.0",
|
||||||
|
"domhandler": "^5.0.3"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/fb55/domutils?sponsor=1"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/svgo/node_modules/entities": {
|
||||||
|
"version": "4.5.0",
|
||||||
|
"resolved": "https://registry.npmjs.org/entities/-/entities-4.5.0.tgz",
|
||||||
|
"integrity": "sha512-V0hjH4dGPh9Ao5p0MoRY6BVqtwCjhz6vI5LT8AJ55H+4g9/4vbHx1I54fS0XuclLhDHArPQCiMjDxjaL8fPxhw==",
|
||||||
|
"dev": true,
|
||||||
|
"license": "BSD-2-Clause",
|
||||||
|
"engines": {
|
||||||
|
"node": ">=0.12"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/fb55/entities?sponsor=1"
|
||||||
|
}
|
||||||
|
},
|
||||||
"node_modules/tapable": {
|
"node_modules/tapable": {
|
||||||
"version": "2.3.0",
|
"version": "2.3.0",
|
||||||
"resolved": "https://registry.npmjs.org/tapable/-/tapable-2.3.0.tgz",
|
"resolved": "https://registry.npmjs.org/tapable/-/tapable-2.3.0.tgz",
|
||||||
|
|||||||
@@ -26,6 +26,7 @@
|
|||||||
"style-loader": "4.0.0",
|
"style-loader": "4.0.0",
|
||||||
"webpack": "5.95.0",
|
"webpack": "5.95.0",
|
||||||
"webpack-cli": "5.1.4",
|
"webpack-cli": "5.1.4",
|
||||||
"webpack-dev-server": "5.1.0"
|
"webpack-dev-server": "5.1.0",
|
||||||
|
"@svgr/webpack": "8.1.0"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,12 @@
|
|||||||
|
import React from 'react';
|
||||||
|
|
||||||
|
export const AppShellContext = React.createContext(null);
|
||||||
|
|
||||||
|
export const useAppShell = () => {
|
||||||
|
const context = React.useContext(AppShellContext);
|
||||||
|
if (!context) {
|
||||||
|
throw new Error('AppShellContext not found. Ensure routes are nested under AppLayout.');
|
||||||
|
}
|
||||||
|
return context;
|
||||||
|
};
|
||||||
|
|
||||||
+214
-21
@@ -17,7 +17,116 @@ export const getAssetFromVersion = (currentVersion, assetType) => {
|
|||||||
return getAssetFromGroup(currentVersion.assets, assetType);
|
return getAssetFromGroup(currentVersion.assets, assetType);
|
||||||
};
|
};
|
||||||
|
|
||||||
export const resolveDocumentAssetUrl = (doc, type, { ensureAssetUrl, getAsset, ensureOptions } = {}) => {
|
const normalizeAssetObjects = (objects) => {
|
||||||
|
if (!Array.isArray(objects)) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
return objects
|
||||||
|
.filter((entry) => Number.isInteger(entry?.ordinal))
|
||||||
|
.slice()
|
||||||
|
.sort((a, b) => a.ordinal - b.ordinal);
|
||||||
|
};
|
||||||
|
|
||||||
|
const mergeAssetObjects = (existingObjects, incomingObjects) => {
|
||||||
|
const merged = new Map();
|
||||||
|
|
||||||
|
normalizeAssetObjects(existingObjects).forEach((entry) => {
|
||||||
|
merged.set(entry.ordinal, { ...entry });
|
||||||
|
});
|
||||||
|
|
||||||
|
normalizeAssetObjects(incomingObjects).forEach((entry) => {
|
||||||
|
const current = merged.get(entry.ordinal) || {};
|
||||||
|
merged.set(entry.ordinal, { ...current, ...entry });
|
||||||
|
});
|
||||||
|
|
||||||
|
return [...merged.entries()]
|
||||||
|
.sort((a, b) => a[0] - b[0])
|
||||||
|
.map(([, value]) => value);
|
||||||
|
};
|
||||||
|
|
||||||
|
export class AssetView {
|
||||||
|
constructor(asset) {
|
||||||
|
this.asset = asset || null;
|
||||||
|
this._objectsRef = null;
|
||||||
|
this._sortedObjects = [];
|
||||||
|
}
|
||||||
|
|
||||||
|
getCardinality() {
|
||||||
|
if (!this.asset) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
const reported = Number(this.asset.cardinality);
|
||||||
|
if (Number.isFinite(reported) && reported > 0) {
|
||||||
|
return reported;
|
||||||
|
}
|
||||||
|
|
||||||
|
const objectsCount = this.getObjects().length;
|
||||||
|
if (objectsCount > 0) {
|
||||||
|
return objectsCount;
|
||||||
|
}
|
||||||
|
|
||||||
|
return this.asset.metadata ? 1 : 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
getObjects() {
|
||||||
|
if (!this.asset || !Array.isArray(this.asset.objects) || this.asset.objects.length === 0) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this._objectsRef === this.asset.objects) {
|
||||||
|
return this._sortedObjects;
|
||||||
|
}
|
||||||
|
|
||||||
|
this._objectsRef = this.asset.objects;
|
||||||
|
this._sortedObjects = normalizeAssetObjects(this.asset.objects);
|
||||||
|
return this._sortedObjects;
|
||||||
|
}
|
||||||
|
|
||||||
|
getObject(ordinal = 1) {
|
||||||
|
const fromObjects = this.getObjects().find((entry) => entry.ordinal === ordinal);
|
||||||
|
if (fromObjects) {
|
||||||
|
return fromObjects;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ordinal === 1 && this.asset) {
|
||||||
|
if (this.asset.url || this.asset.metadata) {
|
||||||
|
return {
|
||||||
|
ordinal: 1,
|
||||||
|
url: this.asset.url || null,
|
||||||
|
metadata: this.asset.metadata || null,
|
||||||
|
expires_at: this.asset.expiresAt ?? null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
getPrimaryObject() {
|
||||||
|
return this.getObject(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
getPrimaryMetadata() {
|
||||||
|
return this.getPrimaryObject()?.metadata || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
getPrimaryUrl() {
|
||||||
|
return this.getPrimaryObject()?.url || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
hasObject(ordinal) {
|
||||||
|
return Boolean(this.getObject(ordinal));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const createAssetView = (asset) => new AssetView(asset);
|
||||||
|
|
||||||
|
export const resolveDocumentAssetUrl = (
|
||||||
|
doc,
|
||||||
|
type,
|
||||||
|
{ ensureAssetUrl, getAsset, ensureOptions, objectOrdinal = 1 } = {},
|
||||||
|
) => {
|
||||||
if (!doc || !type) {
|
if (!doc || !type) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
@@ -25,15 +134,32 @@ export const resolveDocumentAssetUrl = (doc, type, { ensureAssetUrl, getAsset, e
|
|||||||
if (!asset) {
|
if (!asset) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
const view = createAssetView(asset);
|
||||||
|
const object = view.getObject(objectOrdinal);
|
||||||
|
const url = object?.url || (objectOrdinal === 1 ? view.getPrimaryUrl() : null);
|
||||||
|
const expiresAt = typeof object?.expires_at === 'number'
|
||||||
|
? object.expires_at
|
||||||
|
: objectOrdinal === 1 && typeof asset.expiresAt === 'number'
|
||||||
|
? asset.expiresAt
|
||||||
|
: null;
|
||||||
const now = Date.now();
|
const now = Date.now();
|
||||||
const expiresAt = typeof asset.expiresAt === 'number' ? asset.expiresAt : null;
|
if (url && (!expiresAt || expiresAt > now)) {
|
||||||
const hasFreshUrl = asset.url && (!expiresAt || expiresAt > now);
|
return url;
|
||||||
if (hasFreshUrl) {
|
|
||||||
return asset.url;
|
|
||||||
}
|
}
|
||||||
if (doc.id && asset.id && typeof ensureAssetUrl === 'function') {
|
if (doc.id && asset.id && typeof ensureAssetUrl === 'function') {
|
||||||
const force = Boolean(asset.url && expiresAt && expiresAt <= now);
|
const force = Boolean(url && expiresAt && expiresAt <= now);
|
||||||
const options = ensureOptions ? { ...ensureOptions, force } : { force };
|
const options = {
|
||||||
|
force,
|
||||||
|
start: objectOrdinal,
|
||||||
|
limit: 1,
|
||||||
|
...(ensureOptions || {}),
|
||||||
|
};
|
||||||
|
if (!options.start) {
|
||||||
|
options.start = objectOrdinal;
|
||||||
|
}
|
||||||
|
if (!options.limit) {
|
||||||
|
options.limit = 1;
|
||||||
|
}
|
||||||
ensureAssetUrl(doc.id, asset, options).catch(() => {});
|
ensureAssetUrl(doc.id, asset, options).catch(() => {});
|
||||||
}
|
}
|
||||||
return null;
|
return null;
|
||||||
@@ -63,6 +189,10 @@ class AssetManager {
|
|||||||
}
|
}
|
||||||
const cached = this.assetCache.get(asset.id);
|
const cached = this.assetCache.get(asset.id);
|
||||||
if (!cached) {
|
if (!cached) {
|
||||||
|
const normalized = mergeAssetObjects(null, asset.objects);
|
||||||
|
if (normalized.length) {
|
||||||
|
return { ...asset, objects: normalized };
|
||||||
|
}
|
||||||
return asset;
|
return asset;
|
||||||
}
|
}
|
||||||
const merged = { ...cached, ...asset };
|
const merged = { ...cached, ...asset };
|
||||||
@@ -76,6 +206,10 @@ class AssetManager {
|
|||||||
merged.expiresAt = cachedExpires;
|
merged.expiresAt = cachedExpires;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
const mergedObjects = mergeAssetObjects(cached.objects, asset.objects);
|
||||||
|
if (mergedObjects.length) {
|
||||||
|
merged.objects = mergedObjects;
|
||||||
|
}
|
||||||
return merged;
|
return merged;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -172,24 +306,53 @@ class AssetManager {
|
|||||||
return next;
|
return next;
|
||||||
}
|
}
|
||||||
|
|
||||||
ensureAsset(documentId, asset, { force = false } = {}) {
|
ensureAsset(documentId, asset, { force = false, start = null, limit = null } = {}) {
|
||||||
if (!documentId || !asset?.id) {
|
if (!documentId || !asset?.id) {
|
||||||
return Promise.resolve(asset || null);
|
return Promise.resolve(asset || null);
|
||||||
}
|
}
|
||||||
|
|
||||||
const assetExpiresAt = typeof asset.expiresAt === 'number' ? asset.expiresAt : null;
|
const requestedStart = Number.isInteger(start) && start > 0 ? start : 1;
|
||||||
if (!force && asset?.url && (!assetExpiresAt || assetExpiresAt > Date.now())) {
|
const requestedLimit = Number.isInteger(limit) && limit > 0 ? limit : 1;
|
||||||
this.rememberAsset(asset);
|
const requestedEnd = requestedStart + requestedLimit - 1;
|
||||||
return Promise.resolve(asset);
|
|
||||||
}
|
|
||||||
|
|
||||||
const cached = this.assetCache.get(asset.id);
|
const baseAsset = this.assetCache.get(asset.id) || asset;
|
||||||
|
const view = createAssetView(baseAsset);
|
||||||
|
const assetExpiresAt = typeof baseAsset.expiresAt === 'number' ? baseAsset.expiresAt : null;
|
||||||
const now = Date.now();
|
const now = Date.now();
|
||||||
if (!force && cached && cached.expiresAt && cached.expiresAt > now && cached.url) {
|
|
||||||
return Promise.resolve({ ...asset, ...cached });
|
const isOrdinalSatisfied = (ordinal) => {
|
||||||
|
const object = view.getObject(ordinal);
|
||||||
|
if (!object) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (!object.url) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (typeof object.expires_at === 'number') {
|
||||||
|
return object.expires_at > now;
|
||||||
|
}
|
||||||
|
if (ordinal === 1 && baseAsset.url && (!assetExpiresAt || assetExpiresAt > now)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
};
|
||||||
|
|
||||||
|
let needsFetch = force;
|
||||||
|
if (!needsFetch) {
|
||||||
|
for (let ordinal = requestedStart; ordinal <= requestedEnd; ordinal += 1) {
|
||||||
|
if (!isOrdinalSatisfied(ordinal)) {
|
||||||
|
needsFetch = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const inflightKey = `${documentId}:${asset.id}`;
|
if (!needsFetch) {
|
||||||
|
this.rememberAsset(baseAsset);
|
||||||
|
return Promise.resolve(baseAsset);
|
||||||
|
}
|
||||||
|
|
||||||
|
const inflightKey = `${documentId}:${asset.id}:${start ?? 'd'}:${limit ?? 'd'}`;
|
||||||
if (!force && this.assetInflight.has(inflightKey)) {
|
if (!force && this.assetInflight.has(inflightKey)) {
|
||||||
return this.assetInflight.get(inflightKey);
|
return this.assetInflight.get(inflightKey);
|
||||||
}
|
}
|
||||||
@@ -198,14 +361,44 @@ class AssetManager {
|
|||||||
return Promise.reject(new Error('AssetManager API client is not configured.'));
|
return Promise.reject(new Error('AssetManager API client is not configured.'));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const params = {};
|
||||||
|
if (Number.isInteger(start) && start > 0) {
|
||||||
|
params.start = start;
|
||||||
|
}
|
||||||
|
if (Number.isInteger(limit) && limit > 0) {
|
||||||
|
params.limit = limit;
|
||||||
|
}
|
||||||
|
|
||||||
|
const requestConfig = Object.keys(params).length ? { params } : undefined;
|
||||||
|
|
||||||
const request = this.api
|
const request = this.api
|
||||||
.get(`/documents/${documentId}/assets/${asset.id}`)
|
.get(`/assets/${asset.id}`, requestConfig)
|
||||||
.then(({ data }) => {
|
.then(({ data }) => {
|
||||||
|
const incomingObjects = Array.isArray(data.objects) ? data.objects : [];
|
||||||
|
const cachedEntry = this.assetCache.get(asset.id) || baseAsset;
|
||||||
|
const mergedObjects = mergeAssetObjects(cachedEntry?.objects, incomingObjects);
|
||||||
|
const combined = { ...cachedEntry, ...asset, ...data, objects: mergedObjects };
|
||||||
|
const view = createAssetView(combined);
|
||||||
|
const primaryObject = view.getPrimaryObject();
|
||||||
|
const expiresAt = typeof primaryObject?.expires_at === 'number'
|
||||||
|
? primaryObject.expires_at
|
||||||
|
: Date.now() + this.assetPresignTtlMs;
|
||||||
|
const cardinality = (() => {
|
||||||
|
const reported = Number(data.cardinality ?? asset.cardinality ?? cachedEntry?.cardinality);
|
||||||
|
const objectsCount = mergedObjects.length;
|
||||||
|
if (Number.isFinite(reported) && reported > 0) {
|
||||||
|
return Math.max(reported, objectsCount) || null;
|
||||||
|
}
|
||||||
|
return objectsCount || null;
|
||||||
|
})();
|
||||||
|
|
||||||
const entry = {
|
const entry = {
|
||||||
...asset,
|
...combined,
|
||||||
...data,
|
cardinality,
|
||||||
expiresAt: Date.now() + this.assetPresignTtlMs,
|
url: view.getPrimaryUrl(),
|
||||||
|
expiresAt,
|
||||||
};
|
};
|
||||||
|
|
||||||
this.rememberAsset(entry);
|
this.rememberAsset(entry);
|
||||||
return entry;
|
return entry;
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
<svg height="128px" viewBox="0 0 128 128" width="128px" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
|
<svg viewBox="0 0 128 128" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
|
||||||
<linearGradient id="a" gradientTransform="matrix(0.45451 0 0 0.455522 -1210.292114 616.172607)" gradientUnits="userSpaceOnUse" x1="2689.251953" x2="2918.069824" y1="-1106.802979" y2="-1106.802979">
|
<linearGradient id="a" gradientTransform="matrix(0.45451 0 0 0.455522 -1210.292114 616.172607)" gradientUnits="userSpaceOnUse" x1="2689.251953" x2="2918.069824" y1="-1106.802979" y2="-1106.802979">
|
||||||
<stop offset="0" stop-color="#62a0ea"/>
|
<stop offset="0" stop-color="#62a0ea"/>
|
||||||
<stop offset="0.0576991" stop-color="#afd4ff"/>
|
<stop offset="0.0576991" stop-color="#afd4ff"/>
|
||||||
|
Before Width: | Height: | Size: 1.9 KiB After Width: | Height: | Size: 1.8 KiB |
@@ -0,0 +1,19 @@
|
|||||||
|
<svg xmlns="http://www.w3.org/2000/svg" width="100%" height="100%" viewBox="0 0 128 128" xml:space="preserve" style="fill-rule:evenodd;clip-rule:evenodd;stroke-linejoin:round;stroke-miterlimit:2;">
|
||||||
|
<defs>
|
||||||
|
<filter id="cornerShadow" x="-20%" y="-20%" width="200%" height="200%">
|
||||||
|
<feDropShadow dx="-4" dy="4" stdDeviation="8" flood-opacity="0.24"/>
|
||||||
|
</filter>
|
||||||
|
<linearGradient id="cornerGradient" gradientUnits="userSpaceOnUse" x1="32" y1="0" x2="0" y2="64">
|
||||||
|
<stop offset="0%" stop-color="#ffffff"/>
|
||||||
|
<stop offset="100%" stop-color="#f3f3f3"/>
|
||||||
|
</linearGradient>
|
||||||
|
</defs>
|
||||||
|
<g filter="url(#cornerShadow)">
|
||||||
|
<g transform="matrix(-1.81626,-1.81626,0.460023,-0.460023,180.663,203.387)">
|
||||||
|
<path d="M70.488,10.11L89.954,86.966L51.022,86.966L70.488,10.11Z" fill="url(#cornerGradient)"/>
|
||||||
|
</g>
|
||||||
|
<g transform="matrix(1.81626,1.81626,-0.460023,0.460023,4.6259,-132.676)">
|
||||||
|
<path d="M70.488,10.11L89.954,86.966L51.022,86.966L70.488,10.11Z" fill="#fdfdfd"/>
|
||||||
|
</g>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
After Width: | Height: | Size: 1.1 KiB |
@@ -1,15 +1,25 @@
|
|||||||
import React, { useCallback, useEffect, useMemo, useState } from 'react';
|
import React, { useCallback, useEffect, useMemo, useState } from 'react';
|
||||||
import { createPortal } from 'react-dom';
|
import { createPortal } from 'react-dom';
|
||||||
import { DownloadIcon, EditIcon } from '../ui/icons';
|
import { DownloadIcon, EditIcon, ArrowLeftIcon, ArrowRightIcon } from '../ui/icons';
|
||||||
import { getTagColorStyle } from '../utils/colors';
|
import { getTagColorStyle } from '../utils/colors';
|
||||||
import { formatFileSize } from '../utils/format';
|
import { formatFileSize } from '../utils/format';
|
||||||
import { resolveDocumentAssetUrl } from '../asset_manager';
|
import { resolveDocumentAssetUrl, createAssetView } from '../asset_manager';
|
||||||
|
import { useAssetNavigator } from '../hooks/useAssetNavigator';
|
||||||
import { CORRESPONDENT_ROLES } from '../constants/correspondents';
|
import { CORRESPONDENT_ROLES } from '../constants/correspondents';
|
||||||
|
|
||||||
const MAX_PREVIEW_STACK_ITEMS = 15;
|
const MAX_PREVIEW_STACK_ITEMS = 15;
|
||||||
|
|
||||||
const normalizeRole = (role) => (role || '').toLowerCase();
|
const normalizeRole = (role) => (role || '').toLowerCase();
|
||||||
|
|
||||||
|
const derivePreviewOrientation = (metadata) => {
|
||||||
|
const width = Number(metadata?.width);
|
||||||
|
const height = Number(metadata?.height);
|
||||||
|
if (Number.isFinite(width) && Number.isFinite(height) && width > 0 && height > 0) {
|
||||||
|
return width >= height ? 'landscape' : 'portrait';
|
||||||
|
}
|
||||||
|
return 'landscape';
|
||||||
|
};
|
||||||
|
|
||||||
const formatRoleLabel = (role) => {
|
const formatRoleLabel = (role) => {
|
||||||
const normalized = normalizeRole(role);
|
const normalized = normalizeRole(role);
|
||||||
if (!normalized) return 'Other';
|
if (!normalized) return 'Other';
|
||||||
@@ -435,20 +445,34 @@ const DetailPanel = ({
|
|||||||
setOcrOpen(false);
|
setOcrOpen(false);
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
|
const singlePreviewNavigator = useAssetNavigator({
|
||||||
|
document: singleDoc,
|
||||||
|
assetType: 'preview',
|
||||||
|
ensureAssetUrl,
|
||||||
|
getAsset: getDocumentAsset,
|
||||||
|
prefetch: 3,
|
||||||
|
});
|
||||||
|
|
||||||
const makePreviewItem = useCallback(
|
const makePreviewItem = useCallback(
|
||||||
(doc) => {
|
(doc, ordinal = 1) => {
|
||||||
if (!doc) return null;
|
if (!doc) return null;
|
||||||
const url = resolveDocumentAssetUrl(doc, 'preview', {
|
const asset = getDocumentAsset(doc, 'preview');
|
||||||
ensureAssetUrl,
|
const assetView = createAssetView(asset);
|
||||||
getAsset: getDocumentAsset,
|
const object = assetView.getObject(ordinal);
|
||||||
});
|
let url = object?.url || null;
|
||||||
|
if (!url) {
|
||||||
|
url = resolveDocumentAssetUrl(doc, 'preview', {
|
||||||
|
ensureAssetUrl,
|
||||||
|
getAsset: getDocumentAsset,
|
||||||
|
ensureOptions: { start: ordinal, limit: 1 },
|
||||||
|
objectOrdinal: ordinal,
|
||||||
|
});
|
||||||
|
}
|
||||||
if (!url) {
|
if (!url) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
const asset = getDocumentAsset(doc, 'preview');
|
const metadata = object?.metadata || assetView.getPrimaryMetadata() || {};
|
||||||
const width = Number(asset?.metadata?.width) || 0;
|
const orientation = derivePreviewOrientation(metadata);
|
||||||
const height = Number(asset?.metadata?.height) || 0;
|
|
||||||
const orientation = width > 0 && height > 0 ? (width >= height ? 'landscape' : 'portrait') : 'landscape';
|
|
||||||
return {
|
return {
|
||||||
id: doc.id,
|
id: doc.id,
|
||||||
url,
|
url,
|
||||||
@@ -475,32 +499,74 @@ const DetailPanel = ({
|
|||||||
return ordered;
|
return ordered;
|
||||||
}, [selectedDocuments]);
|
}, [selectedDocuments]);
|
||||||
|
|
||||||
|
const stackTopDocument = stackDocuments[0] || null;
|
||||||
|
const stackPreviewNavigator = useAssetNavigator({
|
||||||
|
document: stackTopDocument,
|
||||||
|
assetType: 'preview',
|
||||||
|
ensureAssetUrl,
|
||||||
|
getAsset: getDocumentAsset,
|
||||||
|
prefetch: 3,
|
||||||
|
});
|
||||||
|
|
||||||
const singlePreviewItems = useMemo(() => {
|
const singlePreviewItems = useMemo(() => {
|
||||||
if (!singleDoc) return [];
|
if (!singleDoc) return [];
|
||||||
const item = makePreviewItem(singleDoc);
|
const url = singlePreviewNavigator.currentUrl;
|
||||||
return item ? [item] : [];
|
if (!url) {
|
||||||
}, [singleDoc, makePreviewItem]);
|
return [];
|
||||||
|
|
||||||
const stackPreviews = useMemo(
|
|
||||||
() =>
|
|
||||||
stackDocuments
|
|
||||||
.map((doc) => makePreviewItem(doc))
|
|
||||||
.filter(Boolean),
|
|
||||||
[stackDocuments, makePreviewItem],
|
|
||||||
);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
if (!ensureAssetUrl) {
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
|
const orientation = derivePreviewOrientation(singlePreviewNavigator.currentMetadata);
|
||||||
|
return [
|
||||||
|
{
|
||||||
|
id: singleDoc.id,
|
||||||
|
url,
|
||||||
|
orientation,
|
||||||
|
alt: singleDoc.title || singleDoc.original_name || 'Document preview',
|
||||||
|
},
|
||||||
|
];
|
||||||
|
}, [singleDoc, singlePreviewNavigator.currentUrl, singlePreviewNavigator.currentMetadata]);
|
||||||
|
|
||||||
stackDocuments.forEach((doc) => {
|
const stackPreviews = useMemo(() => {
|
||||||
resolveDocumentAssetUrl(doc, 'preview', {
|
if (!stackDocuments.length) {
|
||||||
ensureAssetUrl,
|
return [];
|
||||||
getAsset: getDocumentAsset,
|
}
|
||||||
});
|
return stackDocuments
|
||||||
});
|
.map((doc) => {
|
||||||
}, [stackDocuments, ensureAssetUrl, getDocumentAsset]);
|
if (!doc) return null;
|
||||||
|
if (stackTopDocument && doc.id === stackTopDocument.id) {
|
||||||
|
const url = stackPreviewNavigator.currentUrl;
|
||||||
|
if (!url) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const orientation = derivePreviewOrientation(stackPreviewNavigator.currentMetadata);
|
||||||
|
return {
|
||||||
|
id: doc.id,
|
||||||
|
url,
|
||||||
|
orientation,
|
||||||
|
alt: doc.title || doc.original_name || 'Document preview',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
return makePreviewItem(doc, 1);
|
||||||
|
})
|
||||||
|
.filter(Boolean);
|
||||||
|
}, [
|
||||||
|
stackDocuments,
|
||||||
|
stackTopDocument,
|
||||||
|
stackPreviewNavigator.currentUrl,
|
||||||
|
stackPreviewNavigator.currentMetadata,
|
||||||
|
makePreviewItem,
|
||||||
|
]);
|
||||||
|
|
||||||
|
const singleCardinality = singlePreviewNavigator.cardinality;
|
||||||
|
const singleEffectiveCardinality = singleCardinality || (singlePreviewNavigator.currentUrl ? 1 : 0);
|
||||||
|
const singleHasPreview = Boolean(singlePreviewNavigator.currentUrl);
|
||||||
|
|
||||||
|
const topDocId = stackTopDocument?.id || null;
|
||||||
|
const topCardinality = stackPreviewNavigator.cardinality;
|
||||||
|
const topEffectiveCardinality = topCardinality || (stackPreviewNavigator.currentUrl ? 1 : 0);
|
||||||
|
const topHasPreview = Boolean(stackPreviewNavigator.currentUrl);
|
||||||
|
const topOrdinal = stackPreviewNavigator.ordinal;
|
||||||
|
const topCanGoPrev = stackPreviewNavigator.canGoPrev;
|
||||||
|
const topCanGoNext = stackPreviewNavigator.canGoNext;
|
||||||
|
|
||||||
const bulkTagUnion = useMemo(() => {
|
const bulkTagUnion = useMemo(() => {
|
||||||
if (!selectedDocuments.length) return [];
|
if (!selectedDocuments.length) return [];
|
||||||
@@ -649,21 +715,74 @@ const DetailPanel = ({
|
|||||||
? new Date(singleDoc.issued_at).toLocaleString()
|
? new Date(singleDoc.issued_at).toLocaleString()
|
||||||
: '—';
|
: '—';
|
||||||
const tagsForDoc = Array.isArray(singleDoc.tags) ? singleDoc.tags : [];
|
const tagsForDoc = Array.isArray(singleDoc.tags) ? singleDoc.tags : [];
|
||||||
|
const pageCountRaw = singleDoc.current_version?.metadata?.page_count;
|
||||||
|
const pageCountValue =
|
||||||
|
typeof pageCountRaw === 'number'
|
||||||
|
? pageCountRaw
|
||||||
|
: pageCountRaw != null && pageCountRaw !== ''
|
||||||
|
? Number.parseInt(pageCountRaw, 10)
|
||||||
|
: null;
|
||||||
|
const hasPageCount = Number.isFinite(pageCountValue) && pageCountValue >= 0;
|
||||||
const metadata =
|
const metadata =
|
||||||
singleDoc.metadata && Object.keys(singleDoc.metadata).length > 0 ? singleDoc.metadata : null;
|
singleDoc.metadata && Object.keys(singleDoc.metadata).length > 0 ? singleDoc.metadata : null;
|
||||||
|
const effectiveCardinality = singleEffectiveCardinality;
|
||||||
|
const canGoPrev = singlePreviewNavigator.canGoPrev;
|
||||||
|
const canGoNext = singlePreviewNavigator.canGoNext;
|
||||||
|
const hasPreviewImage = singleHasPreview;
|
||||||
|
const interceptNavPointer = (event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<>
|
<>
|
||||||
<div>
|
<div className="preview-pane preview-pane--stack">
|
||||||
<div className="preview-pane preview-pane--stack">
|
<PreviewStack
|
||||||
<PreviewStack
|
items={singlePreviewItems}
|
||||||
items={singlePreviewItems}
|
maxItems={1}
|
||||||
maxItems={1}
|
emptyMessage="Preview loading…"
|
||||||
emptyMessage="Preview loading…"
|
onItemActivate={handlePreviewActivate}
|
||||||
onItemActivate={handlePreviewActivate}
|
onOpenPreview={onOpenPreview}
|
||||||
onOpenPreview={onOpenPreview}
|
activeItemId={activePreviewId}
|
||||||
activeItemId={activePreviewId}
|
/>
|
||||||
/>
|
{hasPreviewImage && (effectiveCardinality > 1 || canGoPrev || canGoNext) ? (
|
||||||
</div>
|
<div className="preview-pane__nav preview-pane__nav--overlay">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className="preview-pane__nav-button preview-pane__nav-button--prev"
|
||||||
|
onClick={(event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
singlePreviewNavigator.goPrev();
|
||||||
|
}}
|
||||||
|
onPointerDown={interceptNavPointer}
|
||||||
|
onPointerUp={interceptNavPointer}
|
||||||
|
onMouseDown={interceptNavPointer}
|
||||||
|
onMouseUp={interceptNavPointer}
|
||||||
|
disabled={!canGoPrev}
|
||||||
|
aria-label="Previous preview"
|
||||||
|
>
|
||||||
|
<ArrowLeftIcon />
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className="preview-pane__nav-button preview-pane__nav-button--next"
|
||||||
|
onClick={(event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
singlePreviewNavigator.goNext();
|
||||||
|
}}
|
||||||
|
onPointerDown={interceptNavPointer}
|
||||||
|
onPointerUp={interceptNavPointer}
|
||||||
|
onMouseDown={interceptNavPointer}
|
||||||
|
onMouseUp={interceptNavPointer}
|
||||||
|
disabled={!canGoNext}
|
||||||
|
aria-label="Next preview"
|
||||||
|
>
|
||||||
|
<ArrowRightIcon />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
</div>
|
</div>
|
||||||
<div className="doc-title-row">
|
<div className="doc-title-row">
|
||||||
{isEditingTitle ? (
|
{isEditingTitle ? (
|
||||||
@@ -728,6 +847,11 @@ const DetailPanel = ({
|
|||||||
<div>
|
<div>
|
||||||
<strong>Issued:</strong> {issuedAt}
|
<strong>Issued:</strong> {issuedAt}
|
||||||
</div>
|
</div>
|
||||||
|
{hasPageCount ? (
|
||||||
|
<div>
|
||||||
|
<strong>Pages:</strong> {pageCountValue}
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
<div>
|
<div>
|
||||||
<strong>Original filename:</strong>{' '}
|
<strong>Original filename:</strong>{' '}
|
||||||
{singleDoc.original_name}
|
{singleDoc.original_name}
|
||||||
@@ -858,19 +982,66 @@ const DetailPanel = ({
|
|||||||
const renderBulk = () => {
|
const renderBulk = () => {
|
||||||
const countLabel = `${selectedCount} document${selectedCount === 1 ? '' : 's'}`;
|
const countLabel = `${selectedCount} document${selectedCount === 1 ? '' : 's'}`;
|
||||||
const sizeLabel = stackTotalSizeBytes ? formatFileSize(stackTotalSizeBytes) : '—';
|
const sizeLabel = stackTotalSizeBytes ? formatFileSize(stackTotalSizeBytes) : '—';
|
||||||
|
const topDoc = stackTopDocument;
|
||||||
|
const topDocIdLocal = topDocId;
|
||||||
|
const topOrdinal = stackPreviewNavigator.ordinal;
|
||||||
|
const topCardinalityLocal = topEffectiveCardinality;
|
||||||
|
const topHasPreview = Boolean(stackPreviewNavigator.currentUrl);
|
||||||
|
const topCanGoPrev = stackPreviewNavigator.canGoPrev;
|
||||||
|
const topCanGoNext = stackPreviewNavigator.canGoNext;
|
||||||
|
const interceptTopNavPointer = (event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<>
|
<>
|
||||||
<div>
|
<div className="preview-pane preview-pane--stack">
|
||||||
<div className="preview-pane preview-pane--stack">
|
<PreviewStack
|
||||||
<PreviewStack
|
items={stackPreviews}
|
||||||
items={stackPreviews}
|
emptyMessage="No previews available."
|
||||||
emptyMessage="No previews available."
|
onItemActivate={handlePreviewActivate}
|
||||||
onItemActivate={handlePreviewActivate}
|
onOpenPreview={onOpenPreview}
|
||||||
onOpenPreview={onOpenPreview}
|
activeItemId={activePreviewId}
|
||||||
activeItemId={activePreviewId}
|
/>
|
||||||
/>
|
{topDocIdLocal && topHasPreview && (topCardinalityLocal > 1 || topCanGoPrev || topCanGoNext) ? (
|
||||||
</div>
|
<div className="preview-pane__nav preview-pane__nav--overlay">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className="preview-pane__nav-button preview-pane__nav-button--prev"
|
||||||
|
onClick={(event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
stackPreviewNavigator.goPrev();
|
||||||
|
}}
|
||||||
|
onPointerDown={interceptTopNavPointer}
|
||||||
|
onPointerUp={interceptTopNavPointer}
|
||||||
|
onMouseDown={interceptTopNavPointer}
|
||||||
|
onMouseUp={interceptTopNavPointer}
|
||||||
|
disabled={!topCanGoPrev}
|
||||||
|
aria-label="Previous preview"
|
||||||
|
>
|
||||||
|
<ArrowLeftIcon />
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className="preview-pane__nav-button preview-pane__nav-button--next"
|
||||||
|
onClick={(event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
stackPreviewNavigator.goNext();
|
||||||
|
}}
|
||||||
|
onPointerDown={interceptTopNavPointer}
|
||||||
|
onPointerUp={interceptTopNavPointer}
|
||||||
|
onMouseDown={interceptTopNavPointer}
|
||||||
|
onMouseUp={interceptTopNavPointer}
|
||||||
|
disabled={!topCanGoNext}
|
||||||
|
aria-label="Next preview"
|
||||||
|
>
|
||||||
|
<ArrowRightIcon />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
</div>
|
</div>
|
||||||
<h3 style={{ margin: 0 }}>{countLabel}</h3>
|
<h3 style={{ margin: 0 }}>{countLabel}</h3>
|
||||||
<div className="meta">
|
<div className="meta">
|
||||||
|
|||||||
@@ -1,13 +1,50 @@
|
|||||||
import React, { useCallback, useEffect, useMemo, useRef } from 'react';
|
import React, { useCallback, useEffect, useMemo, useRef } from 'react';
|
||||||
import { resolveDocumentAssetUrl } from '../asset_manager';
|
import { getAssetFromVersion, resolveDocumentAssetUrl, createAssetView } from '../asset_manager';
|
||||||
import { getTagColorStyle } from '../utils/colors';
|
import { getTagColorStyle } from '../utils/colors';
|
||||||
import { DownloadIcon, EditIcon, ViewListIcon, ViewGridIcon } from '../ui/icons';
|
import { DownloadIcon, EditIcon, ViewListIcon, ViewGridIcon, FolderIcon } from '../ui/icons';
|
||||||
|
|
||||||
const FOLDER_ICON_SRC = '/folder.svg';
|
|
||||||
|
|
||||||
const TAG_MIME_TYPES = ['application/x-papercrate-tag', 'text/papercrate-tag'];
|
const TAG_MIME_TYPES = ['application/x-papercrate-tag', 'text/papercrate-tag'];
|
||||||
|
const DEFAULT_GRID_ICON_SIZE = 96;
|
||||||
|
const DEFAULT_GRID_TITLE_SIZE = '11px';
|
||||||
|
const LIST_ICON_SIZE = 48;
|
||||||
|
|
||||||
const DocumentThumbnailImage = ({ document, ensureAssetUrl, getDocumentAsset, alt }) => {
|
const getPageCount = (doc) =>
|
||||||
|
Number.isFinite(doc?.current_version?.metadata?.page_count)
|
||||||
|
? doc.current_version.metadata.page_count
|
||||||
|
: null;
|
||||||
|
|
||||||
|
const DocumentThumbnailImage = ({
|
||||||
|
document,
|
||||||
|
ensureAssetUrl,
|
||||||
|
getDocumentAsset,
|
||||||
|
alt,
|
||||||
|
maxSize = LIST_ICON_SIZE,
|
||||||
|
}) => {
|
||||||
|
const resolvedMaxSize = Math.max(1, Math.round(maxSize || 1));
|
||||||
|
const thumbnailAsset = useMemo(
|
||||||
|
() => getAssetFromVersion(document?.current_version, 'thumbnail'),
|
||||||
|
[document?.current_version],
|
||||||
|
);
|
||||||
|
const thumbnailView = useMemo(() => createAssetView(thumbnailAsset), [thumbnailAsset]);
|
||||||
|
const primaryMetadata = thumbnailView.getPrimaryMetadata() || {};
|
||||||
|
const assetWidth = Number(primaryMetadata?.width);
|
||||||
|
const assetHeight = Number(primaryMetadata?.height);
|
||||||
|
|
||||||
|
const dimensions = useMemo(() => {
|
||||||
|
if (!Number.isFinite(assetWidth) || assetWidth <= 0 || !Number.isFinite(assetHeight) || assetHeight <= 0) {
|
||||||
|
return { width: resolvedMaxSize, height: resolvedMaxSize };
|
||||||
|
}
|
||||||
|
const scale = Math.min(1, resolvedMaxSize / assetWidth, resolvedMaxSize / assetHeight);
|
||||||
|
return {
|
||||||
|
width: Math.max(1, Math.round(assetWidth * scale)),
|
||||||
|
height: Math.max(1, Math.round(assetHeight * scale)),
|
||||||
|
};
|
||||||
|
}, [assetWidth, assetHeight, resolvedMaxSize]);
|
||||||
|
|
||||||
|
const innerStyle = useMemo(
|
||||||
|
() => ({ width: `${dimensions.width}px`, height: `${dimensions.height}px` }),
|
||||||
|
[dimensions.height, dimensions.width],
|
||||||
|
);
|
||||||
const url = useMemo(
|
const url = useMemo(
|
||||||
() =>
|
() =>
|
||||||
resolveDocumentAssetUrl(document, 'thumbnail', {
|
resolveDocumentAssetUrl(document, 'thumbnail', {
|
||||||
@@ -17,19 +54,28 @@ const DocumentThumbnailImage = ({ document, ensureAssetUrl, getDocumentAsset, al
|
|||||||
[document, ensureAssetUrl, getDocumentAsset],
|
[document, ensureAssetUrl, getDocumentAsset],
|
||||||
);
|
);
|
||||||
|
|
||||||
|
const pageCount = getPageCount(document);
|
||||||
|
const showMultiPageBadge = Number.isFinite(pageCount) && pageCount > 1;
|
||||||
|
const innerClasses = ['document-thumbnail-inner'];
|
||||||
|
if (showMultiPageBadge) {
|
||||||
|
innerClasses.push('document-thumbnail-inner--multipage');
|
||||||
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="document-thumbnail-wrapper">
|
<div className="document-thumbnail-wrapper">
|
||||||
{url ? (
|
<div className={innerClasses.join(' ')} style={innerStyle}>
|
||||||
<img
|
{url ? (
|
||||||
src={url}
|
<img
|
||||||
alt={alt || ''}
|
src={url}
|
||||||
className="document-thumbnail"
|
alt={alt || ''}
|
||||||
draggable={false}
|
className="document-thumbnail"
|
||||||
onDragStart={(event) => event.preventDefault()}
|
draggable={false}
|
||||||
/>
|
onDragStart={(event) => event.preventDefault()}
|
||||||
) : (
|
/>
|
||||||
<div className="thumb-placeholder">DOC</div>
|
) : (
|
||||||
)}
|
<div className="thumb-placeholder">DOC</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
@@ -97,6 +143,7 @@ const DocumentsTable = ({
|
|||||||
);
|
);
|
||||||
const scrollRef = useRef(null);
|
const scrollRef = useRef(null);
|
||||||
const isGridView = viewMode === 'grid';
|
const isGridView = viewMode === 'grid';
|
||||||
|
const gridIconSize = DEFAULT_GRID_ICON_SIZE;
|
||||||
const handleSetViewMode = useCallback(
|
const handleSetViewMode = useCallback(
|
||||||
(nextMode) => {
|
(nextMode) => {
|
||||||
if (!onViewModeChange) {
|
if (!onViewModeChange) {
|
||||||
@@ -226,6 +273,13 @@ const DocumentsTable = ({
|
|||||||
[onClearSelection],
|
[onClearSelection],
|
||||||
);
|
);
|
||||||
|
|
||||||
|
const showDefaultEmptyState = !showingSearchResults && !subfolders.length && rows.length === 0;
|
||||||
|
const showListSearchEmptyState =
|
||||||
|
showingSearchResults && rows.length === 0 && !isGridView && !isSearchLoading;
|
||||||
|
const showGridSearchEmptyState =
|
||||||
|
isGridView && showingSearchResults && rows.length === 0 && !isSearchLoading;
|
||||||
|
const showSearchHint = showingSearchResults && rows.length > 0;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<section
|
<section
|
||||||
className={`documents-panel column documents-panel--view-${isGridView ? 'grid' : 'list'}`}
|
className={`documents-panel column documents-panel--view-${isGridView ? 'grid' : 'list'}`}
|
||||||
@@ -295,11 +349,20 @@ const DocumentsTable = ({
|
|||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
{showDefaultEmptyState && (
|
||||||
|
<div className="empty-state">
|
||||||
|
Drop files anywhere or onto a folder to upload documents.
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
{showGridSearchEmptyState && (
|
||||||
|
<div className="empty-state empty-state--global">
|
||||||
|
No documents match the current filters.
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
<div className="column-body">
|
<div className="column-body">
|
||||||
<div
|
<div
|
||||||
ref={scrollRef}
|
ref={scrollRef}
|
||||||
className="documents-scroll"
|
className="documents-scroll"
|
||||||
tabIndex={0}
|
|
||||||
onFocus={(event) => {
|
onFocus={(event) => {
|
||||||
if (event.target === scrollRef.current) {
|
if (event.target === scrollRef.current) {
|
||||||
onDocumentListFocus?.();
|
onDocumentListFocus?.();
|
||||||
@@ -315,15 +378,12 @@ const DocumentsTable = ({
|
|||||||
}}
|
}}
|
||||||
aria-activedescendant={isGridView ? undefined : activeDescendantId}
|
aria-activedescendant={isGridView ? undefined : activeDescendantId}
|
||||||
>
|
>
|
||||||
{!showingSearchResults && !subfolders.length && rows.length === 0 ? (
|
{showDefaultEmptyState ? null : isGridView ? (
|
||||||
<div className="empty-state">
|
|
||||||
Drop files anywhere or onto a folder to upload documents.
|
|
||||||
</div>
|
|
||||||
) : isGridView ? (
|
|
||||||
<div
|
<div
|
||||||
className="documents-grid"
|
className="documents-grid"
|
||||||
role="list"
|
role="list"
|
||||||
onClick={handleGridBackgroundClick}
|
onClick={handleGridBackgroundClick}
|
||||||
|
style={{ '--documents-grid-icon-size': `${gridIconSize}px` }}
|
||||||
>
|
>
|
||||||
{!showingSearchResults &&
|
{!showingSearchResults &&
|
||||||
subfolders.map((folder) => {
|
subfolders.map((folder) => {
|
||||||
@@ -370,10 +430,9 @@ const DocumentsTable = ({
|
|||||||
}}
|
}}
|
||||||
>
|
>
|
||||||
<div className="folder-card__icon">
|
<div className="folder-card__icon">
|
||||||
<img
|
<FolderIcon
|
||||||
src={FOLDER_ICON_SRC}
|
|
||||||
alt="Folder"
|
|
||||||
className="folder-card__icon-svg"
|
className="folder-card__icon-svg"
|
||||||
|
size={gridIconSize}
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
<div className="folder-card__meta">
|
<div className="folder-card__meta">
|
||||||
@@ -399,6 +458,7 @@ const DocumentsTable = ({
|
|||||||
className={cardClasses.join(' ')}
|
className={cardClasses.join(' ')}
|
||||||
role="listitem"
|
role="listitem"
|
||||||
id={`document-card-${doc.id}`}
|
id={`document-card-${doc.id}`}
|
||||||
|
data-doc-id={doc.id}
|
||||||
onClick={(event) => onDocumentRowClick(doc.id, event)}
|
onClick={(event) => onDocumentRowClick(doc.id, event)}
|
||||||
onDoubleClick={() => onDocumentOpen(doc.id)}
|
onDoubleClick={() => onDocumentOpen(doc.id)}
|
||||||
draggable
|
draggable
|
||||||
@@ -416,6 +476,7 @@ const DocumentsTable = ({
|
|||||||
ensureAssetUrl={ensureAssetUrl}
|
ensureAssetUrl={ensureAssetUrl}
|
||||||
getDocumentAsset={getDocumentAsset}
|
getDocumentAsset={getDocumentAsset}
|
||||||
alt={`Thumbnail for ${doc.title || doc.original_name}`}
|
alt={`Thumbnail for ${doc.title || doc.original_name}`}
|
||||||
|
maxSize={gridIconSize}
|
||||||
/>
|
/>
|
||||||
<div className="document-card__meta">
|
<div className="document-card__meta">
|
||||||
<div
|
<div
|
||||||
@@ -442,7 +503,6 @@ const DocumentsTable = ({
|
|||||||
}
|
}
|
||||||
}}
|
}}
|
||||||
role="button"
|
role="button"
|
||||||
tabIndex={0}
|
|
||||||
draggable
|
draggable
|
||||||
onDragStart={(event) => {
|
onDragStart={(event) => {
|
||||||
event.stopPropagation();
|
event.stopPropagation();
|
||||||
@@ -550,10 +610,9 @@ const DocumentsTable = ({
|
|||||||
>
|
>
|
||||||
<td className="thumb-cell">
|
<td className="thumb-cell">
|
||||||
<div className="thumb-icon">
|
<div className="thumb-icon">
|
||||||
<img
|
<FolderIcon
|
||||||
src={FOLDER_ICON_SRC}
|
|
||||||
alt="Folder"
|
|
||||||
className="thumb-icon__image"
|
className="thumb-icon__image"
|
||||||
|
size={32}
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
</td>
|
</td>
|
||||||
@@ -615,6 +674,7 @@ const DocumentsTable = ({
|
|||||||
key={doc.id}
|
key={doc.id}
|
||||||
className={rowClasses.join(' ')}
|
className={rowClasses.join(' ')}
|
||||||
id={`document-row-${doc.id}`}
|
id={`document-row-${doc.id}`}
|
||||||
|
data-doc-id={doc.id}
|
||||||
onClick={(event) => onDocumentRowClick(doc.id, event)}
|
onClick={(event) => onDocumentRowClick(doc.id, event)}
|
||||||
onDoubleClick={() => onDocumentOpen(doc.id)}
|
onDoubleClick={() => onDocumentOpen(doc.id)}
|
||||||
draggable
|
draggable
|
||||||
@@ -679,7 +739,6 @@ const DocumentsTable = ({
|
|||||||
}
|
}
|
||||||
}}
|
}}
|
||||||
role="button"
|
role="button"
|
||||||
tabIndex={0}
|
|
||||||
draggable
|
draggable
|
||||||
onDragStart={(event) => {
|
onDragStart={(event) => {
|
||||||
event.stopPropagation();
|
event.stopPropagation();
|
||||||
@@ -764,23 +823,18 @@ const DocumentsTable = ({
|
|||||||
</table>
|
</table>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
{rows.length === 0 && showingSearchResults && !isGridView && !isSearchLoading && (
|
{showListSearchEmptyState && (
|
||||||
<div className="empty-state">No documents match the current filters.</div>
|
<div className="empty-state">No documents match the current filters.</div>
|
||||||
)}
|
)}
|
||||||
{showingSearchResults && rows.length > 0 && (
|
{showSearchHint && (
|
||||||
<div className="search-hint">
|
<div className="search-hint">
|
||||||
Showing {rows.length} document{rows.length === 1 ? '' : 's'} in {currentFolderName} and subfolders.
|
Showing {rows.length} document{rows.length === 1 ? '' : 's'} in {currentFolderName} and subfolders.
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
{isGridView && showingSearchResults && rows.length === 0 && !isSearchLoading && (
|
|
||||||
<div className="empty-state empty-state--global">
|
|
||||||
No documents match the current filters.
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</section>
|
</section>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|
||||||
export default DocumentsTable;
|
export default DocumentsTable;
|
||||||
export { DocumentThumbnailImage };
|
export { DocumentThumbnailImage };
|
||||||
|
|||||||
@@ -0,0 +1,133 @@
|
|||||||
|
import { useCallback, useEffect, useMemo, useState } from 'react';
|
||||||
|
import { createAssetView } from '../asset_manager';
|
||||||
|
|
||||||
|
const clampOrdinalValue = (value, cardinality, defaultOrdinal) => {
|
||||||
|
const raw = Number.isFinite(value) ? value : defaultOrdinal;
|
||||||
|
let next = Math.max(1, Math.floor(raw));
|
||||||
|
if (cardinality && cardinality > 0) {
|
||||||
|
next = Math.min(next, cardinality);
|
||||||
|
}
|
||||||
|
return next;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const useAssetNavigator = ({
|
||||||
|
document,
|
||||||
|
assetType,
|
||||||
|
ensureAssetUrl,
|
||||||
|
getAsset,
|
||||||
|
prefetch = 2,
|
||||||
|
defaultOrdinal = 1,
|
||||||
|
}) => {
|
||||||
|
const documentId = document?.id || null;
|
||||||
|
|
||||||
|
const asset = useMemo(() => {
|
||||||
|
if (!document || typeof getAsset !== 'function') {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return getAsset(document, assetType);
|
||||||
|
}, [document, assetType, getAsset]);
|
||||||
|
|
||||||
|
const view = useMemo(() => createAssetView(asset), [asset]);
|
||||||
|
const cardinality = view.getCardinality();
|
||||||
|
|
||||||
|
const [ordinal, setOrdinalInternal] = useState(defaultOrdinal);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
setOrdinalInternal(defaultOrdinal);
|
||||||
|
}, [documentId, assetType, defaultOrdinal]);
|
||||||
|
|
||||||
|
const setOrdinal = useCallback(
|
||||||
|
(next) => {
|
||||||
|
setOrdinalInternal((prev) => {
|
||||||
|
const target = typeof next === 'function' ? next(prev) : next;
|
||||||
|
return clampOrdinalValue(target, cardinality, defaultOrdinal);
|
||||||
|
});
|
||||||
|
},
|
||||||
|
[cardinality, defaultOrdinal],
|
||||||
|
);
|
||||||
|
|
||||||
|
const goPrev = useCallback(() => setOrdinal((value) => value - 1), [setOrdinal]);
|
||||||
|
const goNext = useCallback(() => setOrdinal((value) => value + 1), [setOrdinal]);
|
||||||
|
|
||||||
|
const objects = view.getObjects();
|
||||||
|
const currentObject = view.getObject(ordinal);
|
||||||
|
const currentUrl = currentObject?.url || view.getPrimaryUrl();
|
||||||
|
const currentMetadata = currentObject?.metadata || view.getPrimaryMetadata() || null;
|
||||||
|
|
||||||
|
const canGoPrev = ordinal > 1;
|
||||||
|
const canGoNext = cardinality ? ordinal < cardinality : true;
|
||||||
|
|
||||||
|
const ordinalsNeedingLoad = useMemo(() => {
|
||||||
|
const missing = [];
|
||||||
|
if (!asset) {
|
||||||
|
return missing;
|
||||||
|
}
|
||||||
|
const maxOrdinal = cardinality && cardinality > 0
|
||||||
|
? Math.min(cardinality, ordinal + Math.max(1, prefetch) - 1)
|
||||||
|
: ordinal + Math.max(1, prefetch) - 1;
|
||||||
|
|
||||||
|
for (let ord = ordinal; ord <= maxOrdinal; ord += 1) {
|
||||||
|
const object = view.getObject(ord);
|
||||||
|
if (!object?.url) {
|
||||||
|
missing.push(ord);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return missing;
|
||||||
|
}, [asset, view, ordinal, prefetch, cardinality]);
|
||||||
|
|
||||||
|
const fetchStart = ordinalsNeedingLoad.length ? ordinalsNeedingLoad[0] : null;
|
||||||
|
const fetchEnd = ordinalsNeedingLoad.length ? ordinalsNeedingLoad[ordinalsNeedingLoad.length - 1] : null;
|
||||||
|
const fetchLimit = fetchStart && fetchEnd ? fetchEnd - fetchStart + 1 : null;
|
||||||
|
|
||||||
|
const [loading, setLoading] = useState(false);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (!documentId || !asset || !ensureAssetUrl) {
|
||||||
|
setLoading(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!fetchStart || !fetchLimit) {
|
||||||
|
setLoading(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let cancelled = false;
|
||||||
|
setLoading(true);
|
||||||
|
ensureAssetUrl(documentId, asset, {
|
||||||
|
start: fetchStart,
|
||||||
|
limit: fetchLimit,
|
||||||
|
})
|
||||||
|
.catch(() => {})
|
||||||
|
.finally(() => {
|
||||||
|
if (!cancelled) {
|
||||||
|
setLoading(false);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
cancelled = true;
|
||||||
|
};
|
||||||
|
}, [documentId, asset, ensureAssetUrl, fetchStart, fetchLimit]);
|
||||||
|
|
||||||
|
return {
|
||||||
|
document,
|
||||||
|
documentId,
|
||||||
|
asset,
|
||||||
|
assetType,
|
||||||
|
ordinal,
|
||||||
|
setOrdinal,
|
||||||
|
goPrev,
|
||||||
|
goNext,
|
||||||
|
canGoPrev,
|
||||||
|
canGoNext,
|
||||||
|
cardinality,
|
||||||
|
currentObject,
|
||||||
|
currentUrl,
|
||||||
|
currentMetadata,
|
||||||
|
objects,
|
||||||
|
isLoading: loading,
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
export default useAssetNavigator;
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user