tree-sitter-qtpl/grammar.js at master · ezbuy/tree-sitter-qtpl · GitHub

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
112
113
114
115
116
117
118
/**
 * @file the tree-sitter binding for valyala/quicktemplate
 * @author Nace Sc <scbizu@gmail.com>
 * @license MIT
 */

/// <reference types="tree-sitter-cli/dsl" />
// @ts-check

module.exports = grammar({
  name: "qtpl",

  extras: ($) => [/\s/],

  rules: {
    source_file: ($) => repeat($._node),

    _node: ($) =>
      choice(
        $.html,
        $.comment_block,
        $.func_statement,
        $.for_statement,
        $.if_statement,
        $.switch_statement,
        $.plain_statement,
        $.stripspace_statement,
        $.output_statement,
        $.go_code,
        $.go_import
      ),

    func_statement: ($) =>
      seq(
        /\{%-?\s*func/,
        field("signature", $.go_fragment),
        /-?%\}/,
        repeat($._node),
        /\{%-?\s*endfunc\s*-?%\}/
      ),

    for_statement: ($) =>
      seq(
        /\{%-?\s*for/,
        field("condition", $.go_fragment),
        /-?%\}/,
        repeat($._node),
        /\{%-?\s*endfor\s*-?%\}/
      ),

    if_statement: ($) =>
      seq(
        /\{%-?\s*if/,
        field("condition", $.go_fragment),
        /-?%\}/,
        repeat($._node),
        repeat($.elseif_statement),
        optional($.else_statement),
        /\{%-?\s*endif\s*-?%\}/
      ),

    elseif_statement: ($) =>
      seq(
        /\{%-?\s*elseif/,
        field("condition", $.go_fragment),
        /-?%\}/,
        repeat($._node)
      ),

    else_statement: ($) => seq(/\{%-?\s*else\s*-?%\}/, repeat($._node)),

    switch_statement: ($) =>
      prec.left(
        1,
        seq(
          /\{%-?\s*switch/,
          field("expression", $.go_fragment),
          /-?%\}/,
          repeat($.case_statement),
          /\{%-?\s*endswitch\s*-?%\}/
        )
      ),

    case_statement: ($) =>
      prec.right(
        0,
        seq(/\{%-?\s*(case|default)/, $.go_fragment, /-?%\}/, repeat($._node))
      ),

    plain_statement: ($) =>
      seq(/\{%-?\s*plain\s*-?%\}/, repeat($._node), /\{%-?\s*endplain\s*-?%\}/),

    stripspace_statement: ($) =>
      seq(
        /\{%-?\s*(stripspace|collapsespace)\s*-?%\}/,
        repeat($._node),
        /\{%-?\s*end(stripspace|collapsespace)\s*-?%\}/
      ),

    output_statement: ($) =>
      seq(
        /\{%-?(s|(du?l?)|(f(\.[0-9]+)?)|z|((q|j|u)z?)|v|-)?=?/,
        $.go_fragment,
        /-?%\}/
      ),

    go_code: ($) => seq(/\{%-?\s*code/, field("body", $.go_fragment), /-?%\}/),
    go_import: ($) =>
      seq(/\{%-?\s*import/, field("body", $.go_fragment), /-?%\}/),

    go_fragment: () => prec.right(repeat1(choice(/[^%]/, /%[^}]/))),

    comment_block: ($) =>
      seq("{% comment %}", /(?:.|\s)*?/, "{% endcomment %}"),

    html: () => /([^{]|\{[^%])+/,
  },
});