ps-parser 1.0.1

The Powershell Parser
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

use super::{MethodError, MethodResult, PsString, Val};

impl PsString {
    pub(super) fn pad_left(&self, args: Vec<Val>) -> MethodResult<Val> {
        let PsString(mut input) = self.clone();

        if args.len() != 1 {
            //something wrong
            return Err(MethodError::new_incorrect_args("PadLeft", args));
        }

        let Val::Int(width) = args[0] else {
            return Err(MethodError::new_incorrect_args("PadLeft", args));
        };

        let padding = width.saturating_sub(input.len() as i64);
        if padding > 0 {
            input.insert_str(0, &" ".repeat(padding as usize));
        }

        Ok(Val::String(PsString(input)))
    }

    pub(super) fn pad_right(&self, args: Vec<Val>) -> MethodResult<Val> {
        let PsString(mut input) = self.clone();

        if args.len() != 1 {
            //something wrong
            return Err(MethodError::new_incorrect_args("PadRight", args));
        }

        let Val::Int(width) = args[0] else {
            return Err(MethodError::new_incorrect_args("PadRight", args));
        };

        let padding = width.saturating_sub(input.len() as i64);
        if padding > 0 {
            input.push_str(&" ".repeat(padding as usize));
        }

        Ok(Val::String(PsString(input)))
    }
}

#[cfg(test)]
mod tests {
    use crate::{PowerShellSession, PsValue};

    #[test]
    fn pad_left() {
        let mut p = PowerShellSession::new();
        let input = r#"
$string = 'hello'
$string = $string.padleft(2)
$string"#;
        let script_res = p.parse_script(input).unwrap();
        assert_eq!(script_res.result(), PsValue::String("hello".to_string()));

        let input = r#"
$string = 'hello'
$string = $string.padleft(10)
$string"#;
        let script_res = p.parse_script(input).unwrap();
        assert_eq!(
            script_res.result(),
            PsValue::String("     hello".to_string())
        );

        let input = r#"
$string = 'hello'
$string = $string.padleft()
$string"#;
        let script_res = p.parse_script(input).unwrap();
        assert_eq!(
            script_res.result(),
            PsValue::String("\"hello\".padleft()".to_string())
        );
        assert_eq!(
            script_res.errors()[0].to_string(),
            "MethodError: Incorrect arguments \"[]\" for method \"PadLeft\"".to_string()
        );
    }

    #[test]
    fn pad_right() {
        let mut p = PowerShellSession::new();
        let input = r#"
$string = 'hello'
$string = $string.padright(10)
$string"#;
        let script_res = p.parse_script(input).unwrap();
        assert_eq!(
            script_res.result(),
            PsValue::String("hello     ".to_string())
        );

        let input = r#"
$string = 'hello'
$string = $string.padright()
$string"#;
        let script_res = p.parse_script(input).unwrap();
        assert_eq!(
            script_res.result(),
            PsValue::String("\"hello\".padright()".to_string())
        );
        assert_eq!(
            script_res.errors()[0].to_string(),
            "MethodError: Incorrect arguments \"[]\" for method \"PadRight\"".to_string()
        );
    }
}