autosway 0.3.0

Automation program
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

use super::*;
use std::{fs, io::Write, path::PathBuf};

pub struct FSFile {
    pub file: File,
}

impl From<File> for FSFile {
    fn from(value: File) -> Self {
        Self { file: value }
    }
}

impl LuaUserData for FSFile {
    fn add_methods<M: LuaUserDataMethods<Self>>(methods: &mut M) {
        methods.add_method_mut("write", |_, this, data: String| {
            this.file.write_all(data.as_bytes())?;
            Ok(())
        });

        methods.add_method_mut("read_line", |lua, this, _: ()| {
            this.file.read_line_trim()?.into_lua(lua)
        });

        methods.add_method_mut("read_all", |lua, this, _: ()| {
            this.file.read_everything()?.into_lua(lua)
        });

        methods.add_method_mut("flush", |_, this, _: ()| {
            this.file.flush()?;
            Ok(())
        });

        methods.add_method_mut("is_symlink", |_, this, _: ()| {
            Ok(this.file.metadata()?.is_symlink())
        });

        methods.add_method_mut("is_file", |_, this, _: ()| {
            Ok(this.file.metadata()?.is_file())
        });
    }
}

pub struct FS;
impl Module for FS {
    fn load(&self, state: &mut State) -> Result<(), LuaError> {
        let module = state.lua.create_table()?;

        module.set(
            "create",
            state
                .lua
                .create_function(|_, path: PathBuf| match fs::File::create(path) {
                    Ok(f) => Ok(FSFile::from(f)),
                    Err(e) => Err(e.into()),
                })?,
        )?;

        module.set(
            "open",
            state
                .lua
                .create_function(|_, path: PathBuf| match fs::File::open(path) {
                    Ok(f) => Ok(FSFile::from(f)),
                    Err(e) => Err(e.into()),
                })?,
        )?;

        create_module(&state.lua, "autosway.fs", module)?;

        Ok(())
    }
}