Added missing modes on import

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main.cpp b/src/main.cpp
index 7b4d6f9..7f6267b 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -81,11 +81,20 @@ int main(int argc, char *argv[]) {
     const auto modding_import = modding_root->add_subcommand("import", "Import - Import modified files to generate C code\n");
     const auto modding_export = modding_root->add_subcommand("export", "Export - Export modified files to a folder\n");
 
+    modding_import->add_option("mode", mode, "code, binary or header")->required();
     modding_import->add_option("<baserom.z64>", filename, "")->required()->check(CLI::ExistingFile);
 
     modding_import->parse_complete_callback([&] {
         const auto instance = Companion::Instance = new Companion(filename, false, debug, true);
-        instance->Init(ExportType::Code);
+        if (mode == "code") {
+            instance->Init(ExportType::Code);
+        } else if (mode == "binary") {
+            instance->Init(ExportType::Binary);
+        } else if (mode == "header") {
+            instance->Init(ExportType::Header);
+        } else {
+            std::cout << "Invalid mode" << std::endl;
+        }
     });
 
     modding_export->add_option("<baserom.z64>", filename, "")->required()->check(CLI::ExistingFile);