robost-stdlib 0.1.0

Built-in node library for Rust RPA: image matching, clipboard, Excel, HTTP, scheduling, and more
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

// Built-in scenario node library.
// Called via `library: <name>` in scenario YAML.
// All functions share: inputs map → outputs map.

mod archive;
#[cfg(feature = "db")]
mod db;
mod excel;
#[cfg(feature = "ftp")]
mod ftp;
mod keychain;
mod list;
mod mail;
mod notify;
mod number;
mod pdf;

use serde_json::Value;
use std::collections::HashMap;

pub type NodeResult = Result<HashMap<String, Value>, NodeError>;

#[derive(Debug, thiserror::Error)]
pub enum NodeError {
    #[error("missing required input: {0}")]
    MissingInput(String),
    #[error("node error: {0}")]
    Other(String),
}

pub(crate) fn get_str(inputs: &HashMap<String, Value>, key: &str) -> Result<String, NodeError> {
    inputs
        .get(key)
        .and_then(|v| v.as_str())
        .map(str::to_owned)
        .ok_or_else(|| NodeError::MissingInput(key.to_owned()))
}

pub(crate) fn opt_str(inputs: &HashMap<String, Value>, key: &str) -> Option<String> {
    inputs.get(key).and_then(|v| v.as_str()).map(str::to_owned)
}

/// Dispatch a built-in library node by name.
///
/// Name format: `"category.function"`, e.g. `"mail.smtp_send"`, `"list.sort"`.
pub fn dispatch(name: &str, inputs: HashMap<String, Value>) -> NodeResult {
    match name {
        // --- mail ---
        "mail.smtp_send" => mail::smtp_send(inputs),
        "mail.imap_receive" => mail::imap_receive(inputs),

        // --- excel ---
        "excel.add_sheet" => excel::add_sheet(inputs),
        "excel.delete_sheet" => excel::delete_sheet(inputs),
        "excel.rename_sheet" => excel::rename_sheet(inputs),
        "excel.copy_sheet" => excel::copy_sheet(inputs),
        "excel.save_as" => excel::save_as(inputs),
        "excel.list_sheets" => excel::list_sheets(inputs),
        "excel.write_range" => excel::write_range(inputs),
        "excel.set_formula" => excel::set_formula(inputs),

        // --- list ---
        "list.sort" => list::sort(inputs),
        "list.unique" => list::unique(inputs),
        "list.reverse" => list::reverse(inputs),
        "list.length" => list::length(inputs),
        "list.get" => list::get(inputs),
        "list.slice" => list::slice(inputs),
        "list.flatten" => list::flatten(inputs),

        // --- number ---
        "number.format" => number::format(inputs),
        "number.round" => number::round(inputs),
        "number.to_percent" => number::to_percent(inputs),

        // --- db (optional feature) ---
        #[cfg(feature = "db")]
        "db.query" => db::query(inputs),
        #[cfg(feature = "db")]
        "db.query_one" => db::query_one(inputs),
        #[cfg(feature = "db")]
        "db.execute" => db::execute(inputs),

        // --- archive ---
        "archive.compress" => archive::compress(inputs),
        "archive.extract" => archive::extract(inputs),
        "archive.list" => archive::list(inputs),

        // --- ftp ---
        #[cfg(feature = "ftp")]
        "ftp.upload" => ftp::upload(inputs),
        #[cfg(feature = "ftp")]
        "ftp.download" => ftp::download(inputs),
        #[cfg(feature = "ftp")]
        "ftp.list" => ftp::list(inputs),
        #[cfg(feature = "ftp")]
        "ftp.delete" => ftp::delete(inputs),
        #[cfg(feature = "ftp")]
        "ftp.mkdir" => ftp::mkdir(inputs),

        // --- pdf ---
        "pdf.extract_text" => pdf::extract_text(inputs),
        "pdf.page_count" => pdf::page_count(inputs),

        // --- notify (Slack / Teams webhook) ---
        "notify.slack_send" => notify::slack_send(inputs),
        "notify.teams_send" => notify::teams_send(inputs),

        // --- keychain ---
        "keychain.get" => keychain::keychain_get(inputs),
        "keychain.set" => keychain::keychain_set(inputs),
        "keychain.delete" => keychain::keychain_delete(inputs),

        other => Err(NodeError::Other(format!("unknown built-in node: {other}"))),
    }
}