diff options
Diffstat (limited to 'src/exporters/baseexporter.cpp')
| -rw-r--r-- | src/exporters/baseexporter.cpp | 172 |
1 files changed, 172 insertions, 0 deletions
diff --git a/src/exporters/baseexporter.cpp b/src/exporters/baseexporter.cpp new file mode 100644 index 0000000..60dc5a1 --- /dev/null +++ b/src/exporters/baseexporter.cpp @@ -0,0 +1,172 @@ +/*************************************************************************** +* Copyright (C) 2003-2005 by * +* Cyril Bosselut ([email protected]) * +* Jason Kivlighn ([email protected]) * +* * +* 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) any later version. * +***************************************************************************/ + +#include "baseexporter.h" + +#include <tqfile.h> +#include <tqfileinfo.h> + +#include <tdeaboutdata.h> +#include <kdebug.h> +#include <tdelocale.h> +#include <tdeglobal.h> +#include <tdemessagebox.h> +#include <ktar.h> +#include <kstandarddirs.h> + +#include "backends/recipedb.h" + +BaseExporter::BaseExporter( const TQString& _filename, const TQString &format ) : + file( 0 ), + tar_file( 0 ), + filename( _filename ), + m_progress_dlg( 0 ), + compress(false) +{ + //automatically append extension + TQString extension = format.right( format.length()-2 ); + if ( filename.right( extension.length() ) != extension ) + filename += "." + extension; +} + +BaseExporter::~BaseExporter() +{ + delete file; + delete tar_file; +} + +int BaseExporter::headerFlags() const +{ + return RecipeDB::None; +} + +void BaseExporter::setCompressed( bool b ) +{ + compress = b; +} + +void BaseExporter::exporter( const TQValueList<int> &ids, RecipeDB *database, KProgressDialog *progress_dlg ) +{ + m_progress_dlg = progress_dlg; + + if ( createFile() ) + saveToFile( ids, database ); + else + kdDebug() << "no output file has been selected for export." << endl; +} + +void BaseExporter::exporter( int id, RecipeDB *database, KProgressDialog *progress_dlg ) +{ + TQValueList<int> single_recipe_list; + single_recipe_list << id ; + exporter( single_recipe_list, database, progress_dlg ); +} + +void BaseExporter::writeStream( TQTextStream &stream, const RecipeList &recipe_list ) +{ + stream << createHeader(recipe_list); + stream << createContent(recipe_list); + stream << createFooter(); +} + +bool BaseExporter::createFile() +{ + if ( file ) + return true; + + if ( !filename.isEmpty() ) { + if ( compress ) { + tar_file = new KTar( filename, "application/x-gzip" ); + TQFileInfo fi( filename ); + file = new TQFile( locateLocal( "tmp",fi.fileName()+"ml" ) ); + } + else + file = new TQFile(filename); + + return (file != 0); + } + else + return false; +} + +TQString BaseExporter::fileName() const +{ + return filename; +} + +void BaseExporter::saveToFile( const TQValueList<int> &ids, RecipeDB *database ) +{ + if ( file->open( IO_WriteOnly ) ) { + if ( m_progress_dlg ) + m_progress_dlg->progressBar()->setTotalSteps( ids.count()/progressInterval() ); + + TQValueList<int> ids_copy = ids; + TQTextStream stream( file ); + stream.setEncoding( TQTextStream::UnicodeUTF8 ); + + RecipeList recipe_list; + if ( headerFlags() != RecipeDB::None ) { + database->loadRecipes( &recipe_list, headerFlags(), ids ); + } + stream << createHeader( recipe_list ); + + recipe_list.clear(); + for ( uint i = 0; i < ids.count(); i += progressInterval() ) { + TQValueList<int> sub_list; + for ( int sub_i = 0; sub_i < progressInterval(); ++sub_i ) { + if ( ids_copy.count() == 0 ) break; + + sub_list << *ids_copy.begin(); + ids_copy.remove( ids_copy.begin() ); + } + + RecipeList recipe_list; + database->loadRecipes( &recipe_list, supportedItems(), sub_list ); + + TQString content = createContent( recipe_list ); + if ( !content.isEmpty() ) + stream << content; + + if ( m_progress_dlg && m_progress_dlg->wasCancelled() ) + break; + + if ( m_progress_dlg ) { + m_progress_dlg->progressBar()->advance( progressInterval() ); + kapp->processEvents(); + } + } + + stream << createFooter(); + + if ( tar_file && tar_file->open( IO_WriteOnly ) ) { + //close, which flushes the buffer, and then open for reading + file->close(); + file->open( IO_ReadOnly ); + + TQFileInfo fi( file->name() ); + TQByteArray data = file->readAll(); + tar_file->writeFile( fi.fileName(), fi.owner(), fi.group(), data.size(), data ); + tar_file->close(); + } + + file->close(); + } +} + +TQString BaseExporter::krecipes_version() const +{ + TDEInstance * this_instance = TDEGlobal::instance(); + if ( this_instance && this_instance->aboutData() ) + return this_instance->aboutData() ->version(); + + return TQString::null; //Oh, well. We couldn't get the version. +} + |