text_document/
operation.rs

1//! Typed long operation handle.
2
3use std::thread;
4use std::time::Duration;
5
6use anyhow::Result;
7
8use frontend::AppContext;
9
10/// Function that polls the long-operation manager for a result.
11type ResultFn<T> = Box<dyn Fn(&AppContext, &str) -> Option<Result<T>> + Send>;
12
13/// Shared state for a single long operation.
14pub(crate) struct OperationState {
15    ctx: AppContext,
16}
17
18impl OperationState {
19    pub fn new(ctx: &AppContext) -> Self {
20        Self { ctx: ctx.clone() }
21    }
22}
23
24/// A handle to a running long operation (Markdown/HTML import, DOCX export).
25///
26/// Provides typed access to progress, cancellation, and the result.
27/// Progress events are also emitted via [`DocumentEvent::LongOperationProgress`](crate::DocumentEvent::LongOperationProgress)
28/// and [`DocumentEvent::LongOperationFinished`](crate::DocumentEvent::LongOperationFinished)
29/// for the callback/polling path.
30///
31/// Retrieve the result via [`wait()`](Self::wait) (blocking, consumes the handle)
32/// or [`try_result()`](Self::try_result) (non-blocking, can be called repeatedly).
33pub struct Operation<T> {
34    id: String,
35    state: OperationState,
36    result_fn: ResultFn<T>,
37}
38
39impl<T> Operation<T> {
40    pub(crate) fn new(id: String, ctx: &AppContext, result_fn: ResultFn<T>) -> Self {
41        Self {
42            id,
43            state: OperationState::new(ctx),
44            result_fn,
45        }
46    }
47
48    /// The operation ID (for matching with [`DocumentEvent`](crate::DocumentEvent) variants).
49    pub fn id(&self) -> &str {
50        &self.id
51    }
52
53    /// Get the current progress, if available.
54    /// Returns `(percent, message)` where percent is 0.0–100.0.
55    pub fn progress(&self) -> Option<(f64, String)> {
56        let mgr = self
57            .state
58            .ctx
59            .long_operation_manager
60            .lock()
61            .ok()
62            .or_else(|| {
63                // Mutex poisoned — recover by re-locking through the poison.
64                match self.state.ctx.long_operation_manager.lock() {
65                    Ok(g) => Some(g),
66                    Err(e) => Some(e.into_inner()),
67                }
68            })?;
69        mgr.get_operation_progress(&self.id)
70            .map(|p| (p.percentage as f64, p.message.unwrap_or_default()))
71    }
72
73    /// Returns `true` if the operation has finished (success or failure).
74    pub fn is_done(&self) -> bool {
75        (self.result_fn)(&self.state.ctx, &self.id).is_some()
76    }
77
78    /// Cancel the operation. No-op if already finished.
79    pub fn cancel(&self) {
80        if let Ok(mgr) = self.state.ctx.long_operation_manager.lock() {
81            mgr.cancel_operation(&self.id);
82        }
83    }
84
85    /// Block the calling thread until the operation completes and return
86    /// the typed result. Consumes the handle.
87    pub fn wait(self) -> Result<T> {
88        loop {
89            if let Some(result) = (self.result_fn)(&self.state.ctx, &self.id) {
90                return result;
91            }
92            thread::sleep(Duration::from_millis(50));
93        }
94    }
95
96    /// Block until the operation completes or the timeout expires.
97    /// Returns `None` if the timeout elapsed before the operation finished.
98    pub fn wait_timeout(self, timeout: Duration) -> Option<Result<T>> {
99        let deadline = std::time::Instant::now() + timeout;
100        loop {
101            if let Some(result) = (self.result_fn)(&self.state.ctx, &self.id) {
102                return Some(result);
103            }
104            if std::time::Instant::now() >= deadline {
105                return None;
106            }
107            let remaining = deadline.saturating_duration_since(std::time::Instant::now());
108            thread::sleep(remaining.min(Duration::from_millis(50)));
109        }
110    }
111
112    /// Non-blocking: returns the result if the operation has completed,
113    /// `None` if still running. Can be called repeatedly.
114    pub fn try_result(&mut self) -> Option<Result<T>> {
115        (self.result_fn)(&self.state.ctx, &self.id)
116    }
117}
118
119// ── Result types ────────────────────────────────────────────────
120
121/// Result of a Markdown import (`set_markdown`).
122#[derive(Debug, Clone)]
123pub struct MarkdownImportResult {
124    pub block_count: usize,
125}
126
127/// Result of an HTML import (`set_html`).
128#[derive(Debug, Clone)]
129pub struct HtmlImportResult {
130    pub block_count: usize,
131}
132
133/// Result of a DOCX export (`to_docx`).
134#[derive(Debug, Clone)]
135pub struct DocxExportResult {
136    pub file_path: String,
137    pub paragraph_count: usize,
138}
text_document/operation.rs

text_document/
operation.rs
