Add support for IF statements

b0o · Nov 22, 2024 · 4e7848d · 4e7848d
1 parent cd8d4d4
commit 4e7848d
Show file tree

Hide file tree

Showing 5 changed files with 177,044 additions and 171,796 deletions.
diff --git a/grammar.js b/grammar.js
@@ -129,7 +129,8 @@ module.exports = grammar(Python, {
     _compound_statement: ($, original) =>
       choice(
         original,
-        $.def_statement,
+        $.DEF_statement,
+        $.IF_statement,
         $.cdef_statement,
         $.ctypedef_statement,
         $.property_definition,
@@ -148,7 +149,7 @@ module.exports = grammar(Python, {
         ),
       ),
 
-    def_statement: $ =>
+    DEF_statement: $ =>
       seq(
         "DEF",
         field("name", $.identifier),
@@ -157,6 +158,28 @@ module.exports = grammar(Python, {
         $._newline,
       ),
 
+    IF_statement: $ => seq(
+      'IF',
+      field('condition', $.expression),
+      ':',
+      field('consequence', $._suite),
+      repeat(field('alternative', $.ELIF_clause)),
+      optional(field('alternative', $.ELSE_clause)),
+    ),
+
+    ELIF_clause: $ => seq(
+      'ELIF',
+      field('condition', $.expression),
+      ':',
+      field('consequence', $._suite),
+    ),
+
+    ELSE_clause: $ => seq(
+      'ELSE',
+      ':',
+      field('body', $._suite),
+    ),
+
     cdef_statement: $ =>
       seq(
         choice("cdef", "cpdef"),

diff --git a/src/grammar.json b/src/grammar.json
diff --git a/src/node-types.json b/src/node-types.json