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

use rnix::types::{Lambda, TypedNode};
use rnix::SyntaxKind::*;

/// Pretty-prints the arguments to a function
pub fn pprint_args(lambda: &Lambda) -> String {
    // TODO: handle docs directly on NODE_IDENT args (uncommon case)
    let mut lambda = lambda.clone();
    let mut out = String::new();
    loop {
        let arg = lambda.arg().unwrap();
        match arg.kind() {
            NODE_IDENT => {
                out.push_str(&arg.to_string());
                out.push_str(": ");
                let body = lambda.body().unwrap();
                if body.kind() == NODE_LAMBDA {
                    lambda = Lambda::cast(body).unwrap();
                } else {
                    break;
                }
            }
            NODE_PATTERN => {
                out.push_str(&arg.to_string());
                out.push_str(": ");
                break;
            }
            t => {
                unreachable!("unhandled arg type {:?}", t);
            }
        }
    }
    out.push_str("...");
    out

    //pprint_arg(lambda.arg());
}