ryzom/tools/client/ryzom_installer/src/filescopier.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 #include "stdpch.h"
  18 #include "filescopier.h"
  19 #include "utils.h"
  20 #include "operation.h"
  21
  22 #include "nel/misc/path.h"
  23
  24 #ifdef DEBUG_NEW
  25         #define new DEBUG_NEW
  26 #endif
  27
  28 CFilesCopier::CFilesCopier(IOperationProgressListener *listener):m_listener(listener)
  29 {
  30 }
  31
  32 CFilesCopier::~CFilesCopier()
  33 {
  34 }
  35
  36 void CFilesCopier::setSourceDirectory(const QString &src)
  37 {
  38         m_sourceDirectory = src;
  39 }
  40
  41 void CFilesCopier::setDestinationDirectory(const QString &dst)
  42 {
  43         m_destinationDirectory = dst;
  44 }
  45
  46 void CFilesCopier::setIncludeFilter(const QStringList &filter)
  47 {
  48         m_includeFilter = filter;
  49 }
  50
  51 void CFilesCopier::addFile(const QString &filename)
  52 {
  53         m_files << filename;
  54 }
  55
  56 bool CFilesCopier::exec()
  57 {
  58         if (m_sourceDirectory.isEmpty() || m_destinationDirectory.isEmpty()) return false;
  59
  60         if (m_listener) m_listener->operationPrepare();
  61
  62         QDir().mkpath(m_destinationDirectory);
  63
  64         FilesToCopy files;
  65
  66         // create the list of files to copy
  67         CFilesCopier::getFilesList(files);
  68
  69         // copy them
  70         return copyFiles(files);
  71 }
  72
  73 void CFilesCopier::getFile(const QFileInfo &fileInfo, const QDir &srcDir, FilesToCopy &files) const
  74 {
  75         // full path to file
  76         QString fullPath = fileInfo.absoluteFilePath();
  77
  78         QString relativePath = srcDir.relativeFilePath(fullPath);
  79
  80         QFileInfo relativeFileInfo(relativePath);
  81
  82         // correct absolute path
  83         if (relativeFileInfo.isAbsolute())
  84         {
  85                 relativePath = relativeFileInfo.fileName();
  86         }
  87
  88         // full path where to copy file
  89         QString dstPath = m_destinationDirectory + "/" + relativePath;
  90
  91         if (fileInfo.isDir())
  92         {
  93                 // create directory
  94                 QDir().mkpath(dstPath);
  95
  96                 QDir subDir(fullPath);
  97
  98                 // get list of all files in directory
  99                 QFileInfoList entries = subDir.entryInfoList(QDir::AllEntries | QDir::NoDotAndDotDot);
 100
 101                 // proces seach file recursively
 102                 foreach(const QFileInfo &entry, entries)
 103                 {
 104                         getFile(entry, srcDir, files);
 105                 }
 106         }
 107         else
 108         {
 109                 // add the file to list with all useful information
 110                 FileToCopy file;
 111                 file.filename = fileInfo.fileName();
 112                 file.src = fileInfo.filePath();
 113                 file.dst = dstPath;
 114                 file.size = fileInfo.size();
 115                 file.date = fileInfo.lastModified().toTime_t();
 116                 file.permissions = fileInfo.permissions();
 117
 118                 files << file;
 119         }
 120 }
 121
 122 void CFilesCopier::getFilesList(FilesToCopy &files) const
 123 {
 124         QDir srcDir(m_sourceDirectory);
 125
 126         // only copy all files from filter
 127         QFileInfoList entries = srcDir.entryInfoList(m_includeFilter);
 128
 129         foreach(const QFileInfo &entry, entries)
 130         {
 131                 getFile(entry, srcDir, files);
 132         }
 133
 134         // copy additional files
 135         foreach(const QString &fullpath, m_files)
 136         {
 137                 QFileInfo fileInfo(fullpath);
 138
 139                 getFile(fileInfo, srcDir, files);
 140         }
 141 }
 142
 143 bool CFilesCopier::copyFiles(const FilesToCopy &files)
 144 {
 145         qint64 totalSize = 0;
 146
 147         foreach(const FileToCopy &file, files)
 148         {
 149                 totalSize += file.size;
 150         }
 151
 152         if (m_listener)
 153         {
 154                 m_listener->operationInit(0, totalSize);
 155                 m_listener->operationStart();
 156         }
 157
 158         qint64 processedSize = 0;
 159
 160         foreach(const FileToCopy &file, files)
 161         {
 162                 if (m_listener && m_listener->operationShouldStop())
 163                 {
 164                         m_listener->operationStop();
 165                         return true;
 166                 }
 167
 168                 if (m_listener) m_listener->operationProgress(processedSize, file.filename);
 169
 170                 QFileInfo dstFileInfo(file.dst);
 171
 172                 if (dstFileInfo.size() != file.size || dstFileInfo.lastModified().toTime_t() != file.date)
 173                 {
 174                         // force deleting previous file since it was incomplete
 175                         QFile::remove(file.dst);
 176
 177                         if (!QFile::copy(file.src, file.dst))
 178                         {
 179                                 if (m_listener) m_listener->operationFail(QApplication::tr("Unable to copy file %1 to %2").arg(file.src).arg(file.dst));
 180                                 return false;
 181                         }
 182
 183                         if (!QFile::setPermissions(file.dst, file.permissions))
 184                         {
 185                                 nlwarning("Unable to change permissions of %s", Q2C(file.dst));
 186                         }
 187
 188                         if (!NLMISC::CFile::setFileModificationDate(qToUtf8(file.dst), file.date))
 189                         {
 190                                 nlwarning("Unable to change date of %s", Q2C(file.dst));
 191                         }
 192                 }
 193
 194                 processedSize += file.size;
 195         }
 196
 197         // wait 1 second to be sure all files have been copied (because to disk cache)
 198         QThread::sleep(1);
 199
 200         if (m_listener)
 201         {
 202                 m_listener->operationSuccess(totalSize);
 203         }
 204
 205         return true;
 206 }