src/base/torrentfileguard.cpp

   1 /*
   2  * Bittorrent Client using Qt and libtorrent.
   3  * Copyright (C) 2016  Eugene Shalygin <eugene.shalygin@gmail.com>
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License
   7  * as published by the Free Software Foundation; either version 2
   8  * of the License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18  *
  19  * In addition, as a special exception, the copyright holders give permission to
  20  * link this program with the OpenSSL project's "OpenSSL" library (or with
  21  * modified versions of it that use the same license as the "OpenSSL" library),
  22  * and distribute the linked executables. You must obey the GNU General Public
  23  * License in all respects for all of the code used other than "OpenSSL".  If you
  24  * modify file(s), you may extend this exception to your version of the file(s),
  25  * but you are not obligated to do so. If you do not wish to do so, delete this
  26  * exception statement from your version.
  27  */
  28
  29 #include "torrentfileguard.h"
  30
  31 #include "settingvalue.h"
  32 #include "utils/fs.h"
  33
  34 FileGuard::FileGuard(const QString &path)
  35     : m_path {path}
  36     , m_remove {true}
  37 {
  38 }
  39
  40 void FileGuard::setAutoRemove(const bool remove) noexcept
  41 {
  42     m_remove = remove;
  43 }
  44
  45 FileGuard::~FileGuard()
  46 {
  47     if (m_remove && !m_path.isEmpty())
  48         Utils::Fs::forceRemove(m_path); // forceRemove() checks for file existence
  49 }
  50
  51 TorrentFileGuard::TorrentFileGuard(const QString &path, const TorrentFileGuard::AutoDeleteMode mode)
  52     : FileGuard {mode != Never ? path : QString()}
  53     , m_mode {mode}
  54     , m_wasAdded {false}
  55 {
  56 }
  57
  58 TorrentFileGuard::TorrentFileGuard(const QString &path)
  59     : TorrentFileGuard {path, autoDeleteMode()}
  60 {
  61 }
  62
  63 TorrentFileGuard::~TorrentFileGuard()
  64 {
  65     if (!m_wasAdded && (m_mode != Always))
  66         setAutoRemove(false);
  67 }
  68
  69 void TorrentFileGuard::markAsAddedToSession()
  70 {
  71     m_wasAdded = true;
  72 }
  73
  74 TorrentFileGuard::AutoDeleteMode TorrentFileGuard::autoDeleteMode()
  75 {
  76     return autoDeleteModeSetting().get(AutoDeleteMode::Never);
  77 }
  78
  79 void TorrentFileGuard::setAutoDeleteMode(TorrentFileGuard::AutoDeleteMode mode)
  80 {
  81     autoDeleteModeSetting() = mode;
  82 }
  83
  84 SettingValue<TorrentFileGuard::AutoDeleteMode> &TorrentFileGuard::autoDeleteModeSetting()
  85 {
  86     static SettingValue<AutoDeleteMode> setting {"Core/AutoDeleteAddedTorrentFile"};
  87     return setting;
  88 }