Add output option to codegen

thelang-io · May 7, 2023 · 5c730e6 · 5c730e6
1 parent b92119a
commit 5c730e6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/Codegen.cpp b/src/Codegen.cpp
@@ -94,7 +94,7 @@ void Codegen::compile (
 ) {
   auto code = std::get<0>(result);
   auto flags = std::get<1>(result);
-  auto f = std::ofstream("build/output.c");
+  auto f = std::ofstream(path + ".c");
 
   f << code;
   f.close();
@@ -123,10 +123,10 @@ void Codegen::compile (
     }
   }
 
-  auto cmd = compiler + " build/output.c " + libraries + "-w -o " + path + flagsStr + (debug ? " -g" : "");
+  auto cmd = compiler + " " + path + ".c " + libraries + "-w -o " + path + flagsStr + (debug ? " -g" : "");
   auto returnCode = std::system(cmd.c_str());
 
-  std::filesystem::remove("build/output.c");
+  std::filesystem::remove(path + ".c");
 
   if (returnCode != 0) {
     throw Error("failed to compile generated code");

diff --git a/src/main.cpp b/src/main.cpp
@@ -31,6 +31,7 @@ int main (int argc, char *argv[]) {
     auto isLex = false;
     auto isParse = false;
     auto fileName = std::optional<std::string>{};
+    auto output = std::string("build/a.out");
     auto platform = std::string("default");
 
     for (auto i = 1; i < argc; i++) {
@@ -57,7 +58,9 @@ int main (int argc, char *argv[]) {
         auto optName = arg.substr(2, arg.find('=') - 2);
         auto optVal = arg.substr(arg.find('=') + 1);
 
-        if (optName == "platform") {
+        if (optName == "output") {
+          output = optVal;
+        } else if (optName == "platform") {
           if (std::set<std::string>{"linux", "macos", "windows"}.contains(optVal)) {
             platform = optVal;
           } else {
@@ -122,7 +125,7 @@ int main (int argc, char *argv[]) {
       return EXIT_SUCCESS;
     }
 
-    Codegen::compile("build/a.out", result, platform);
+    Codegen::compile(output, result, platform);
     return EXIT_SUCCESS;
   } catch (const Error &err) {
     std::cerr << "Error: " << err.what() << std::endl;