ExpressionDialog.cpp

/*
 * Copyright (C) 2006 - 2025 Evan Teran <evan.teran@gmail.com>
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "ExpressionDialog.h"
#include "Expression.h"
#include "ISymbolManager.h"
#include "Symbol.h"
#include "edb.h"

#include <QCompleter>
#include <QPushButton>

ExpressionDialog::ExpressionDialog(const QString &title, const QString &prompt, QWidget *parent, Qt::WindowFlags f)
	: QDialog(parent, f) {

	setWindowTitle(title);

	layout_     = new QVBoxLayout(this);
	labelText_  = new QLabel(prompt, this);
	labelError_ = new QLabel(this);
	expression_ = new QLineEdit(this);
	buttonBox_  = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel, Qt::Horizontal, this);

	connect(buttonBox_, &QDialogButtonBox::accepted, this, &ExpressionDialog::accept);
	connect(buttonBox_, &QDialogButtonBox::rejected, this, &ExpressionDialog::reject);

	layout_->addWidget(labelText_);
	layout_->addWidget(expression_);
	layout_->addWidget(labelError_);
	layout_->addWidget(buttonBox_);

	paletteError_.setColor(QPalette::WindowText, Qt::red);
	labelError_->setPalette(paletteError_);

	buttonBox_->button(QDialogButtonBox::Ok)->setEnabled(false);

	setLayout(layout_);

	connect(expression_, &QLineEdit::textChanged, this, &ExpressionDialog::on_text_changed);
	expression_->selectAll();

	std::vector<Symbol> symbols = edb::v1::symbol_manager().symbols();
	QStringList allLabels;

	for (const Symbol &sym : symbols) {
		allLabels.append(sym.name_no_prefix);
	}

	allLabels.append(edb::v1::symbol_manager().labels().values());

	auto completer = new QCompleter(allLabels, this);
	expression_->setCompleter(completer);
	allLabels.clear();
}

void ExpressionDialog::on_text_changed(const QString &text) {
	QHash<edb::address_t, QString> labels = edb::v1::symbol_manager().labels();
	edb::address_t resAddr                = labels.key(text);

	bool retval = false;

	if (resAddr) {
		lastAddress_ = resAddr;
		retval       = true;
	} else {
		Expression<edb::address_t> expr(text, edb::v1::get_variable, edb::v1::get_value);

		Result<edb::address_t, ExpressionError> address = expr.evaluate();
		if (address) {
			labelError_->clear();
			retval       = true;
			lastAddress_ = *address;
		} else {
			labelError_->setText(address.error().what());
			retval = false;
		}
	}

	buttonBox_->button(QDialogButtonBox::Ok)->setEnabled(retval);
}

edb::address_t ExpressionDialog::getAddress() {
	return lastAddress_;
}