From 1fdc150116cad39aae5c5da407c3312b47a59e3a Mon Sep 17 00:00:00 2001
From: Unit 193 <unit193@unit193.net>
Date: Fri, 17 Dec 2021 07:06:30 -0500
Subject: New upstream version 3.3.3+dfsg1.

---
 SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h')

diff --git a/SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h b/SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h
index e374312..daa226c 100644
--- a/SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h
+++ b/SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h
@@ -12,6 +12,7 @@
 namespace Ui {
     class NewConstraintDialog;
 }
+class QCommandLinkButton;
 
 class GUI_API_EXPORT NewConstraintDialog : public QDialog
 {
@@ -25,6 +26,7 @@ class GUI_API_EXPORT NewConstraintDialog : public QDialog
         ~NewConstraintDialog();
 
         SqliteStatement* getConstraint();
+        void disableMode(ConstraintDialog::Constraint constraintType);
         int exec();
 
     protected:
@@ -34,7 +36,7 @@ class GUI_API_EXPORT NewConstraintDialog : public QDialog
         void init();
         void initTable();
         void initColumn();
-        void addButton(const Icon& icon, const QString text, const char* slot);
+        QCommandLinkButton* addButton(const Icon& icon, const QString text, const char* slot);
         int createColumnConstraint(ConstraintDialog::Constraint constraintType);
         int createTableConstraint(ConstraintDialog::Constraint constraintType);
         int editConstraint();
@@ -47,6 +49,7 @@ class GUI_API_EXPORT NewConstraintDialog : public QDialog
         QPointer<SqliteCreateTable> createTable;
         QPointer<SqliteCreateTable::Column> columnStmt;
         ConstraintDialog* constraintDialog = nullptr;
+        QHash<ConstraintDialog::Constraint, QCommandLinkButton*> modeToButton;
 
     private slots:
         void createTablePk();
@@ -60,6 +63,7 @@ class GUI_API_EXPORT NewConstraintDialog : public QDialog
         void createColumnNotNull();
         void createColumnDefault();
         void createColumnCollate();
+        void createColumnGenerated();
 };
 
 #endif // NEWCONSTRAINTDIALOG_H
-- 
cgit v1.2.3