#include "courtroom.h"

#include "aoemotebutton.h"
#include "options.h"

void Courtroom::initialize_emotes()
{
  ui_emotes = new QWidget(this);
  ui_emotes->setObjectName("ui_emotes");

  ui_emote_left = new AOButton(ao_app, this);
  ui_emote_left->setObjectName("ui_emote_left");
  ui_emote_right = new AOButton(ao_app, this);
  ui_emote_right->setObjectName("ui_emote_right");

  ui_emote_dropdown = new QComboBox(this);
  ui_emote_dropdown->setContextMenuPolicy(Qt::CustomContextMenu);
  ui_emote_dropdown->setObjectName("ui_emote_dropdown");

  emote_menu = new QMenu(this);
  emote_menu->setObjectName("ui_emote_menu");

  emote_preview = new AOEmotePreview(ao_app, this);
  emote_preview->setObjectName("ui_emote_preview");
  emote_preview->resize(256, 192);

  connect(ui_emote_left, &AOButton::clicked, this, &Courtroom::on_emote_left_clicked);
  connect(ui_emote_right, &AOButton::clicked, this, &Courtroom::on_emote_right_clicked);

  connect(ui_emote_dropdown, QOverload<int>::of(&QComboBox::activated), this, &Courtroom::on_emote_dropdown_changed);
  connect(ui_emote_dropdown, &AOEmoteButton::customContextMenuRequested, this, &Courtroom::show_emote_menu);

  connect(ui_pre, QOverload<int>::of(&QCheckBox::stateChanged), this, &Courtroom::update_emote_preview);
  connect(ui_flip, &AOButton::clicked, this, &Courtroom::update_emote_preview);
  connect(ui_pair_offset_spinbox, QOverload<int>::of(&QSpinBox::valueChanged), this, &Courtroom::update_emote_preview);
  connect(ui_pair_vert_offset_spinbox, QOverload<int>::of(&QSpinBox::valueChanged), this, &Courtroom::update_emote_preview);
}

void Courtroom::refresh_emotes()
{
  // Should properly refresh the emote list
  qDeleteAll(ui_emote_list.begin(), ui_emote_list.end());
  ui_emote_list.clear();

  set_size_and_pos(ui_emotes, "emotes");

  set_size_and_pos(ui_emote_left, "emote_left");
  ui_emote_left->setImage("arrow_left");

  set_size_and_pos(ui_emote_right, "emote_right");
  ui_emote_right->setImage("arrow_right");

  QPoint f_spacing = ao_app->get_button_spacing("emote_button_spacing", "courtroom_design.ini");
  QPoint p_point = ao_app->get_button_spacing("emote_button_size", "courtroom_design.ini");

  if (ui_emotes->width() == 0 || ui_emotes->height() == 0)
  { // Workaround for a nasty crash
    ui_emotes->hide();
    return;
  }

  const int button_width = p_point.x();
  int x_spacing = f_spacing.x();
  int x_mod_count = 0;

  const int button_height = p_point.y();
  int y_spacing = f_spacing.y();
  int y_mod_count = 0;

  emote_columns = ((ui_emotes->width() - button_width) / (x_spacing + button_width)) + 1;
  emote_rows = ((ui_emotes->height() - button_height) / (y_spacing + button_height)) + 1;

  max_emotes_on_page = emote_columns * emote_rows;

  QString selected_image = ao_app->get_image_suffix(ao_app->get_theme_path("emote_selected", ""), true);

  for (int i = 0; i < max_emotes_on_page; ++i)
  {
    int x_pos = (button_width + x_spacing) * x_mod_count;
    int y_pos = (button_height + y_spacing) * y_mod_count;

    AOEmoteButton *f_emote = new AOEmoteButton(i, button_width, button_height, ao_app, ui_emotes);
    f_emote->setSelectedImage(selected_image);
    f_emote->move(x_pos, y_pos);
    ui_emote_list.append(f_emote);

    f_emote->setContextMenuPolicy(Qt::CustomContextMenu);
    connect(f_emote, &AOEmoteButton::emoteClicked, this, &Courtroom::on_emote_clicked);
    connect(f_emote, &AOEmoteButton::customContextMenuRequested, this, &Courtroom::show_emote_menu);

    ++x_mod_count;

    if (x_mod_count == emote_columns)
    {
      ++y_mod_count;
      x_mod_count = 0;
    }
  }
}

void Courtroom::set_emote_page()
{
  if (m_cid == -1)
  {
    return;
  }

  int total_emotes = ao_app->get_emote_number(current_char);

  ui_emote_left->hide();
  ui_emote_right->hide();

  for (AOEmoteButton *i_button : std::as_const(ui_emote_list))
  {
    i_button->hide();
  }

  int total_pages = total_emotes / max_emotes_on_page;
  int emotes_on_page = 0;

  if (total_emotes % max_emotes_on_page != 0)
  {
    ++total_pages;
    // i. e. not on the last page
    if (total_pages > current_emote_page + 1)
    {
      emotes_on_page = max_emotes_on_page;
    }
    else
    {
      emotes_on_page = total_emotes % max_emotes_on_page;
    }
  }
  else
  {
    emotes_on_page = max_emotes_on_page;
  }

  if (total_pages > current_emote_page + 1)
  {
    ui_emote_right->show();
  }

  if (current_emote_page > 0)
  {
    ui_emote_left->show();
  }

  for (int n_emote = 0; n_emote < emotes_on_page && n_emote < ui_emote_list.size(); ++n_emote)
  {
    int n_real_emote = n_emote + current_emote_page * max_emotes_on_page;
    AOEmoteButton *f_emote = ui_emote_list.at(n_emote);

    if (n_real_emote == current_emote)
    {
      f_emote->setImage(current_char, n_real_emote, true);
    }
    else
    {
      f_emote->setImage(current_char, n_real_emote, false);
    }

    f_emote->show();
    f_emote->setToolTip(QString::number(n_real_emote + 1) + ": " + ao_app->get_emote_comment(current_char, n_real_emote));
  }
}

void Courtroom::set_emote_dropdown()
{
  ui_emote_dropdown->clear();

  int total_emotes = ao_app->get_emote_number(current_char);

  for (int n = 0; n < total_emotes; ++n)
  {
    ui_emote_dropdown->addItem(QString::number(n + 1) + ": " + ao_app->get_emote_comment(current_char, n));
    QString icon_path = ao_app->get_image_suffix(ao_app->get_character_path(current_char, "emotions/button" + QString::number(n + 1) + "_off"));
    ui_emote_dropdown->setItemIcon(n, QIcon(icon_path));
  }
  if (current_emote > -1 && current_emote < ui_emote_dropdown->count())
  {
    ui_emote_dropdown->setCurrentIndex(current_emote);
  }
}

void Courtroom::select_emote(int p_id)
{
  int min = current_emote_page * max_emotes_on_page;
  int max = (max_emotes_on_page - 1) + current_emote_page * max_emotes_on_page;

  if (current_emote >= min && current_emote <= max)
  {
    ui_emote_list.at(current_emote % max_emotes_on_page)->setImage(current_char, current_emote, false);
  }

  int old_emote = current_emote;

  current_emote = p_id;

  if (current_emote >= min && current_emote <= max)
  {
    ui_emote_list.at(current_emote % max_emotes_on_page)->setImage(current_char, current_emote, true);
  }

  int emote_mod = ao_app->get_emote_mod(current_char, current_emote);

  if (old_emote == current_emote)
  {
    ui_pre->setChecked(!ui_pre->isChecked());
  }
  else if (!Options::getInstance().clearPreOnPlayEnabled())
  {
    if (emote_mod == PREANIM || emote_mod == PREANIM_ZOOM)
    {
      ui_pre->setChecked(true);
    }
    else
    {
      ui_pre->setChecked(false);
    }
  }

  ui_emote_dropdown->setCurrentIndex(current_emote);
  update_emote_preview();
  ui_ic_chat_message->setFocus();
}

void Courtroom::update_emote_preview()
{
  if (!emote_preview->isVisible())
  {
    return;
  }

  QString pre = ao_app->get_pre_emote(current_char, current_emote);
  if (ui_pre->isChecked() && !pre.isEmpty() && pre != "-")
  {
    preview_emote(pre, kal::CharacterAnimationLayer::PreEmote);
  }
  else
  {
    preview_emote(ao_app->get_emote(current_char, current_emote), kal::CharacterAnimationLayer::IdleEmote);
  }
}

void Courtroom::on_emote_clicked(int p_id)
{
  select_emote(p_id + max_emotes_on_page * current_emote_page);
}

void Courtroom::show_emote_menu(const QPoint &pos)
{
  QWidget *button = qobject_cast<QWidget *>(sender());
  int id = current_emote;
  if (qobject_cast<AOEmoteButton *>(button))
  {
    AOEmoteButton *emote_button = qobject_cast<AOEmoteButton *>(sender());
    id = emote_button->id();
  }
  int emote_num = id + max_emotes_on_page * current_emote_page;
  emote_menu->clear();
  emote_menu->setDefaultAction(emote_menu->addAction("Preview Selected", this, [this] {
    emote_preview->show();
    emote_preview->raise();
    emote_preview->updateViewportGeometry();
    update_emote_preview();
  }));
  QString prefix;
  QString f_pre = ao_app->get_pre_emote(current_char, emote_num);
  if (!f_pre.isEmpty() && f_pre != "-")
  {
    emote_menu->addAction("Preview pre: " + f_pre, this, [this, f_pre] { preview_emote(f_pre, kal::CharacterAnimationLayer::PreEmote); });
  }

  QString f_emote = ao_app->get_emote(current_char, emote_num);
  if (!f_emote.isEmpty())
  {
    emote_menu->addAction("Preview idle: " + f_emote, this, [this, f_emote] { preview_emote(f_emote, kal::CharacterAnimationLayer::IdleEmote); });
    emote_menu->addAction("Preview talk: " + f_emote, this, [this, f_emote] { preview_emote(f_emote, kal::CharacterAnimationLayer::TalkEmote); });
    QStringList c_paths = {ao_app->get_image_suffix(ao_app->get_character_path(current_char, "(c)" + f_emote)), ao_app->get_image_suffix(ao_app->get_character_path(current_char, "(c)/" + f_emote))};
    // if there is a (c) animation
    if (file_exists(ao_app->find_image(c_paths)))
    {
      emote_menu->addAction("Preview segway: " + f_emote, this, [this, f_emote] { preview_emote(f_emote, kal::CharacterAnimationLayer::PostEmote); });
    }
  }
  emote_menu->popup(button->mapToGlobal(pos));
}

void Courtroom::preview_emote(QString f_emote, kal::CharacterAnimationLayer::EmoteType emoteType)
{
  emote_preview->show();
  emote_preview->raise();
  emote_preview->updateViewportGeometry();
  emote_preview->display(current_char, f_emote, emoteType, ui_flip->isChecked(), ui_pair_offset_spinbox->value(), ui_pair_vert_offset_spinbox->value());
}

void Courtroom::on_emote_left_clicked()
{
  --current_emote_page;

  set_emote_page();

  ui_ic_chat_message->setFocus();
}

void Courtroom::on_emote_right_clicked()
{
  ++current_emote_page;

  set_emote_page();

  ui_ic_chat_message->setFocus();
}

void Courtroom::on_emote_dropdown_changed(int p_index)
{
  select_emote(p_index);
}