kkk

Author: juleswritescode

crates/pgls_treesitter_grammar/grammar.js

@@ -2618,33 +2618,33 @@ module.exports = grammar({
         partialSeq(
           field("left", $.lhs_column_list),
           "=",
-          field(
-            "end",
-            choice(
-              wrapped_in_parenthesis(
-                comma_list(
-                  choice($._expression, $.all_fields, $.keyword_default),
-                  true,
-                ),
-              ),
-              partialSeq(
-                $.keyword_row,
-                wrapped_in_parenthesis(
-                  comma_list(
-                    choice($._expression, $.all_fields, $.keyword_default),
-                    false,
-                  ),
-                ),
-              ),
-              $.subquery,
-            ),
-          ),
+          field("end", $.rhs_column_list),
         ),
       ),
 
     lhs_column_list: ($) =>
       partialSeq("(", comma_list($._column_indirection, false), ")"),
 
+    rhs_column_list: ($) =>
+      choice(
+        wrapped_in_parenthesis(
+          comma_list(
+            choice($._expression, $.all_fields, $.keyword_default),
+            true,
+          ),
+        ),
+        partialSeq(
+          $.keyword_row,
+          wrapped_in_parenthesis(
+            comma_list(
+              choice($._expression, $.all_fields, $.keyword_default),
+              false,
+            ),
+          ),
+        ),
+        $.subquery,
+      ),
+
     table_option: ($) =>
       choice(
         seq(

crates/pgls_treesitter_grammar/tests/snapshots/grammar_tests__test_2_sql3.snap

@@ -18,14 +18,15 @@ program [0..55] 'update auth.users set (email, id) = ('my@mail.com', 1);'
         assignment [22..54] '(email, id) = ('my@mail.com', 1)' (@end)
           lhs_column_list [22..33] '(email, id)' (@left)
             ( [22..23] '('
-            column_identifier [23..28] 'email' (@end)
+            column_identifier [23..28] 'email'
             , [28..29] ','
-            column_identifier [30..32] 'id' (@end)
+            column_identifier [30..32] 'id'
             ) [32..33] ')'
           = [34..35] '='
-          ( [36..37] '(' (@right)
-          literal [37..50] ''my@mail.com'' (@right)
-          , [50..51] ',' (@right)
-          literal [52..53] '1' (@right)
-          ) [53..54] ')' (@end)
+          rhs_column_list [36..54] '('my@mail.com', 1)' (@end)
+            ( [36..37] '('
+            literal [37..50] ''my@mail.com''
+            , [50..51] ','
+            literal [52..53] '1'
+            ) [53..54] ')' (@end)
   ; [54..55] ';'