About to break stuff

Author: Jim Hamill

ExplorerWidget.cpp

@@ -6,6 +6,7 @@
 #include <QSqlQuery>
 #include <QSqlError>
 #include <QInputDialog>
+#include <QMenu>
 
 ExplorerWidget::ExplorerWidget(QWidget *parent) :
     QWidget(parent),
@@ -14,6 +15,15 @@ ExplorerWidget::ExplorerWidget(QWidget *parent) :
     m_schemaWidget(NULL)
 {
     ui->setupUi(this);
+
+    // Connect actions
+    connect(ui->actionAdd_Table, SIGNAL(triggered()), this, SLOT(addTable()));
+    connect(ui->actionRemove_Table, SIGNAL(triggered()), this, SLOT(removeTable()));
+    connect(ui->actionRename_Table, SIGNAL(triggered()), this, SLOT(renameTable()));
+
+    // Connect buttons
+    connect(ui->addButton, SIGNAL(clicked()), this, SLOT(addTable()));
+    connect(ui->removeButton, SIGNAL(clicked()), this, SLOT(removeTable()));
 }
 
 ExplorerWidget::~ExplorerWidget()
@@ -29,6 +39,16 @@ void ExplorerWidget::init(QSqlDatabase *database, Extension *extension)
     // Set the extension
     m_extension = extension;
 
+    // Hide add button if it's not supported
+    if (!m_extension || !m_extension->hasCapability(ADD_TABLE)) {
+        ui->addButton->setVisible(false);
+    }
+
+    // Hide remove button if it's not supported
+    if (!m_extension || !m_extension->hasCapability(REMOVE_TABLE)) {
+        ui->removeButton->setVisible(false);
+    }
+
     // Set the database for child widgets
     ui->browseTab->setDatabase(m_database);
 
@@ -69,42 +89,69 @@ void ExplorerWidget::on_tableListWidget_itemActivated(QListWidgetItem *item)
     m_schemaWidget->setTable(item->text());
 }
 
-void ExplorerWidget::on_removeButton_clicked()
+void ExplorerWidget::addTable()
+{
+    // Get table name
+    QString table = QInputDialog::getText(this, "Create Table", "Enter table name");
+
+    // Create table if extension supports it
+    m_extension->createTable(table);
+
+    // Refresh explorer
+    refresh();
+}
+
+void ExplorerWidget::removeTable()
 {
     // Display messagebox for confirmation
     if (QMessageBox::Yes == QMessageBox::question(this, "Drop Table", "Are you sure you want to drop this table?")) {
 
         // Get table name
         QString table = ui->tableListWidget->currentItem()->text();
 
-        // Drop table if extension supports it
-        if (m_extension) {
-            m_extension->removeTable(table);
-        }
-
-        else {
-            QMessageBox::warning(this, "Sorry", "This functionality is not available for this database type yet");
-        }
+        // Drop table
+        m_extension->removeTable(table);
 
         // Refresh explorer
         refresh();
     }
 }
 
-void ExplorerWidget::on_addButton_clicked()
+void ExplorerWidget::renameTable()
 {
-    // Get table name
-    QString table = QInputDialog::getText(this, "Create Table", "Enter table name");
+    // Get old table name
+    QString from = ui->tableListWidget->currentItem()->text();
 
-    // Drop table if extension supports it
-    if (m_extension) {
-        m_extension->createTable(table);
-    }
+    // Get new table name
+    QString to = QInputDialog::getText(this, "Rename table", "Enter new table name");
 
-    else {
-        QMessageBox::warning(this, "Sorry", "This functionality is not available for this database type yet");
-    }
+    // Rename table
+    m_extension->renameTable(from, to);
 
     // Refresh explorer
     refresh();
 }
+
+void ExplorerWidget::on_tableListWidget_customContextMenuRequested(const QPoint &pos)
+{
+    // Setup context menu
+    QMenu *menu = new QMenu(this);
+
+    // If our extension can add tables
+    if (m_extension && m_extension->hasCapability(ADD_TABLE)) {
+        menu->addAction(ui->actionAdd_Table);
+    }
+
+    // If our extension can remove tables
+    if (m_extension && m_extension->hasCapability(REMOVE_TABLE)) {
+        menu->addAction(ui->actionRemove_Table);
+    }
+
+    // If our extension can rename tables
+    if (m_extension && m_extension->hasCapability(RENAME_TABLE)) {
+        menu->addAction(ui->actionRename_Table);
+    }
+
+    // Show the menu
+    menu->popup(ui->tableListWidget->viewport()->mapToGlobal(pos));
+}

ExplorerWidget.h

@@ -22,14 +22,14 @@ class ExplorerWidget : public QWidget
     void init(QSqlDatabase *database, Extension *extension = NULL);
 
 public slots:
+    void addTable();
+    void removeTable();
+    void renameTable();
     void refresh();
 
 private slots:
     void on_tableListWidget_itemActivated(QListWidgetItem *item);
-
-    void on_removeButton_clicked();
-
-    void on_addButton_clicked();
+    void on_tableListWidget_customContextMenuRequested(const QPoint &pos);
 
 private:
     Ui::ExplorerWidget *ui;

ExplorerWidget.ui

@@ -34,6 +34,9 @@
           <height>16777215</height>
          </size>
         </property>
+        <property name="contextMenuPolicy">
+         <enum>Qt::CustomContextMenu</enum>
+        </property>
        </widget>
       </item>
       <item>
@@ -81,6 +84,30 @@
     </widget>
    </item>
   </layout>
+  <action name="actionAdd_Table">
+   <property name="icon">
+    <iconset theme="list-add"/>
+   </property>
+   <property name="text">
+    <string>Add Table</string>
+   </property>
+  </action>
+  <action name="actionRemove_Table">
+   <property name="icon">
+    <iconset theme="list-remove"/>
+   </property>
+   <property name="text">
+    <string>Remove Table</string>
+   </property>
+  </action>
+  <action name="actionRename_Table">
+   <property name="icon">
+    <iconset theme="document-edit"/>
+   </property>
+   <property name="text">
+    <string>Rename Table</string>
+   </property>
+  </action>
  </widget>
  <customwidgets>
   <customwidget>

Extension/Extension.cpp

@@ -39,6 +39,26 @@ Extension::~Extension()
 
 }
 
+int Extension::hasCapability(Capability capability)
+{
+    return false;
+}
+
+int Extension::importDatabase()
+{
+    return false;
+}
+
+int Extension::exportDatabase()
+{
+    return false;
+}
+
+int Extension::clearDatabase()
+{
+    return false;
+}
+
 int Extension::createTable(QString table)
 {
     return false;
@@ -49,6 +69,21 @@ int Extension::removeTable(QString table)
     return false;
 }
 
+int Extension::renameTable(QString from, QString to)
+{
+    return false;
+}
+
+int Extension::addColumn(QString table)
+{
+    return false;
+}
+
+int Extension::removeColumn(QString table, QString column)
+{
+    return false;
+}
+
 void Extension::addRelations(QSqlRelationalTableModel *model)
 {
 

Extension/Extension.h

@@ -26,6 +26,20 @@ class ExtensionTab
     QWidget *m_widget;
 };
 
+enum Capability
+{
+    IMPORT_DATABASE = 1,
+    EXPORT_DATABASE = 2,
+    CLEAR_DATABASE = 3,
+
+    ADD_TABLE = 11,
+    REMOVE_TABLE = 12,
+    RENAME_TABLE = 13,
+
+    ADD_COLUMN = 21,
+    REMOVE_COLUMN = 22
+};
+
 class Extension : public QObject
 {
 Q_OBJECT
@@ -34,8 +48,22 @@ Q_OBJECT
     explicit Extension(QObject *parent = 0, QSqlDatabase *database = new QSqlDatabase());
     ~Extension();
 
+    // Defines capabilities
+    virtual int hasCapability(Capability capability);
+
+    // Database functions
+    virtual int importDatabase();
+    virtual int exportDatabase();
+    virtual int clearDatabase();
+
+    // Table functions
     virtual int createTable(QString table);
     virtual int removeTable(QString table);
+    virtual int renameTable(QString from, QString to);
+
+    // Schema functions
+    virtual int addColumn(QString table);
+    virtual int removeColumn(QString table, QString column);
 
     virtual void addRelations(QSqlRelationalTableModel *model);