Better handling of shebangs and comments at the top of a file.

compatibility/compatibility.nom

@@ -93,6 +93,7 @@ externally [..]
             set %with_upgraded_args 's metatable to (%tree 's metatable)
             %tree = (call %UPGRADES.%ver with [%with_upgraded_args, %end_version])
     
+    %tree.shebang = "#!/usr/bin/env nomsu -V\%end_version"
     return %tree
 
 externally (%tree upgraded from %start_version) means (..)

nomsu.4.peg

@@ -6,13 +6,13 @@ file:
     {:curr_indent: %nil :}
     !.
 
-shebang: "#!" (!"nomsu" [^%nl])* "nomsu" ws+ "-V" ws* {:version: [0-9.]+ :} [^%nl]*
+shebang: "#!" (!"nomsu" [^%nl])* "nomsu" ws+ "-V" ws* {:version: [0-9.]+ :} [^%nl]* (%nl / !.)
 
 eof: !.
 
 file_chunks (FileChunks):
     {:curr_indent: ' '* :}
-    shebang? comment? blank_lines?
+    {:shebang: shebang :}?
     (top_block (nl_nodent section_division top_block)*)
     blank_lines?
     ws* unexpected_chunk?

nomsu_decompiler.lua

@@ -223,6 +223,9 @@ tree_to_nomsu = function(tree)
   end
   local _exp_0 = tree.type
   if "FileChunks" == _exp_0 then
+    if tree.shebang then
+      nomsu:append(tree.shebang, "\n")
+    end
     local should_clump
     should_clump = function(prev_line, line)
       if prev_line.type == "Action" and line.type == "Action" then

nomsu_decompiler.moon

@@ -172,6 +172,8 @@ tree_to_nomsu = (tree)->
 
     switch tree.type
         when "FileChunks"
+            if tree.shebang
+                nomsu\append tree.shebang, "\n"
             should_clump = (prev_line, line)->
                 if prev_line.type == "Action" and line.type == "Action"
                     if prev_line.stub == "use" then return line.stub == "use"

tools/upgrade.nom

@@ -45,9 +45,7 @@ for %path in %args:
         %tree = (%code parsed)
         %uptree = (..)
             %tree upgraded from (%start_version or (%tree.version or (Nomsu version))) to %version
-        %text = "\
-            ..#!/usr/bin/env nomsu -V\%version
-            \(%uptree as nomsu)"
+        %text = (%uptree as nomsu)
         
         when:
             %inplace: