/***
  (C) 2025 Amlal El Mahrouss
 */

#ifdef __APPLE__

#include <LibCompiler/Defines.h>
#include <LibDebugger/POSIXMachContract.h>
#include <Vendor/Dialogs.h>
#include <cstdint>
#include <iostream>
#include <string>

static BOOL                                  kKeepRunning = false;
static LibDebugger::POSIX::POSIXMachContract kDebugger;
static LibDebugger::ProcessID                kPID           = 0L;
static LibDebugger::CAddress                 kActiveAddress = nullptr;
static std::string                           kPath          = "";

#define kBlank "\e[0;30m"
#define kRed "\e[0;31m"
#define kWhite "\e[0;97m"

#define kStdOut (std::cout << kRed << "dbg: " << kWhite)

/// @internal
/// @brief Handles CTRL-C signal on debugger.
static void dbgi_ctrlc_handler(std::int32_t _) {
  if (!kPID) {
    return;
  }

  auto list = kDebugger.Get();

  kDebugger.Break();

  pfd::notify("Debugger Event", "Breakpoint hit!");

  kKeepRunning = false;
}

LIBCOMPILER_MODULE(DebuggerMachPOSIX) {
  pfd::notify("Debugger Event",
              "Userland Debugger\n(C) 2025 Amlal El Mahrouss, all rights reserved.");

  if (argc >= 3 && std::string(argv[1]) == "-p" && argv[2] != nullptr) {
    kPath = argv[2];
    kDebugger.SetPath(kPath);

    kStdOut << "[+] Path set to: " << kPath << "\n";
  }

  ::signal(SIGINT, dbgi_ctrlc_handler);

  while (YES) {
    if (kKeepRunning) {
      continue;
    }

    std::string cmd;
    std::getline(std::cin, cmd);

    if (cmd == "c" || cmd == "cont" || cmd == "continue") {
      if (kDebugger.Continue()) {
        kKeepRunning = true;

        kStdOut << "[+] Continuing...\n";

        pfd::notify("Debugger Event", "Continuing...");
      }
    }

    if (cmd == "d" || cmd == "detach") kDebugger.Detach();

    if (cmd == "start") {
      kStdOut << "[?] Enter a argument to use: ";
      std::getline(std::cin, cmd);

      kDebugger.Attach(kPath, cmd, kPID);
    }

    if (cmd == "exit") {
      if (kPID > 0) kDebugger.Detach();

      break;
    }

    if (cmd == "break" || cmd == "b") {
      kStdOut << "[?] Enter a symbol to break on: ";

      std::getline(std::cin, cmd);

      if (kDebugger.Breakpoint(cmd)) {
        pfd::notify("Debugger Event", "Add Breakpoint at: " + cmd);
      }
    }
  }

  return EXIT_SUCCESS;
}

#endif