main.cc

Go to the documentation of this file.

#if __APPLE__
#include "app/platform/app_delegate.h"
#endif
 
#ifdef __EMSCRIPTEN__
#include <emscripten.h>
#include "app/platform/wasm/wasm_bootstrap.h"
#include "app/platform/wasm/wasm_collaboration.h"
#endif
 
#define IMGUI_DEFINE_MATH_OPERATORS
#include <algorithm>
#include <chrono>
#include <fstream>
#include <iostream>
#include <set>
#include <thread>
 
#include "absl/debugging/symbolize.h"
#include "absl/strings/ascii.h"
#include "absl/strings/str_split.h"
#include "absl/strings/strip.h"
#include "app/application.h"
#include "app/controller.h"
#include "app/editor/editor_manager.h"
#include "app/emu/debug/symbol_provider.h"
#include "app/startup_flags.h"
#if !defined(__EMSCRIPTEN__) && defined(YAZE_HTTP_API_ENABLED)
#include "app/service/editor_http_server.h"
#endif
#include "core/features.h"
#include "util/crash_handler.h"
#include "util/flag.h"
#include "util/log.h"
#include "util/platform_paths.h"
#include "yaze.h"
 
// ============================================================================
// Global Accessors for WASM Integration
DEFINE_FLAG(std::string, log_file, "", "Output log file path for debugging.");
DEFINE_FLAG(std::string, rom_file, "", "ROM file to load on startup.");
DEFINE_FLAG(bool, debug, false, "Enable debug logging and verbose output.");
DEFINE_FLAG(std::string, log_level, "info",
            "Minimum log level: debug, info, warn, error, or fatal.");
DEFINE_FLAG(bool, log_to_console, false,
            "Mirror logs to stderr even when writing to a file.");
DEFINE_FLAG(
    std::string, log_categories, "",
    "Comma-separated list of log categories to enable or disable. "
    "Prefix with '-' to disable a category. "
    "Example: \"Room,DungeonEditor\" (allowlist) or \"-Input,-Graphics\" "
    "(blocklist).");
 
// Navigation flags
DEFINE_FLAG(
    std::string, editor, "",
    "The editor to open on startup (e.g., Dungeon, Overworld, Assembly).");
 
DEFINE_FLAG(std::string, open_panels, "",
            "Comma-separated list of panel IDs to open (e.g. "
            "'dungeon.room_list,emulator.cpu_debugger')");
 
// UI visibility flags
DEFINE_FLAG(std::string, startup_welcome, "auto",
            "Welcome screen behavior at startup: auto, show, or hide.");
DEFINE_FLAG(std::string, startup_dashboard, "auto",
            "Dashboard panel behavior at startup: auto, show, or hide.");
DEFINE_FLAG(std::string, startup_sidebar, "auto",
            "Panel sidebar visibility at startup: auto, show, or hide.");
DEFINE_FLAG(std::string, asset_mode, "auto",
            "Asset load mode: auto, full, or lazy.");
 
DEFINE_FLAG(int, room, -1, "Open Dungeon Editor at specific room ID (0-295).");
DEFINE_FLAG(int, map, -1, "Open Overworld Editor at specific map ID (0-159).");
 
// AI Agent API flags
DEFINE_FLAG(bool, enable_api, false, "Enable the AI Agent API server.");
DEFINE_FLAG(bool, editor_control, false,
            "Enable guarded editor commands on the loopback HTTP API.");
DEFINE_FLAG(int, api_port, 8080, "Port for the AI Agent API server.");
 
DEFINE_FLAG(bool, headless, false,
            "Run in headless mode without a GUI window.");
DEFINE_FLAG(bool, service, false,
            "Run in service mode (GUI backend initialized but window hidden).");
DEFINE_FLAG(
    bool, server, false,
    "Run in server mode (implies --enable_api, --enable_test_harness). "
    "Defaults to --headless unless --service or --no-headless is specified.");
 
#ifdef YAZE_WITH_GRPC
// gRPC test harness flags
DEFINE_FLAG(bool, enable_test_harness, false,
            "Start gRPC test harness server for automated GUI testing.");
 
DEFINE_FLAG(int, test_harness_port, 50052,
            "Port for Unified gRPC server (default: 50052).");
DEFINE_FLAG(std::string, backend, "internal",
            "Emulator backend for gRPC service: 'internal' or 'mesen'.");
#endif
 
// Symbol Export Flags
DEFINE_FLAG(std::string, export_symbols, "",
            "Export symbols to file (requires --rom_file).");
DEFINE_FLAG(std::string, symbol_format, "mesen",
            "Format for symbol export: mesen, wla, asar, bsnes.");
DEFINE_FLAG(std::string, load_symbols, "",
            "Load symbol file (.mlb, .sym, .asm) on startup.");
DEFINE_FLAG(std::string, load_asar_symbols, "",
            "Load Asar symbols from directory on startup.");
DEFINE_FLAG(bool, export_symbols_fast, false,
            "Export symbols without initializing UI. Requires --load_symbols "
            "or --load_asar_symbols.");
 
// ============================================================================
// Global Accessors for WASM Integration
// These are used by yaze_debug_inspector.cc and wasm_terminal_bridge.cc
// ============================================================================
namespace yaze {
namespace emu {
class Emulator;
}
namespace editor {
class EditorManager;
}
}  // namespace yaze
 
namespace yaze::app {
 
yaze::emu::Emulator* GetGlobalEmulator() {
  auto* ctrl = yaze::Application::Instance().GetController();
  if (ctrl && ctrl->editor_manager()) {
    return &ctrl->editor_manager()->emulator();
  }
  return nullptr;
}
 
yaze::editor::EditorManager* GetGlobalEditorManager() {
  auto* ctrl = yaze::Application::Instance().GetController();
  if (ctrl) {
    return ctrl->editor_manager();
  }
  return nullptr;
}
 
}  // namespace yaze::app
 
namespace {
 
yaze::util::LogLevel ParseLogLevelFlag(const std::string& raw_level,
                                       bool debug_flag) {
  if (debug_flag) {
    return yaze::util::LogLevel::YAZE_DEBUG;
  }
 
  const std::string lower = absl::AsciiStrToLower(raw_level);
  if (lower == "debug") {
    return yaze::util::LogLevel::YAZE_DEBUG;
  }
  if (lower == "warn" || lower == "warning") {
    return yaze::util::LogLevel::WARNING;
  }
  if (lower == "error") {
    return yaze::util::LogLevel::ERROR;
  }
  if (lower == "fatal") {
    return yaze::util::LogLevel::FATAL;
  }
  return yaze::util::LogLevel::INFO;
}
 
std::set<std::string> ParseLogCategories(const std::string& raw) {
  std::set<std::string> categories;
  for (absl::string_view token :
       absl::StrSplit(raw, ',', absl::SkipWhitespace())) {
    if (!token.empty()) {
      categories.insert(std::string(absl::StripAsciiWhitespace(token)));
    }
  }
  return categories;
}
 
const char* LogLevelToString(yaze::util::LogLevel level) {
  switch (level) {
    case yaze::util::LogLevel::YAZE_DEBUG:
      return "debug";
    case yaze::util::LogLevel::INFO:
      return "info";
    case yaze::util::LogLevel::WARNING:
      return "warn";
    case yaze::util::LogLevel::ERROR:
      return "error";
    case yaze::util::LogLevel::FATAL:
      return "fatal";
  }
  return "info";
}
 
std::vector<std::string> ParseCommaList(const std::string& raw) {
  std::vector<std::string> tokens;
  for (absl::string_view token :
       absl::StrSplit(raw, ',', absl::SkipWhitespace())) {
    if (!token.empty()) {
      tokens.emplace_back(absl::StripAsciiWhitespace(token));
    }
  }
  return tokens;
}
 
// Handles --help/-h and --version before normal flag parsing so they work
// regardless of any other arguments. Returns true if handled (caller exits).
bool HandleHelpOrVersion(int argc, char** argv) {
  for (int i = 1; i < argc; ++i) {
    const std::string arg = argv[i];
    if (arg == "--version") {
      std::cout << "yaze " << YAZE_VERSION_STRING << "\n";
      return true;
    }
    if (arg == "--help" || arg == "-h") {
      std::cout << argv[0] << " (yaze " << YAZE_VERSION_STRING << ")\n"
                << "Usage: " << argv[0] << " [flags]\n\nFlags:\n";
      auto flags = yaze::util::global_flag_registry()->AllFlags();
      std::sort(flags.begin(), flags.end(),
                [](const yaze::util::IFlag* a, const yaze::util::IFlag* b) {
                  return a->name() < b->name();
                });
      for (const auto* flag : flags) {
        std::cout << "  " << flag->name() << "\n      " << flag->help() << "\n";
      }
      return true;
    }
  }
  return false;
}
 
}  // namespace
 
void TickFrame() {
  yaze::Application::Instance().Tick();
}
 
// Helper Functions
 
void SetupCrashHandling() {
#ifndef __EMSCRIPTEN__
  yaze::util::CrashHandler::Initialize(YAZE_VERSION_STRING);
  yaze::util::CrashHandler::CleanupOldLogs(5);
#endif
}
 
yaze::util::LogLevel ResolveLogConfig(bool debug_flag,
                                      bool log_to_console_flag) {
  yaze::util::LogLevel log_level =
      ParseLogLevelFlag(FLAGS_log_level->Get(), debug_flag);
  std::set<std::string> log_categories =
      ParseLogCategories(FLAGS_log_categories->Get());
 
  std::string log_path = FLAGS_log_file->Get();
  if (log_path.empty()) {
    auto logs_dir = yaze::util::PlatformPaths::GetAppDataSubdirectory("logs");
    if (logs_dir.ok()) {
      log_path = (*logs_dir / "yaze.log").string();
    }
  }
 
  yaze::util::LogManager::instance().configure(log_level, log_path,
                                               log_categories);
 
  if (debug_flag || log_to_console_flag) {
    yaze::core::FeatureFlags::get().kLogToConsole = true;
  }
  if (debug_flag) {
    LOG_INFO("Main", "🚀 YAZE started in debug mode");
  }
  LOG_INFO("Main",
           "Logging configured (level=%s, file=%s, console=%s, categories=%zu)",
           LogLevelToString(log_level),
           log_path.empty() ? "<stderr>" : log_path.c_str(),
           (debug_flag || log_to_console_flag) ? "on" : "off",
           log_categories.size());
  return log_level;
}
 
yaze::AppConfig BuildAppConfig(yaze::util::LogLevel log_level,
                               std::string log_path) {
  yaze::AppConfig config;
 
  bool server_mode = FLAGS_server->Get();
  bool service_mode = FLAGS_service->Get();
 
  config.headless = FLAGS_headless->Get() || (server_mode && !service_mode);
  config.service_mode = service_mode;
  config.editor_control = FLAGS_editor_control->Get();
  config.enable_api = FLAGS_enable_api->Get() || server_mode || service_mode ||
                      config.editor_control;
 
#ifdef YAZE_WITH_GRPC
  config.enable_test_harness =
      FLAGS_enable_test_harness->Get() || server_mode || service_mode;
  config.test_harness_port = FLAGS_test_harness_port->Get();
  config.backend = FLAGS_backend->Get();
#endif
 
  config.rom_file = FLAGS_rom_file->Get();
  config.log_file = log_path;
  config.debug = (log_level == yaze::util::LogLevel::YAZE_DEBUG);
  config.log_categories = FLAGS_log_categories->Get();
  config.startup_editor = FLAGS_editor->Get();
  config.jump_to_room = FLAGS_room->Get();
  config.jump_to_map = FLAGS_map->Get();
  config.api_port = FLAGS_api_port->Get();
 
  config.welcome_mode =
      yaze::StartupVisibilityFromString(FLAGS_startup_welcome->Get());
  config.dashboard_mode =
      yaze::StartupVisibilityFromString(FLAGS_startup_dashboard->Get());
  config.sidebar_mode =
      yaze::StartupVisibilityFromString(FLAGS_startup_sidebar->Get());
  config.asset_load_mode =
      yaze::AssetLoadModeFromString(FLAGS_asset_mode->Get());
  if (config.asset_load_mode == yaze::AssetLoadMode::kAuto) {
    config.asset_load_mode = (config.headless || server_mode || service_mode)
                                 ? yaze::AssetLoadMode::kLazy
                                 : yaze::AssetLoadMode::kFull;
  }
 
  if (!FLAGS_open_panels->Get().empty()) {
    config.open_panels = ParseCommaList(FLAGS_open_panels->Get());
  }
  return config;
}
 
bool RunSymbolExportFastPath() {
  if (!FLAGS_export_symbols->Get().empty() &&
      FLAGS_export_symbols_fast->Get()) {
    yaze::emu::debug::SymbolProvider symbols;
    bool loaded = false;
 
    if (!FLAGS_load_symbols->Get().empty()) {
      LOG_INFO("Main", "Loading symbols from %s...",
               FLAGS_load_symbols->Get().c_str());
      auto status = symbols.LoadSymbolFile(FLAGS_load_symbols->Get());
      if (!status.ok()) {
        LOG_ERROR("Main", "Failed to load symbols: %s",
                  status.ToString().c_str());
      } else {
        loaded = true;
      }
    }
 
    if (!FLAGS_load_asar_symbols->Get().empty()) {
      LOG_INFO("Main", "Loading Asar symbols from %s...",
               FLAGS_load_asar_symbols->Get().c_str());
      auto status =
          symbols.LoadAsarAsmDirectory(FLAGS_load_asar_symbols->Get());
      if (!status.ok()) {
        LOG_ERROR("Main", "Failed to load Asar symbols: %s",
                  status.ToString().c_str());
      } else {
        loaded = true;
      }
    }
 
    if (!loaded) {
      LOG_ERROR(
          "Main",
          "No symbols loaded. Use --load_symbols or --load_asar_symbols.");
      return false;
    }
 
    LOG_INFO("Main", "Exporting symbols to %s...",
             FLAGS_export_symbols->Get().c_str());
    yaze::emu::debug::SymbolFormat format =
        yaze::emu::debug::SymbolFormat::kMesen;
    std::string format_str = absl::AsciiStrToLower(FLAGS_symbol_format->Get());
    if (format_str == "asar")
      format = yaze::emu::debug::SymbolFormat::kAsar;
    else if (format_str == "wla")
      format = yaze::emu::debug::SymbolFormat::kWlaDx;
    else if (format_str == "bsnes")
      format = yaze::emu::debug::SymbolFormat::kBsnes;
 
    auto export_or = symbols.ExportSymbols(format);
    if (export_or.ok()) {
      std::ofstream out(FLAGS_export_symbols->Get());
      if (out.is_open()) {
        out << *export_or;
        LOG_INFO("Main", "Symbols exported successfully (fast path)");
        return true;
      }
      LOG_ERROR("Main", "Failed to open output file: %s",
                FLAGS_export_symbols->Get().c_str());
      return false;
    }
    LOG_ERROR("Main", "Failed to export symbols: %s",
              export_or.status().ToString().c_str());
    return false;
  }
  return false;
}
 
// Main Entry Point
 
int main(int argc, char** argv) {
  absl::InitializeSymbolizer(argv[0]);
 
  if (HandleHelpOrVersion(argc, argv)) {
    return EXIT_SUCCESS;
  }
 
  SetupCrashHandling();
 
  yaze::util::FlagParser parser(yaze::util::global_flag_registry());
  RETURN_IF_EXCEPTION(parser.Parse(argc, argv));
 
  // Set up logging
  const bool debug_flag = FLAGS_debug->Get();
  const bool log_to_console_flag = FLAGS_log_to_console->Get();
  auto log_level = ResolveLogConfig(debug_flag, log_to_console_flag);
  std::string log_path = FLAGS_log_file->Get();
  if (log_path.empty()) {
    auto logs_dir = yaze::util::PlatformPaths::GetAppDataSubdirectory("logs");
    if (logs_dir.ok()) {
      log_path = (*logs_dir / "yaze.log").string();
    }
  }
 
  // Build AppConfig from flags
  auto config = BuildAppConfig(log_level, log_path);
 
  // Fast symbol export path (no UI initialization)
  if (RunSymbolExportFastPath()) {
    return EXIT_SUCCESS;
  }
#if !defined(__EMSCRIPTEN__)
  // If fast export failed but was requested, and we are not emscripten, we should probably exit or continue?
  // The helper returns false if not requested OR if failed.
  // If requested and failed, it logs errors.
  // To match original logic:
  if (!FLAGS_export_symbols->Get().empty() &&
      FLAGS_export_symbols_fast->Get()) {
    return EXIT_FAILURE;
  }
#endif
 
#ifdef __APPLE__
  if (!config.headless) {
    return yaze_run_cocoa_app_delegate(config);
  }
#endif
 
#if defined(_WIN32) && !defined(__EMSCRIPTEN__)
  SDL_SetMainReady();
#endif
 
#ifdef __EMSCRIPTEN__
  yaze::app::wasm::InitializeWasmPlatform();
 
  // Store config for deferred initialization
  static yaze::AppConfig s_wasm_config = config;
  static bool s_wasm_initialized = false;
 
  // Main loop that handles deferred initialization for filesystem readiness
  auto WasmMainLoop = []() {
    // Wait for filesystem to be ready before initializing application
    if (!s_wasm_initialized) {
      if (yaze::app::wasm::IsFileSystemReady()) {
        LOG_INFO("Main", "Filesystem ready, initializing application...");
        yaze::Application::Instance().Initialize(s_wasm_config);
        s_wasm_initialized = true;
      } else {
        // Still waiting for filesystem - do nothing this frame
        return;
      }
    }
 
    // Normal tick once initialized
    TickFrame();
  };
 
  // Use 0 for frame rate to enable requestAnimationFrame (better performance)
  // The third parameter (1) simulates infinite loop
  emscripten_set_main_loop(WasmMainLoop, 0, 1);
#else
  // Desktop Main Loop (Linux/Windows)
 
  // API Server
#if defined(YAZE_HTTP_API_ENABLED)
  std::unique_ptr<yaze::cli::api::HttpServer> api_server;
#endif
#if !defined(YAZE_HTTP_API_ENABLED)
  if (config.enable_api) {
    LOG_WARN("Main",
             "HTTP API requested but not enabled at build time "
             "(set -DYAZE_ENABLE_HTTP_API=ON and rebuild).");
  }
#endif
 
  yaze::Application::Instance().Initialize(config);
 
#if defined(YAZE_HTTP_API_ENABLED)
  api_server = yaze::app::SetupEditorHttpServer(config);
#endif
 
  // Handle symbol loading if requested
  auto* ctrl = yaze::Application::Instance().GetController();
  if (ctrl && ctrl->editor_manager()) {
    auto& symbols = ctrl->editor_manager()->emulator().symbol_provider();
 
    if (!FLAGS_load_symbols->Get().empty()) {
      LOG_INFO("Main", "Loading symbols from %s...",
               FLAGS_load_symbols->Get().c_str());
      auto status = symbols.LoadSymbolFile(FLAGS_load_symbols->Get());
      if (!status.ok()) {
        LOG_ERROR("Main", "Failed to load symbols: %s",
                  status.ToString().c_str());
      }
    }
 
    if (!FLAGS_load_asar_symbols->Get().empty()) {
      LOG_INFO("Main", "Loading Asar symbols from %s...",
               FLAGS_load_asar_symbols->Get().c_str());
      auto status =
          symbols.LoadAsarAsmDirectory(FLAGS_load_asar_symbols->Get());
      if (!status.ok()) {
        LOG_ERROR("Main", "Failed to load Asar symbols: %s",
                  status.ToString().c_str());
      }
    }
  }
 
  // Handle symbol export if requested (GUI mode)
  if (!FLAGS_export_symbols->Get().empty()) {
    LOG_INFO("Main", "Exporting symbols to %s...",
             FLAGS_export_symbols->Get().c_str());
 
    auto* ctrl = yaze::Application::Instance().GetController();
    if (ctrl && ctrl->editor_manager()) {
      auto* manager = ctrl->editor_manager();
 
      // Attempt to find symbols from the current session
      // For now, we'll assume they are in the emulator's symbol provider
      auto& symbols = manager->emulator().symbol_provider();
 
      yaze::emu::debug::SymbolFormat format =
          yaze::emu::debug::SymbolFormat::kMesen;
      std::string format_str =
          absl::AsciiStrToLower(FLAGS_symbol_format->Get());
      if (format_str == "asar")
        format = yaze::emu::debug::SymbolFormat::kAsar;
      else if (format_str == "wla")
        format = yaze::emu::debug::SymbolFormat::kWlaDx;
      else if (format_str == "bsnes")
        format = yaze::emu::debug::SymbolFormat::kBsnes;
 
      auto export_or = symbols.ExportSymbols(format);
      if (export_or.ok()) {
        std::ofstream out(FLAGS_export_symbols->Get());
        if (out.is_open()) {
          out << *export_or;
          LOG_INFO("Main", "Symbols exported successfully");
        } else {
          LOG_ERROR("Main", "Failed to open output file: %s",
                    FLAGS_export_symbols->Get().c_str());
        }
      } else {
        LOG_ERROR("Main", "Failed to export symbols: %s",
                  export_or.status().ToString().c_str());
      }
    }
 
#if defined(YAZE_HTTP_API_ENABLED)
    if (api_server)
      api_server->Stop();
#endif
    yaze::Application::Instance().Shutdown();
    return EXIT_SUCCESS;
  }
 
  if (config.headless) {
    LOG_INFO("Main", "Running in HEADLESS mode (no GUI window)");
    // Optimized headless loop
    while (yaze::Application::Instance().GetController()->IsActive()) {
      yaze::Application::Instance().Tick();
      // Sleep to reduce CPU usage in headless mode
      // 60Hz = ~16ms, but we can sleep longer if just serving API/gRPC
      std::this_thread::sleep_for(std::chrono::milliseconds(16));
    }
  } else {
    // Normal GUI loop (also used for Service Mode with hidden window)
    if (config.service_mode) {
      LOG_INFO("Main", "Running in SERVICE mode (Hidden GUI window)");
    }
    while (yaze::Application::Instance().GetController()->IsActive()) {
      TickFrame();
    }
  }
 
#if defined(YAZE_HTTP_API_ENABLED)
  if (api_server) {
    api_server->Stop();
  }
#endif
 
  yaze::Application::Instance().Shutdown();
 
#endif  // __EMSCRIPTEN__
 
  return EXIT_SUCCESS;
}