fix: env_var & = in flags

Author: blindFS

languages/nu.scm

@@ -27,15 +27,20 @@
   "match"
   "try"
   "while"
+  (env_var)
 ] @append_space
 
 [
-  "="
   (match_guard)
   (short_flag)
   (long_flag)
 ] @prepend_space
 
+(decl_alias "=" @prepend_space)
+(stmt_let "=" @prepend_space)
+(stmt_mut "=" @prepend_space)
+(stmt_const "=" @prepend_space)
+
 (param_value
   "=" @append_space
 )
@@ -112,7 +117,7 @@
 )
 
 ;; special flags
-(_
+(
   [
     (short_flag)
     (long_flag)
@@ -174,6 +179,16 @@
   (block) @prepend_space
 )
 
+(returns
+  ":"? @do_nothing
+) @prepend_space
+
+(returns
+  type: _ @append_space
+  .
+  type: _
+)
+
 (decl_use
   (_) @prepend_space
 )

test/expected_command.nu

@@ -7,5 +7,5 @@ ls | get -i name
   ls
   | get -i name
 )
-^$cmd --arg1 -arg2 arg=value arg=($arg3)
+FOO=BAR BAR=BAZ ^$cmd --arg1=val1 -arg2 arg=value arg=($arg3)
 cat unknown.txt o+e> (null-device)

test/input_command.nu

@@ -5,5 +5,5 @@ ls | get -i name
 # external command
 ^git add (ls
   | get -i name)
-^$cmd  --arg1 -arg2  arg=value arg=($arg3)
+FOO=BAR  BAR=BAZ  ^$cmd  --arg1=val1 -arg2  arg=value arg=($arg3)
 cat unknown.txt o+e> (null-device)