1 files changed, 70 insertions, 1 deletions
diff --git a/src/main.cpp b/src/main.cpp
index f35f93ce..55b6f9d4 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -4,6 +4,7 @@
 #include "parse/parseerror.hpp"
 #include "ast/ast.hpp"
 #include <serialiser_texttree.hpp>
+#include <cstring>
 
 extern AST::Crate Parse_Crate(::std::string mainfile);
 extern void ResolvePaths(AST::Crate& crate);
@@ -12,11 +13,71 @@ extern AST::Flat Convert_Flatten(const AST::Crate& crate);
 /// main!
 int main(int argc, char *argv[])
 {
+    const char *infile = NULL;
+    ::std::string   outfile;
+    const char *crate_path = ".";
+    const char *emit_type = "c";
+    for( int i = 1; i < argc; i ++ )
+    {
+        const char* arg = argv[i];
+        
+        if( arg[0] != '-' )
+        {
+            infile = arg;
+        }
+        else if( arg[1] != '-' )
+        {
+            arg ++; // eat '-'
+            for( ; *arg; arg ++ )
+            {
+                switch(*arg)
+                {
+                // "-o <file>" : Set output file
+                case 'o':
+                    if( i == argc - 1 ) {
+                        // TODO: BAIL!
+                        return 1;
+                    }
+                    outfile = argv[++i];
+                    break;
+                default:
+                    return 1;
+                }
+            }
+        }
+        else
+        {
+            if( strcmp(arg, "--crate-path") == 0 ) {
+                if( i == argc - 1 ) {
+                    // TODO: BAIL!
+                    return 1;
+                }
+                crate_path = argv[++i];
+            }
+            else if( strcmp(arg, "--emit") == 0 ) {
+                if( i == argc - 1 ) {
+                    // TODO: BAIL!
+                    return 1;
+                }
+                emit_type = argv[++i];
+            }
+            else {
+                return 1;
+            }
+        }
+    }
+    
+    if( outfile == "" )
+    {
+        outfile = infile;
+        outfile += ".o";
+    }
+    
     Serialiser_TextTree s_tt(::std::cout);
     Serialiser& s = s_tt;
     try
     {
-        AST::Crate crate = Parse_Crate("samples/1.rs");
+        AST::Crate crate = Parse_Crate(infile);
 
         s << crate;
     
@@ -26,6 +87,13 @@ int main(int argc, char *argv[])
 
         // Typecheck / type propagate module (type annotations of all values)
 
+        if( strcmp(emit_type, "ast") == 0 )
+        {
+            ::std::ofstream os(outfile);
+            Serialiser_TextTree os_tt(os);
+            ((Serialiser&)os_tt) << crate;
+            return 0;
+        }
         // Flatten modules into "mangled" set
         AST::Flat flat_crate = Convert_Flatten(crate);
 
@@ -35,6 +103,7 @@ int main(int argc, char *argv[])
     catch(const ParseError::Base& e)
     {
         ::std::cerr << "Parser Error: " << e.what() << ::std::endl;
+        return 2;
     }
     return 0;
 }