src/app/filelogger.cpp

   1 /*
   2  * Bittorrent Client using Qt and libtorrent.
   3  * Copyright (C) 2016  sledgehammer999 <hammered999@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 "filelogger.h"
  30
  31 #include <chrono>
  32
  33 #include <QDateTime>
  34 #include <QDir>
  35 #include <QTextStream>
  36 #include <QVector>
  37
  38 #include "base/global.h"
  39 #include "base/logger.h"
  40 #include "base/utils/fs.h"
  41
  42 namespace
  43 {
  44     const std::chrono::seconds FLUSH_INTERVAL {2};
  45 }
  46
  47 FileLogger::FileLogger(const Path &path, const bool backup
  48                        , const int maxSize, const bool deleteOld, const int age
  49                        , const FileLogAgeType ageType)
  50     : m_backup(backup)
  51     , m_maxSize(maxSize)
  52 {
  53     m_flusher.setInterval(FLUSH_INTERVAL);
  54     m_flusher.setSingleShot(true);
  55     connect(&m_flusher, &QTimer::timeout, this, &FileLogger::flushLog);
  56
  57     changePath(path);
  58     if (deleteOld)
  59         this->deleteOld(age, ageType);
  60
  61     const Logger *const logger = Logger::instance();
  62     for (const Log::Msg &msg : asConst(logger->getMessages()))
  63         addLogMessage(msg);
  64
  65     connect(logger, &Logger::newLogMessage, this, &FileLogger::addLogMessage);
  66 }
  67
  68 FileLogger::~FileLogger()
  69 {
  70     closeLogFile();
  71 }
  72
  73 void FileLogger::changePath(const Path &newPath)
  74 {
  75     // compare paths as strings to perform case sensitive comparison on all the platforms
  76     if (newPath.data() == m_path.parentPath().data())
  77         return;
  78
  79     closeLogFile();
  80
  81     m_path = newPath / Path(u"qbittorrent.log"_qs);
  82     m_logFile.setFileName(m_path.data());
  83
  84     Utils::Fs::mkpath(newPath);
  85     openLogFile();
  86 }
  87
  88 void FileLogger::deleteOld(const int age, const FileLogAgeType ageType)
  89 {
  90     const QDateTime date = QDateTime::currentDateTime();
  91     const QDir dir {m_path.parentPath().data()};
  92     const QFileInfoList fileList = dir.entryInfoList(QStringList(u"qbittorrent.log.bak*"_qs)
  93         , (QDir::Files | QDir::Writable), (QDir::Time | QDir::Reversed));
  94
  95     for (const QFileInfo &file : fileList)
  96     {
  97         QDateTime modificationDate = file.lastModified();
  98         switch (ageType)
  99         {
 100         case DAYS:
 101             modificationDate = modificationDate.addDays(age);
 102             break;
 103         case MONTHS:
 104             modificationDate = modificationDate.addMonths(age);
 105             break;
 106         default:
 107             modificationDate = modificationDate.addYears(age);
 108         }
 109         if (modificationDate > date)
 110             break;
 111         Utils::Fs::removeFile(Path(file.absoluteFilePath()));
 112     }
 113 }
 114
 115 void FileLogger::setBackup(const bool value)
 116 {
 117     m_backup = value;
 118 }
 119
 120 void FileLogger::setMaxSize(const int value)
 121 {
 122     m_maxSize = value;
 123 }
 124
 125 void FileLogger::addLogMessage(const Log::Msg &msg)
 126 {
 127     if (!m_logFile.isOpen()) return;
 128
 129     QTextStream stream(&m_logFile);
 130 #if (QT_VERSION < QT_VERSION_CHECK(6, 0, 0))
 131     stream.setCodec("UTF-8");
 132 #endif
 133
 134     switch (msg.type)
 135     {
 136     case Log::INFO:
 137         stream << "(I) ";
 138         break;
 139     case Log::WARNING:
 140         stream << "(W) ";
 141         break;
 142     case Log::CRITICAL:
 143         stream << "(C) ";
 144         break;
 145     default:
 146         stream << "(N) ";
 147     }
 148
 149     stream << QDateTime::fromMSecsSinceEpoch(msg.timestamp).toString(Qt::ISODate) << " - " << msg.message << '\n';
 150
 151     if (m_backup && (m_logFile.size() >= m_maxSize))
 152     {
 153         closeLogFile();
 154         int counter = 0;
 155         Path backupLogFilename = m_path + ".bak";
 156
 157         while (backupLogFilename.exists())
 158         {
 159             ++counter;
 160             backupLogFilename = m_path + ".bak" + QString::number(counter);
 161         }
 162
 163         Utils::Fs::renameFile(m_path, backupLogFilename);
 164         openLogFile();
 165     }
 166     else
 167     {
 168         if (!m_flusher.isActive())
 169             m_flusher.start();
 170     }
 171 }
 172
 173 void FileLogger::flushLog()
 174 {
 175     if (m_logFile.isOpen())
 176         m_logFile.flush();
 177 }
 178
 179 void FileLogger::openLogFile()
 180 {
 181     if (!m_logFile.open(QIODevice::WriteOnly | QIODevice::Append | QIODevice::Text)
 182         || !m_logFile.setPermissions(QFile::ReadOwner | QFile::WriteOwner))
 183     {
 184         m_logFile.close();
 185         LogMsg(tr("An error occurred while trying to open the log file. Logging to file is disabled."), Log::CRITICAL);
 186     }
 187 }
 188
 189 void FileLogger::closeLogFile()
 190 {
 191     m_flusher.stop();
 192     m_logFile.close();
 193 }