cli improvements

src/compiler/NubLang.CLI/Program.cs

@@ -15,37 +15,24 @@ var options = new Options();
 for (var i = 0; i < args.Length; i++)
 {
     var arg = args[i];
-
-    if (arg == "-o")
+    switch (arg)
     {
-        i++;
-        if (i >= args.Length - 1)
+        case "-o":
         {
-            Console.Error.WriteLine($"{arg} must be followed by a value");
-            return 1;
-        }
+            ++i;
+            if (i >= args.Length)
+            {
+                return 1;
+            }
 
-        options.OutputPath = args[i];
-    }
-    else if (arg == "-c")
-    {
-        options.Link = false;
-    }
-    else if (arg == "-ssa")
-    {
-        options.EmitSsa = true;
-    }
-    else if (arg == "-asm")
-    {
-        options.EmitAsm = true;
-    }
-    else if (arg == "-obj")
-    {
-        options.EmitObj = true;
-    }
-    else
-    {
-        options.Files.Add(new SourceFile(arg));
+            options.OutputPath = args[i];
+            break;
+        }
+        default:
+        {
+            options.Files.Add(new SourceFile(arg));
+            break;
+        }
     }
 }
 
@@ -104,35 +91,31 @@ var objectFiles = new List<string>();
 for (var i = 0; i < typedSyntaxTrees.Count; i++)
 {
     var syntaxTree = typedSyntaxTrees[i];
-    var outFileName = Path.ChangeExtension(options.Files[i].Path, null);
+    var outFileName = Path.Combine("build", "code", Path.ChangeExtension(options.Files[i].Path, null));
+
+    var outFileDir = Path.GetDirectoryName(outFileName);
+    if (!string.IsNullOrEmpty(outFileDir))
+    {
+        Directory.CreateDirectory(outFileDir);
+    }
 
     var generator = new QBEGenerator(syntaxTree, typedDefinitionTable);
     var ssa = generator.Emit();
 
-    var ssaFilePath = $"{outFileName}.ssa";
+    var ssaFilePath = Path.ChangeExtension(outFileName, "ssa");
     File.WriteAllText(ssaFilePath, ssa);
 
-    var asmFilePath = $"{outFileName}.s";
+    var asmFilePath = Path.ChangeExtension(outFileName, "s");
     var qbeSuccess = await QBE.Invoke(ssaFilePath, asmFilePath);
 
-    if (!options.EmitSsa)
-    {
-        File.Delete(ssaFilePath);
-    }
-
     if (!qbeSuccess)
     {
         return 1;
     }
 
-    var objFilePath = $"{outFileName}.o";
+    var objFilePath = Path.ChangeExtension(outFileName, "o");
     var asmSuccess = await GCC.Assemble(asmFilePath, objFilePath);
 
-    if (!options.EmitAsm)
-    {
-        File.Delete(asmFilePath);
-    }
-
     if (!asmSuccess)
     {
         return 1;
@@ -141,50 +124,43 @@ for (var i = 0; i < typedSyntaxTrees.Count; i++)
     objectFiles.Add(objFilePath);
 }
 
-var runtimeFilePath = "libruntime_x64.a";
+const string runtimeName = "libruntime_x64.a";
 
 var resources = Assembly.GetExecutingAssembly().GetManifestResourceNames();
-var runtime = resources.First(r => r.EndsWith(runtimeFilePath));
+var runtime = resources.First(r => r.EndsWith(runtimeName));
 
 await using var reader = Assembly.GetExecutingAssembly().GetManifestResourceStream(runtime);
 
 if (reader == null)
 {
-    Console.Error.WriteLine($"Cannot open stream to '{runtimeFilePath}'");
+    Console.Error.WriteLine($"Cannot open stream to '{runtimeName}'");
     return 1;
 }
 
-await using var writer = new FileStream(runtimeFilePath, FileMode.Create);
+var runtimePath = Path.Combine("build", "runtime", runtimeName);
+var runtimeDir = Path.GetDirectoryName(runtimePath);
+if (!string.IsNullOrEmpty(runtimeDir))
+{
+    Directory.CreateDirectory(runtimeDir);
+}
+
+await using var writer = new FileStream(runtimePath, FileMode.Create);
 
 reader.CopyTo(writer);
 
-objectFiles.Add(runtimeFilePath);
+objectFiles.Add(runtimePath);
 
-if (options.Link)
+var outPath = options.OutputPath ?? Path.Combine("build", "out.a");
+var outDir = Path.GetDirectoryName(outPath);
+if (!string.IsNullOrEmpty(outDir))
 {
-    var outPath = options.OutputPath ?? "out";
-    var linkResult = await GCC.Link(outPath, objectFiles);
-    if (!linkResult)
-    {
-        return 1;
-    }
-}
-else
-{
-    var outPath = options.OutputPath ?? "out.a";
-    var archiveResult = await Archive.Invoke(outPath, objectFiles);
-    if (!archiveResult)
-    {
-        return 1;
-    }
+    Directory.CreateDirectory(outDir);
 }
 
-if (!options.EmitObj)
+var archiveResult = await Archive.Invoke(outPath, objectFiles);
+if (!archiveResult)
 {
-    foreach (var objectFile in objectFiles)
-    {
-        File.Delete(objectFile);
-    }
+    return 1;
 }
 
 return 0;