ast-grep-language 0.42.2

Search and Rewrite code at large scale using precise AST pattern
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

#![cfg(test)]

use super::*;

fn test_match(query: &str, source: &str) {
  use crate::test::test_match_lang;
  test_match_lang(query, source, Haskell);
}

fn test_non_match(query: &str, source: &str) {
  use crate::test::test_non_match_lang;
  test_non_match_lang(query, source, Haskell);
}

#[test]
fn test_haskell_str() {
  test_match("return $A", "return 3");
  test_match(r#""abc""#, r#""abc""#);
  test_match("$A $B", "f x");
  test_match("$A ($B $C)", "f (x y)");
  test_match("let $A = $B in $A + $A", "let x = 3 in x + x");
  test_non_match("$A $B", "f");
  test_non_match("$A + $A", "3 + 4");
  test_non_match("$A ($B $C)", "f x y");
}

fn test_replace(src: &str, pattern: &str, replacer: &str) -> String {
  use crate::test::test_replace_lang;
  test_replace_lang(src, pattern, replacer, Haskell)
}

#[test]
fn test_haskell_replace() {
  let ret = test_replace(
    r#"
fibonacci :: [Int]
fibonacci =
  1 : 1 : zipWith (+) fibonacci (tail fibonacci)
"#,
    r#"$F = $$$BODY"#,
    r#"$F = undefined"#,
  );
  assert_eq!(
    ret,
    r#"
fibonacci :: [Int]
fibonacci = undefined
"#
  );

  let ret = test_replace(
    r#"
flip :: (a -> b -> c) -> b -> a -> c
flip f x y = f y x
"#,
    r#"$F :: $A -> $B"#,
    r#"$F :: ($B) -> $A"#,
  );
  assert_eq!(
    ret,
    r#"
flip :: (b -> a -> c) -> (a -> b -> c)
flip f x y = f y x
"#
  );
}