/* SampleHive
 * Copyright (C) 2021  Apoorv Singh
 * A simple, modern audio sample browser/manager for GNU/Linux.
 *
 * This file is a part of SampleHive
 *
 * SampleHive is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * SampleHive is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "GUI/Trash.hpp"
#include "Database/Database.hpp"
#include "Utility/ControlIDs.hpp"
#include "Utility/HiveData.hpp"
#include "Utility/Log.hpp"
#include "Utility/Paths.hpp"
#include "Utility/Signal.hpp"
#include "Utility/Serialize.hpp"
#include "Utility/Utils.hpp"

#include <exception>

#include <wx/gdicmn.h>
#include <wx/menu.h>
#include <wx/msgdlg.h>

cTrashPanel::cTrashPanel(wxWindow* window)
    : wxPanel(window, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL),
      m_pWindow(window)
{
    m_pMainSizer = new wxBoxSizer(wxVERTICAL);
    m_pTrashSizer = new wxBoxSizer(wxVERTICAL);
    m_pButtonSizer = new wxBoxSizer(wxHORIZONTAL);

    m_pTrash = new wxTreeCtrl(this, SampleHive::ID::BC_Trash, wxDefaultPosition, wxDefaultSize,
                              wxTR_NO_BUTTONS | wxTR_HIDE_ROOT | wxTR_MULTIPLE);

    // Setting m_Trash to accept files to be dragged and dropped on it
    m_pTrash->DragAcceptFiles(true);

    m_pTrashSizer->Add(m_pTrash, wxSizerFlags(1).Expand());

    m_pRestoreTrashedItemButton = new wxButton(this, SampleHive::ID::BC_RestoreTrashedItem, _("Restore sample"),
                                               wxDefaultPosition, wxDefaultSize, 0);
    m_pRestoreTrashedItemButton->SetToolTip(_("Restore selected sample"));

    m_pButtonSizer->Add(m_pRestoreTrashedItemButton, wxSizerFlags(1).Expand());

    // Addubg root to TrashedItems
    m_TrashRoot = m_pTrash->AddRoot("Trash");

    m_pTrash->Connect(wxEVT_DROP_FILES, wxDropFilesEventHandler(cTrashPanel::OnDragAndDropToTrash), NULL, this);
    Bind(wxEVT_TREE_ITEM_RIGHT_CLICK, &cTrashPanel::OnShowTrashContextMenu, this, SampleHive::ID::BC_Trash);
    Bind(wxEVT_BUTTON, &cTrashPanel::OnClickRestoreTrashItem, this, SampleHive::ID::BC_RestoreTrashedItem);

    m_pMainSizer->Add(m_pTrashSizer, wxSizerFlags(1).Expand());
    m_pMainSizer->Add(m_pButtonSizer, wxSizerFlags(0).Expand());

    // Sizer for trash pane
    this->SetSizer(m_pMainSizer);
    m_pMainSizer->Fit(this);
    m_pMainSizer->SetSizeHints(this);
    m_pMainSizer->Layout();
}

void cTrashPanel::OnDragAndDropToTrash(wxDropFilesEvent& event)
{
    SampleHive::cSerializer serializer;
    cDatabase db;

    if (event.GetNumberOfFiles() > 0)
    {
        wxFileDataObject file_data;
        wxArrayString files;

        wxDataViewItemArray items;
        int rows = SampleHive::cHiveData::Get().GetListCtrlSelections(items);

        wxString msg;

        wxDataViewItem root = wxDataViewItem(wxNullPtr);
        wxDataViewItem container, child;

        for (int i = 0; i < rows; i++)
        {
            int item_row = SampleHive::cHiveData::Get().GetListCtrlRowFromItem(items, i);

            wxString text_value = SampleHive::cHiveData::Get().GetListCtrlTextValue(item_row, 1);

            std::string multi_selection = serializer.DeserializeShowFileExtension() ?
                SampleHive::cHiveData::Get().GetListCtrlTextValue(item_row, 1).BeforeLast('.').ToStdString() :
                SampleHive::cHiveData::Get().GetListCtrlTextValue(item_row, 1).ToStdString() ;

            file_data.AddFile(multi_selection);

            files = file_data.GetFilenames();

            if (db.GetFavoriteColumnValueByFilename(files[i].ToStdString()))
            {
                SampleHive::cHiveData::Get().ListCtrlSetVariant(wxVariant(wxBitmap(ICON_STAR_EMPTY_16px, wxBITMAP_TYPE_PNG)), item_row, 0);

                db.UpdateFavoriteColumn(files[i].ToStdString(), 0);

                for (int j = 0; j < SampleHive::cHiveData::Get().GetHiveChildCount(root); j++)
                {
                    container = SampleHive::cHiveData::Get().GetHiveNthChild(root, j);

                    for (int k = 0; k < SampleHive::cHiveData::Get().GetHiveChildCount(container); k++)
                    {
                        child = SampleHive::cHiveData::Get().GetHiveNthChild(container, k);

                        wxString child_text = serializer.DeserializeShowFileExtension() ?
                            SampleHive::cHiveData::Get().GetHiveItemText(child).BeforeLast('.') :
                            SampleHive::cHiveData::Get().GetHiveItemText(child);

                        if (child_text == files[i])
                        {
                            SampleHive::cHiveData::Get().HiveDeleteItem(child);
                            break;
                        }
                    }
                }
            }

            db.UpdateTrashColumn(files[i].ToStdString(), 1);
            db.UpdateHiveName(files[i].ToStdString(),
                              SampleHive::cHiveData::Get().GetHiveItemText(SampleHive::cHiveData::Get().GetFavoritesHive()).ToStdString());

            m_pTrash->AppendItem(m_TrashRoot, text_value);

            SampleHive::cHiveData::Get().ListCtrlDeleteItem(item_row);

            msg = wxString::Format(_("%s sent to trash"), text_value);
        }

        if (!msg.IsEmpty())
            SampleHive::cSignal::SendInfoBarMessage(msg, wxICON_ERROR, *this);
    }
}

void cTrashPanel::OnShowTrashContextMenu(wxTreeEvent& event)
{
    SampleHive::cSerializer serializer;
    cDatabase db;

    wxTreeItemId selected_trashed_item = event.GetItem();

    wxMenu menu;

    menu.Append(SampleHive::ID::MN_DeleteTrash, _("Delete from database"), _("Delete the selected sample(s) from database"));
    menu.Append(SampleHive::ID::MN_RestoreTrashedItem, _("Restore sample"), _("Restore the selected sample(s) back to library"));

    if (selected_trashed_item.IsOk())
    {
        switch (m_pTrash->GetPopupMenuSelectionFromUser(menu, event.GetPoint()))
        {
            case SampleHive::ID::MN_DeleteTrash:
            {
                wxString trashed_item_name = serializer.DeserializeShowFileExtension() ?
                    m_pTrash->GetItemText(selected_trashed_item).BeforeLast('.') :
                    m_pTrash->GetItemText(selected_trashed_item);

                db.RemoveSampleFromDatabase(trashed_item_name.ToStdString());

                m_pTrash->Delete(selected_trashed_item);

                SH_LOG_INFO("{} deleted from trash and databse", trashed_item_name);
            }
            break;
            case SampleHive::ID::MN_RestoreTrashedItem:
            {
                wxArrayTreeItemIds selected_item_ids;
                m_pTrash->GetSelections(selected_item_ids);

                wxFileDataObject file_data;
                wxArrayString files;

                wxString selected_item_text;
                std::string filename;

                for (size_t i = 0; i < selected_item_ids.GetCount(); i++)
                {
                    selected_item_text = m_pTrash->GetItemText(selected_item_ids[i]);

                    filename = SampleHive::cUtils::Get().GetFilenamePathAndExtension(selected_item_text).Filename;

                    file_data.AddFile(filename);

                    files = file_data.GetFilenames();

                    db.UpdateTrashColumn(files[i].ToStdString(), 0);

                    try
                    {
                        wxVector<wxVector<wxVariant>> dataset;

                        if (db.RestoreFromTrashByFilename(files[i].ToStdString(),
                                                          dataset,
                                                          serializer.DeserializeShowFileExtension(),
                                                          ICON_STAR_FILLED_16px, ICON_STAR_EMPTY_16px).empty())
                        {
                            SH_LOG_INFO("Error! Database is empty.");
                        }
                        else
                        {
                            for (auto data : dataset)
                            {
                                SampleHive::cHiveData::Get().ListCtrlAppendItem(data);
                            }
                        }
                    }
                    catch (std::exception& e)
                    {
                        SH_LOG_ERROR("Error loading data. {}", e.what());
                    }

                    m_pTrash->Delete(selected_item_ids[i]);

                    SH_LOG_INFO("{} restored from trash", files[i]);
                }
            }
            break;
            default:
                break;
        }
    }
}

void cTrashPanel::OnClickRestoreTrashItem(wxCommandEvent& event)
{
    SampleHive::cSerializer serializer;
    cDatabase db;

    wxArrayTreeItemIds selected_item_ids;
    m_pTrash->GetSelections(selected_item_ids);

    wxFileDataObject file_data;
    wxArrayString files;

    wxString selected_item_text;
    std::string filename;

    if (m_pTrash->GetChildrenCount(m_TrashRoot) == 0)
    {
        wxMessageBox(_("Trash is empty, nothing to restore!"), wxMessageBoxCaptionStr, wxOK | wxCENTRE, this);
        return;
    }

    if (selected_item_ids.IsEmpty())
    {
        wxMessageBox(_("No item selected, try selected a item first."), wxMessageBoxCaptionStr, wxOK | wxCENTRE, this);
        return;
    }

    for (size_t i = 0; i < selected_item_ids.GetCount(); i++)
    {
        selected_item_text = m_pTrash->GetItemText(selected_item_ids[i]);

        filename = SampleHive::cUtils::Get().GetFilenamePathAndExtension(selected_item_text).Filename;

        file_data.AddFile(filename);

        files = file_data.GetFilenames();

        db.UpdateTrashColumn(files[i].ToStdString(), 0);

        try
        {
            wxVector<wxVector<wxVariant>> dataset;

            if (db.RestoreFromTrashByFilename(files[i].ToStdString(), dataset,
                                              serializer.DeserializeShowFileExtension(),
                                              ICON_STAR_FILLED_16px, ICON_STAR_EMPTY_16px).empty())
            {
                SH_LOG_INFO("Error! Database is empty.");
            }
            else
            {
                for (auto data : dataset)
                {
                    SampleHive::cHiveData::Get().ListCtrlAppendItem(data);
                }
            }
        }
        catch (std::exception& e)
        {
            SH_LOG_ERROR("Error loading data. {}", e.what());
        }

        m_pTrash->Delete(selected_item_ids[i]);
    }
}

cTrashPanel::~cTrashPanel()
{

}