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 namespace
  35 {
  36     SettingValue<TorrentFileGuard::AutoDeleteMode> autoDeleteModeSetting()
  37     {
  38         return SettingValue<TorrentFileGuard::AutoDeleteMode> {u"Core/AutoDeleteAddedTorrentFile"_s};
  39     }
  40 }
  41
  42 FileGuard::FileGuard(const Path &path)
  43     : m_path {path}
  44     , m_remove {true}
  45 {
  46 }
  47
  48 void FileGuard::setAutoRemove(const bool remove) noexcept
  49 {
  50     m_remove = remove;
  51 }
  52
  53 FileGuard::~FileGuard()
  54 {
  55     if (m_remove && !m_path.isEmpty())
  56         Utils::Fs::removeFile(m_path); // removeFile() checks for file existence
  57 }
  58
  59 TorrentFileGuard::TorrentFileGuard(const Path &path, const TorrentFileGuard::AutoDeleteMode mode)
  60     : FileGuard {mode != Never ? path : Path()}
  61     , m_mode {mode}
  62 {
  63 }
  64
  65 TorrentFileGuard::TorrentFileGuard(const Path &path)
  66     : TorrentFileGuard {path, autoDeleteMode()}
  67 {
  68 }
  69
  70 TorrentFileGuard::~TorrentFileGuard()
  71 {
  72     if (!m_wasAdded && (m_mode != Always))
  73         setAutoRemove(false);
  74 }
  75
  76 void TorrentFileGuard::markAsAddedToSession()
  77 {
  78     m_wasAdded = true;
  79 }
  80
  81 TorrentFileGuard::AutoDeleteMode TorrentFileGuard::autoDeleteMode()
  82 {
  83     return autoDeleteModeSetting().get(AutoDeleteMode::Never);
  84 }
  85
  86 void TorrentFileGuard::setAutoDeleteMode(const TorrentFileGuard::AutoDeleteMode mode)
  87 {
  88     autoDeleteModeSetting() = mode;
  89 }