#include <std_include.hpp>
#include "loader/component_loader.hpp"

#include "command.hpp"
#include "network.hpp"

#include <utils/hook.hpp>
#include <utils/string.hpp>

namespace network {
namespace {
void cl_co_op_connect(const command::params& params) {
  // Valid format is IP Port
  if (params.size() < 3) {
    return;
  }

  const auto input = std::strtol(params.get(2), nullptr, 10);
  const auto port = ::htons(static_cast<std::uint16_t>(input));

  command::execute(
      utils::string::va("connect_coop {0} {1}", params.get(1), port));
}
} // namespace

class component final : public component_interface {
public:
  void post_load() override { patch_sp(); }

private:
  static void patch_sp() {
    // Ignore protocol mismatch
    utils::hook::set<std::uint8_t>(0x65D0C2, 0xEB);
    utils::hook::set<std::uint8_t>(0x65D0ED, 0xEB);

    // Disable matchmaking stuff
    utils::hook::set<std::uint8_t>(0x43BAE0, 0xEB);

    // Ignore 'MAX_PACKET_USERCMDS'
    utils::hook::set<std::uint8_t>(0x4B1436, 0xEB);

    // Disable IWNet stuff
    utils::hook::set<std::uint8_t>(0x44E824, 0xEB);
    utils::hook::set<std::uint8_t>(0x44E808, 0xEB);

    // Disable MP packet handler
    utils::hook::set<std::uint8_t>(0x65E717, 0xEB);

    // Disable LSP packet handler (suppress print message)
    utils::hook::nop(0x65E3C4, 5);

    // Avoid spam
    utils::hook(0x65E786, game::Com_DPrintf, HOOK_CALL).install()->quick();
    utils::hook(0x65D659, game::Com_DPrintf, HOOK_CALL).install()->quick();

    // Disable BigShort in NET_AdrToString
    utils::hook(0x4BF4D1, game::ShortNoSwap, HOOK_CALL).install()->quick();
    utils::hook(0x4BF50C, game::ShortNoSwap, HOOK_CALL).install()->quick();

    // Fix NET_StringToAdr (Only known workaround that actually works)
    utils::hook::set<const char*>(0x475417, "connect_coop");
    command::add("connect", cl_co_op_connect);

    // Disable BigShort in GetLocalAddressForEncryptedConnection
    utils::hook(0x4CF42D, game::ShortNoSwap, HOOK_CALL).install()->quick();

    // Parse port as unsigned short in Net_AddrToString
    utils::hook::set<const char*>(0x4BF4F3, "%u.%u.%u.%u:%hu");

    // Force Win socket initialization
    utils::hook::nop(0x42B649, 2);

    // Kill LSP
    utils::hook::set<std::uint8_t>(0x4553F0, 0xC3); // Hello
    utils::hook::set<std::uint8_t>(0x428D00, 0xC3); // LSP_SendLogRequest
    utils::hook::set<std::uint8_t>(0x4D13C0, 0xC3); // LSP_ParsePacket
    utils::hook::set<std::uint8_t>(0x66D440, 0xC3); // LSP_AddKeepAlive
    utils::hook::nop(0x494E68, 5);                  // Don't create LSP socket
  }
};
} // namespace network

REGISTER_COMPONENT(network::component)