You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
134 lines
4.7 KiB
134 lines
4.7 KiB
/***************************************************************************
|
|
copyright : (C) 2003-2006 by Robby Stephenson
|
|
email : robby@periapsis.org
|
|
***************************************************************************/
|
|
|
|
/***************************************************************************
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of version 2 of the GNU General Public License as *
|
|
* published by the Free Software Foundation; *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
#include "tellicozipexporter.h"
|
|
#include "tellicoxmlexporter.h"
|
|
#include "../collection.h"
|
|
#include "../imagefactory.h"
|
|
#include "../image.h"
|
|
#include "../filehandler.h"
|
|
#include "../stringset.h"
|
|
#include "../tellico_debug.h"
|
|
#include "../progressmanager.h"
|
|
|
|
#include <tdelocale.h>
|
|
#include <tdeconfig.h>
|
|
#include <kzip.h>
|
|
#include <tdeapplication.h>
|
|
|
|
#include <tqdom.h>
|
|
#include <tqbuffer.h>
|
|
|
|
using Tellico::Export::TellicoZipExporter;
|
|
|
|
TQString TellicoZipExporter::formatString() const {
|
|
return i18n("Tellico Zip File");
|
|
}
|
|
|
|
TQString TellicoZipExporter::fileFilter() const {
|
|
return i18n("*.tc *.bc|Tellico Files (*.tc)") + TQChar('\n') + i18n("*|All Files");
|
|
}
|
|
|
|
bool TellicoZipExporter::exec() {
|
|
m_cancelled = false;
|
|
Data::CollPtr coll = collection();
|
|
if(!coll) {
|
|
return false;
|
|
}
|
|
|
|
// TODO: maybe need label?
|
|
ProgressItem& item = ProgressManager::self()->newProgressItem(this, TQString(), true);
|
|
item.setTotalSteps(100);
|
|
connect(&item, TQT_SIGNAL(signalCancelled(ProgressItem*)), TQT_SLOT(slotCancel()));
|
|
ProgressItem::Done done(this);
|
|
|
|
TellicoXMLExporter exp;
|
|
exp.setEntries(entries());
|
|
exp.setURL(url()); // needed in case of relative URL values
|
|
long opt = options();
|
|
opt |= Export::ExportUTF8; // always export to UTF-8
|
|
opt |= Export::ExportImages; // always list the images in the xml
|
|
opt &= ~Export::ExportProgress; // don't show progress for xml export
|
|
exp.setOptions(opt);
|
|
exp.setIncludeImages(false); // do not include the images themselves in XML
|
|
TQCString xml = exp.exportXML().toCString(); // encoded in utf-8
|
|
ProgressManager::self()->setProgress(this, 5);
|
|
|
|
TQByteArray data;
|
|
TQBuffer buf(data);
|
|
|
|
if(m_cancelled) {
|
|
return true; // intentionally cancelled
|
|
}
|
|
|
|
KZip zip(TQT_TQIODEVICE(&buf));
|
|
zip.open(IO_WriteOnly);
|
|
zip.writeFile(TQString::fromLatin1("tellico.xml"), TQString(), TQString(), xml.length(), xml);
|
|
|
|
if(m_includeImages) {
|
|
ProgressManager::self()->setProgress(this, 10);
|
|
// gonna be lazy and just increment progress every 3 images
|
|
// it might be less, might be more
|
|
uint j = 0;
|
|
const TQString imagesDir = TQString::fromLatin1("images/");
|
|
StringSet imageSet;
|
|
Data::FieldVec imageFields = coll->imageFields();
|
|
// already took 10%, only 90% left
|
|
const uint stepSize = TQMAX(1, (coll->entryCount() * imageFields.count()) / 90);
|
|
for(Data::EntryVec::ConstIterator it = entries().begin(); it != entries().end() && !m_cancelled; ++it) {
|
|
for(Data::FieldVec::Iterator fIt = imageFields.begin(); fIt != imageFields.end(); ++fIt, ++j) {
|
|
const TQString id = it->field(fIt);
|
|
if(id.isEmpty() || imageSet.has(id)) {
|
|
continue;
|
|
}
|
|
const Data::ImageInfo& info = ImageFactory::imageInfo(id);
|
|
if(info.linkOnly) {
|
|
myLog() << "TellicoZipExporter::exec() - not copying linked image: " << id << endl;
|
|
continue;
|
|
}
|
|
const Data::Image& img = ImageFactory::imageById(id);
|
|
// if no image, continue
|
|
if(img.isNull()) {
|
|
kdWarning() << "TellicoZipExporter::exec() - no image found for " << fIt->title() << " field" << endl;
|
|
kdWarning() << "...for the entry titled " << it->title() << endl;
|
|
continue;
|
|
}
|
|
TQByteArray ba = img.byteArray();
|
|
// myDebug() << "TellicoZipExporter::data() - adding image id = " << it->field(fIt) << endl;
|
|
zip.writeFile(imagesDir + id, TQString(), TQString(), ba.size(), ba);
|
|
imageSet.add(id);
|
|
if(j%stepSize == 0) {
|
|
ProgressManager::self()->setProgress(this, TQMIN(10+j/stepSize, 99));
|
|
kapp->processEvents();
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
ProgressManager::self()->setProgress(this, 80);
|
|
}
|
|
|
|
zip.close();
|
|
if(m_cancelled) {
|
|
return true;
|
|
}
|
|
|
|
bool success = FileHandler::writeDataURL(url(), data, options() & Export::ExportForce);
|
|
return success;
|
|
}
|
|
|
|
void TellicoZipExporter::slotCancel() {
|
|
m_cancelled = true;
|
|
}
|
|
|
|
#include "tellicozipexporter.moc"
|