fix: add command parser for main

Author: vansangpfiev

engine/main.cc

@@ -2,9 +2,11 @@
 #include <drogon/drogon.h>
 #include <climits>  // for PATH_MAX
 #include <iostream>
+#include "controllers/command_line_parser.h"
 #include "cortex-common/cortexpythoni.h"
 #include "utils/cortex_utils.h"
 #include "utils/dylib.h"
+#include "utils/archive_utils.h"
 
 #if defined(__APPLE__) && defined(__MACH__)
 #include <libgen.h>  // for dirname()
@@ -20,6 +22,18 @@
 #endif
 
 int main(int argc, char* argv[]) {
+  // Create logs/ folder and setup log to file
+  std::filesystem::create_directory(cortex_utils::logs_folder);
+  trantor::AsyncFileLogger asyncFileLogger;
+  asyncFileLogger.setFileName(cortex_utils::logs_base_name);
+  asyncFileLogger.startLogging();
+  trantor::Logger::setOutputFunction(
+      [&](const char* msg, const uint64_t len) {
+        asyncFileLogger.output(msg, len);
+      },
+      [&]() { asyncFileLogger.flush(); });
+  asyncFileLogger.setFileSizeLimit(cortex_utils::log_file_size_limit);
+
   // Check if this process is for python execution
   if (argc > 1) {
     if (strcmp(argv[1], "--run_python_file") == 0) {
@@ -41,11 +55,17 @@ int main(int argc, char* argv[]) {
     }
   }
 
+  if (argc > 1) {
+    CommandLineParser clp;
+    clp.SetupCommand(argc, argv);
+    return 0;
+  }
+
   int thread_num = 1;
   std::string host = "127.0.0.1";
   int port = 3928;
 
-  // Number of cortex-cpp threads
+  // Number of cortex.cpp threads
   if (argc > 1) {
     thread_num = std::atoi(argv[1]);
   }
@@ -64,9 +84,9 @@ int main(int argc, char* argv[]) {
   int drogon_thread_num = std::max(thread_num, logical_cores);
   // cortex_utils::nitro_logo();
 #ifdef CORTEX_CPP_VERSION
-  LOG_INFO << "cortex-cpp version: " << CORTEX_CPP_VERSION;
+  LOG_INFO << "cortex.cpp version: " << CORTEX_CPP_VERSION;
 #else
-  LOG_INFO << "cortex-cpp version: undefined";
+  LOG_INFO << "cortex.cpp version: undefined";
 #endif
 
   LOG_INFO << "Server started, listening at: " << host << ":" << port;
@@ -78,4 +98,4 @@ int main(int argc, char* argv[]) {
   drogon::app().run();
 
   return 0;
-}
+}