{

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->setMargin(0);

setLayout(layout);

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

{

tablesComboBox->setCurrentIndex(-1);

idsComboBox->setCurrentIndex(-1);

clauseEdit->clear();

});

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

[=](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)

.arg(id)

.arg(clauses)

.trimmed();

: QStyledItemDelegate(parent)

, m_db(db)

, m_table(table)

{

for(auto it=m_db.schemata.constBegin();it!=m_db.schemata.constEnd();++it)

{

for(auto jt=it->constBegin();jt!=it->constEnd();++jt)

{

if((*jt)->type() == sqlb::Object::Types::Table)

{

QString tableName = (*jt)->name();

m_tablesIds.insert(tableName, std::dynamic_pointer_cast<sqlb::Table>(*jt)->fieldNames());

}

}

}

{

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

int column = index.row(); // weird? I know right

const sqlb::Field& field = m_table.fields.at(column);

auto fk = std::dynamic_pointer_cast<sqlb::ForeignKeyClause>(m_table.constraint({field.name()}, sqlb::Constraint::ForeignKeyConstraintType));

if (fk) {

fkEditor->tablesComboBox->setCurrentText(fk->table());

fkEditor->clauseEdit->setText(fk->constraint());

if (!fk->columns().isEmpty())

fkEditor->idsComboBox->setCurrentText(fk->columns().first());

} else {

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

}

{

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

QString sql = fkEditor->getSql();

int column = index.row();

const sqlb::Field& field = m_table.fields.at(column);

if (sql.isEmpty()) {

// Remove the foreign key

m_table.removeConstraints({field.name()}, sqlb::Constraint::ConstraintTypes::ForeignKeyConstraintType);

} else {

// Set the foreign key

sqlb::ForeignKeyClause* fk = new sqlb::ForeignKeyClause;

const QString table = fkEditor->tablesComboBox->currentText();

const QString id = fkEditor->idsComboBox->currentText();

const QString clause = fkEditor->clauseEdit->text();

fk->setTable(table);

if (!id.isEmpty())

fk->setColumns({id});

if (!clause.trimmed().isEmpty()) {

fk->setConstraint(clause);

}

m_table.setConstraint({field.name()}, sqlb::ConstraintPtr(fk));

}

model->setData(index, sql);

{

Q_UNUSED(index)

editor->setGeometry(option.rect);

{

// this is used for recursive table constraints when

// table column references column within same table

m_tablesIds.remove(oldTableName);

m_tablesIds.insert(m_table.name(), m_table.fieldNames());