#include "Render/FontCache.h"
#include "Render/TextContext.h"
#include "Resource/BitmapResource.h"
#include "Resource/ResourceMgr.h"
#include "UI/UIBmpTextButton.h"

#include <bit>
#include <cstring>

namespace {
  int Add(int left, int right) {
    return std::bit_cast<int>(static_cast<unsigned>(left) + static_cast<unsigned>(right));
  }
}

// OFFSET: 0x0084ED30
UIBmpTextButton::UIBmpTextButton(UIWindowContext& windows, UIButtonContext& controls)
    : UIBitmapButton(windows, controls) {
}

// OFFSET: 0x00852360
// FUNCTION: fn_10708
UIBmpTextButton::~UIBmpTextButton() = default;

// OFFSET: 0x007C2340
// FUNCTION: fn_09323
void UIBmpTextButton::SetText(std::string value) {
  this->text = value;
  this->Invalidate();
}

// OFFSET: 0x0085BDE0
// FUNCTION: fn_10799
void UIBmpTextButton::Paint() {
  this->FillRegion(0, 0, this->geometry.width, this->geometry.height, 0xFFFF00FFu);
  const BitmapButtonTextStyle* style = nullptr;

  for (const auto& candidate : this->styles) {
    if (
      (this->m_disabled == 0 && candidate.state == this->geometry.unknown_0030) ||
      (this->m_disabled == 1 && candidate.state == 3)
    ) {
      style = &candidate;
      break;
    }
  }

  const unsigned state = this->geometry.unknown_0030;
  const bool disabled = this->m_disabled == 1;

  if (disabled || state <= 2) {
    const char* path = this->m_names[disabled ? 3 : state].c_str();
    auto* resource = static_cast<CBitmapRes*>(this->m_buttonContext->resources.Load(path));
    this->DrawBitmap(0, 0, resource ? &resource->image : nullptr, false);

    if (!this->text.empty()) {
      const auto draw = [&](int x, int y, int height, unsigned color, unsigned char bold) {
        this->DrawTextAt(
          this->m_buttonContext->fonts, this->m_buttonContext->locale, x, y, this->text.c_str(), 0, 0, height, color,
          bold, 0
        );
      };

      if (style) {
        const unsigned shadowColor = disabled ? 0xE2E2E2u : state == 0 ? 0xD7D7D7u : state == 1 ? 0xC7A796u : 0xFFFFFFu;
        const int offset = !disabled && (state == 0 || state == 1) ? 1 : 0;

        if (style->shadow == 1) {
          draw(Add(style->x, offset), Add(style->y, offset), style->height, shadowColor, disabled ? 0 : style->bold);
        }

        const int pressedOffset = disabled || state == 2 ? 1 : 0;
        draw(Add(style->x, pressedOffset), Add(style->y, pressedOffset), style->height, style->color, style->bold);
      } else {
        const int width = this->TextWidth(
          this->m_buttonContext->fonts, this->m_buttonContext->locale, this->text.c_str(), 0, 0, 12, 0, 0
        );
        int height;

        {
          TextContext context(*this->geometry.surface, this->m_buttonContext->fonts, this->m_buttonContext->locale);
          context.SetFont(0, 12, 0, 0);
          height = context.Measure(this->text.c_str(), static_cast<int>(std::strlen(this->text.c_str()))).cy;
        }

        const int x = Add(this->geometry.width, -width) / 2;
        const int y = Add(this->geometry.height, -height) / 2;
        const bool pressed = disabled || state == 2;
        const unsigned shadowColor = disabled ? 0xE2E2E2u : state == 0 ? 0xD7D7D7u : state == 1 ? 0xC7A796u : 0xFFFFFFu;
        draw(Add(x, pressed ? 0 : 1), Add(y, pressed ? 0 : 1), 12, shadowColor, 0);
        draw(Add(x, pressed ? 1 : 0), Add(y, pressed ? 1 : 0), 12, disabled ? 0x6F6F6Fu : 0, 0);
      }
    }
  }

  if (this->m_invalidateParent == 1 && this->parent) {
    static_cast<UIWindow*>(this->parent)->Invalidate();
  }
}