Add button works

Author: Jim Hamill

BrowseWidget.cpp

@@ -86,8 +86,10 @@ void BrowseWidget::on_removeButton_clicked()
 
 void BrowseWidget::on_addButton_clicked()
 {
-    // Set the edit model to not save
-    m_model->setEditStrategy(QSqlTableModel::OnManualSubmit);
+    // Set the edit model to not save (only change if not already set, otherwise it crashes)
+    if (m_model->editStrategy() != QSqlTableModel::OnManualSubmit) {
+        m_model->setEditStrategy(QSqlTableModel::OnManualSubmit);
+    }
 
     // Create a new record
     QSqlRecord newRecord;

DatabaseConnectionWidget.cpp

@@ -93,17 +93,6 @@ bool DatabaseConnectionWidget::connectToDatabase(QString name, QString driver, Q
         return false;
     }
 
-    // Setup Explorer Widget
-    ui->explorerTab->setDatabase(&m_database);
-
-    // Setup Query Widget
-    ui->queryTab->setDatabase(&m_database);
-    connect(ui->queryTab, SIGNAL(refreshNeeded()), this, SLOT(refresh()));
-
-    // Setup SQL Widget
-    ui->sqlTab->setDatabase(m_database);
-    connect(ui->sqlTab, SIGNAL(refreshNeeded()), this, SLOT(refresh()));
-
     // If there's extra support for this driver, add it
     if (driver == "QMYSQL" || driver == "QMYSQL3") {
 
@@ -123,6 +112,17 @@ bool DatabaseConnectionWidget::connectToDatabase(QString name, QString driver, Q
         }
     }
 
+    // Setup Explorer Widget
+    ui->explorerTab->init(&m_database, m_extension);
+
+    // Setup Query Widget
+    ui->queryTab->setDatabase(&m_database);
+    connect(ui->queryTab, SIGNAL(refreshNeeded()), this, SLOT(refresh()));
+
+    // Setup SQL Widget
+    ui->sqlTab->setDatabase(m_database);
+    connect(ui->sqlTab, SIGNAL(refreshNeeded()), this, SLOT(refresh()));
+
     return true;
 }
 

ExplorerWidget.cpp

@@ -5,10 +5,12 @@
 #include <QMessageBox>
 #include <QSqlQuery>
 #include <QSqlError>
+#include <QInputDialog>
 
 ExplorerWidget::ExplorerWidget(QWidget *parent) :
     QWidget(parent),
-    ui(new Ui::ExplorerWidget)
+    ui(new Ui::ExplorerWidget),
+    m_extension(NULL)
 {
     ui->setupUi(this);
 }
@@ -18,11 +20,14 @@ ExplorerWidget::~ExplorerWidget()
     delete ui;
 }
 
-void ExplorerWidget::setDatabase(QSqlDatabase *database)
+void ExplorerWidget::init(QSqlDatabase *database, Extension *extension)
 {
     // Set the database
     m_database = database;
 
+    // Set the extension
+    m_extension = extension;
+
     // Set the database for child widgets
     ui->browseTab->setDatabase(m_database);
     ui->schemaTab->setDatabase(m_database);
@@ -58,15 +63,34 @@ void ExplorerWidget::on_removeButton_clicked()
         // Get table name
         QString table = ui->tableListWidget->currentItem()->text();
 
-        // Run the drop query
-        QSqlQuery query = m_database->exec("DROP TABLE " + table);
+        // Drop table if extension supports it
+        if (m_extension) {
+            m_extension->removeTable(table);
+        }
 
-        // Check if there was an error and display it if there was
-        if (query.lastError().type() != QSqlError::NoError) {
-            QMessageBox::critical(this, "Operation failed", query.lastError().text());
+        else {
+            QMessageBox::warning(this, "Sorry", "This functionality is not available for this database type yet");
         }
 
         // Refresh explorer
         refresh();
     }
 }
+
+void ExplorerWidget::on_addButton_clicked()
+{
+    // Get table name
+    QString table = QInputDialog::getText(this, "Create Table", "Enter table name");
+
+    // Drop table if extension supports it
+    if (m_extension) {
+        m_extension->createTable(table);
+    }
+
+    else {
+        QMessageBox::warning(this, "Sorry", "This functionality is not available for this database type yet");
+    }
+
+    // Refresh explorer
+    refresh();
+}

ExplorerWidget.h

@@ -5,6 +5,8 @@
 #include <QSqlDatabase>
 #include <QListWidgetItem>
 
+#include "Extension/Extension.h"
+
 namespace Ui {
 class ExplorerWidget;
 }
@@ -17,7 +19,7 @@ class ExplorerWidget : public QWidget
     explicit ExplorerWidget(QWidget *parent = 0);
     ~ExplorerWidget();
 
-    void setDatabase(QSqlDatabase *database);
+    void init(QSqlDatabase *database, Extension *extension = NULL);
 
 public slots:
     void refresh();
@@ -27,11 +29,16 @@ private slots:
 
     void on_removeButton_clicked();
 
+    void on_addButton_clicked();
+
 private:
     Ui::ExplorerWidget *ui;
 
     // Database
     QSqlDatabase *m_database;
+
+    // Extension
+    Extension *m_extension;
 };
 
 #endif // EXPLORERWIDGET_H

ExplorerWidget.ui

@@ -39,7 +39,7 @@
       <item>
        <layout class="QHBoxLayout" name="horizontalLayout">
         <item>
-         <widget class="QPushButton" name="pushButton_2">
+         <widget class="QPushButton" name="addButton">
           <property name="text">
            <string>Add</string>
           </property>

Extension/Extension.cpp

@@ -39,6 +39,16 @@ Extension::~Extension()
 
 }
 
+int Extension::createTable(QString table)
+{
+    return false;
+}
+
+int Extension::removeTable(QString table)
+{
+    return false;
+}
+
 QList<ExtensionTab*> Extension::getTabs()
 {
     return m_tabs;

Extension/Extension.h

@@ -31,6 +31,9 @@ Q_OBJECT
     explicit Extension(QObject *parent = 0, QSqlDatabase *database = new QSqlDatabase());
     ~Extension();
 
+    virtual int createTable(QString table);
+    virtual int removeTable(QString table);
+
     virtual QList<ExtensionTab*> getTabs();
     virtual QList<QToolBar*> getToolBars();
 

Extension/MySQLExtension.cpp

@@ -7,6 +7,9 @@
 #include <QDir>
 #include <QProcess>
 #include <QMessageBox>
+#include <QInputDialog>
+#include <QSqlQuery>
+#include <QSqlError>
 
 MySQLExtension::MySQLExtension(QObject *parent, QSqlDatabase *database) :
     Extension(parent, database)
@@ -50,3 +53,19 @@ MySQLExtension::~MySQLExtension()
     delete m_backupAction;
     delete m_restoreAction;
 }
+
+int MySQLExtension::createTable(QString table)
+{
+    // Run the drop query
+    QSqlQuery query = m_database->exec("CREATE TABLE " + table + "(id INT NOT NULL AUTO_INCREMENT, PRIMARY KEY(id))");
+
+    return true;
+}
+
+int MySQLExtension::removeTable(QString table)
+{
+    // Run the drop query
+    QSqlQuery query = m_database->exec("DROP TABLE " + table);
+
+    return true;
+}

Extension/MySQLExtension.h

@@ -16,6 +16,9 @@ Q_OBJECT
     explicit MySQLExtension(QObject *parent = 0, QSqlDatabase *database = new QSqlDatabase());
     ~MySQLExtension();
 
+    int createTable(QString table);
+    int removeTable(QString table);
+
 private:
     ExtensionTab *m_toolsTab;
     QToolBar m_toolsToolBar;

SqlWidget.cpp

@@ -88,7 +88,7 @@ void SqlWidget::on_runSqlButton_clicked()
     }
 
     // Let the user know the code has been run
-    QMessageBox::information(this, "SQL Executed Successfully", "SQL Code has been executed successfully.");
+    QMessageBox::information(this, "SQL Executed Successfully", "SQL has been executed successfully.");
 
     // Emit a signal that the database needs refreshing
     emit refreshNeeded();