qrlkit 0.5.1

Quick Resource Locator: open named URLs from imported TOML files
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

use anyhow::{Result, ensure};

// Replace only complete, unquoted words. Paths are supplied as arguments, never
// interpolated into shell code. Leave quoted strings and escaped words intact.
pub fn expand(hook: &str, placeholder: &str, replacement: &str) -> Result<String> {
    ensure!(
        !hook.contains(['\0', '\n', '\r']),
        "{placeholder} hook cannot contain NUL or newlines"
    );
    let mut output = String::new();
    let mut word = String::new();
    let mut quote = None;
    let mut escaped = false;
    let mut found = false;
    let flush = |word: &mut String, output: &mut String, found: &mut bool| {
        if word == placeholder {
            output.push_str(replacement);
            *found = true;
        } else {
            output.push_str(word);
        }
        word.clear();
    };
    for ch in hook.chars() {
        if escaped {
            word.push(ch);
            escaped = false;
        } else if ch == '\\' && quote != Some('\'') {
            word.push(ch);
            escaped = true;
        } else if let Some(delimiter) = quote {
            word.push(ch);
            if ch == delimiter {
                quote = None;
            }
        } else if ch == '\'' || ch == '"' {
            quote = Some(ch);
            word.push(ch);
        } else if ch.is_whitespace() || ";&|()<>{}".contains(ch) {
            flush(&mut word, &mut output, &mut found);
            output.push(ch);
        } else {
            word.push(ch);
        }
    }
    ensure!(
        quote.is_none() && !escaped,
        "Unclosed quote or trailing escape in {placeholder} hook"
    );
    flush(&mut word, &mut output, &mut found);
    ensure!(
        found,
        "Hook needs a standalone unquoted {placeholder} argument"
    );
    Ok(output)
}