#include "Render/Renderer.h"
#include "Client/Mode/ModeMessage.h"
#include "UI/Window/WindowId.h"
#include "UI/Window/UIEnums.h"
#include "UI/Menu/UIMenuIconWnd.h"

#include "Client/Application/ApplicationWindow.h"
#include "Client/Mode/CMode.h"
#include "Client/Mode/ModeMgr.h"
#include "Client/Replay/Replay.h"
#include "Client/Session/Session.h"
#include "Contents/EntryQueue/EntryQueueMgr.h"
#include "Contents/Guild/GuildMgr.h"
#include "Contents/Tables/MessageTable.h"
#include "Network/Connection/Connection.h"
#include "Network/Packets/PacketId.h"
#include "Network/Packets/PacketLengths.h"
#include "Script/LuaScript.h"
#include "UI/Navigation/UINavigationWnd.h"
#include "UI/Window/UIWindowMgr.h"
#include "Utility/Text/FormatString.h"

#include <windows.h>
#include <shellapi.h>

#include <algorithm>
#include <cstdint>
#include <cstring>
#include <string>
#include "Contents/Guild/GuildInfo.h"

#include "Resource/MSGSTRINGID.h"
#pragma pack(push, 1)
struct MenuIconBankingCheckPacket {
  /* 0000 */ std::int16_t id;
  /* 0002 */ std::uint32_t aid;
};

struct MenuIconOpenUiPacket {
  /* 0000 */ std::int16_t id;
  /* 0002 */ unsigned char type;
};
#pragma pack(pop)


// OFFSET: 0x0084A1E0
// FUNCTION: fn_10606
UIMenuIcon::UIMenuIcon() {}

// OFFSET: 0x0084A240
// FUNCTION: fn_10607
UIMenuIconWnd::UIMenuIconWnd()
    : UIFrameWnd(1), openButton(nullptr), notificationOpenButton(nullptr), closeButton(nullptr) {}

// OFFSET: 0x0084A850
// FUNCTION: fn_10612
bool UIMenuIconWnd::ToggleWindow(int id) {
  const bool closed = g_windowMgr.CloseWindow(id);
  if (!closed) {
    g_windowMgr.EnsureWindow(id);
  }

  return !closed;
}

// OFFSET: 0x0084A880
// FUNCTION: fn_10613
UIWindow* UIMenuIconWnd::HitTest(int x, int y) {
  if (!this->CanHitTest() || this->geometry.visible == 0 || x < this->geometry.x) {
    return nullptr;
  }

  if (!g_session.m_showMenuIcon) {
    if (x >= this->geometry.x + this->geometry.width || y < this->geometry.y || y >= this->geometry.y + this->notificationOpenButton->geometry.height) {
      return nullptr;
    }
    if (this->notifiedCommands.size() == 0) {
      return this->openButton;
    }
    return this->notificationOpenButton;
  }

  if (x >= this->geometry.x + this->geometry.width || y < this->geometry.y || y >= this->geometry.y + this->geometry.height) {
    return nullptr;
  }

  for (auto child = this->m_children.rbegin(); child != this->m_children.rend(); ++child) {
    UIWindow* hit = (*child)->HitTest((x - this->geometry.x), (y - this->geometry.y));
    if (hit) {
      return hit;
    }
  }

  return this;
}

// OFFSET: 0x0084A960
// FUNCTION: fn_10614
bool UIMenuIconWnd::IsTransmitMouseInput() {
  return true;
}

// OFFSET: 0x0084A970
// FUNCTION: fn_10615
void UIMenuIconWnd::Move(int x, int y) {
  this->geometry.x = x;
  this->geometry.y = y - 4;
  g_windowMgr.StoreSnapRectangle(this);
}

// OFFSET: 0x0084A9A0
// FUNCTION: fn_10616
void UIMenuIconWnd::OnCreate(int width, int height) {
  const int ids[24] = {192, 195, 194, 196, 379, 199, 373, 376, 361, 219, 430, 193, 461, 399, 476, 473, 511, 512, 370, 540, 544, 567, 581, 587};
  std::string names[24] = {
    "status", "equip", "item", "skill", "booking", "party", "guild", "battle", "quest", "map", "navigation", "option",
    "bank", "rec", "mail", "achievement", "tip", "shop", "keyboard", "attendance", "adventurerAgency", "repute", "adventureguide", "probability",
  };

  unsigned int position = 0;
  for (int index = 0; index < 24; ++index) {
    if (ids[index] == 192 && g_session.IsDoramJob(g_session.m_job) == true) {
      names[index] = "status_doram";
    }

    if (ids[index] == 379 || ids[index] == 600) {
      continue;
    }

    UIMenuIcon* icon = new UIMenuIcon();
    icon->SetBitmap(FormatOwnedString("유저인터페이스\\menu_icon\\bt_%s.bmp", names[index].c_str()).c_str(), 0, 0);
    icon->SetBitmap(FormatOwnedString("유저인터페이스\\menu_icon\\bt_%s.bmp", names[index].c_str()).c_str(), 1, 0);
    icon->SetBitmap(FormatOwnedString("유저인터페이스\\menu_icon\\bt_%s_press.bmp", names[index].c_str()).c_str(), 2, 0);
    icon->SetSize(icon->BitmapWidth(), icon->BitmapHeight());
    const int row = position / 5 * 43;
    const int x = position % 5 * 42 + 9;
    icon->Move(x, row + 23);
    icon->SetId(ids[index]);
    icon->Enable(0);
    this->AddChild(icon);
    this->menuButtons[ids[index]] = icon;

    switch (ids[index]) {
    case ID_STATUS:
    case ID_ITEMS:
    case ID_SKILL:
    case 473:
    case 476: {
      UIMenuIcon* badge = new UIMenuIcon();
      badge->SetBitmap(FormatOwnedString("유저인터페이스\\menu_icon\\bt_%s_new.bmp", names[index].c_str()).c_str(), 0, 0);
      badge->SetBitmap(FormatOwnedString("유저인터페이스\\menu_icon\\bt_%s_new.bmp", names[index].c_str()).c_str(), 1, 0);
      badge->SetBitmap(FormatOwnedString("유저인터페이스\\menu_icon\\bt_%s_new_press.bmp", names[index].c_str()).c_str(), 2, 0);
      badge->SetSize(badge->BitmapWidth(), badge->BitmapHeight());
      badge->Move(x, row + 17);
      badge->SetId(ids[index]);
      badge->Enable(0);
      this->AddChild(badge);
      this->notificationButtons[ids[index]] = badge;
      break;
    }
    }

    ++position;
  }

  this->openButton = new UIBitmapButton();
  this->openButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_normal.bmp", 0, 0);
  this->openButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_over.bmp", 1, 0);
  this->openButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_press.bmp", 2, 0);
  this->openButton->SetSize(this->openButton->BitmapWidth(), this->openButton->BitmapHeight());
  this->openButton->Move(0, 4);
  this->openButton->SetId(510);
  this->openButton->Enable(0);
  this->AddChild(this->openButton);
  this->notificationOpenButton = new UIBitmapButton();
  this->notificationOpenButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_new_normal.bmp", 0, 0);
  this->notificationOpenButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_new_over.bmp", 1, 0);
  this->notificationOpenButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_new_press.bmp", 2, 0);
  this->notificationOpenButton->SetSize(this->notificationOpenButton->BitmapWidth(), this->notificationOpenButton->BitmapHeight());
  this->notificationOpenButton->Move(0, 0);
  this->notificationOpenButton->SetId(510);
  this->notificationOpenButton->Enable(0);
  this->AddChild(this->notificationOpenButton);
  this->closeButton = new UIBitmapButton();
  this->closeButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_close_normal.bmp", 0, 0);
  this->closeButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_close_over.bmp", 1, 0);
  this->closeButton->SetBitmap("유저인터페이스\\menu_icon\\bt_menu_close_press.bmp", 2, 0);
  this->closeButton->SetSize(this->closeButton->BitmapWidth(), this->closeButton->BitmapHeight());
  this->closeButton->Move(0, 4);
  this->closeButton->SetId(201);
  this->closeButton->Enable(0);
  this->AddChild(this->closeButton);
  this->SendMsg(this, UM_COMMAND, g_session.m_showMenuIcon == true ? 510 : 201, 0, 0, 0);
  if (this->menuButtons.size() > 20) {
    this->Resize(220, 238);
  }
}

// OFFSET: 0x0084B940
// FUNCTION: fn_10617
void UIMenuIconWnd::Paint() {
  this->m_primitives.clear();
  this->openButton->Enable(0);
  this->notificationOpenButton->Enable(0);
  this->closeButton->Enable(0);

  UIButton* toggle;
  if (g_session.m_showMenuIcon) {
    toggle = this->closeButton;
  } else if (this->notifiedCommands.size() == 0) {
    toggle = this->openButton;
  } else {
    toggle = this->notificationOpenButton;
  }
  toggle->Enable(1);

  for (auto it = this->menuButtons.begin(); it != this->menuButtons.end(); ++it) {
    if (!g_session.m_showMenuIcon) {
      it->second->Enable(0);
      if (this->notificationButtons.find(it->first) != this->notificationButtons.end()) {
        this->notificationButtons[it->first]->Enable(0);
      }
    } else {
      this->DrawNamed(g_renderer->primitives, 0, 13, "유저인터페이스\\menu_icon\\bg_menu.tga", 0xFFFFFFFF, 0, 0);
      const bool pinned = std::find(this->notifiedCommands.begin(), this->notifiedCommands.end(), it->first) != this->notifiedCommands.end();
      it->second->Enable(!pinned);
      if (this->notificationButtons.find(it->first) != this->notificationButtons.end()) {
        this->notificationButtons[it->first]->Enable(pinned);
      }
    }
  }
}

// OFFSET: 0x0084BD20
// FUNCTION: fn_10618
void UIMenuIconWnd::OnLBtnDown(int x, int y) {}

// OFFSET: 0x0084BD30
// FUNCTION: fn_10619
void UIMenuIconWnd::OnLBtnUp(int x, int y) {}

// OFFSET: 0x0084BD40
// FUNCTION: fn_10620
void UIMenuIcon::OnMouseMove(int x, int y) {
  int behaviorId = 10041;
  int message = 4698;
  bool known = true;
  switch (this->geometry.id) {
  case ID_STATUS:
    behaviorId = 11;
    message = 105;
    break;
  case 193:
    message = 265;
    break;
  case ID_ITEMS:
    behaviorId = 8;
    message = 106;
    break;
  case ID_EQUIP:
    behaviorId = 10;
    message = 104;
    break;
  case ID_SKILL:
    behaviorId = 37;
    message = 283;
    break;
  case ID_FRIEND:
    behaviorId = 69;
    message = 103;
    break;
  case ID_MAP:
    behaviorId = 140;
    message = 2859;
    break;
  case 361:
    behaviorId = 134;
    message = 1317;
    break;
  case 370:
    message = 3128;
    break;
  case 373:
    behaviorId = 59;
    message = 3134;
    break;
  case 376:
    behaviorId = 157;
    message = 2003;
    break;
  case 399:
    behaviorId = 198;
    message = 2350;
    break;
  case 430:
    behaviorId = 203;
    message = 2353;
    break;
  case 461:
    behaviorId = 251;
    message = 3135;
    break;
  case 473:
    behaviorId = 270;
    message = 2646;
    break;
  case 476:
    behaviorId = 263;
    message = 3136;
    break;
  case 511:
    behaviorId = 315;
    message = 3239;
    break;
  case 512:
    behaviorId = 318;
    message = 3137;
    break;
  case 540:
    behaviorId = 328;
    message = 3473;
    break;
  case 544:
    behaviorId = 324;
    message = 3514;
    break;
  case 567:
    behaviorId = 346;
    message = 3827;
    break;
  case 581:
    behaviorId = 10002;
    message = 4053;
    break;
  case 587:
    message = 4119;
    break;
  default:
    known = false;
    break;
  }

  if (known) {
    const char* behavior = nullptr;
    g_session.m_hotkeyScript->Call("GetBeHaviorOfWindow", "d>s", behaviorId, &behavior);
    if (std::strlen(behavior) == 0) {
      this->m_tooltip = LookupMessage(message);
    } else {
      const std::string text = FormatOwnedString("%s (%s)", LookupMessage(message), behavior);
      this->m_tooltip = text;
    }
  }

  if (this->geometry.id == 193) {
    const std::string text = FormatOwnedString("%s (%s)", LookupMessage(message), "Esc");
    this->m_tooltip = text;
  }
}

// OFFSET: 0x0084C250
// FUNCTION: fn_10621
void UIMenuIconWnd::OnMouseMove(int x, int y) {}

// OFFSET: 0x0084C260
// FUNCTION: fn_10622
int UIMenuIconWnd::SendMsg(UIWindow* sender, int message, std::intptr_t argument1, std::intptr_t argument2, std::intptr_t argument3, std::intptr_t argument4) {
  if (message == UM_COMMAND) {
    if (!Replay::Instance().AllowsEvent(argument1)) {
      switch (argument1) {
      case ID_CLOSE:
        g_session.m_showMenuIcon = false;
        this->Invalidate();
        break;
      case ID_NOTICE: {
        const int id = argument2;
        auto found = std::find(this->notifiedCommands.begin(), this->notifiedCommands.end(), id);
        if (argument3 > 0) {
          if (found == this->notifiedCommands.end()) {
            this->notifiedCommands.push_back(id);
          }
        } else if (found != this->notifiedCommands.end()) {
          this->notifiedCommands.erase(found);
        }
        this->Invalidate();
        break;
      }
      case ID_STATUS:
        this->ToggleWindow(11);
        break;
      case ID_EQUIP:
        this->ToggleWindow(10);
        break;
      case ID_ITEMS:
        this->ToggleWindow(8);
        break;
      case ID_SKILL:
        this->ToggleWindow(37);
        break;
      case ID_FRIEND:
        this->ToggleWindow(69);
        break;
      case 373:
        this->ToggleWindow(static_cast<int>(g_guildInfo.m_guildId) < 1 ? 212 : 59);
        break;
      case ID_QUEST:
        this->ToggleWindow(10008);
        break;
      case ID_MAP:
        this->ToggleWindow(140);
        break;
      case 430:
        if (g_windowMgr.navigationWnd) {
          g_windowMgr.navigationWnd->SendMsg(nullptr, UM_COMMAND, ID_CLOSE, 0, 0, 0);
        } else {
          g_windowMgr.EnsureWindow(WID_NAVIGATIONV4WND);
        }
        break;
      case ID_OPTION:
        this->ToggleWindow(155);
        break;
      case 399:
        if (Replay::Instance().state == 1) {
          this->ToggleWindow(198);
        } else {
          g_windowMgr.Confirm("REC End please", 0, true, false, 0, "Message", 280, 120, 0);
        }
        break;
      case 370:
        this->ToggleWindow(303);
        break;
      case 461: {
        MenuIconBankingCheckPacket packet;
        packet.id = ID_CZ_REQ_BANKING_CHECK;
        packet.aid = g_session.m_aid;
        GetRagConnection().Send(g_packetLengths.FixedLength(packet.id), &packet);
        break;
      }
      case 510:
        g_session.m_showMenuIcon = true;
        this->Invalidate();
        break;
      case 476:
        if (g_windowMgr.LookupWindow(WID_REFININGWND)) {
          g_windowMgr.SendNotice(UIM_PUSHINTOCHATHISTORY, reinterpret_cast<std::intptr_t>(LookupMessage(MSI_AURA_EFFECT_OFF)), 255, 0, 0, 0);
        } else if (g_windowMgr.LookupWindow(WID_GRADEENCHANTWND)) {
          g_windowMgr.SendNotice(UIM_PUSHINTOCHATHISTORY, reinterpret_cast<std::intptr_t>(LookupMessage(MSI_BROADCASTING_GRADE_ENCHANT_SUCCESS)), 255, 0, 0, 0);
        } else if (g_windowMgr.LookupWindow(WID_ENCHANTUI)) {
          g_windowMgr.SendNotice(UIM_PUSHINTOCHATHISTORY, reinterpret_cast<std::intptr_t>(LookupMessage(MSI_ENCHANT_REQUEST)), 255, 0, 0, 0);
        } else {
          this->ToggleWindow(263);
        }
        break;
      case 473:
        this->ToggleWindow(270);
        break;
      case 511:
        this->ToggleWindow(315);
        break;
      case 512:
        g_modeMgr.GetGameMode()->SendMsg(MM_REQUEST_CASH_SHOP, 0, 0, 0, 0);
        break;
      case 540:
        if (g_windowMgr.LookupWindow(WID_CHECKATTENDANCEWND)) {
          g_windowMgr.CloseWindow(WID_CHECKATTENDANCEWND);
        } else {
          MenuIconOpenUiPacket packet;
          packet.id = ID_CZ_UI_OPEN;
          packet.type = 5;
          GetRagConnection().Send(g_packetLengths.FixedLength(packet.id), &packet);
        }
        break;
      case 544:
        this->ToggleWindow(324);
        break;
      case 567:
        this->ToggleWindow(346);
        break;
      case 581:
        this->ToggleWindow(10002);
        break;
      case 587:
        if (g_windowMgr.Confirm(LookupMessage(MSI_GENDER_CHANGE_FAILED_CAUSE_MONSTER_TRANSFORM), 2, true, false, 0, "Message", 280, 120, 0) == 187) {
          ShellExecuteA(hWnd, "open", LookupMessage(MSI_GENDER_CHANGE_FAILED_CAUSE_MARRIED), nullptr, nullptr, SW_SHOWNORMAL);
        }
        break;
      }
    }
  } else if (message == UM_REFRESH) {
    this->SendMsg(nullptr, UM_COMMAND, g_session.m_showMenuIcon == true ? 201 : 510, 0, 0, 0);
  }

  if (message == UM_COMMAND && argument1 == 376 && !Replay::Instance().AllowsEvent(argument1) && g_entryQueueMgr) {
    if (reinterpret_cast<const unsigned char*>(g_entryQueueMgr)[0x3C] == 1) {
      g_entryQueueMgr->SendRankingRequest();
    }
    this->ToggleWindow(157);
  }

  return UIWindow::SendMsg(sender, message, argument1, argument2, argument3, argument4);
}