Hashbang

src/compiler/Exec_phr.sig

@@ -1,5 +1,5 @@
 val execToplevelPhrase: Asynt.Dec -> unit;
 val quietdec : bool ref
-
+val hashbang : bool ref
 
 

src/compiler/Exec_phr.sml

@@ -8,6 +8,10 @@ open Symtable Rtvals Load_phr;
 
 val quietdec = ref false ;
 
+(* Is enabled if running with -hashbang arg: *)
+
+val hashbang = ref false ;
+
 (* Executing a top-level declaration. *)
 
 local

src/compiler/Lexer.lex

@@ -9,11 +9,11 @@ datatype lexingMode =
     NORMALlm
   | QUOTElm
   | ANTIQUOTElm
+  | HASHBANGlm
 
 val lexingMode = ref NORMALlm
 
 val parCount = Stack.new() : int Stack.t
-
 fun resetLexerState() =
 (
   lexingMode := NORMALlm;
@@ -207,6 +207,10 @@ fun scanString scan lexbuf =
   setLexStartPos lexbuf (!savedLexemeStart - getLexAbsPos lexbuf)
 )
 
+(* enable support for Shebang/Hashbang *)
+fun enableHashbang b = (if b
+                        then lexingMode := HASHBANGlm
+                        else lexingMode := NORMALlm; ());
 }
 
 rule Token = parse
@@ -221,12 +225,16 @@ rule Token = parse
                case !lexingMode of
                    NORMALlm =>
                      QUOTER (get_stored_string())
+                 | HASHBANGlm =>
+                     QUOTER (get_stored_string())
                  | ANTIQUOTElm =>
                      QUOTEM (get_stored_string())
                  | QUOTElm =>
                      fatalError "Token")
           | ANTIQUOTElm =>
               AntiQuotation lexbuf
+          | HASHBANGlm =>
+              Hashbang lexbuf
       }
 
 and TokenN = parse
@@ -461,6 +469,12 @@ and AntiQuotation = parse
       { 
         skipString "ill-formed antiquotation" SkipQuotation lexbuf
       }
+and Hashbang = parse
+   "#!" [^`\n` `\r`]*
+      { lexingMode := NORMALlm; Token lexbuf }
+  | ""
+      { lexingMode := NORMALlm; Token lexbuf }
+
 ;
 
 

src/compiler/Lexer.sig

@@ -1,3 +1,6 @@
+(* Lexer.sig *)
+
+val enableHashbang  : bool -> unit;
 val quotation : bool ref;
 val resetLexerState : unit -> unit;
 val Token : Lexing.lexbuf -> Parser.token;

src/compiler/Maint.sml

@@ -3,6 +3,7 @@
 open List BasicIO Nonstdio;
 open Miscsys Memory Fnlib Config Mixture Location Units Smlperv Rtvals  Smltop;
 open Types (* cvr *);
+open Lexer  (*to set hashbang*);
 val initialFiles = ref ([] : string list);
 
 (* Initial loop *)
@@ -72,6 +73,13 @@ fun set_value_polymorphism b _ =
 fun set_quietdec b _ =
   Exec_phr.quietdec := b;
 
+fun set_hashbang b _ =
+  let in
+    Lexer.enableHashbang b;
+    Exec_phr.quietdec := b;
+    Exec_phr.hashbang := b
+  end;
+
 fun add_include d =
   load_path := !load_path @ [d];
 
@@ -117,6 +125,8 @@ fun main () =
                ("-perv",      Arg.String perv_set),
                ("-imptypes",  Arg.Unit (set_value_polymorphism false)),
                ("-valuepoly", Arg.Unit (set_value_polymorphism true)),
+               ("-hashbang",  Arg.Unit (set_hashbang true)),
+               ("-s",         Arg.Unit (set_hashbang true)),
                ("-quietdec",  Arg.Unit (set_quietdec true)),
                ("-msgstyle",  Arg.String set_msgstyle),
                ("-m",         Arg.String set_msgstyle),
@@ -148,7 +158,8 @@ fun main () =
     resetTypes(); 
     Miscsys.catch_interrupt true;
     input_lexbuf := Compiler.createLexerStream std_in;
-    (initial_loop() handle EndOfFile => ());
+    (initial_loop() handle EndOfFile =>
+         if !Exec_phr.hashbang then BasicIO.exit 0 else ());
     main_loop()
   end
   handle

src/launch/mosml.tpl

@@ -24,6 +24,9 @@ while : ; do
     -quietdec)
       options="$options -quietdec"
       ;;
+    -s|-hashbang)
+      options="$options -hashbang"
+      ;;
     -valuepoly)
       options="$options -valuepoly"
       ;;