v0.1.0-pre.059

kb-app-demo-desktop/src/demo_core_extraction.rs

@@ -1,5 +1,5 @@
 // file: kb-app-demo-desktop/src/demo_core_extraction.rs
-// version: 9
+// version: 10
 
 //! Tauri commands and UI payloads for canonical transaction to core extraction.
 
@@ -158,6 +158,70 @@ impl std::ops::Drop for crate::DemoCoreExtractionRunGuard<'_> {
     }
 }
 
+pub(crate) fn demo_core_extraction_options_payload(
+    state: &crate::AppState,
+) -> crate::DemoCoreExtractionOptionsPayload {
+    return crate::DemoCoreExtractionOptionsPayload {
+        processor_version: kb_pipeline::CORE_EXTRACTION_PROCESSOR_VERSION.to_string(),
+        default_limit: 100,
+        default_max_concurrent_extractions: 4,
+        running: state.demo_core_extraction_running().load(std::sync::atomic::Ordering::Acquire),
+    };
+}
+
+pub(crate) async fn execute_demo_core_extraction(
+    app_handle: tauri::AppHandle,
+    state: &crate::AppState,
+    request: crate::DemoCoreExtractionRequest,
+) -> std::result::Result<crate::DemoCoreExtractionSummaryPayload, std::string::String> {
+    let acquire_result = state.demo_core_extraction_running().compare_exchange(
+        false,
+        true,
+        std::sync::atomic::Ordering::AcqRel,
+        std::sync::atomic::Ordering::Acquire,
+    );
+    if acquire_result.is_err() {
+        return std::result::Result::Err(
+            "a core extraction campaign is already running".to_string(),
+        );
+    }
+    let _run_guard = crate::DemoCoreExtractionRunGuard {
+        running: state.demo_core_extraction_running(),
+    };
+    state
+        .demo_core_extraction_cancel_requested()
+        .store(false, std::sync::atomic::Ordering::Release);
+    let pipeline_request_result = crate::build_demo_core_extraction_pipeline_request(request);
+    let pipeline_request = match pipeline_request_result {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(error) => return std::result::Result::Err(error),
+    };
+    let store_result = crate::connect_postgres_store(state.active_profile()).await;
+    let store = match store_result {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(error) => return std::result::Result::Err(error),
+    };
+    let observer = crate::DemoCoreExtractionObserver {
+        app_handle,
+        cancel_requested: state.demo_core_extraction_cancel_requested(),
+    };
+    let summary_result =
+        kb_pipeline::execute_core_extraction(&store, &pipeline_request, &observer).await;
+    let summary = match summary_result {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(error) => return std::result::Result::Err(error.to_string()),
+    };
+    return std::result::Result::Ok(crate::demo_core_extraction_summary_payload(summary));
+}
+
+pub(crate) fn cancel_demo_core_extraction(state: &crate::AppState) -> bool {
+    let running = state.demo_core_extraction_running().load(std::sync::atomic::Ordering::Acquire);
+    state
+        .demo_core_extraction_cancel_requested()
+        .store(true, std::sync::atomic::Ordering::Release);
+    return running;
+}
+
 pub(crate) fn build_demo_core_extraction_pipeline_request(
     request: crate::DemoCoreExtractionRequest,
 ) -> std::result::Result<kb_pipeline::CoreExtractionRequest, std::string::String> {