project 0.1.2

Project automation powered by Rust and Lua
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

use super::join_path;
use crate::App;
use std::path::{Path, PathBuf};
use tsuki::context::{Args, Context, Ret};
use tsuki::{Lua, Module, Ref, Table, fp};

/// Implementation of [Module] for `path` API.
pub struct PathModule;

impl Module<App> for PathModule {
    const NAME: &str = "path";

    type Inst<'a> = Ref<'a, Table<App>>;

    fn open(self, lua: &Lua<App>) -> Result<Self::Inst<'_>, Box<dyn core::error::Error>> {
        let m = lua.create_table();

        m.set_str_key("basename", fp!(basename));
        m.set_str_key("dirname", fp!(dirname));
        m.set_str_key("join", fp!(join));

        Ok(m)
    }
}

fn basename(cx: Context<App, Args>) -> Result<Context<App, Ret>, Box<dyn std::error::Error>> {
    let path = cx.arg(1);
    let path = path
        .to_str()?
        .as_utf8()
        .ok_or_else(|| path.error("expect UTF-8 string"))?;
    let path = Path::new(path);

    if let Some(v) = path.file_name() {
        // We requires argument to be UTF-8 so this will never fails.
        cx.push_str(v.to_str().unwrap())?;
    }

    Ok(cx.into())
}

fn dirname(cx: Context<App, Args>) -> Result<Context<App, Ret>, Box<dyn std::error::Error>> {
    let path = cx.arg(1);
    let path = path
        .to_str()?
        .as_utf8()
        .ok_or_else(|| path.error("expect UTF-8 string"))?;
    let path = Path::new(path);

    if let Some(v) = path.parent() {
        // We requires argument to be UTF-8 so this will never fails.
        cx.push_str(v.to_str().unwrap())?;
    }

    Ok(cx.into())
}

fn join(cx: Context<App, Args>) -> Result<Context<App, Ret>, Box<dyn std::error::Error>> {
    let mut path = PathBuf::new();

    join_path(&cx, |_, c| {
        path.push(c);
        Ok(())
    })?;

    // join_path() requires all components to be UTF-8 so this should never fails.
    cx.push_str(path.to_str().unwrap())?;

    Ok(cx.into())
}