File indexing completed on 2024-05-12 16:46:08

 /***************************************************************************
     Copyright (C) 2009 Robby Stephenson <robby@periapsis.org>
  ***************************************************************************/
 
 /***************************************************************************
  *                                                                         *
  *   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) version 3 or any later version        *
  *   accepted by the membership of KDE e.V. (or its successor approved     *
  *   by the membership of KDE e.V.), which shall act as a proxy            *
  *   defined in Section 14 of version 3 of the license.                    *
  *                                                                         *
  *   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, see <http://www.gnu.org/licenses/>. *
  *                                                                         *
  ***************************************************************************/
 
 #include "imagedirectory.h"
 #include "image.h"
 #include "../core/filehandler.h"
 #include "../tellico_debug.h"
 
 #include <KZip>
 
 #include <QFile>
 #include <QDir>
 #include <QUrl>
 #include <QTemporaryDir>
 
 using namespace Tellico;
 using Tellico::ImageStorage;
 using Tellico::ImageDirectory;
 using Tellico::TemporaryImageDirectory;
 using Tellico::ImageZipArchive;
 
 ImageDirectory::ImageDirectory() : ImageStorage(), m_pathExists(false), m_dir(nullptr) {
 }
 
 ImageDirectory::ImageDirectory(const QString& path_) : ImageStorage() , m_dir(nullptr) {
   setPath(path_);
 }
 
 ImageDirectory::~ImageDirectory() {
   delete m_dir;
   m_dir = nullptr;
 }
 
 QString ImageDirectory::path() {
   return m_path;
 }
 
 void ImageDirectory::setPath(const QString& path_) {
   m_path = path_;
   QDir dir(m_path);
   m_pathExists = dir.exists();
 }
 
 bool ImageDirectory::hasImage(const QString& id_) {
   return m_pathExists && QFile::exists(path() + id_);
 }
 
 Tellico::Data::Image* ImageDirectory::imageById(const QString& id_) {
   if(!hasImage(id_)) {
     return nullptr;
   }
 
   Data::Image* img = new Data::Image(path() + id_, id_);
   if(!img) {
     myLog() << "image not found:" << (path() + id_);
     return nullptr;
   }
   if(img->isNull()) {
     myLog() << "image found but null:" << (path() + id_);
     delete img;
     return nullptr;
   }
   return img;
 }
 
 bool ImageDirectory::writeImage(const Data::Image& img_) {
   const QString path = this->path(); // virtual function, so don't assume m_path is correct
   if(!m_pathExists) {
     if(path.isEmpty()) {
       // an empty path means the file hasn't been saved yet
       if(!m_dir) {
         m_dir = new QTemporaryDir(); // default is to auto-delete, aka autoRemove()
         // in KDE4, the way this worked included the final slash.
         ImageDirectory::setPath(m_dir->path() + QLatin1Char('/'));
       }
       return writeImage(img_);
     }
     QDir dir(path);
     if(dir.mkdir(path)) {
 //      myLog() << "created" << path;
     } else {
       myWarning() << "unable to create dir:" << path;
     }
     m_pathExists = true;
   }
   QUrl target = QUrl::fromLocalFile(path);
   target.setPath(target.path() + img_.id());
   return FileHandler::writeDataURL(target, img_.byteArray(), true /* force */);
 }
 
 bool ImageDirectory::removeImage(const QString& id_) {
   return m_pathExists && QFile::remove(path() + id_);
 }
 
 TemporaryImageDirectory::TemporaryImageDirectory() : ImageDirectory(), m_dir(nullptr) {
 }
 
 TemporaryImageDirectory::~TemporaryImageDirectory() {
   purge();
 }
 
 void TemporaryImageDirectory::purge() {
   delete m_dir;
   m_dir = nullptr;
 }
 
 QString TemporaryImageDirectory::path() {
   if(!m_dir) {
     m_dir = new QTemporaryDir(); // default is to auto-delete, aka autoRemove()
     // in KDE4, the way this worked included the final slash.
     ImageDirectory::setPath(m_dir->path() + QLatin1Char('/'));
   }
   return ImageDirectory::path();
 }
 
 void TemporaryImageDirectory::setPath(const QString& path) {
   Q_UNUSED(path);
   Q_ASSERT(path.isEmpty()); // should never be called, that's why it's private
 }
 
 ImageZipArchive::ImageZipArchive() : ImageStorage(), m_imgDir(nullptr) {
 }
 
 ImageZipArchive::~ImageZipArchive() {
 }
 
 void ImageZipArchive::setZip(std::unique_ptr<KZip> zip_) {
   m_images.clear();
   m_zip = std::move(zip_);
   m_imgDir = nullptr;
 
   const KArchiveDirectory* dir = m_zip->directory();
   if(!dir) {
     m_zip.reset();
     return;
   }
   const KArchiveEntry* imgDirEntry = dir->entry(QStringLiteral("images"));
   if(!imgDirEntry || !imgDirEntry->isDirectory()) {
     m_zip.reset();
     return;
   }
   m_imgDir = static_cast<const KArchiveDirectory*>(imgDirEntry);
   m_images.add(m_imgDir->entries());
 }
 
 bool ImageZipArchive::hasImage(const QString& id_) {
   return m_images.has(id_);
 }
 
 Tellico::Data::Image* ImageZipArchive::imageById(const QString& id_) {
   if(!hasImage(id_)) {
     return nullptr;
   }
   Data::Image* img = nullptr;
   const KArchiveEntry* file = m_imgDir->entry(id_);
   if(file && file->isFile()) {
     img = new Data::Image(static_cast<const KArchiveFile*>(file)->data(),
                           id_.section(QLatin1Char('.'), -1).toUpper(), id_);
   }
   // might be unexpected behavior, but in order to delete the zip object after
   // all images are read, we need to consider the image gone now
   m_images.remove(id_);
   if(m_images.isEmpty()) {
     m_zip.reset();
     m_imgDir = nullptr;
   }
   if(!img) {
     myLog() << "image not found:" << id_;
     return nullptr;
   }
   if(img->isNull()) {
     myLog() << "image found but null:" << id_;
     delete img;
     return nullptr;
   }
   return img;
 }