From 9b976a6c4add3719b8f9afef53578ff778b50aa1 Mon Sep 17 00:00:00 2001
From: tv <tv@krebsco.de>
Date: Sun, 1 Nov 2026 22:29:54 +0100
Subject: api: add support for counters

---
 src/api/store.rs | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

(limited to 'src/api/store.rs')

diff --git a/src/api/store.rs b/src/api/store.rs
index 8b09487..7039008 100644
--- a/src/api/store.rs
+++ b/src/api/store.rs
@@ -77,6 +77,13 @@ macro_rules! read_at {
     };
 }
 
+#[derive(Debug, Deserialize)]
+pub(crate) struct PostQuery {
+    /// Adds this to the counter at the path instead of inserting the body
+    /// (`store::increment_path`).
+    increment: Option<i64>,
+}
+
 #[derive(Debug, Deserialize)]
 pub(crate) struct DeleteQuery {
     #[serde(default)]
@@ -355,11 +362,14 @@ fn body_value(headers: &HeaderMap, body: &Bytes) -> Result<JsonValue, StatusCode
 /// `POST /store/{key}`: inserts the body into a list, `key` ending in the index it
 /// goes to (`store::insert_path`: the items from there on move up; the length, or `-1`,
 /// appends); `201` with where it went (`{"index": n}`). Not a list there, or an index
-/// past its end: `409`, saying which.
+/// past its end: `409`, saying which. With `?increment=n`, adds `n` to the counter at
+/// `key` instead (`store::increment_path`), answering with its value then; not a
+/// counter there: `409`.
 pub(crate) async fn post_store_entry(
     State(state): State<AppState>,
     Extension(user): Extension<AuthenticatedUser>,
     Path(key): Path<String>,
+    Query(query): Query<PostQuery>,
     RawQuery(raw_query): RawQuery,
     headers: HeaderMap,
     body: Bytes,
@@ -368,9 +378,22 @@ pub(crate) async fn post_store_entry(
         return Err(StatusCode::SERVICE_UNAVAILABLE.into());
     }
     let path = parse_write_path(&state, &user, &key, "POST")?;
-    if let Some(redirect) = to_positive(&state, &path, true, None, raw_query.as_deref()).await? {
+    // An increment names an item (`-1`: the last); an insert where it goes (`-1`: the end).
+    let insertion = query.increment.is_none();
+    if let Some(redirect) =
+        to_positive(&state, &path, insertion, None, raw_query.as_deref()).await?
+    {
         return Ok(redirect);
     }
+    if let Some(by) = query.increment {
+        let value = state
+            .mutate_doc_for(Some(user.0.clone()), move |doc| {
+                store::increment_path(doc, &path, by)
+            })
+            .await
+            .map_err(|e| store_status(e, "Failed to increment a counter"))?;
+        return response_for_value(&JsonValue::from(value), headers.get(ACCEPT));
+    }
     let index = path.last().and_then(|leaf| store::list_index(leaf));
     let value = body_value(&headers, &body)?;
     let inserted = state
-- 
cgit v1.3.1