From 6ce50e22500dea575677cea2a2e3d98c0ad22696 Mon Sep 17 00:00:00 2001
From: Jack Lilhammers <jack.lilhammers@gmail.com>
Date: Tue, 7 Jul 2020 16:50:17 +0200
Subject: [PATCH] Draft of support for the unique constraint (#95)

---
 src/generators/sqlgeneratorbase.cpp |  4 +++-
 src/generators/sqlitegenerator.cpp  |  3 +++
 src/tablemodel.cpp                  | 10 +++++++++-
 src/tablemodel.h                    |  3 ++-
 4 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/generators/sqlgeneratorbase.cpp b/src/generators/sqlgeneratorbase.cpp
index e1484c1..4307ed9 100644
--- a/src/generators/sqlgeneratorbase.cpp
+++ b/src/generators/sqlgeneratorbase.cpp
@@ -147,7 +147,9 @@ QString SqlGeneratorBase::fieldDeclare(FieldModel *field)
     QString type = fieldType(field);
     if (type.isEmpty())
         return type;
-    return field->name + " " + type + (field->notNull ? " NOT NULL" : "");
+    return field->name + " " + type
+            + (field->notNull ? " NOT NULL" : "")
+            + (field->isUnique ? " UNIQUE" : "");
 }
 
 QStringList SqlGeneratorBase::constraints(TableModel *table)
diff --git a/src/generators/sqlitegenerator.cpp b/src/generators/sqlitegenerator.cpp
index bc2f921..2ab1826 100644
--- a/src/generators/sqlitegenerator.cpp
+++ b/src/generators/sqlitegenerator.cpp
@@ -104,6 +104,9 @@ QString SqliteGenerator::fieldDeclare(FieldModel *field)
     if (field->notNull)
         type.append(" NOT NULL");
 
+    if (field->isUnique)
+        type.append(" UNIQUE");
+
     return field->name + " " + type;
 }
 
diff --git a/src/tablemodel.cpp b/src/tablemodel.cpp
index d8d8b43..1c9ddef 100644
--- a/src/tablemodel.cpp
+++ b/src/tablemodel.cpp
@@ -73,7 +73,7 @@ FieldModel *TableModel::field(const QString &name) const
     foreach (FieldModel *f, _fields)
         if(f->name == name)
             return f;
-    
+
     return nullptr;
 }
 
@@ -251,6 +251,9 @@ TableModel::TableModel(const QJsonObject &json, const QString &tableName) : _typ
         if(fieldObject.contains(__nut_NOT_NULL))
             f->notNull = fieldObject.value(__nut_NOT_NULL).toBool();
 
+        if(fieldObject.contains(__nut_UNIQUE))
+            f->isUnique = fieldObject.value(__nut_UNIQUE).toBool();
+
         if(fieldObject.contains(__nut_LEN))
             f->length = fieldObject.value(__nut_LEN).toInt();
 
@@ -288,6 +291,9 @@ QJsonObject TableModel::toJson() const
         if(f->notNull)
             fieldObj.insert(__nut_NOT_NULL, f->notNull);
 
+        if(f->isUnique)
+            fieldObj.insert(__nut_UNIQUE, f->isUnique);
+
         if(!f->defaultValue.isNull())
             fieldObj.insert(__nut_DEFAULT_VALUE, f->defaultValue);
 
@@ -359,6 +365,7 @@ FieldModel::FieldModel(const QJsonObject &json)
     type = static_cast<QMetaType::Type>(json.value(__TYPE).toInt());
     length = json.value(__nut_LEN).toInt();
     notNull = json.value(__nut_NOT_NULL).toBool();
+    isUnique = json.value(__nut_UNIQUE).toBool();
     isAutoIncrement = json.value(__nut_AUTO_INCREMENT).toBool();
     isPrimaryKey = json.value(__nut_PRIMARY_KEY).toBool();
     defaultValue = json.value(__nut_DEFAULT_VALUE).toString();
@@ -372,6 +379,7 @@ QJsonObject FieldModel::toJson() const
     fieldObj.insert(__TYPE, QString(QVariant::typeToName(type)));
     fieldObj.insert(__nut_LEN, length);
     fieldObj.insert(__nut_NOT_NULL, notNull);
+    fieldObj.insert(__nut_UNIQUE, isUnique);
     fieldObj.insert(__nut_AUTO_INCREMENT, isAutoIncrement);
     fieldObj.insert(__nut_PRIMARY_KEY, isPrimaryKey);
     fieldObj.insert(__nut_DEFAULT_VALUE, defaultValue);
diff --git a/src/tablemodel.h b/src/tablemodel.h
index 8a1bb8d..1fa89ae 100644
--- a/src/tablemodel.h
+++ b/src/tablemodel.h
@@ -55,7 +55,8 @@ struct FieldModel{
                 && type == f.type
                 && length == f.length
                 && defaultValue == f.defaultValue
-                && notNull == f.notNull;
+                && notNull == f.notNull
+                && isUnique == f.isUnique;
 
         return b;
     }