File indexing completed on 2023-10-01 08:44:29

 /***************************************************************************
     The network search widget dock widget 
                              -------------------
     begin                : Fri Mai 04 2018
     copyright            : (C) 2018-2019 by Alexander Reinholdt
     email                : alexander.reinholdt@kdemail.net
  ***************************************************************************/
 
 /***************************************************************************
  *   This program 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 2 of the License, or     *
  *   (at your option) any later version.                                   *
  *                                                                         *
  *   This program 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, write to the                         *
  *   Free Software Foundation, Inc., 51 Franklin Street, Suite 500, Boston,*
  *   MA 02110-1335, USA                                                    *
  ***************************************************************************/
 
 #ifdef HAVE_CONFIG_H
 #include <config.h>
 #endif
 
 // application specific includes
 #include "smb4ksharesviewdockwidget.h"
 #include "smb4ksharesviewitem.h"
 #include "core/smb4ksettings.h"
 #include "core/smb4kmounter.h"
 #include "core/smb4ksynchronizer.h"
 #include "core/smb4kbookmarkhandler.h"
 #include "core/smb4kshare.h"
 #include "core/smb4khardwareinterface.h"
 
 #if defined(Q_OS_LINUX)
 #include "smb4kmountsettings_linux.h"
 #elif defined(Q_OS_FREEBSD) || defined(Q_OS_NETBSD)
 #include "smb4kmountsettings_bsd.h"
 #endif
 
 // Qt includes
 #include <QActionGroup>
 #include <QMenu>
 #include <QApplication>
 #include <QDropEvent>
 
 // KDE includes
 #include <KI18n/KLocalizedString>
 #include <KIconThemes/KIconLoader>
 #include <KWidgetsAddons/KMessageBox>
 #include <KIOWidgets/KIO/DropJob>
 #include <KCoreAddons/KJobUiDelegate>
 
 
 Smb4KSharesViewDockWidget::Smb4KSharesViewDockWidget(const QString& title, QWidget* parent)
 : QDockWidget(title, parent)
 {
   //
   // Set the shares view
   // 
   m_sharesView = new Smb4KSharesView(this);
   setWidget(m_sharesView);
   
   //
   // The action collection
   // 
   m_actionCollection = new KActionCollection(this);
   
   //
   // The context menu
   // 
   m_contextMenu = new KActionMenu(this);
   
   //
   // Set up the actions
   //
   setupActions();
 
   //
   // Load the settings
   // 
   loadSettings();
   
   //
   // Connections
   //
   connect(m_sharesView, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(slotContextMenuRequested(QPoint)));
   connect(m_sharesView, SIGNAL(itemActivated(QListWidgetItem*)), this, SLOT(slotItemActivated(QListWidgetItem*)));
   connect(m_sharesView, SIGNAL(itemSelectionChanged()), this, SLOT(slotItemSelectionChanged()));
   connect(m_sharesView, SIGNAL(acceptedDropEvent(Smb4KSharesViewItem*,QDropEvent*)), this, SLOT(slotDropEvent(Smb4KSharesViewItem*,QDropEvent*)));
   
   connect(Smb4KMounter::self(), SIGNAL(mounted(SharePtr)), this, SLOT(slotShareMounted(SharePtr)));
   connect(Smb4KMounter::self(), SIGNAL(unmounted(SharePtr)), this, SLOT(slotShareUnmounted(SharePtr)));
   connect(Smb4KMounter::self(), SIGNAL(updated(SharePtr)), this, SLOT(slotShareUpdated(SharePtr)));
   
   connect(KIconLoader::global(), SIGNAL(iconChanged(int)), this, SLOT(slotIconSizeChanged(int)));
 }
 
 
 Smb4KSharesViewDockWidget::~Smb4KSharesViewDockWidget()
 {
 }
 
 
 void Smb4KSharesViewDockWidget::loadSettings()
 {
   //
   // Adjust the view according to the setting chosen
   //
   switch (Smb4KSettings::sharesViewMode())
   {
     case Smb4KSettings::EnumSharesViewMode::IconView:
     {
       int iconSize = KIconLoader::global()->currentSize(KIconLoader::Desktop);
       m_sharesView->setViewMode(Smb4KSharesView::IconMode, iconSize);
       break;
     }
     case Smb4KSettings::EnumSharesViewMode::ListView:
     {
       int iconSize = KIconLoader::global()->currentSize(KIconLoader::Small);
       m_sharesView->setViewMode(Smb4KSharesView::ListMode, iconSize);
       break;
     }
     default:
     {
       break;
     }
   }
   
   //
   // Adjust the unmount actions if needed
   // 
   if (!m_sharesView->selectedItems().isEmpty())
   {
     QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
     
     if (selectedItems.size() == 1)
     {
       Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItems.first());
       m_actionCollection->action("unmount_action")->setEnabled((!item->shareItem()->isForeign() || Smb4KMountSettings::unmountForeignShares()));
     }
     else if (selectedItems.size() > 1)
     {
       int foreign = 0;
       
       for (QListWidgetItem *selectedItem : selectedItems)
       {
         Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
         
         if (item && item->shareItem()->isForeign())
         {
           foreign++;
         }
       }
       
       m_actionCollection->action("unmount_action")->setEnabled(((selectedItems.size() > foreign) || Smb4KMountSettings::unmountForeignShares()));
     }
   }
 
   actionCollection()->action("unmount_all_action")->setEnabled(((!onlyForeignMountedShares() || Smb4KMountSettings::unmountForeignShares()) && m_sharesView->count() != 0));
 }
 
 
 void Smb4KSharesViewDockWidget::saveSettings()
 {
   //
   // Not used at the moment
   // 
 }
 
 
 KActionCollection *Smb4KSharesViewDockWidget::actionCollection()
 {
   return m_actionCollection;
 }
 
 
 void Smb4KSharesViewDockWidget::setupActions()
 {
   //
   // The 'View Modes' submenu and the respective actions
   //
   KActionMenu *viewModesMenu = new KActionMenu(KDE::icon("view-choose"), i18n("View Modes"), this);
   
   QActionGroup *viewModesGroup = new QActionGroup(this);
   viewModesGroup->setExclusive(true);
   connect(viewModesGroup, SIGNAL(triggered(QAction*)), this, SLOT(slotViewModeChanged(QAction*)));
   
   QAction *iconViewAction = new QAction(KDE::icon("view-list-icons"), i18n("Icon View"), this);
   iconViewAction->setObjectName("icon_view_action");
   iconViewAction->setCheckable(true);
   viewModesGroup->addAction(iconViewAction);
   viewModesMenu->addAction(iconViewAction);
   
   QAction *listViewAction = new QAction(KDE::icon("view-list-details"), i18n("List View"), this);
   listViewAction->setObjectName("list_view_action");
   listViewAction->setCheckable(true);
   viewModesGroup->addAction(listViewAction);
   viewModesMenu->addAction(listViewAction);
   
   switch (Smb4KSettings::sharesViewMode())
   {
     case Smb4KSettings::EnumSharesViewMode::IconView:
     {
       iconViewAction->setChecked(true);
       break;
     }
     case Smb4KSettings::EnumSharesViewMode::ListView:
     {
       listViewAction->setChecked(true);
       break;
     }
     default:
     {
       break;
     }
   }
   
   //
   // The Unmount action
   // 
   QAction *unmountAction = new QAction(KDE::icon("media-eject"), i18n("&Unmount"), this);
   connect(unmountAction, SIGNAL(triggered(bool)), this, SLOT(slotUnmountActionTriggered(bool)));
   
   //
   // The Unmount All action
   // 
   QAction *unmountAllAction = new QAction(KDE::icon("system-run"), i18n("U&nmount All"), this);
   connect(unmountAllAction, SIGNAL(triggered(bool)), this, SLOT(slotUnmountAllActionTriggered(bool)));
   
   //
   // The Add Bookmark action
   // 
   QAction *bookmarkAction = new QAction(KDE::icon("bookmark-new"), i18n("Add &Bookmark"), this);
   connect(bookmarkAction, SIGNAL(triggered(bool)), this, SLOT(slotBookmarkActionTriggered(bool)));
 
   //
   // The Synchronize action
   // 
   QAction *synchronizeAction = new QAction(KDE::icon("folder-sync"), i18n("S&ynchronize"), this);
   connect(synchronizeAction, SIGNAL(triggered(bool)), this, SLOT(slotSynchronizeActionTriggered(bool)));
 
   // 
   // The Open with Konsole action
   // 
   QAction *konsoleAction = new QAction(KDE::icon("utilities-terminal"), i18n("Open with Konso&le"), this);
   connect(konsoleAction, SIGNAL(triggered(bool)), this, SLOT(slotKonsoleActionTriggered(bool)));
 
   QAction *filemanagerAction = new QAction(KDE::icon("system-file-manager"), i18n("Open with F&ile Manager"), this);
   connect(filemanagerAction, SIGNAL(triggered(bool)), this, SLOT(slotFileManagerActionTriggered(bool)));
   
   //
   // Three separators
   // 
   QAction *separator1 = new QAction(this);
   separator1->setSeparator(true);
   
   QAction *separator2 = new QAction(this);
   separator2->setSeparator(true);
   
   QAction *separator3 = new QAction(this);
   separator3->setSeparator(true);
   
   
   //
   // Add the actions
   // 
   m_actionCollection->addAction("shares_view_modes", viewModesMenu);
   m_actionCollection->addAction("shares_separator1", separator1);
   m_actionCollection->addAction("unmount_action", unmountAction);
   m_actionCollection->addAction("unmount_all_action", unmountAllAction);
   m_actionCollection->addAction("shares_separator2", separator2);
   m_actionCollection->addAction("bookmark_action", bookmarkAction);
   m_actionCollection->addAction("synchronize_action", synchronizeAction);
   m_actionCollection->addAction("shares_separator3", separator3);
   m_actionCollection->addAction("konsole_action", konsoleAction);
   m_actionCollection->addAction("filemanager_action", filemanagerAction);
 
   //   
   // Set the shortcuts
   //
   m_actionCollection->setDefaultShortcut(unmountAction, QKeySequence(Qt::CTRL+Qt::Key_U));
   m_actionCollection->setDefaultShortcut(unmountAllAction, QKeySequence(Qt::CTRL+Qt::Key_N));
   m_actionCollection->setDefaultShortcut(bookmarkAction, QKeySequence(Qt::CTRL+Qt::Key_B));
   m_actionCollection->setDefaultShortcut(synchronizeAction, QKeySequence(Qt::CTRL+Qt::Key_Y));
   m_actionCollection->setDefaultShortcut(konsoleAction, QKeySequence(Qt::CTRL+Qt::Key_L));
   m_actionCollection->setDefaultShortcut(filemanagerAction, QKeySequence(Qt::CTRL+Qt::Key_I));
   
   // 
   // Disable all actions
   // 
   unmountAction->setEnabled(false);
   unmountAllAction->setEnabled(false);
   bookmarkAction->setEnabled(false);
   synchronizeAction->setEnabled(false);
   konsoleAction->setEnabled(false);
   filemanagerAction->setEnabled(false);
 
   // 
   // Plug the actions into the context menu
   //
   for (QAction *a : m_actionCollection->actions())
   {
     m_contextMenu->addAction(a);
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotContextMenuRequested(const QPoint& pos)
 {
   m_contextMenu->menu()->popup(m_sharesView->viewport()->mapToGlobal(pos));
 }
 
 
 void Smb4KSharesViewDockWidget::slotItemActivated(QListWidgetItem* /*item*/)
 {
   // 
   // Do not execute the item when keyboard modifiers were pressed
   // or the mouse button is not the left one.
   // 
   if (QApplication::keyboardModifiers() == Qt::NoModifier)
   {
     slotFileManagerActionTriggered(false);
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotItemSelectionChanged()
 {
   QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
 
   if (selectedItems.size() == 1)
   {
     Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItems.first());
     bool syncRunning = Smb4KSynchronizer::self()->isRunning(item->shareItem());
 
     m_actionCollection->action("unmount_action")->setEnabled((!item->shareItem()->isForeign() || Smb4KMountSettings::unmountForeignShares()));
     m_actionCollection->action("bookmark_action")->setEnabled(true);
 
     if (!item->shareItem()->isInaccessible())
     {
       m_actionCollection->action("synchronize_action")->setEnabled(!QStandardPaths::findExecutable("rsync").isEmpty() && !syncRunning);
       m_actionCollection->action("konsole_action")->setEnabled(!QStandardPaths::findExecutable("konsole").isEmpty());
       m_actionCollection->action("filemanager_action")->setEnabled(true);
     }
     else
     {
       m_actionCollection->action("synchronize_action")->setEnabled(false);
       m_actionCollection->action("konsole_action")->setEnabled(false);
       m_actionCollection->action("filemanager_action")->setEnabled(false);
     }
   }
   else if (selectedItems.size() > 1)
   {
     int syncsRunning = 0;
     int inaccessible = 0;
     int foreign = 0;
     
     for (QListWidgetItem *selectedItem : selectedItems)
     {
       Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
       
       if (item)
       {
         // Is the share synchronized at the moment?
         if (Smb4KSynchronizer::self()->isRunning(item->shareItem()))
         {
           syncsRunning += 1;
         }
         
         // Is the share inaccessible at the moment?
         if (item->shareItem()->isInaccessible())
         {
           inaccessible += 1;
         }
         
         // Was the share being mounted by another user?
         if (item->shareItem()->isForeign())
         {
           foreign += 1;
         }
       }
     }
     
     m_actionCollection->action("unmount_action")->setEnabled(((selectedItems.size() > foreign) || Smb4KMountSettings::unmountForeignShares()));
     m_actionCollection->action("bookmark_action")->setEnabled(true);
 
     if (selectedItems.size() > inaccessible)
     {
       m_actionCollection->action("synchronize_action")->setEnabled(!QStandardPaths::findExecutable("rsync").isEmpty() && (selectedItems.size() > syncsRunning));
       m_actionCollection->action("konsole_action")->setEnabled(!QStandardPaths::findExecutable("konsole").isEmpty());
       m_actionCollection->action("filemanager_action")->setEnabled(true);
     }
     else
     {
       m_actionCollection->action("synchronize_action")->setEnabled(false);
       m_actionCollection->action("konsole_action")->setEnabled(false);
       m_actionCollection->action("filemanager_action")->setEnabled(false);
     }
   }
   else
   {
     m_actionCollection->action("unmount_action")->setEnabled(false);
     m_actionCollection->action("bookmark_action")->setEnabled(false);
     m_actionCollection->action("synchronize_action")->setEnabled(false);
     m_actionCollection->action("konsole_action")->setEnabled(false);
     m_actionCollection->action("filemanager_action")->setEnabled(false);
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotDropEvent(Smb4KSharesViewItem* item, QDropEvent* e)
 {
   if (item && e)
   {
     if (e->mimeData()->hasUrls())
     {
       if (Smb4KHardwareInterface::self()->isOnline())
       {
         QUrl dest = QUrl::fromLocalFile(item->shareItem()->path());
         KIO::DropJob *job = drop(e, dest, KIO::DefaultFlags);
         job->uiDelegate()->setAutoErrorHandlingEnabled(true);
         job->uiDelegate()->setAutoWarningHandlingEnabled(true);
       }
       else
       {
         KMessageBox::sorry(m_sharesView, i18n("<qt>There is no active connection to the share <b>%1</b>! You cannot drop any files here.</qt>", item->shareItem()->displayString()));
       }
     }
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotViewModeChanged(QAction* action)
 {
   //
   // Set the new view mode
   //
   if (action->objectName() == "icon_view_action")
   {
     Smb4KSettings::setSharesViewMode(Smb4KSettings::EnumSharesViewMode::IconView);
   }
   else if (action->objectName() ==  "list_view_action")
   {
     Smb4KSettings::setSharesViewMode(Smb4KSettings::EnumSharesViewMode::ListView);
   }
   
   //
   // Save settings
   //
   Smb4KSettings::self()->save();
   
   //
   // Load the settings
   //
   loadSettings();
 }
 
 
 void Smb4KSharesViewDockWidget::slotShareMounted(const SharePtr& share)
 {
   //
   // Add the share to the shares view
   // 
   if (share)
   {
     // Add the item
     (void) new Smb4KSharesViewItem(m_sharesView, share);
     
     // Sort the view
     m_sharesView->sortItems(Qt::AscendingOrder);
     
     // Enable/disable the 'Unmount All' action
     actionCollection()->action("unmount_all_action")->setEnabled(
       ((!onlyForeignMountedShares() || Smb4KMountSettings::unmountForeignShares()) && m_sharesView->count() != 0));
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotShareUnmounted(const SharePtr& share)
 {
   //
   // Remove the share from the shares view
   // 
   if (share)
   {
     // Get the item and delete it. Take care of the current item, if necessary.
     for (int i = 0; i < m_sharesView->count(); ++i)
     {
       Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(m_sharesView->item(i));
           
       if (item && (item->shareItem()->path() == share->path() || item->shareItem()->canonicalPath() == share->canonicalPath()))
       {
         if (item == m_sharesView->currentItem())
         {
           m_sharesView->setCurrentItem(0);
         }
             
         delete m_sharesView->takeItem(i);
         break;
       }
       else
       {
         continue;
       }
     }
      
     // Enable/disable the 'Unmount All' action
     actionCollection()->action("unmount_all_action")->setEnabled(
       ((!onlyForeignMountedShares() || Smb4KMountSettings::unmountForeignShares()) && m_sharesView->count() != 0));
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotShareUpdated(const SharePtr& share)
 {
   //
   // Get the share item and updated it
   // 
   if (share)
   {
     for (int i = 0; i < m_sharesView->count(); ++i)
     {
       Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(m_sharesView->item(i));
           
       if (item && (item->shareItem()->path() == share->path() || item->shareItem()->canonicalPath() == share->canonicalPath()))
       {
         item->update();
         break;
       }
       else
       {
         continue;
       }
     }
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotUnmountActionTriggered(bool /*checked*/)
 {
   QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
   QList<SharePtr> shares;
   
   for (QListWidgetItem *selectedItem : selectedItems)
   {
     Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
 
     if (item)
     {
       shares << item->shareItem();
     }
   }
 
   Smb4KMounter::self()->unmountShares(shares, false);
 }
 
 
 void Smb4KSharesViewDockWidget::slotUnmountAllActionTriggered(bool /*checked*/)
 {
   Smb4KMounter::self()->unmountAllShares(false);
 }
 
 
 void Smb4KSharesViewDockWidget::slotBookmarkActionTriggered(bool /*checked*/)
 {
   QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
   QList<SharePtr> shares;
 
   for (QListWidgetItem *selectedItem : selectedItems)
   {
     Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
     shares << item->shareItem();
   }
 
   Smb4KBookmarkHandler::self()->addBookmarks(shares);
 }
 
 
 void Smb4KSharesViewDockWidget::slotSynchronizeActionTriggered(bool /*checked*/)
 {
   QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
   
   for (QListWidgetItem *selectedItem : selectedItems)
   {
     Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
 
     if (item && !item->shareItem()->isInaccessible() && !Smb4KSynchronizer::self()->isRunning(item->shareItem()))
     {
       Smb4KSynchronizer::self()->synchronize(item->shareItem());
     }
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotKonsoleActionTriggered(bool /*checked*/)
 {
   QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
 
   for (QListWidgetItem *selectedItem : selectedItems)
   {
     Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
 
     if (item && !item->shareItem()->isInaccessible())
     {
       openShare(item->shareItem(), Konsole);
     }
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotFileManagerActionTriggered(bool /*checked*/)
 {
   QList<QListWidgetItem *> selectedItems = m_sharesView->selectedItems();
 
   for (QListWidgetItem *selectedItem : selectedItems)
   {
     Smb4KSharesViewItem *item = static_cast<Smb4KSharesViewItem *>(selectedItem);
 
     if (item && !item->shareItem()->isInaccessible())
     {
       openShare(item->shareItem(), FileManager);
     }
   }
 }
 
 
 void Smb4KSharesViewDockWidget::slotIconSizeChanged(int group)
 {
   //
   // Change the icon size depending of the view mode
   // 
   if (group == KIconLoader::Desktop && Smb4KSettings::sharesViewMode() == Smb4KSettings::EnumSharesViewMode::IconView)
   {
     int iconSize = KIconLoader::global()->currentSize(KIconLoader::Desktop);
     m_sharesView->setIconSize(QSize(iconSize, iconSize));
   }
   else if (group == KIconLoader::Small && Smb4KSettings::sharesViewMode() == Smb4KSettings::EnumSharesViewMode::ListView)
   {
     int iconSize = KIconLoader::global()->currentSize(KIconLoader::Small);
     m_sharesView->setIconSize(QSize(iconSize, iconSize));
   }
 }