dolphin/Source/Core/DolphinQt/Settings/TriforceBaseboardSettingsDialog.cpp

// Copyright 2026 Dolphin Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#include "DolphinQt/Settings/TriforceBaseboardSettingsDialog.h"

#include <QDialogButtonBox>
#include <QGroupBox>
#include <QHeaderView>
#include <QLineEdit>
#include <QPushButton>
#include <QTableWidget>
#include <QVBoxLayout>

#include <fmt/format.h>
#include <fmt/ranges.h>

#include "Common/Logging/Log.h"

#include "Core/Config/MainSettings.h"
#include "Core/HW/DVD/AMMediaboard.h"

#include "DolphinQt/QtUtils/QtUtils.h"

static constexpr int COLUMN_EMULATED = 0;
static constexpr int COLUMN_REAL = 1;
static constexpr int COLUMN_DESCRIPTION = 2;

TriforceBaseboardSettingsDialog::TriforceBaseboardSettingsDialog(QWidget* parent) : QDialog{parent}
{
  setWindowTitle(tr("Triforce Baseboard"));

  auto* const dialog_layout = new QVBoxLayout{this};

  auto* const ip_override_group = new QGroupBox{tr("IP Address Overrides")};
  dialog_layout->addWidget(ip_override_group);

  auto* const ip_override_layout = new QVBoxLayout{ip_override_group};

  m_ip_overrides_table = new QTableWidget;
  ip_override_layout->addWidget(m_ip_overrides_table);

  m_ip_overrides_table->setColumnCount(3);
  m_ip_overrides_table->setHorizontalHeaderLabels({tr("Emulated"), tr("Real"), tr("Description")});
  m_ip_overrides_table->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
  m_ip_overrides_table->setSizeAdjustPolicy(QTableWidget::AdjustToContents);

  m_ip_overrides_table->setEditTriggers(QAbstractItemView::DoubleClicked |
                                        QAbstractItemView::EditKeyPressed);

  auto* const button_box = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
  dialog_layout->addWidget(button_box);

  auto* const load_default_button = new QPushButton(tr("Default"));
  connect(load_default_button, &QPushButton::clicked, this,
          &TriforceBaseboardSettingsDialog::LoadDefault);

  auto* const clear_button = new QPushButton(tr("Clear"));
  connect(clear_button, &QPushButton::clicked, this, &TriforceBaseboardSettingsDialog::OnClear);

  button_box->addButton(load_default_button, QDialogButtonBox::ActionRole);
  button_box->addButton(clear_button, QDialogButtonBox::ActionRole);

  connect(button_box, &QDialogButtonBox::accepted, this, &TriforceBaseboardSettingsDialog::accept);
  connect(button_box, &QDialogButtonBox::rejected, this, &TriforceBaseboardSettingsDialog::reject);

  connect(this, &TriforceBaseboardSettingsDialog::accepted, this,
          &TriforceBaseboardSettingsDialog::SaveConfig);

  connect(m_ip_overrides_table, &QTableWidget::itemChanged, this, [this](QTableWidgetItem* item) {
    const int row_count = m_ip_overrides_table->rowCount();
    const bool is_last_row = item->row() == row_count - 1;
    if (is_last_row)
    {
      if (!item->text().isEmpty())
        m_ip_overrides_table->insertRow(row_count);
    }
    else if (item->column() != COLUMN_DESCRIPTION && item->text().isEmpty())
    {
      // Just remove inner rows that have text erased.
      // This UX could be less weird, but it's good enough for now.
      m_ip_overrides_table->removeRow(item->row());
    }
  });

  LoadConfig();

  QtUtils::AdjustSizeWithinScreen(this);
}

void TriforceBaseboardSettingsDialog::LoadConfig()
{
  m_ip_overrides_table->setRowCount(0);

  // Add the final empty row.
  m_ip_overrides_table->insertRow(0);

  int row = 0;

  const auto ip_overrides_str = Config::Get(Config::MAIN_TRIFORCE_IP_OVERRIDES);
  for (auto&& ip_pair : ip_overrides_str | std::views::split(','))
  {
    const auto ip_pair_str = std::string_view{ip_pair};
    const auto parsed = AMMediaboard::ParseIPOverride(ip_pair_str);
    if (!parsed.has_value())
    {
      ERROR_LOG_FMT(COMMON, "Bad IP pair string: {}", ip_pair_str);
      continue;
    }

    m_ip_overrides_table->insertRow(row);

    m_ip_overrides_table->setItem(row, COLUMN_EMULATED,
                                  new QTableWidgetItem(QString::fromUtf8(parsed->original)));
    m_ip_overrides_table->setItem(row, COLUMN_REAL,
                                  new QTableWidgetItem(QString::fromUtf8(parsed->replacement)));
    m_ip_overrides_table->setItem(row, COLUMN_DESCRIPTION,
                                  new QTableWidgetItem(QString::fromUtf8(parsed->description)));

    ++row;
  }
}

void TriforceBaseboardSettingsDialog::SaveConfig()
{
  // Ignore our empty row.
  const int row_count = m_ip_overrides_table->rowCount() - 1;

  std::vector<std::string> replacements(row_count);
  for (int row = 0; row < row_count; ++row)
  {
    auto* const original_item = m_ip_overrides_table->item(row, COLUMN_EMULATED);
    auto* const replacement_item = m_ip_overrides_table->item(row, COLUMN_REAL);

    // Skip incomplete rows.
    if (original_item == nullptr || replacement_item == nullptr)
      continue;

    replacements[row] = fmt::format("{}={}", StripWhitespace(original_item->text().toStdString()),
                                    replacement_item->text().toStdString());

    auto* const description_item = m_ip_overrides_table->item(row, COLUMN_DESCRIPTION);
    if (description_item == nullptr)
      continue;

    const auto description = description_item->text().toStdString();

    if (!description.empty())
      (replacements[row] += ' ') += description;
  }

  Config::SetBaseOrCurrent(Config::MAIN_TRIFORCE_IP_OVERRIDES,
                           fmt::format("{}", fmt::join(replacements, ",")));
}

void TriforceBaseboardSettingsDialog::LoadDefault()
{
  // This alters the config before "OK" is pressed. Bad UX..

  Config::SetBaseOrCurrent(Config::MAIN_TRIFORCE_IP_OVERRIDES, Config::DefaultState{});

  LoadConfig();
}

void TriforceBaseboardSettingsDialog::OnClear()
{
  Config::SetBaseOrCurrent(Config::MAIN_TRIFORCE_IP_OVERRIDES, "");
  LoadConfig();
}