1 files changed, 299 insertions, 0 deletions
diff --git a/SQLiteStudio3/guiSQLiteStudio/constraints/tablepkanduniquepanel.cpp b/SQLiteStudio3/guiSQLiteStudio/constraints/tablepkanduniquepanel.cpp
new file mode 100644
index 0000000..f2a0ada
--- /dev/null
+++ b/SQLiteStudio3/guiSQLiteStudio/constraints/tablepkanduniquepanel.cpp
@@ -0,0 +1,299 @@
+#include "tablepkanduniquepanel.h"
+#include "ui_tablepkanduniquepanel.h"
+#include "parser/keywords.h"
+#include "schemaresolver.h"
+#include "uiutils.h"
+#include <QGridLayout>
+#include <QHBoxLayout>
+#include <QCheckBox>
+#include <QComboBox>
+#include <QLabel>
+#include <QSignalMapper>
+#include <QDebug>
+#include <QScrollBar>
+
+TablePrimaryKeyAndUniquePanel::TablePrimaryKeyAndUniquePanel(QWidget *parent) :
+    ConstraintPanel(parent),
+    ui(new Ui::TablePrimaryKeyAndUniquePanel)
+{
+    ui->setupUi(this);
+    init();
+}
+
+TablePrimaryKeyAndUniquePanel::~TablePrimaryKeyAndUniquePanel()
+{
+    delete ui;
+}
+
+void TablePrimaryKeyAndUniquePanel::changeEvent(QEvent *e)
+{
+    QWidget::changeEvent(e);
+    switch (e->type()) {
+        case QEvent::LanguageChange:
+            ui->retranslateUi(this);
+            break;
+        default:
+            break;
+    }
+}
+
+void TablePrimaryKeyAndUniquePanel::init()
+{
+    columnsLayout = new QGridLayout();
+    ui->scrollAreaWidgetContents->setLayout(columnsLayout);
+
+    connect(ui->namedCheckBox, SIGNAL(toggled(bool)), this, SIGNAL(updateValidation()));
+    connect(ui->namedLineEdit, SIGNAL(textChanged(QString)), this, SIGNAL(updateValidation()));
+
+    ui->conflictComboBox->addItems(getConflictAlgorithms());
+
+    columnSignalMapping = new QSignalMapper(this);
+    connect(columnSignalMapping, SIGNAL(mapped(int)), this, SLOT(updateColumnState(int)));
+
+    connect(ui->namedCheckBox, SIGNAL(toggled(bool)), this, SLOT(updateState()));
+    connect(ui->conflictCheckBox, SIGNAL(toggled(bool)), this, SLOT(updateState()));
+    updateState();
+}
+
+void TablePrimaryKeyAndUniquePanel::readCollations()
+{
+    SchemaResolver resolver(db);
+    QStringList collList = resolver.getCollations();
+
+    if (collList.size() > 0)
+        collList.prepend("");
+
+    collations.setStringList(collList);
+}
+
+void TablePrimaryKeyAndUniquePanel::buildColumn(SqliteCreateTable::Column* column, int row)
+{
+    int col = 0;
+
+    QCheckBox* check = new QCheckBox(column->name);
+    columnsLayout->addWidget(check, row, col++);
+    columnSignalMapping->setMapping(check, row);
+    connect(check, SIGNAL(toggled(bool)), columnSignalMapping, SLOT(map()));
+    connect(check, SIGNAL(toggled(bool)), this, SIGNAL(updateValidation()));
+
+    QComboBox* collation = nullptr;
+    if (!constraint.isNull() && constraint->dialect == Dialect::Sqlite3)
+    {
+        collation = new QComboBox();
+        collation->setMaximumWidth(ui->colHdrCollation->width());
+        collation->setMinimumWidth(ui->colHdrCollation->width() - ui->scrollArea->verticalScrollBar()->width());
+        collation->setEditable(true);
+        collation->lineEdit()->setPlaceholderText(tr("Collate", "table constraints"));
+        collation->setModel(&collations);
+        columnsLayout->addWidget(collation, row, col++);
+    }
+
+    QComboBox* sortOrder = new QComboBox();
+    sortOrder->setFixedWidth(ui->colHdrSort->width());
+    sortOrder->setToolTip(tr("Sort order", "table constraints"));
+    columnsLayout->addWidget(sortOrder, row, col++);
+
+    QStringList sortList = {"", sqliteSortOrder(SqliteSortOrder::ASC), sqliteSortOrder(SqliteSortOrder::DESC)};
+    sortOrder->addItems(sortList);
+
+    totalColumns++;
+
+    updateColumnState(row);
+}
+
+int TablePrimaryKeyAndUniquePanel::getColumnIndex(const QString& colName)
+{
+    QWidget* item = nullptr;
+    QCheckBox* cb = nullptr;
+    for (int i = 0; i < totalColumns; i++)
+    {
+        item = columnsLayout->itemAtPosition(i, 0)->widget();
+        cb = qobject_cast<QCheckBox*>(item);
+        if (cb->text().compare(colName, Qt::CaseInsensitive) == 0)
+            return i;
+    }
+    return -1;
+}
+
+void TablePrimaryKeyAndUniquePanel::updateColumnState(int colIdx)
+{
+    QWidget* item = columnsLayout->itemAtPosition(colIdx, 0)->widget();
+    QCheckBox* cb = qobject_cast<QCheckBox*>(item);
+    bool enable = cb->isChecked();
+
+    item = columnsLayout->itemAtPosition(colIdx, 1)->widget();
+    qobject_cast<QComboBox*>(item)->setEnabled(enable);
+
+    if (!constraint.isNull() && constraint->dialect == Dialect::Sqlite3)
+    {
+        item = columnsLayout->itemAtPosition(colIdx, 2)->widget();
+        qobject_cast<QComboBox*>(item)->setEnabled(enable);
+    }
+
+    updateState();
+}
+
+void TablePrimaryKeyAndUniquePanel::updateState()
+{
+    ui->namedLineEdit->setEnabled(ui->namedCheckBox->isChecked());
+    ui->conflictComboBox->setEnabled(ui->conflictCheckBox->isChecked());
+}
+
+void TablePrimaryKeyAndUniquePanel::constraintAvailable()
+{
+    readCollations();
+    buildColumns();
+    readConstraint();
+}
+
+bool TablePrimaryKeyAndUniquePanel::validate()
+{
+    bool countOk  = false;
+    QWidget* item = nullptr;
+    QCheckBox* cb = nullptr;
+    for (int i = 0; i < totalColumns; i++)
+    {
+        item = columnsLayout->itemAtPosition(i, 0)->widget();
+        cb = qobject_cast<QCheckBox*>(item);
+        if (cb->isChecked())
+        {
+            countOk = true;
+            break;
+        }
+    }
+
+    bool nameOk = true;
+    if (ui->namedCheckBox->isChecked() && ui->namedLineEdit->text().isEmpty())
+        nameOk = false;
+
+    setValidState(ui->groupBox, countOk, tr("Select at least one column."));
+    setValidState(ui->namedLineEdit, nameOk, tr("Enter a name of the constraint."));
+
+    return countOk && nameOk;
+}
+
+
+void TablePrimaryKeyAndUniquePanel::storeConfiguration()
+{
+    if (constraint.isNull())
+        return;
+
+    SqliteCreateTable::Constraint* constr = dynamic_cast<SqliteCreateTable::Constraint*>(constraint.data());
+
+    // Name
+    constr->name = QString::null;
+    if (ui->namedCheckBox->isChecked())
+        constr->name = ui->namedLineEdit->text();
+
+    // On conflict
+    if (ui->conflictCheckBox->isChecked())
+        constr->onConflict = sqliteConflictAlgo(ui->conflictComboBox->currentText());
+
+    // Columns
+    foreach (SqliteIndexedColumn* idxCol, constr->indexedColumns)
+        delete idxCol;
+
+    constr->indexedColumns.clear();
+
+    QCheckBox* check = nullptr;
+    QComboBox* combo = nullptr;
+    SqliteIndexedColumn* idxCol = nullptr;
+    QString name;
+    QString collate;
+    SqliteSortOrder sortOrder;
+    for (int i = 0; i < totalColumns; i++)
+    {
+        check = dynamic_cast<QCheckBox*>(columnsLayout->itemAtPosition(i, 0)->widget());
+        if (!check->isChecked())
+            continue;
+
+        name = check->text();
+
+        if (constr->dialect == Dialect::Sqlite3)
+        {
+            combo = dynamic_cast<QComboBox*>(columnsLayout->itemAtPosition(i, 1)->widget());
+            collate = combo->currentText();
+            if (collate.isEmpty())
+                collate = QString::null;
+
+            combo = dynamic_cast<QComboBox*>(columnsLayout->itemAtPosition(i, 2)->widget());
+            sortOrder = sqliteSortOrder(combo->currentText());
+        }
+        else
+        {
+            combo = dynamic_cast<QComboBox*>(columnsLayout->itemAtPosition(i, 1)->widget());
+            sortOrder = sqliteSortOrder(combo->currentText());
+        }
+
+        idxCol = new SqliteIndexedColumn(name, collate, sortOrder);
+        idxCol->setParent(constr);
+        constr->indexedColumns << idxCol;
+    }
+}
+
+void TablePrimaryKeyAndUniquePanel::readConstraint()
+{
+    if (constraint.isNull())
+        return;
+
+    SqliteCreateTable::Constraint* constr = dynamic_cast<SqliteCreateTable::Constraint*>(constraint.data());
+
+    // Name
+    if (!constr->name.isNull())
+    {
+        ui->namedCheckBox->setChecked(true);
+        ui->namedLineEdit->setText(constr->name);
+    }
+
+    // On conflict
+    if (constr->onConflict != SqliteConflictAlgo::null)
+    {
+        ui->conflictCheckBox->setChecked(true);
+        ui->conflictComboBox->setCurrentText(sqliteConflictAlgo(constr->onConflict));
+    }
+
+    // Columns
+    int idx;
+    QCheckBox* check = nullptr;
+    QComboBox* combo = nullptr;
+    foreach (SqliteIndexedColumn* idxCol, constr->indexedColumns)
+    {
+        idx = getColumnIndex(idxCol->name);
+        if (idx < 0)
+            continue;
+
+        check = dynamic_cast<QCheckBox*>(columnsLayout->itemAtPosition(idx, 0)->widget());
+        check->setChecked(true);
+
+        if (constr->dialect == Dialect::Sqlite3)
+        {
+            combo = dynamic_cast<QComboBox*>(columnsLayout->itemAtPosition(idx, 1)->widget());
+            combo->setCurrentText(idxCol->collate);
+
+            combo = dynamic_cast<QComboBox*>(columnsLayout->itemAtPosition(idx, 2)->widget());
+            combo->setCurrentText(sqliteSortOrder(idxCol->sortOrder));
+        }
+        else
+        {
+            combo = dynamic_cast<QComboBox*>(columnsLayout->itemAtPosition(idx, 1)->widget());
+            combo->setCurrentText(sqliteSortOrder(idxCol->sortOrder));
+        }
+    }
+
+    if (constr->dialect == Dialect::Sqlite2)
+    {
+        ui->colHdrCollation->setVisible(false);
+    }
+}
+
+void TablePrimaryKeyAndUniquePanel::buildColumns()
+{
+    totalColumns = 0;
+    if (constraint.isNull())
+        return;
+
+    SqliteCreateTable* createTable = dynamic_cast<SqliteCreateTable*>(constraint->parentStatement());
+    int row = 0;
+    foreach (SqliteCreateTable::Column* column, createTable->columns)
+        buildColumn(column, row++);
+}