atlas_vm 0.5.2

The Atlas77 Virtual Machine
Documentation 
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

use crate::errors::RuntimeError;
use crate::memory::object_map::ObjectKind;
use crate::memory::vm_data::VMData;
use crate::runtime::vm_state::VMState;
use crate::CallBack;

pub const FILE_FUNCTIONS: [(&str, CallBack); 5] = [
    ("read_dir", read_dir),
    ("read_file", read_file),
    ("write_file", write_file),
    ("file_exists", file_exists),
    ("remove_file", remove_file),
];

pub fn read_dir(state: VMState) -> Result<VMData, RuntimeError> {
    let path_ptr = state.stack.pop_with_rc(state.object_map)?.as_object();
    let raw_path = state.object_map.get(path_ptr)?;
    let path = raw_path.string();

    let entries = std::fs::read_dir(path).unwrap();
    let mut list = Vec::new();
    for entry in entries {
        let entry = entry.unwrap();
        let path = entry.path();
        let path_str = path.to_str().unwrap();
        let obj_idx = state.object_map.put(ObjectKind::String(path_str.to_string()));
        match obj_idx {
            Ok(index) => list.push(VMData::new_string(index)),
            Err(_) => return Err(RuntimeError::OutOfMemory),
        }
    }

    let list_idx = state.object_map.put(ObjectKind::List(list));
    match list_idx {
        Ok(index) => Ok(VMData::new_list(index)),
        Err(_) => Err(RuntimeError::OutOfMemory),
    }
}

pub fn read_file(state: VMState) -> Result<VMData, RuntimeError> {
    let path_ptr = state.stack.pop_with_rc(state.object_map)?.as_object();
    let raw_path = state.object_map.get(path_ptr)?;
    let path = raw_path.string();

    let content = std::fs::read_to_string(path).unwrap();
    let obj_idx = state.object_map.put(ObjectKind::String(content));
    match obj_idx {
        Ok(index) => Ok(VMData::new_string(index)),
        Err(_) => Err(RuntimeError::OutOfMemory),
    }
}

pub fn write_file(state: VMState) -> Result<VMData, RuntimeError> {
    let content_ptr = state.stack.pop_with_rc(state.object_map)?.as_object();
    let path_ptr = state.stack.pop_with_rc(state.object_map)?.as_object();

    let path = state.object_map.get(path_ptr)?.string().clone();
    let raw_content = state.object_map.get(content_ptr)?;
    let content = raw_content.string();

    std::fs::write(path, content).unwrap();
    Ok(VMData::new_unit())
}

pub fn file_exists(state: VMState) -> Result<VMData, RuntimeError> {
    let path_ptr = state.stack.pop_with_rc(state.object_map)?.as_object();
    let raw_path = state.object_map.get(path_ptr)?;
    let path = raw_path.string();

    let exists = std::path::Path::new(&path).exists();
    Ok(VMData::new_bool(exists))
}

pub fn remove_file(state: VMState) -> Result<VMData, RuntimeError> {
    let path_ptr = state.stack.pop_with_rc(state.object_map)?.as_object();
    let raw_path = state.object_map.get(path_ptr)?;
    let path = raw_path.string();

    std::fs::remove_file(path).unwrap();
    Ok(VMData::new_unit())
}