Can now click on a folder to filter the results and the folders are persistent now.

src/Database.cpp

@@ -1,3 +1,4 @@
+#include <deque>
 #include <exception>
 
 #include <wx/log.h>
@@ -18,10 +19,10 @@ Database::~Database()
 
 }
 
-void Database::CreateDatabase()
+void Database::CreateTableSamples()
 {
     /* Create SQL statement */
-    std::string sample = "CREATE TABLE IF NOT EXISTS SAMPLES("
+    std::string samples = "CREATE TABLE IF NOT EXISTS SAMPLES("
         "FAVORITE       INT     NOT NULL,"
         "FILENAME       TEXT    NOT NULL,"
         "EXTENSION      TEXT    NOT NULL,"
@@ -47,11 +48,55 @@ void Database::CreateDatabase()
             wxLogDebug("Opening DB..");
         }
 
-        rc = sqlite3_exec(m_Database, sample.c_str(), NULL, 0, &m_ErrMsg);
+        rc = sqlite3_exec(m_Database, samples.c_str(), NULL, 0, &m_ErrMsg);
 
         if (rc != SQLITE_OK)
         {
-            wxMessageDialog msgDialog(NULL, "Error! Cannot create table.",
+            wxMessageDialog msgDialog(NULL, "Error! Cannot create table samples.",
+                                      "Error", wxOK | wxICON_ERROR);
+            msgDialog.ShowModal();
+            sqlite3_free(m_ErrMsg);
+        }
+        else
+        {
+            wxLogDebug("Table created successfully.");
+        }
+
+        rc = sqlite3_close(m_Database);
+
+        if (rc == SQLITE_OK)
+            wxLogDebug("DB Closed..");
+        else
+            wxLogDebug("Error! Cannot close DB, Error code: %d, Error message: %s", rc, m_ErrMsg);
+    }
+    catch (const std::exception &exception)
+    {
+        wxLogDebug(exception.what());
+    }
+}
+
+void Database::CreateTableCollections()
+{
+    /* Create SQL statement */
+    std::string collections = "CREATE TABLE IF NOT EXISTS COLLECTIONS(FOLDERNAME TEXT NOT NULL);";
+
+    try
+    {
+        if (sqlite3_open("sample.hive", &m_Database) != SQLITE_OK)
+        {
+            wxLogDebug("Error opening DB");
+            throw sqlite3_errmsg(m_Database);
+        }
+        else
+        {
+            wxLogDebug("Opening DB..");
+        }
+
+        rc = sqlite3_exec(m_Database, collections.c_str(), NULL, 0, &m_ErrMsg);
+
+        if (rc != SQLITE_OK)
+        {
+            wxMessageDialog msgDialog(NULL, "Error! Cannot create table collections.",
                                       "Error", wxOK | wxICON_ERROR);
             msgDialog.ShowModal();
             sqlite3_free(m_ErrMsg);
@@ -75,7 +120,7 @@ void Database::CreateDatabase()
 }
 
 //Loops through a Sample array and adds them to the database
-void Database::InsertSamples(std::vector<Sample> samples)
+void Database::InsertIntoSamples(std::vector<Sample> samples)
 {
     try
     {
@@ -186,6 +231,104 @@ void Database::InsertSamples(std::vector<Sample> samples)
     }
 }
 
+void Database::InsertIntoCollections(const std::string& folderName)
+{
+    try
+    {
+        if (sqlite3_open("sample.hive", &m_Database) != SQLITE_OK)
+        {
+            wxLogDebug("Error opening DB");
+            throw sqlite3_errmsg(m_Database);
+        }
+        else
+        {
+            wxLogDebug("Opening DB..");
+        }
+
+        std::string insert = "INSERT INTO COLLECTIONS(FOLDERNAME) VALUES(?);";
+
+        rc = sqlite3_prepare_v2(m_Database, insert.c_str(), insert.size(), &m_Stmt, NULL);
+
+        // rc = sqlite3_exec(m_Database, "BEGIN TRANSACTION", NULL, NULL, &m_ErrMsg);
+
+        if (rc != SQLITE_OK)
+            wxLogDebug("Cannot prepare sql statement..");
+
+        // Sample sample;
+
+        // std::string filename;
+        // std::string file_extension;
+        // std::string sample_pack;
+        // std::string type;
+        // std::string path;
+
+        // for(unsigned int i = 0; i < samples.size(); i++)
+        // {
+        //     sample = samples[i];
+
+        //     filename = sample.GetFilename();
+        //     file_extension = sample.GetFileExtension();
+        //     sample_pack = sample.GetSamplePack();
+        //     type = sample.GetType();
+        //     path = sample.GetPath();
+
+        //     std::string folder = "Favourites";
+
+        // rc = sqlite3_bind_int(m_Stmt, 1, sample.GetFavorite());
+        rc = sqlite3_bind_text(m_Stmt, 1, folderName.c_str(), folderName.size(), SQLITE_STATIC);
+
+        rc = sqlite3_step(m_Stmt);
+            // rc = sqlite3_clear_bindings(m_Stmt);
+            // rc = sqlite3_reset(m_Stmt);
+        // }
+
+        // rc = sqlite3_exec(m_Database, "END TRANSACTION", NULL, NULL, &m_ErrMsg);
+
+        rc = sqlite3_finalize(m_Stmt);
+
+        if (rc != SQLITE_OK)
+        {
+            wxLogDebug("Error! Cannot insert data into table. Error code: %d: Msg: %s", rc , sqlite3_errmsg(m_Database));
+        }
+        else
+        {
+            wxLogDebug("Data inserted successfully.");
+        }
+
+        if (rc == SQLITE_BUSY)
+            wxLogDebug("SQLITE_BUSY");
+        if (rc == SQLITE_ABORT)
+            wxLogDebug("SQLITE_ABORT");
+        if (rc == SQLITE_NOMEM)
+            wxLogDebug("SQLITE_NOMEM");
+        if (rc == SQLITE_LOCKED)
+            wxLogDebug("SQLITE_LOCKED");
+        if (rc == SQLITE_IOERR)
+            wxLogDebug("SQLITE_IOERR");
+        if (rc == SQLITE_CORRUPT)
+            wxLogDebug("SQLITE_CORRUPT");
+        if (rc == SQLITE_READONLY)
+            wxLogDebug("SQLITE_READONLY");
+        if (rc == SQLITE_ERROR)
+            wxLogDebug("SQLITE_ERROR");
+        if (rc == SQLITE_PERM)
+            wxLogDebug("SQLITE_PERM");
+        if (rc == SQLITE_INTERNAL)
+            wxLogDebug("SQLITE_INTERNAL");
+
+        rc = sqlite3_close(m_Database);
+
+        if (rc == SQLITE_OK)
+            wxLogDebug("DB Closed..");
+        else
+            wxLogDebug("Error! Cannot close DB, Error code: %d, Error message: %s", rc, m_ErrMsg);
+    }
+    catch (const std::exception &exception)
+    {
+        wxLogDebug(exception.what());
+    }
+}
+
 void Database::UpdateFolder(const std::string& folderName)
 {
     try
@@ -555,6 +698,45 @@ void Database::RemoveSampleFromDatabase(const std::string& filename)
     }
 }
 
+void Database::RemoveFolderFromCollections(const std::string& folderName)
+{
+    try
+    {
+        rc = sqlite3_open("sample.hive", &m_Database);
+
+        std::string remove = "DELETE FROM COLLECTIONS WHERE FOLDERNAME = ?;";
+
+        rc = sqlite3_prepare_v2(m_Database, remove.c_str(), remove.size(), &m_Stmt, NULL);
+
+        rc = sqlite3_bind_text(m_Stmt, 1, folderName.c_str(), folderName.size(), SQLITE_STATIC);
+
+        if (sqlite3_step(m_Stmt) == SQLITE_DONE)
+        {
+            wxLogDebug("Record found, Deleting..");
+        }
+
+        rc = sqlite3_finalize(m_Stmt);
+
+        if (rc != SQLITE_OK)
+        {
+            wxMessageDialog msgDialog(NULL, "Error! Cannot delete data from table.",
+                                      "Error", wxOK | wxICON_ERROR);
+            msgDialog.ShowModal();
+            sqlite3_free(m_ErrMsg);
+        }
+        else
+        {
+            wxLogDebug("Deleted data from table successfully.");
+        }
+
+        sqlite3_close(m_Database);
+    }
+    catch (const std::exception &exception)
+    {
+        wxLogDebug(exception.what());
+    }
+}
+
 std::string Database::GetSamplePathByFilename(const std::string& filename)
 {
     std::string path;
@@ -660,7 +842,7 @@ Database::LoadDatabase(wxVector<wxVector<wxVariant>>& vecSet,
 
         std::string load = "SELECT FAVORITE, FILENAME, EXTENSION, SAMPLEPACK, \
                             TYPE, CHANNELS, LENGTH, SAMPLERATE, BITRATE, PATH, \
-                            TRASHED FROM SAMPLES;";
+                            TRASHED, FOLDER FROM SAMPLES;";
 
         rc = sqlite3_prepare_v2(m_Database, load.c_str(), load.size(), &m_Stmt, NULL);
 
@@ -681,6 +863,7 @@ Database::LoadDatabase(wxVector<wxVector<wxVariant>>& vecSet,
                 int bitrate = sqlite3_column_int(m_Stmt, 8);
                 wxString path = std::string(reinterpret_cast<const char*>(sqlite3_column_text(m_Stmt, 9)));
                 int trashed = sqlite3_column_int(m_Stmt, 10);
+                wxString folder_name = std::string(reinterpret_cast<const char*>(sqlite3_column_text(m_Stmt, 11)));
 
                 wxVector<wxVariant> vec;
 
@@ -692,7 +875,52 @@ Database::LoadDatabase(wxVector<wxVector<wxVariant>>& vecSet,
                     {
                         vec.push_back(true);
 
-                        favorite_tree.AppendItem(favorite_item, filename);
+                        wxLogDebug("Loading collection items..");
+
+                        std::deque<wxDataViewItem> nodes;
+                        nodes.push_back(favorite_tree.GetNthChild(wxDataViewItem(wxNullPtr), 0));
+
+                        wxDataViewItem current_item, found_item;
+
+                        int row = 0;
+                        int folder_count = favorite_tree.GetChildCount(wxDataViewItem(wxNullPtr));
+
+                        while(!nodes.empty())
+                        {
+                            current_item = nodes.front();
+                            nodes.pop_front();
+
+                            if (favorite_tree.GetItemText(current_item) == folder_name)
+                            {
+                                found_item = current_item;
+                                wxLogDebug("Loading, folder name: %s", folder_name);
+                                break;
+                            }
+
+                            wxDataViewItem child = favorite_tree.GetNthChild(wxDataViewItem(wxNullPtr), 0);
+
+                            while (row < (folder_count - 1))
+                            {
+                                row ++;
+
+                                child = favorite_tree.GetNthChild(wxDataViewItem(wxNullPtr), row);
+                                nodes.push_back(child);
+                            }
+                        }
+
+                        nodes.clear();
+
+                        if (found_item.IsOk())
+                        {
+                            // wxLogDebug("Another folder by the name %s already exist. Please try with a different name.",
+                                       // folder_name);
+                            favorite_tree.AppendItem(found_item, filename);
+                        }
+                        // else
+                        // {
+                        //     favorite_tree.AppendItem(wxDataViewItem(wxNullPtr), folder_name);
+                        // }
+
                     }
                     else
                         vec.push_back(false);
@@ -814,6 +1042,123 @@ Database::FilterDatabaseBySampleName(wxVector<wxVector<wxVariant>>& sampleVec, c
     return sampleVec;
 }
 
+wxVector<wxVector<wxVariant>>
+Database::FilterDatabaseByFolderName(wxVector<wxVector<wxVariant>>& sampleVec, const std::string& folderName)
+{
+    try
+    {
+        if (sqlite3_open("sample.hive", &m_Database) != SQLITE_OK)
+        {
+            wxLogDebug("Error opening DB");
+            throw sqlite3_errmsg(m_Database);
+        }
+
+        std::string filter = "SELECT FAVORITE, FILENAME, SAMPLEPACK, TYPE, \
+                              CHANNELS, LENGTH, SAMPLERATE, BITRATE \
+                              FROM SAMPLES WHERE FOLDER = ? AND FAVORITE = 1;";
+
+        rc = sqlite3_prepare_v2(m_Database, filter.c_str(), filter.size(), &m_Stmt, NULL);
+
+        rc = sqlite3_bind_text(m_Stmt, 1, folderName.c_str(), folderName.size(), SQLITE_STATIC);
+
+        if (rc == SQLITE_OK)
+        {
+            int row = 0;
+
+            while (SQLITE_ROW == sqlite3_step(m_Stmt))
+            {
+                wxLogInfo("Record found, fetching..");
+                int favorite = sqlite3_column_int(m_Stmt, 0);
+                wxString filename = wxString(std::string(reinterpret_cast<const char*>(sqlite3_column_text(m_Stmt, 1))));
+                wxString sample_pack = wxString(std::string(reinterpret_cast<const char*>(sqlite3_column_text(m_Stmt, 2))));
+                wxString sample_type = std::string(reinterpret_cast<const char *>(sqlite3_column_text(m_Stmt, 3)));
+                int channels = sqlite3_column_int(m_Stmt, 4);
+                int length = sqlite3_column_int(m_Stmt, 5);
+                int sample_rate = sqlite3_column_int(m_Stmt, 6);
+                int bitrate = sqlite3_column_int(m_Stmt, 7);
+
+                wxVector<wxVariant> vec;
+
+                if (favorite == 1)
+                    vec.push_back(true);
+                else
+                    vec.push_back(false);
+
+                vec.push_back(filename);
+                vec.push_back(sample_pack);
+                vec.push_back(sample_type);
+                vec.push_back(wxString::Format("%d", channels));
+                vec.push_back(wxString::Format("%d", length));
+                vec.push_back(wxString::Format("%d", sample_rate));
+                vec.push_back(wxString::Format("%d", bitrate));
+
+                sampleVec.push_back(vec);
+
+                row++;
+            }
+        }
+        else
+        {
+            wxMessageDialog msgDialog(NULL, "Error! Cannot filter data from table.",
+                                      "Error", wxOK | wxICON_ERROR);
+            msgDialog.ShowModal();
+            sqlite3_free(m_ErrMsg);
+        }
+
+        rc = sqlite3_finalize(m_Stmt);
+
+        sqlite3_close(m_Database);
+    }
+    catch (const std::exception &exception)
+    {
+        wxLogDebug(exception.what());
+    }
+
+    return sampleVec;
+}
+
+void Database::LoadCollectionFolder(wxDataViewTreeCtrl& treeCtrl)
+{
+    try
+    {
+        if (sqlite3_open("sample.hive", &m_Database) != SQLITE_OK)
+        {
+            wxLogDebug("Error opening DB");
+            throw sqlite3_errmsg(m_Database);
+        }
+
+        std::string select = "SELECT FOLDERNAME FROM COLLECTIONS;";
+
+        rc = sqlite3_prepare_v2(m_Database, select.c_str(), select.size(), &m_Stmt, NULL);
+
+        if (rc == SQLITE_OK)
+        {
+            while (SQLITE_ROW == sqlite3_step(m_Stmt))
+            {
+                wxLogInfo("Record found, fetching..");
+                wxString folder = wxString(std::string(reinterpret_cast<const char*>(sqlite3_column_text(m_Stmt, 0))));
+
+                treeCtrl.AppendContainer(wxDataViewItem(wxNullPtr), folder);
+            }
+        }
+        else
+        {
+            wxMessageDialog msgDialog(NULL, "Error! Cannot load foldername from collection table.",
+                                      "Error", wxOK | wxICON_ERROR);
+            msgDialog.ShowModal();
+            sqlite3_free(m_ErrMsg);
+        }
+
+        rc = sqlite3_finalize(m_Stmt);
+
+        sqlite3_close(m_Database);
+    }
+    catch (const std::exception &exception)
+    {
+        wxLogDebug(exception.what());
+    }
+}
+
 //Compares the input array with the database and removes duplicates.
 wxArrayString Database::CheckDuplicates(const wxArrayString& files)
 {