#include "report_popup.h"

#include <wx/dcbuffer.h>

#include <map>
#include <string>

#include "global.h"
#include "tracker_state.h"

ReportPopup::ReportPopup(wxWindow* parent)
    : wxScrolledCanvas(parent, wxID_ANY) {
  SetBackgroundStyle(wxBG_STYLE_PAINT);

  unchecked_eye_ =
      wxBitmap(wxImage(GetAbsolutePath("assets/unchecked.png").c_str(),
                       wxBITMAP_TYPE_PNG)
                   .Scale(32, 32));
  checked_eye_ = wxBitmap(
      wxImage(GetAbsolutePath("assets/checked.png").c_str(), wxBITMAP_TYPE_PNG)
          .Scale(32, 32));

  SetScrollRate(5, 5);

  SetBackgroundColour(*wxBLACK);
  Hide();

  Bind(wxEVT_PAINT, &ReportPopup::OnPaint, this);
}

void ReportPopup::SetDoorId(int door_id) {
  door_id_ = door_id;

  UpdateIndicators();
}

void ReportPopup::Reset() {
  door_id_ = -1;
}

void ReportPopup::UpdateIndicators() {
  wxMemoryDC mem_dc;

  const std::map<std::string, bool>& report = GetDoorRequirements(door_id_);

  int acc_height = 10;
  int col_width = 0;

  for (const auto& [text, obtained] : report) {
    wxSize item_extent = mem_dc.GetTextExtent(text);
    int item_height = std::max(32, item_extent.GetHeight()) + 10;
    acc_height += item_height;

    if (item_extent.GetWidth() > col_width) {
      col_width = item_extent.GetWidth();
    }
  }

  int item_width = col_width + 10 + 32;
  int full_width = item_width + 20;

  Fit();
  SetVirtualSize(full_width, acc_height);

  rendered_ = wxBitmap(full_width, acc_height);
  mem_dc.SelectObject(rendered_);
  mem_dc.SetPen(*wxTRANSPARENT_PEN);
  mem_dc.SetBrush(*wxBLACK_BRUSH);
  mem_dc.DrawRectangle({0, 0}, {full_width, acc_height});

  mem_dc.SetFont(GetFont());

  int cur_height = 10;

  for (const auto& [text, obtained] : report) {
    wxBitmap* eye_ptr = obtained ? &checked_eye_ : &unchecked_eye_;

    mem_dc.DrawBitmap(*eye_ptr, wxPoint{10, cur_height});

    mem_dc.SetTextForeground(obtained ? *wxWHITE : *wxRED);
    wxSize item_extent = mem_dc.GetTextExtent(text);
    mem_dc.DrawText(
        text, wxPoint{10 + 32 + 10,
                      cur_height + (32 - mem_dc.GetFontMetrics().height) / 2});

    cur_height += 10 + 32;
  }
}

void ReportPopup::OnPaint(wxPaintEvent& event) {
  if (door_id_ != -1) {
    wxBufferedPaintDC dc(this);
    PrepareDC(dc);
    dc.DrawBitmap(rendered_, 0, 0);
  }

  event.Skip();
}