autosway 0.4.1

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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use super::*;
use std::env;
use std::path::PathBuf;
use std::sync::Arc;
use tokio::process;

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

        module.set(
            "exec",
            state.lua.create_async_function(
                |lua, (name, args): (String, Vec<String>)| async move {
                    let exec = match urs::misc::search_in_path(&name) {
                        Some(p) => p,
                        None => return Err(LuaError::runtime("Executable not found")),
                    };

                    let output = match process::Command::new(exec).args(args).output().await {
                        Ok(o) => o,
                        Err(e) => return Err(LuaError::external(Arc::new(e))),
                    };

                    match String::from_utf8(output.stdout) {
                        Ok(string) => Ok(string.trim().into_lua(&lua)?),
                        Err(e) => Err(LuaError::external(Arc::new(e))),
                    }
                },
            )?,
        )?;

        module.set(
            "get_pid",
            state
                .lua
                .create_function(|_, _: ()| Ok(std::process::id()))?,
        )?;

        module.set(
            "get_env",
            state
                .lua
                .create_function(|lua, name: String| match env::var(name) {
                    Ok(v) => Ok(v.into_lua(lua)?),
                    Err(_) => Ok(LuaValue::Nil),
                })?,
        )?;

        module.set(
            "set_env",
            state
                .lua
                .create_function(|_, (name, value): (String, String)| {
                    env::set_var(name, value);
                    Ok(())
                })?,
        )?;

        module.set(
            "current_exe",
            state
                .lua
                .create_function(|lua, _: ()| match env::current_exe() {
                    Ok(p) => Ok(p.into_lua(lua)?),
                    Err(e) => Err(LuaError::external(Arc::new(e))),
                })?,
        )?;

        module.set(
            "get_cwd",
            state
                .lua
                .create_function(|lua, _: ()| match env::current_dir() {
                    Ok(p) => Ok(p.into_lua(lua)?),
                    Err(_) => Ok(LuaValue::Nil),
                })?,
        )?;

        module.set(
            "set_cwd",
            state.lua.create_function(|_, p: PathBuf| {
                env::set_current_dir(p)?;
                Ok(())
            })?,
        )?;

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

        Ok(())
    }
}