func TestCompiler(t *testing.T) { for id, test := range []struct { ast *nodePattern result Glob sep []rune }{ { ast: pattern(&nodeText{text: "abc"}), result: match.NewText("abc"), }, { ast: pattern(&nodeAny{}), sep: separators, result: match.NewAny(separators), }, { ast: pattern(&nodeAny{}), result: match.NewSuper(), }, { ast: pattern(&nodeSuper{}), result: match.NewSuper(), }, { ast: pattern(&nodeSingle{}), sep: separators, result: match.NewSingle(separators), }, { ast: pattern(&nodeRange{ lo: 'a', hi: 'z', not: true, }), result: match.NewRange('a', 'z', true), }, { ast: pattern(&nodeList{ chars: "abc", not: true, }), result: match.NewList([]rune{'a', 'b', 'c'}, true), }, { ast: pattern(&nodeAny{}, &nodeSingle{}, &nodeSingle{}, &nodeSingle{}), sep: separators, result: match.EveryOf{Matchers: match.Matchers{ match.NewMin(3), match.NewContains(string(separators), true), }}, }, { ast: pattern(&nodeAny{}, &nodeSingle{}, &nodeSingle{}, &nodeSingle{}), result: match.NewMin(3), }, { ast: pattern(&nodeAny{}, &nodeText{text: "abc"}, &nodeSingle{}), sep: separators, result: match.NewBTree( match.NewRow( 4, match.Matchers{ match.NewText("abc"), match.NewSingle(separators), }..., ), match.NewAny(separators), nil, ), }, { ast: pattern(&nodeSuper{}, &nodeSingle{}, &nodeText{text: "abc"}, &nodeSingle{}), sep: separators, result: match.NewBTree( match.NewRow( 5, match.Matchers{ match.NewSingle(separators), match.NewText("abc"), match.NewSingle(separators), }..., ), match.NewSuper(), nil, ), }, { ast: pattern(&nodeAny{}, &nodeText{text: "abc"}), result: match.NewSuffix("abc"), }, { ast: pattern(&nodeText{text: "abc"}, &nodeAny{}), result: match.NewPrefix("abc"), }, { ast: pattern(&nodeText{text: "abc"}, &nodeAny{}, &nodeText{text: "def"}), result: match.NewPrefixSuffix("abc", "def"), }, { ast: pattern(&nodeAny{}, &nodeAny{}, &nodeAny{}, &nodeText{text: "abc"}, &nodeAny{}, &nodeAny{}), result: match.NewContains("abc", false), }, { ast: pattern(&nodeAny{}, &nodeAny{}, &nodeAny{}, &nodeText{text: "abc"}, &nodeAny{}, &nodeAny{}), sep: separators, result: match.NewBTree( match.NewText("abc"), match.NewAny(separators), match.NewAny(separators), ), }, { ast: pattern(&nodeSuper{}, &nodeSingle{}, &nodeText{text: "abc"}, &nodeSuper{}, &nodeSingle{}), result: match.NewBTree( match.NewText("abc"), match.NewMin(1), match.NewMin(1), ), }, { ast: pattern(anyOf(&nodeText{text: "abc"})), result: match.NewText("abc"), }, { ast: pattern(anyOf(pattern(anyOf(pattern(&nodeText{text: "abc"}))))), result: match.NewText("abc"), }, { ast: pattern(anyOf( pattern( &nodeText{text: "abc"}, &nodeSingle{}, ), pattern( &nodeText{text: "abc"}, &nodeList{chars: "def"}, ), pattern( &nodeText{text: "abc"}, ), pattern( &nodeText{text: "abc"}, ), )), result: match.NewBTree( match.NewText("abc"), nil, match.AnyOf{Matchers: match.Matchers{ match.NewSingle(nil), match.NewList([]rune{'d', 'e', 'f'}, false), match.NewNothing(), }}, ), }, { ast: pattern( &nodeRange{lo: 'a', hi: 'z'}, &nodeRange{lo: 'a', hi: 'x', not: true}, &nodeAny{}, ), result: match.NewBTree( match.NewRow( 2, match.Matchers{ match.NewRange('a', 'z', false), match.NewRange('a', 'x', true), }..., ), nil, match.NewSuper(), ), }, { ast: pattern(anyOf( pattern( &nodeText{text: "abc"}, &nodeList{chars: "abc"}, &nodeText{text: "ghi"}, ), pattern( &nodeText{text: "abc"}, &nodeList{chars: "def"}, &nodeText{text: "ghi"}, ), )), result: match.NewRow( 7, match.Matchers{ match.NewText("abc"), match.AnyOf{Matchers: match.Matchers{ match.NewList([]rune{'a', 'b', 'c'}, false), match.NewList([]rune{'d', 'e', 'f'}, false), }}, match.NewText("ghi"), }..., ), }, // { // ast: pattern( // anyOf(&nodeText{text: "a"}, &nodeText{text: "b"}), // anyOf(&nodeText{text: "c"}, &nodeText{text: "d"}), // ), // result: match.AnyOf{Matchers: match.Matchers{ // match.NewRow(Matchers: match.Matchers{match.Raw{"a"}, match.Raw{"c", 1}}), // match.NewRow(Matchers: match.Matchers{match.Raw{"a"}, match.Raw{"d"}}), // match.NewRow(Matchers: match.Matchers{match.Raw{"b"}, match.Raw{"c", 1}}), // match.NewRow(Matchers: match.Matchers{match.Raw{"b"}, match.Raw{"d"}}), // }}, // }, } { m, err := compile(test.ast, test.sep) if err != nil { t.Errorf("compilation error: %s", err) continue } if !reflect.DeepEqual(m, test.result) { t.Errorf("#%d results are not equal:\nexp: %#v\nact: %#v", id, test.result, m) continue } } }
func optimize(matcher match.Matcher) match.Matcher { switch m := matcher.(type) { case match.Any: if len(m.Separators) == 0 { return match.NewSuper() } case match.AnyOf: if len(m.Matchers) == 1 { return m.Matchers[0] } return m case match.List: if m.Not == false && len(m.List) == 1 { return match.NewText(string(m.List)) } return m case match.BTree: m.Left = optimize(m.Left) m.Right = optimize(m.Right) r, ok := m.Value.(match.Text) if !ok { return m } leftNil := m.Left == nil rightNil := m.Right == nil if leftNil && rightNil { return match.NewText(r.Str) } _, leftSuper := m.Left.(match.Super) lp, leftPrefix := m.Left.(match.Prefix) _, rightSuper := m.Right.(match.Super) rs, rightSuffix := m.Right.(match.Suffix) if leftSuper && rightSuper { return match.NewContains(r.Str, false) } if leftSuper && rightNil { return match.NewSuffix(r.Str) } if rightSuper && leftNil { return match.NewPrefix(r.Str) } if leftNil && rightSuffix { return match.NewPrefixSuffix(r.Str, rs.Suffix) } if rightNil && leftPrefix { return match.NewPrefixSuffix(lp.Prefix, r.Str) } return m } return matcher }
func TestCompiler(t *testing.T) { for id, test := range []struct { ast *ast.Node result match.Matcher sep []rune }{ { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ), result: match.NewText("abc"), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ), sep: separators, result: match.NewAny(separators), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ), result: match.NewSuper(), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindSuper, nil), ), result: match.NewSuper(), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindSingle, nil), ), sep: separators, result: match.NewSingle(separators), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindRange, ast.Range{ Lo: 'a', Hi: 'z', Not: true, }), ), result: match.NewRange('a', 'z', true), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindList, ast.List{ Chars: "abc", Not: true, }), ), result: match.NewList([]rune{'a', 'b', 'c'}, true), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindSingle, nil), ast.NewNode(ast.KindSingle, nil), ast.NewNode(ast.KindSingle, nil), ), sep: separators, result: match.EveryOf{Matchers: match.Matchers{ match.NewMin(3), match.NewContains(string(separators), true), }}, }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindSingle, nil), ast.NewNode(ast.KindSingle, nil), ast.NewNode(ast.KindSingle, nil), ), result: match.NewMin(3), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindSingle, nil), ), sep: separators, result: match.NewBTree( match.NewRow( 4, match.Matchers{ match.NewText("abc"), match.NewSingle(separators), }..., ), match.NewAny(separators), nil, ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"/"}), ast.NewNode(ast.KindAnyOf, nil, ast.NewNode(ast.KindText, ast.Text{"z"}), ast.NewNode(ast.KindText, ast.Text{"ab"}), ), ast.NewNode(ast.KindSuper, nil), ), sep: separators, result: match.NewBTree( match.NewText("/"), nil, match.NewBTree( match.NewAnyOf(match.NewText("z"), match.NewText("ab")), nil, match.NewSuper(), ), ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindSuper, nil), ast.NewNode(ast.KindSingle, nil), ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindSingle, nil), ), sep: separators, result: match.NewBTree( match.NewRow( 5, match.Matchers{ match.NewSingle(separators), match.NewText("abc"), match.NewSingle(separators), }..., ), match.NewSuper(), nil, ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindText, ast.Text{"abc"}), ), result: match.NewSuffix("abc"), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindAny, nil), ), result: match.NewPrefix("abc"), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindText, ast.Text{"def"}), ), result: match.NewPrefixSuffix("abc", "def"), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindAny, nil), ), result: match.NewContains("abc", false), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindAny, nil), ast.NewNode(ast.KindAny, nil), ), sep: separators, result: match.NewBTree( match.NewText("abc"), match.NewAny(separators), match.NewAny(separators), ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindSuper, nil), ast.NewNode(ast.KindSingle, nil), ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindSuper, nil), ast.NewNode(ast.KindSingle, nil), ), result: match.NewBTree( match.NewText("abc"), match.NewMin(1), match.NewMin(1), ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ), result: match.NewText("abc"), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAnyOf, nil, ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAnyOf, nil, ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ), ), ), ), ), result: match.NewText("abc"), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAnyOf, nil, ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindSingle, nil), ), ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindList, ast.List{Chars: "def"}), ), ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ), ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ), ), ), result: match.NewBTree( match.NewText("abc"), nil, match.AnyOf{Matchers: match.Matchers{ match.NewSingle(nil), match.NewList([]rune{'d', 'e', 'f'}, false), match.NewNothing(), }}, ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindRange, ast.Range{Lo: 'a', Hi: 'z'}), ast.NewNode(ast.KindRange, ast.Range{Lo: 'a', Hi: 'x', Not: true}), ast.NewNode(ast.KindAny, nil), ), result: match.NewBTree( match.NewRow( 2, match.Matchers{ match.NewRange('a', 'z', false), match.NewRange('a', 'x', true), }..., ), nil, match.NewSuper(), ), }, { ast: ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindAnyOf, nil, ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindList, ast.List{Chars: "abc"}), ast.NewNode(ast.KindText, ast.Text{"ghi"}), ), ast.NewNode(ast.KindPattern, nil, ast.NewNode(ast.KindText, ast.Text{"abc"}), ast.NewNode(ast.KindList, ast.List{Chars: "def"}), ast.NewNode(ast.KindText, ast.Text{"ghi"}), ), ), ), result: match.NewRow( 7, match.Matchers{ match.NewText("abc"), match.AnyOf{Matchers: match.Matchers{ match.NewList([]rune{'a', 'b', 'c'}, false), match.NewList([]rune{'d', 'e', 'f'}, false), }}, match.NewText("ghi"), }..., ), }, } { m, err := Compile(test.ast, test.sep) if err != nil { t.Errorf("compilation error: %s", err) continue } if !reflect.DeepEqual(m, test.result) { t.Errorf("[%d] Compile():\nexp: %#v\nact: %#v\n\ngraphviz:\nexp:\n%s\nact:\n%s\n", id, test.result, m, debug.Graphviz("", test.result.(match.Matcher)), debug.Graphviz("", m.(match.Matcher))) continue } } }