bool CZipper::read_as_utf8 (const QString &archname, const QString &fname)
{
QuaZip zip (archname, settings->value ("zip_charset_in", "UTF-8").toString().trimmed());
if (! zip.open (QuaZip::mdUnzip))
return false;
zip.setCurrentFile (fname);
if (! zip.hasCurrentFile())
return false;
QuaZipFileInfo info;
if (! zip.getCurrentFileInfo (&info))
return false;
QuaZipFile file (&zip);
if (! file.open (QIODevice::ReadOnly))
return false;
QByteArray ba = file.readAll();
string_data = QString::fromUtf8 (ba.data());
file.close();
zip.close();
return true;
}
Book Parse (const QString& filename)
{
QuaZip *zip = new QuaZip (filename);
std::shared_ptr<void> scopeGuard (nullptr,
[zip] (void*) { zip->close (); delete zip; });
Book book;
book.OriginalPath_ = filename;
if (!zip->open (QuaZip::mdUnzip))
{
qWarning () << Q_FUNC_INFO
<< "unable to open file "
<< filename
<< "as zip archive";
return book;
}
for (const auto& opfPath : GetOPFPaths (zip))
{
zip->setCurrentFile (opfPath);
QuaZipFile opfFile (zip);
if (!opfFile.open (QIODevice::ReadOnly))
{
qWarning () << Q_FUNC_INFO
<< "unable to open opf file";
return book;
}
QDomDocument document;
QString errorMsg;
int errorLine = -1, errorColumn = -1;
if (!document.setContent (opfFile.readAll (),
&errorMsg, &errorLine, &errorColumn))
{
qWarning () << Q_FUNC_INFO
<< errorMsg
<< "in line:"
<< errorLine
<< "column:"
<< errorColumn;
return book;
}
FillEPubBookInfo (document, book, zip);
book.AddedData_ = QDateTime::currentDateTime ();
book.Rate_ = NoStar;
book.Size_ = QFileInfo (filename).size ();
book.IsValid_ = true;
if (book.TitleInfo_.Title_.isEmpty())
book.TitleInfo_.Title_ = QObject::tr ("Unknown");
Author author;
author.Name_ = QObject::tr ("Unknown");
if (book.TitleInfo_.Authors_.isEmpty ())
book.TitleInfo_.Authors_ << author;
if (book.TitleInfo_.Authors_.at (0).Name_.isEmpty ())
book.TitleInfo_.Authors_ [0] = author;
}
return book;
}
BookParser_f MakeBookParser (const QString& filename)
{
if (filename.endsWith (".fb2"))
return FB2::Parse;
else if (filename.endsWith (".epub"))
return EPUB::Parse;
else if (filename.endsWith (".mobi") ||
filename.endsWith (".prc"))
return Mobi::Parse;
else if (filename.endsWith (".zip"))
{
QuaZip *zip = new QuaZip (filename);
std::shared_ptr<void> scopeGuard (nullptr,
[zip] (void*) { zip->close (); delete zip; });
if (!zip->open (QuaZip::mdUnzip))
{
qWarning () << Q_FUNC_INFO
<< "unable to open file "
<< filename
<< "as zip archive";
return BookParser_f ();
}
for (const auto& info : zip->getFileInfoList ())
{
if (info.name.endsWith (".fb2"))
{
QString fileNameInArch = info.name;
zip->setCurrentFile (fileNameInArch);
QuaZipFile file (zip);
if (!file.open (QIODevice::ReadOnly))
continue;
const quint32 size = info.uncompressedSize;
auto ba = file.readAll ();
return [ba, size, filename, fileNameInArch] (const QString&) -> Book
{
auto book = FB2::ParseFB2Content (ba);
book.AddedData_ = QDateTime::currentDateTime ();
book.Rate_ = NoStar;
book.OriginalPath_ = filename;
book.Size_ = size;
book.Content_ = ba;
if (book.TitleInfo_.Title_.isEmpty())
book.TitleInfo_.Title_ = QObject::tr ("Unknown");
Author author;
author.Name_ = QObject::tr ("Unknown");
if (book.TitleInfo_.Authors_.isEmpty ())
book.TitleInfo_.Authors_ << author;
if (book.TitleInfo_.Authors_.at (0).Name_.isEmpty ())
book.TitleInfo_.Authors_ [0] = author;
return book;
};
}
}
}
return BookParser_f ();
}
//! @brief extract currently opened archive
//! @brief dest path to extract archive to, can be filename when extracting a
//! single file.
//! @brief file file to extract from archive, full archive if empty.
//! @return true on success, false otherwise
bool ZipUtil::extractArchive(QString& dest, QString file)
{
qDebug() << "[ZipUtil] extractArchive" << dest << file;
bool result = true;
if(!m_zip) {
return false;
}
QuaZipFile *currentFile = new QuaZipFile(m_zip);
int entries = m_zip->getEntriesCount();
int current = 0;
// construct the filename when extracting a single file from an archive.
// if the given destination is a full path use it as output name,
// otherwise use it as path to place the file as named in the archive.
QString singleoutfile;
if(!file.isEmpty() && QFileInfo(dest).isDir()) {
singleoutfile = dest + "/" + file;
}
else if(!file.isEmpty()){
singleoutfile = dest;
}
for(bool more = m_zip->goToFirstFile(); more; more = m_zip->goToNextFile())
{
++current;
// if the entry is a path ignore it. Path existence is ensured separately.
if(m_zip->getCurrentFileName().split("/").last() == "")
continue;
// some tools set the MS-DOS file attributes. Check those for D flag,
// since in some cases a folder entry does not end with a /
QuaZipFileInfo fi;
currentFile->getFileInfo(&fi);
if(fi.externalAttr & 0x10) // FAT entry bit 4 indicating directory
continue;
QString outfilename;
if(!singleoutfile.isEmpty()
&& QFileInfo(m_zip->getCurrentFileName()).fileName() == file) {
outfilename = singleoutfile;
}
else if(singleoutfile.isEmpty()) {
outfilename = dest + "/" + m_zip->getCurrentFileName();
}
if(outfilename.isEmpty())
continue;
QFile outputFile(outfilename);
// make sure the output path exists
if(!QDir().mkpath(QFileInfo(outfilename).absolutePath())) {
result = false;
emit logItem(tr("Creating output path failed"), LOGERROR);
qDebug() << "[ZipUtil] creating output path failed for:"
<< outfilename;
break;
}
if(!outputFile.open(QFile::WriteOnly)) {
result = false;
emit logItem(tr("Creating output file failed"), LOGERROR);
qDebug() << "[ZipUtil] creating output file failed:"
<< outfilename;
break;
}
currentFile->open(QIODevice::ReadOnly);
outputFile.write(currentFile->readAll());
if(currentFile->getZipError() != UNZ_OK) {
result = false;
emit logItem(tr("Error during Zip operation"), LOGERROR);
qDebug() << "[ZipUtil] QuaZip error:" << currentFile->getZipError()
<< "on file" << currentFile->getFileName();
break;
}
currentFile->close();
outputFile.close();
emit logProgress(current, entries);
}
delete currentFile;
emit logProgress(1, 1);
return result;
}
