Make more public export endpoints for /set_expiration and /cancel (#37232)

Will be handy for self-hosted. At least gives people something to
curl.

GitOrigin-RevId: 6184ba01e9023b0a052e7940bb30aee815c4035d

Author: nipunn1313

crates/common/src/types/mod.rs

@@ -25,6 +25,7 @@ mod functions;
 mod index;
 mod maybe_value;
 mod object_key;
+mod snapshot_export;
 mod table;
 mod timestamp;
 
@@ -87,6 +88,7 @@ pub use object_key::{
     FullyQualifiedObjectKey,
     ObjectKey,
 };
+pub use snapshot_export::SetExportExpirationRequest;
 pub use table::TableStats;
 #[cfg(any(test, feature = "testing"))]
 pub use timestamp::unchecked_repeatable_ts;

crates/common/src/types/snapshot_export.rs

@@ -0,0 +1,10 @@
+use serde::{
+    Deserialize,
+    Serialize,
+};
+
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(rename_all = "camelCase")]
+pub struct SetExportExpirationRequest {
+    pub expiration_ts_ns: u64,
+}

crates/local_backend/src/router.rs

@@ -110,8 +110,10 @@ use crate::{
         schema_state,
     },
     snapshot_export::{
+        cancel_export,
         get_zip_export,
         request_zip_export,
+        set_export_expiration,
     },
     snapshot_import::{
         cancel_import,
@@ -202,7 +204,9 @@ pub fn router(st: LocalAppState) -> Router {
 
     let snapshot_export_routes = Router::new()
         .route("/request/zip", post(request_zip_export))
-        .route("/zip/{id}", get(get_zip_export));
+        .route("/zip/{id}", get(get_zip_export))
+        .route("/set_expiration/{snapshot_id}", post(set_export_expiration))
+        .route("/cancel/{snapshot_id}", post(cancel_export));
 
     let api_routes = Router::new()
         .merge(cli_routes)

crates/local_backend/src/snapshot_export.rs

@@ -18,18 +18,23 @@ use common::{
     components::ComponentId,
     http::{
         extract::{
+            Json,
             Path,
             Query,
         },
         HttpResponseError,
     },
+    types::SetExportExpirationRequest,
 };
 use either::Either;
 use errors::ErrorMetadata;
 use http::StatusCode;
-use model::exports::types::{
-    ExportFormat,
-    ExportRequestor,
+use model::exports::{
+    types::{
+        ExportFormat,
+        ExportRequestor,
+    },
+    ExportsModel,
 };
 use serde::Deserialize;
 use storage::StorageGetStream;
@@ -118,3 +123,56 @@ pub async fn get_zip_export(
         Body::from_stream(stream),
     ))
 }
+
+#[derive(Deserialize)]
+pub struct SetExportExpirationPathArgs {
+    snapshot_id: String,
+}
+
+#[debug_handler]
+#[fastrace::trace]
+pub async fn set_export_expiration(
+    State(st): State<LocalAppState>,
+    ExtractIdentity(identity): ExtractIdentity,
+    Path(SetExportExpirationPathArgs { snapshot_id }): Path<SetExportExpirationPathArgs>,
+    Json(SetExportExpirationRequest { expiration_ts_ns }): Json<SetExportExpirationRequest>,
+) -> Result<StatusCode, HttpResponseError> {
+    if !(identity.is_system() || identity.is_admin()) {
+        Err(anyhow::anyhow!(ErrorMetadata::forbidden(
+            "SetExportExpirationForbidden",
+            "Must have system or admin identity to set export expiration"
+        )))?;
+    }
+    let snapshot_id: DeveloperDocumentId = snapshot_id
+        .parse::<DeveloperDocumentId>()
+        .map_err(|e| anyhow::anyhow!(e))?;
+    let mut tx = st.application.begin(identity).await?;
+    ExportsModel::new(&mut tx)
+        .set_expiration(snapshot_id, expiration_ts_ns)
+        .await?;
+    st.application.commit(tx, "set_export_expiration").await?;
+    Ok(StatusCode::OK)
+}
+
+#[debug_handler]
+#[fastrace::trace]
+pub async fn cancel_export(
+    State(st): State<LocalAppState>,
+    ExtractIdentity(identity): ExtractIdentity,
+    Path(SetExportExpirationPathArgs { snapshot_id }): Path<SetExportExpirationPathArgs>,
+) -> Result<StatusCode, HttpResponseError> {
+    // This route is accessed directly from the admin dashboard
+    if !(identity.is_system() || identity.is_admin()) {
+        Err(anyhow::anyhow!(ErrorMetadata::forbidden(
+            "CancelExportForbidden",
+            "Must have system or admin identity to cancel cloud export"
+        )))?;
+    }
+    let snapshot_id: DeveloperDocumentId = snapshot_id
+        .parse::<DeveloperDocumentId>()
+        .map_err(|e| anyhow::anyhow!(e))?;
+    let mut tx = st.application.begin(identity).await?;
+    ExportsModel::new(&mut tx).cancel(snapshot_id).await?;
+    st.application.commit(tx, "cancel_export").await?;
+    Ok(StatusCode::OK)
+}