1 files changed, 45 insertions, 0 deletions
diff --git a/crates/turtle/src/atuin_server/handlers/status.rs b/crates/turtle/src/atuin_server/handlers/status.rs
new file mode 100644
index 00000000..0cf2ca1e
--- /dev/null
+++ b/crates/turtle/src/atuin_server/handlers/status.rs
@@ -0,0 +1,45 @@
+use axum::{Json, extract::State, http::StatusCode};
+use tracing::instrument;
+
+use super::{ErrorResponse, ErrorResponseStatus, RespExt};
+use crate::atuin_server::router::{AppState, UserAuth};
+use crate::atuin_server_database::Database;
+
+use crate::atuin_common::api::*;
+
+const VERSION: &str = env!("CARGO_PKG_VERSION");
+
+#[instrument(skip_all, fields(user.id = user.id))]
+pub async fn status<DB: Database>(
+    UserAuth(user): UserAuth,
+    state: State<AppState<DB>>,
+) -> Result<Json<StatusResponse>, ErrorResponseStatus<'static>> {
+    let db = &state.0.database;
+
+    let deleted = db.deleted_history(&user).await.unwrap_or(vec![]);
+
+    let count = match db.count_history_cached(&user).await {
+        // By default read out the cached value
+        Ok(count) => count,
+
+        // If that fails, fallback on a full COUNT. Cache is built on a POST
+        // only
+        Err(_) => match db.count_history(&user).await {
+            Ok(count) => count,
+            Err(_) => {
+                return Err(ErrorResponse::reply("failed to query history count")
+                    .with_status(StatusCode::INTERNAL_SERVER_ERROR));
+            }
+        },
+    };
+
+    tracing::debug!(user = user.username, "requested sync status");
+
+    Ok(Json(StatusResponse {
+        count,
+        deleted,
+        username: user.username,
+        version: VERSION.to_string(),
+        page_size: state.settings.page_size,
+    }))
+}