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

use crate::CommandSender;

fn is_safe_filename_char(c: char) -> bool {
    c.is_alphanumeric() || matches!(c, ' ' | '-' | '_' | '.')
}

/// Replace "dangerous" characters by a safe one.
pub fn santitize_file_name(file_name: &str) -> String {
    file_name.replace(|c: char| !is_safe_filename_char(c), "-")
}

impl CommandSender {
    /// Save some bytes to disk, after first showing a save dialog.
    ///
    /// [This may only be called on the main thread](https://docs.rs/rfd/latest/rfd/#macos-non-windowed-applications-async-and-threading).
    #[allow(clippy::unused_self)] // Not used on Wasm
    pub fn save_file_dialog(
        &self,
        _: re_capabilities::MainThreadToken,
        file_name: &str,
        title: String,
        data: Vec<u8>,
    ) {
        re_tracing::profile_function!();

        let file_name = santitize_file_name(file_name);

        #[cfg(target_arch = "wasm32")]
        {
            // Web
            wasm_bindgen_futures::spawn_local(async move {
                if let Err(err) = async_save_dialog(&file_name, &title, data).await {
                    re_log::error!("File saving failed: {err}");
                } else {
                    re_log::info!("{file_name} saved.");
                };
            });
        }

        #[cfg(not(target_arch = "wasm32"))]
        {
            // Native
            let path = {
                re_tracing::profile_scope!("file_dialog");
                rfd::FileDialog::new()
                    .set_file_name(file_name)
                    .set_title(title)
                    .save_file()
            };
            if let Some(path) = path {
                use crate::SystemCommandSender as _;
                self.send_system(crate::SystemCommand::FileSaver(Box::new(move || {
                    std::fs::write(&path, &data)?;
                    Ok(path)
                })));
            }
        }
    }
}

#[cfg(target_arch = "wasm32")]
async fn async_save_dialog(file_name: &str, title: &str, data: Vec<u8>) -> anyhow::Result<()> {
    use anyhow::Context as _;

    let file_handle = rfd::AsyncFileDialog::new()
        .set_file_name(file_name)
        .set_title(title)
        .save_file()
        .await;

    let Some(file_handle) = file_handle else {
        return Ok(()); // aborted
    };

    file_handle
        .write(data.as_slice())
        .await
        .context("Failed to save")
}