#include "cli/handlers/agent/commands.h"
#include "cli/handlers/agent/todo_commands.h"
#include "cli/cli.h"

#include <string>
#include <vector>

#include "absl/flags/declare.h"
#include "absl/flags/flag.h"
#include "absl/status/status.h"

ABSL_DECLARE_FLAG(bool, quiet);

namespace yaze {
namespace cli {
namespace agent {
namespace {

constexpr absl::string_view kUsage =
  "Usage: agent <subcommand> [options]\n"
  "\n"
  "AI-Powered Agent Subcommands:\n"
  "  simple-chat              Simple text-based chat (recommended for testing)\n"
  "                          Modes: interactive | piped | batch | single-message\n"
  "                          Example: agent simple-chat \"What dungeons exist?\" --rom=zelda3.sfc\n"
  "                          Example: agent simple-chat --rom=zelda3.sfc --ai_provider=ollama\n"
  "                          Example: echo \"List sprites\" | agent simple-chat --rom=zelda3.sfc\n"
  "                          Example: agent simple-chat --file=queries.txt --rom=zelda3.sfc\n"
  "\n"
  "  test-conversation        Run automated test conversation with AI\n"
  "                          Example: agent test-conversation --rom=zelda3.sfc --ai_provider=ollama\n"
  "\n"
  "  chat                    Full FTXUI-based chat interface\n"
  "                          Example: agent chat --rom=zelda3.sfc\n"
  "\n"
  "ROM Inspection Tools (can be called by AI or directly):\n"
  "  resource-list           List labeled resources (dungeons, sprites, etc.)\n"
  "                          Example: agent resource-list --type=dungeon --format=json\n"
  "\n"
  "  resource-search         Search resource labels by fuzzy text\n"
  "                          Example: agent resource-search --query=soldier --type=sprite\n"
  "\n"
  "  dungeon-list-sprites    List sprites in a dungeon room\n"
  "                          Example: agent dungeon-list-sprites --room=5 --format=json\n"
  "\n"
  "  dungeon-describe-room   Summarize metadata for a dungeon room\n"
  "                          Example: agent dungeon-describe-room --room=0x12 --format=text\n"
  "\n"
  "  overworld-find-tile     Search for tile placements in overworld\n"
  "                          Example: agent overworld-find-tile --tile=0x02E --format=json\n"
  "\n"
  "  overworld-describe-map  Get metadata about an overworld map\n"
  "                          Example: agent overworld-describe-map --map=0 --format=json\n"
  "\n"
  "  overworld-list-warps    List entrances/exits/holes in overworld\n"
  "                          Example: agent overworld-list-warps --map=0 --format=json\n"
  "\n"
  "Proposal & Testing Commands:\n"
  "  run                     Execute agent task\n"
  "  plan                    Generate execution plan\n"
  "  diff                    Show ROM differences\n"
  "  accept                  Accept and apply proposal changes\n"
  "  test                    Run agent tests\n"
  "  gui                     Launch GUI components\n"
  "  learn                   Train agent on examples\n"
  "  list                    List available resources\n"
  "  commit                  Commit changes\n"
  "  revert                  Revert changes\n"
  "  describe                Describe agent capabilities\n"
  "  todo                    Manage tasks and project planning\n"
  "\n"
  "Global Options:\n"
  "  --rom=<path>            Path to Zelda3 ROM file (required for most commands)\n"
  "  --ai_provider=<name>    AI provider: mock (default) | ollama | gemini\n"
  "  --ai_model=<name>       Model name (e.g., qwen2.5-coder:7b for Ollama)\n"
  "  --ollama_host=<url>     Ollama server URL (default: http://localhost:11434)\n"
  "  --gemini_api_key=<key>  Gemini API key (or set GEMINI_API_KEY env var)\n"
  "  --format=<type>         Output format: text | markdown | json | compact\n"
  "\n"
  "For more details, see: docs/simple_chat_input_methods.md";

}  // namespace
}  // namespace agent

absl::Status Agent::Run(const std::vector<std::string>& arg_vec) {
  if (arg_vec.empty()) {
    return absl::InvalidArgumentError(std::string(agent::kUsage));
  }

  const std::string& subcommand = arg_vec[0];
  std::vector<std::string> subcommand_args(arg_vec.begin() + 1, arg_vec.end());

  if (subcommand == "run") {
    return agent::HandleRunCommand(subcommand_args, rom_);
  }
  if (subcommand == "plan") {
    return agent::HandlePlanCommand(subcommand_args);
  }
  if (subcommand == "diff") {
    return agent::HandleDiffCommand(rom_, subcommand_args);
  }
  if (subcommand == "accept") {
    return agent::HandleAcceptCommand(subcommand_args, rom_);
  }
  if (subcommand == "test") {
    return agent::HandleTestCommand(subcommand_args);
  }
  if (subcommand == "test-conversation") {
    return agent::HandleTestConversationCommand(subcommand_args);
  }
  if (subcommand == "gui") {
    return agent::HandleGuiCommand(subcommand_args);
  }
  if (subcommand == "learn") {
    return agent::HandleLearnCommand(subcommand_args);
  }
  if (subcommand == "list") {
    return agent::HandleListCommand();
  }
  if (subcommand == "commit") {
    return agent::HandleCommitCommand(rom_);
  }
  if (subcommand == "revert") {
    return agent::HandleRevertCommand(rom_);
  }
  if (subcommand == "describe") {
    return agent::HandleDescribeCommand(subcommand_args);
  }
  if (subcommand == "resource-list") {
    return agent::HandleResourceListCommand(subcommand_args);
  }
  if (subcommand == "resource-search") {
    return agent::HandleResourceSearchCommand(subcommand_args);
  }
  if (subcommand == "dungeon-list-sprites") {
    return agent::HandleDungeonListSpritesCommand(subcommand_args);
  }
  if (subcommand == "dungeon-describe-room") {
    return agent::HandleDungeonDescribeRoomCommand(subcommand_args);
  }
  if (subcommand == "overworld-find-tile") {
    return agent::HandleOverworldFindTileCommand(subcommand_args);
  }
  if (subcommand == "overworld-describe-map") {
    return agent::HandleOverworldDescribeMapCommand(subcommand_args);
  }
  if (subcommand == "overworld-list-warps") {
    return agent::HandleOverworldListWarpsCommand(subcommand_args);
  }
  if (subcommand == "chat") {
    return agent::HandleChatCommand(rom_);
  }
  if (subcommand == "simple-chat") {
    return agent::HandleSimpleChatCommand(subcommand_args, &rom_, absl::GetFlag(FLAGS_quiet));
  }
  if (subcommand == "todo") {
    return handlers::HandleTodoCommand(subcommand_args);
  }
  
  // Hex manipulation commands
  if (subcommand == "hex-read") {
    return agent::HandleHexRead(subcommand_args, &rom_);
  }
  if (subcommand == "hex-write") {
    return agent::HandleHexWrite(subcommand_args, &rom_);
  }
  if (subcommand == "hex-search") {
    return agent::HandleHexSearch(subcommand_args, &rom_);
  }
  
  // Palette manipulation commands
  if (subcommand == "palette-get-colors") {
    return agent::HandlePaletteGetColors(subcommand_args, &rom_);
  }
  if (subcommand == "palette-set-color") {
    return agent::HandlePaletteSetColor(subcommand_args, &rom_);
  }
  if (subcommand == "palette-analyze") {
    return agent::HandlePaletteAnalyze(subcommand_args, &rom_);
  }

  return absl::InvalidArgumentError(std::string(agent::kUsage));
}

}  // namespace cli
}  // namespace yaze