kanri 0.11.0

Manage your projects within the terminal.
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
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
141
142

use mlua::prelude::*;
use std::{
    env,
    io::ErrorKind,
    path::PathBuf,
    process::{Command, Stdio},
};

use crate::terminal::print_action_run;

fn command_error(error: std::io::Error) -> mlua::Error {
    let message = match error.kind() {
        ErrorKind::NotFound => "program not found",
        ErrorKind::Interrupted => "program was interrupted",
        ErrorKind::PermissionDenied => "not enough permissions to execute program",
        _ => "unknown error occurred",
    };
    mlua::Error::runtime(message)
}

fn print_command(cmd: &String, args: &[String]) {
    let mut command: String = String::from(cmd);
    for i in args {
        command.push_str(&format!(" {}", i));
    }

    print_action_run(&command);
}

pub fn create_os_module(
    lua: &Lua,
    current_dir: impl Into<PathBuf>,
    quiet: bool,
) -> LuaResult<LuaTable> {
    let os_table = lua.create_table()?;
    let current_dir = current_dir.into();

    os_table.set("system", lua.create_function(|_, ()| Ok(env::consts::OS))?)?;
    os_table.set("arch", lua.create_function(|_, ()| Ok(env::consts::ARCH))?)?;
    os_table.set(
        "family",
        lua.create_function(|_, ()| Ok(env::consts::FAMILY))?,
    )?;
    os_table.set(
        "exe_suffix",
        lua.create_function(|_, ()| Ok(env::consts::EXE_SUFFIX))?,
    )?;
    os_table.set(
        "dir_separator",
        lua.create_function(|_, ()| Ok(std::path::MAIN_SEPARATOR.to_string()))?,
    )?;
    os_table.set(
        "path_separator",
        lua.create_function(|_, ()| Ok(if cfg!(windows) { ";" } else { ":" }))?,
    )?;
    os_table.set(
        "temp_dir",
        lua.create_function(|_, ()| Ok(env::temp_dir().to_string_lossy().to_string()))?,
    )?;
    os_table.set(
        "env",
        lua.create_function(|_, name: String| Ok(env::var(name).ok()))?,
    )?;

    let cwd = current_dir.clone();
    os_table.set(
        "current_dir",
        lua.create_function(move |_, ()| Ok(cwd.to_string_lossy().to_string()))?,
    )?;

    let exec_dir = current_dir.clone();
    let os_exec = lua.create_function(move |_, (cmd, args): (String, Vec<String>)| {
        if cmd.is_empty() {
            return Err(mlua::Error::runtime("program cannot be empty"));
        }

        let mut command = Command::new(&cmd);

        if quiet {
            command
                .stdin(Stdio::null())
                .stdout(Stdio::null())
                .stderr(Stdio::null());
        }
        command.args(&args).current_dir(&exec_dir);
        if !quiet {
            print_command(&cmd, &args);
        }
        command.status().map(|_| ()).map_err(command_error)
    })?;
    os_table.set("exec", os_exec)?;

    let status_dir = current_dir.clone();
    let os_exec_status = lua.create_function(move |_, (cmd, args): (String, Vec<String>)| {
        if cmd.is_empty() {
            return Err(mlua::Error::runtime("program cannot be empty"));
        }

        let mut command = Command::new(&cmd);
        if quiet {
            command
                .stdin(Stdio::null())
                .stdout(Stdio::null())
                .stderr(Stdio::null());
        }
        command.args(&args).current_dir(&status_dir);
        if !quiet {
            print_command(&cmd, &args);
        }
        command
            .status()
            .map(|status| status.code().unwrap_or(-1))
            .map_err(command_error)
    })?;
    os_table.set("exec_status", os_exec_status)?;

    let output_dir = current_dir.clone();
    let os_exec_output = lua.create_function(move |_, (cmd, args): (String, Vec<String>)| {
        if cmd.is_empty() {
            return Err(mlua::Error::runtime("program cannot be empty"));
        }

        let mut command = Command::new(&cmd);
        if quiet {
            command
                .stdin(Stdio::null())
                .stdout(Stdio::null())
                .stderr(Stdio::null());
        }
        command.args(&args).current_dir(&output_dir);
        if !quiet {
            print_command(&cmd, &args);
        }
        command
            .output()
            .map(|output| String::from_utf8_lossy(&output.stdout).to_string())
            .map_err(command_error)
    })?;
    os_table.set("exec_output", os_exec_output)?;

    Ok(os_table)
}