rong_fs 0.3.1

Filesystem module for RongJS
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
140

use crate::grant_file_access;
use crate::rong_file::RongFile;
use rong::*;
use std::path::PathBuf;
use tokio::io::AsyncWriteExt;

/// Resolve destination to a PathBuf. Accepts a string path or a RongFile object.
fn resolve_dest(dest: &JSValue) -> JSResult<PathBuf> {
    // Try as string first
    if dest.is_string() {
        let path: String = dest.clone().to_rust()?;
        return grant_file_access(&path);
    }

    // Try as RongFile object
    if let Some(obj) = dest.clone().into_object()
        && let Ok(rf) = obj.borrow::<RongFile>()
    {
        return Ok(rf.resolved().clone());
    }

    Err(HostError::new(
        rong::error::E_INVALID_ARG,
        "destination must be a string path or RongFile",
    )
    .with_name("TypeError")
    .into())
}

/// Universal write: Rong.write(dest, data) -> Promise<number>
///
/// dest: string path or RongFile
/// data: string, TypedArray, ArrayBuffer, or RongFile (copy)
async fn rong_write(_ctx: JSContext, dest: JSValue, data: JSValue) -> JSResult<f64> {
    let resolved = resolve_dest(&dest)?;

    // Dispatch on data type
    if data.is_string() {
        // Write string as UTF-8
        let text: String = data.to_rust()?;
        let bytes = text.as_bytes();
        let len = bytes.len();

        let mut file = tokio::fs::OpenOptions::new()
            .write(true)
            .create(true)
            .truncate(true)
            .open(&resolved)
            .await
            .map_err(|e| HostError::new("FS_IO", format!("Failed to open file: {}", e)))?;
        file.write_all(bytes)
            .await
            .map_err(|e| HostError::new("FS_IO", format!("Write failed: {}", e)))?;
        file.flush()
            .await
            .map_err(|e| HostError::new("FS_IO", format!("Flush failed: {}", e)))?;

        return Ok(len as f64);
    }

    if data.is_array_buffer() {
        // Write ArrayBuffer
        let ab: JSArrayBuffer = data.to_rust()?;
        let bytes = ab.as_slice();
        let len = bytes.len();

        let mut file = tokio::fs::OpenOptions::new()
            .write(true)
            .create(true)
            .truncate(true)
            .open(&resolved)
            .await
            .map_err(|e| HostError::new("FS_IO", format!("Failed to open file: {}", e)))?;
        file.write_all(bytes)
            .await
            .map_err(|e| HostError::new("FS_IO", format!("Write failed: {}", e)))?;
        file.flush()
            .await
            .map_err(|e| HostError::new("FS_IO", format!("Flush failed: {}", e)))?;

        return Ok(len as f64);
    }

    // Try as object (TypedArray or RongFile)
    if let Some(obj) = data.clone().into_object() {
        // Try as RongFile (copy semantics)
        let source = {
            if let Ok(rf) = obj.borrow::<RongFile>() {
                Some(rf.resolved().clone())
            } else {
                None
            }
        };
        if let Some(source) = source {
            let bytes_copied = tokio::fs::copy(&source, &resolved)
                .await
                .map_err(|e| HostError::new("FS_IO", format!("Failed to copy file: {}", e)))?;
            return Ok(bytes_copied as f64);
        }

        // Try as TypedArray (Uint8Array, etc.)
        if let Some(ta) = AnyJSTypedArray::from_object(obj)
            && let Some(bytes) = ta.as_bytes()
        {
            let len = bytes.len();

            let mut file = tokio::fs::OpenOptions::new()
                .write(true)
                .create(true)
                .truncate(true)
                .open(&resolved)
                .await
                .map_err(|e| HostError::new("FS_IO", format!("Failed to open file: {}", e)))?;
            file.write_all(bytes)
                .await
                .map_err(|e| HostError::new("FS_IO", format!("Write failed: {}", e)))?;
            file.flush()
                .await
                .map_err(|e| HostError::new("FS_IO", format!("Flush failed: {}", e)))?;

            return Ok(len as f64);
        }
    }

    Err(HostError::new(
        rong::error::E_INVALID_ARG,
        "data must be a string, ArrayBuffer, TypedArray, or RongFile",
    )
    .with_name("TypeError")
    .into())
}

pub(crate) fn init(ctx: &JSContext) -> JSResult<()> {
    let rong = ctx.host_namespace();

    let write_fn = JSFunc::new(ctx, rong_write)?.name("write")?;
    rong.set("write", write_fn)?;

    Ok(())
}