tinymist 0.14.18

An integrated language service for Typst.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

use std::{collections::HashMap, sync::Arc};

use lsp_types::notification::Notification;
use serde::{Deserialize, Serialize};
use serde_json::Value as JsonValue;
use sync_ls::{internal_error, LspClient, LspResult};
use tinymist_preview::{ControlPlaneMessage, Previewer};
use tinymist_std::error::IgnoreLogging;
use tokio::sync::{mpsc, oneshot};

use crate::project::ProjectPreviewState;
use crate::tool::preview::{HttpServer, ProjectPreviewHandler};

pub struct PreviewTab {
    /// Task ID
    pub task_id: String,
    /// Previewer
    pub previewer: Previewer,
    /// Http Server for Previewer
    pub srv: HttpServer,
    /// Control plane message sender
    pub ctl_tx: mpsc::UnboundedSender<ControlPlaneMessage>,
    /// Compile handler
    pub compile_handler: Arc<ProjectPreviewHandler>,
    /// Whether this tab is primary
    pub is_primary: bool,
    /// Whether this tab is background
    pub is_background: bool,
}

pub enum PreviewRequest {
    Started(PreviewTab),
    Kill(String, oneshot::Sender<LspResult<JsonValue>>),
    KillAll(oneshot::Sender<LspResult<JsonValue>>),
    Scroll(String, ControlPlaneMessage),
    ScrollAll(ControlPlaneMessage),
}

pub struct PreviewActor {
    pub client: LspClient,
    pub tabs: HashMap<String, PreviewTab>,
    pub preview_rx: mpsc::UnboundedReceiver<PreviewRequest>,
    /// the watchers for the preview
    pub(crate) watchers: ProjectPreviewState,
}

impl PreviewActor {
    pub async fn run(mut self) {
        while let Some(req) = self.preview_rx.recv().await {
            match req {
                PreviewRequest::Started(tab) => {
                    self.tabs.insert(tab.task_id.clone(), tab);
                }
                PreviewRequest::Kill(task_id, tx) => {
                    self.kill(task_id, tx).await;
                }
                PreviewRequest::Scroll(task_id, req) => {
                    self.scroll(task_id, req).await;
                }
                PreviewRequest::KillAll(tx) => {
                    for task_id in self.tabs.keys().cloned().collect::<Vec<_>>() {
                        let (tx, _rx) = oneshot::channel();
                        self.kill(task_id, tx).await;
                    }
                    let _ = tx.send(Ok(JsonValue::Null));
                }
                PreviewRequest::ScrollAll(req) => {
                    for task_id in self.tabs.keys().cloned().collect::<Vec<_>>() {
                        self.scroll(task_id, req.clone()).await;
                    }
                }
            }
        }
    }

    async fn kill(&mut self, task_id: String, tx: oneshot::Sender<LspResult<JsonValue>>) {
        log::info!("PreviewTask({task_id}): killing");

        if self.tabs.get(&task_id).is_some_and(|tab| tab.is_background) {
            // todo: eliminate this warning in log in future
            log::warn!("PreviewTask({task_id}): cannot kill a background preview");

            let _ = tx.send(Ok(JsonValue::Null));
            return;
        }

        let Some(mut tab) = self.tabs.remove(&task_id) else {
            let _ = tx.send(Err(internal_error("task not found")));
            return;
        };

        // Unregister preview early
        let unregistered = self.watchers.unregister(&tab.compile_handler.project_id);
        if !unregistered {
            log::warn!("PreviewTask({task_id}): failed to unregister preview");
        }

        if tab.is_primary {
            tab.compile_handler.unpin_primary();
        } else {
            tab.compile_handler
                .settle()
                .log_error_with(|| format!("PreviewTask({}): failed to settle", tab.task_id));
        }

        let client = self.client.clone();
        self.client.handle.spawn(async move {
            tab.previewer.stop().await;
            let _ = tab.srv.shutdown_tx.send(());

            // Wait for previewer to stop
            log::info!("PreviewTask({task_id}): wait for previewer to stop");
            tab.previewer.join().await;
            log::info!("PreviewTask({task_id}): wait for static server to stop");
            let _ = tab.srv.join.await;

            log::info!("PreviewTask({task_id}): killed");
            // Send response
            let _ = tx.send(Ok(JsonValue::Null));
            // Send global notification
            client.send_notification::<DisposePreview>(&DisposePreview { task_id });
        });
    }

    async fn scroll(&mut self, task_id: String, req: ControlPlaneMessage) -> Option<()> {
        self.tabs.get(&task_id)?.ctl_tx.send(req).ok()
    }
}

#[derive(Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct DisposePreview {
    task_id: String,
}

impl Notification for DisposePreview {
    type Params = Self;
    const METHOD: &'static str = "tinymist/preview/dispose";
}