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

use std::fs::File;

use crate::common::*;
use mlua::prelude::*;
use urs::io::ReadExtra;

mod modules;

pub trait Module {
    fn load(&self, state: &mut State) -> Result<(), LuaError>;
}

impl State {
    pub fn setup_lua(&mut self) -> Result<(), LuaError> {
        if let Some(dir) = get_config_dir(ConfigDir::KnifeDir) {
            let s = if let Some(s) = dir.to_str() {
                s
            } else {
                log::log!(log::Error, "Could not add knife directory to load path");
                return Err(LuaError::RuntimeError("".to_string()));
            };

            let code = format!("package.path = package.path..';{}/?/init.lua'", s);
            self.lua.load(code).exec()?
        }

        self.load_module(modules::OS)?;
        self.load_module(modules::FS)?;
        self.load_module(modules::I3)?;
        self.load_module(modules::Util)?;
        self.load_module(modules::Desktop)?;
        self.load_module(modules::Multimedia)?;
        self.load_module(modules::Extra)?;

        self.lua.load(include_str!("../lua/globals.lua")).exec()?;

        if let Some(dir) = get_config_dir(ConfigDir::RootDir) {
            if let Ok(mut f) = File::open(dir.join("init.lua")) {
                self.lua.load(f.read_everything()?).exec()?;
            }
        }

        Ok(())
    }

    pub fn load_module(&mut self, m: impl Module) -> Result<(), LuaError> {
        m.load(self)?;
        Ok(())
    }

    pub fn run_knife(&mut self, name: &str) -> Result<(), LuaError> {
        for knife in &self.knifes {
            if knife.name == name {
                let mut f = File::open(&knife.init)?;
                self.lua.load(f.read_everything()?).exec()?;
                return Ok(());
            }
        }

        log::log!(log::Error, "No knife found: {name}");
        Ok(())
    }
}