{

Q_OBJECT

explicit ForeignKeyEditor(QWidget* parent = nullptr)

: QWidget(parent)

, tablesComboBox(new QComboBox(this))

, idsComboBox(new QComboBox(this))

, clauseEdit(new QLineEdit(this))

, m_btnReset(new QPushButton(tr("&Reset"), this))

{

idsComboBox->setSizeAdjustPolicy(QComboBox::AdjustToContents);

clauseEdit->setPlaceholderText(tr("Foreign key clauses (ON UPDATE, ON DELETE etc.)"));

QHBoxLayout* layout = new QHBoxLayout(this);

layout->addWidget(tablesComboBox);

layout->addWidget(idsComboBox);

layout->addWidget(clauseEdit);

layout->addWidget(m_btnReset);

layout->setSpacing(0);

layout->setContentsMargins(0, 0, 0, 0);

setLayout(layout);

connect(m_btnReset, &QPushButton::clicked, this, [&]

{

tablesComboBox->setCurrentIndex(-1);

idsComboBox->setCurrentIndex(-1);

clauseEdit->clear();

});

connect(tablesComboBox, static_cast<void(QComboBox::*)(int)>(&QComboBox::currentIndexChanged), this,

[=](int index)

{

// reset ids combo box

idsComboBox->setCurrentIndex(-1);

// disable clauses editor if none of tables is selected

bool enableClausesEditor = (index!= -1);

clauseEdit->setEnabled(enableClausesEditor);

});

{

if (tablesComboBox->currentText().isEmpty())

return QString();

const QString table = sqlb::escapeIdentifier(tablesComboBox->currentText());

const QString clauses = clauseEdit->text();

QString id = idsComboBox->currentText();

if (!id.isEmpty())

id = QString("(%1)").arg(sqlb::escapeIdentifier(id));

return QString("%1%2 %3").arg(

table,

id,

clauses)

.trimmed();

: QStyledItemDelegate(parent)

, m_db(db)

, m_table(table)

{

for(const auto& it : m_db.schemata)

{

for(const auto& jt : it.second.tables)

{

// Don't insert the current table into the list. The name and fields of the current table are always taken from the m_table reference

if(jt.first != m_table.name())

m_tablesIds.insert({jt.first, jt.second->fieldNames()});

}

}

{

ForeignKeyEditor* fkEditor = static_cast<ForeignKeyEditor*>(editor);

// First try to get the fields list by assuming this is called for a constraints table in which the first column contains a list of fields.

// If this fails, assume this is called for a fields table in which the first column contains the name of a single table field.

sqlb::StringVector fields = index.sibling(index.row(), 0).data(Qt::UserRole).value<sqlb::StringVector>();

if(fields.empty())

fields.push_back(index.sibling(index.row(), 0).data().toString().toStdString());

fkEditor->foreignKey = m_table.foreignKey(fields);

if (fkEditor->foreignKey) {

fkEditor->tablesComboBox->setCurrentText(QString::fromStdString(fkEditor->foreignKey->table()));

fkEditor->clauseEdit->setText(QString::fromStdString(fkEditor->foreignKey->constraint()));

if (!fkEditor->foreignKey->columns().empty())

fkEditor->idsComboBox->setCurrentText(QString::fromStdString(fkEditor->foreignKey->columns().front()));

} else {

fkEditor->tablesComboBox->setCurrentIndex(-1);

}

{

ForeignKeyEditor* fkEditor = static_cast<ForeignKeyEditor*>(editor);

QString sql = fkEditor->getSql();

// First try to get the fields list by assuming this is called for a constraints table in which the first column contains a list of fields.

// If this fails, assume this is called for a fields table in which the first column contains the name of a single table field.

sqlb::StringVector fields = index.sibling(index.row(), 0).data(Qt::UserRole).value<sqlb::StringVector>();

if(fields.empty())

fields.push_back(index.sibling(index.row(), 0).data().toString().toStdString());

if (sql.isEmpty() && fkEditor->foreignKey) {

// Remove the foreign key

m_table.removeConstraint(fkEditor->foreignKey);

} else {

// Create a new foreign key object if required

if(!fkEditor->foreignKey)

{

fkEditor->foreignKey = std::make_shared<sqlb::ForeignKeyClause>();

m_table.addConstraint(fields, fkEditor->foreignKey);

}

// Set data

const auto table = fkEditor->tablesComboBox->currentText().toStdString();

const auto id = fkEditor->idsComboBox->currentText().toStdString();

const auto clause = fkEditor->clauseEdit->text().trimmed().toStdString();

fkEditor->foreignKey->setTable(table);

if(id.empty())

fkEditor->foreignKey->setColumns({});

else

fkEditor->foreignKey->setColumns({id});

fkEditor->foreignKey->setConstraint(clause);

}

model->setData(index, sql);

{

Q_UNUSED(index)

editor->setGeometry(option.rect);