1 files changed, 107 insertions, 0 deletions
diff --git a/SQLiteStudio3/guiSQLiteStudio/constraints/columncollatepanel.cpp b/SQLiteStudio3/guiSQLiteStudio/constraints/columncollatepanel.cpp
new file mode 100644
index 0000000..de78b2b
--- /dev/null
+++ b/SQLiteStudio3/guiSQLiteStudio/constraints/columncollatepanel.cpp
@@ -0,0 +1,107 @@
+#include "columncollatepanel.h"
+#include "ui_columncollatepanel.h"
+#include "parser/ast/sqlitecreatetable.h"
+#include "schemaresolver.h"
+#include "uiutils.h"
+#include <QStringListModel>
+
+ColumnCollatePanel::ColumnCollatePanel(QWidget *parent) :
+    ConstraintPanel(parent),
+    ui(new Ui::ColumnCollatePanel)
+{
+    ui->setupUi(this);
+    init();
+}
+
+ColumnCollatePanel::~ColumnCollatePanel()
+{
+    delete ui;
+}
+
+void ColumnCollatePanel::changeEvent(QEvent *e)
+{
+    QWidget::changeEvent(e);
+    switch (e->type()) {
+        case QEvent::LanguageChange:
+            ui->retranslateUi(this);
+            break;
+        default:
+            break;
+    }
+}
+
+void ColumnCollatePanel::init()
+{
+    collationModel = new QStringListModel(this);
+    ui->collationCombo->setModel(collationModel);
+    connect(ui->namedCheck, SIGNAL(toggled(bool)), this, SIGNAL(updateValidation()));
+    connect(ui->namedEdit, SIGNAL(textChanged(QString)), this, SIGNAL(updateValidation()));
+    connect(ui->collationCombo->lineEdit(), SIGNAL(textChanged(QString)), this, SIGNAL(updateValidation()));
+    connect(ui->namedCheck, SIGNAL(toggled(bool)), this, SLOT(updateState()));
+    updateState();
+}
+
+void ColumnCollatePanel::readConstraint()
+{
+    SqliteCreateTable::Column::Constraint* constr = dynamic_cast<SqliteCreateTable::Column::Constraint*>(constraint.data());
+    ui->collationCombo->setCurrentText(constr->collationName);
+
+    if (!constr->name.isNull())
+    {
+        ui->namedCheck->setChecked(true);
+        ui->namedEdit->setText(constr->name);
+    }
+}
+
+void ColumnCollatePanel::readCollations()
+{
+    SchemaResolver resolver(db);
+    QStringList collList = resolver.getCollations();
+
+    if (collList.size() > 0)
+        collList.prepend("");
+
+    collationModel->setStringList(collList);
+}
+
+void ColumnCollatePanel::updateState()
+{
+    ui->namedEdit->setEnabled(ui->namedCheck->isChecked());
+}
+
+bool ColumnCollatePanel::validate()
+{
+    bool nameOk = true;
+    if (ui->namedCheck->isChecked() && ui->namedEdit->text().isEmpty())
+        nameOk = false;
+
+    bool collationOk = !ui->collationCombo->currentText().isEmpty();
+
+    setValidState(ui->namedEdit, nameOk, tr("Enter a name of the constraint."));
+    setValidState(ui->collationCombo, collationOk, tr("Enter a collation name."));
+
+    return nameOk && collationOk;
+}
+
+void ColumnCollatePanel::constraintAvailable()
+{
+    if (constraint.isNull())
+        return;
+
+    readCollations();
+    readConstraint();
+}
+
+void ColumnCollatePanel::storeConfiguration()
+{
+    if (constraint.isNull())
+        return;
+
+    SqliteCreateTable::Column::Constraint* constr = dynamic_cast<SqliteCreateTable::Column::Constraint*>(constraint.data());
+    constr->type = SqliteCreateTable::Column::Constraint::COLLATE;
+
+    if (ui->namedCheck->isChecked())
+        constr->name = ui->namedEdit->text();
+
+    constr->collationName = ui->collationCombo->currentText();
+}