#include "exports.h"

 #include "branchitem.h"

 #include "file.h"

 #include "linkablemapobj.h"

 #include "misc.h"

 #include "mainwindow.h"

 #include "warningdialog.h"

 #include "xsltproc.h"

 extern Main *mainWindow;

 extern QDir vymBaseDir;

 extern QString vymName;

 ExportBase::ExportBase()

 {

 	indentPerDepth="  ";

 	bool ok;

     tmpDir.setPath (makeTmpDir(ok,"vym-export"));

 	if (!tmpDir.exists() || !ok)

 		QMessageBox::critical( 0, QObject::tr( "Error" ),

 					   QObject::tr("Couldn't access temporary directory\n"));

 	cancelFlag=false;				   

 }

 ExportBase::~ExportBase()

 {

 	// Cleanup tmpdir

 	removeDir (tmpDir);

 }

 void ExportBase::setDir(const QDir &d)

 {

 	outDir=d;

 }

 void ExportBase::setFile (const QString &p)

 {

 	outputFile=p;

 }

 QString ExportBase::getFile ()

 {

 	return outputFile;

 }

 void ExportBase::setModel(VymModel *m)

 {

 	model=m;

 }

 void ExportBase::setCaption (const QString &s)

 {

 	caption=s;

 }

 void ExportBase::addFilter(const QString &s)

 {

 	filter=s;

 }

 bool ExportBase::execDialog()

 {

 	{

 		QFileDialog *fd=new QFileDialog( 0, caption);

 		fd->setFilter (filter);

 		fd->setCaption(caption);

 		fd->setMode( QFileDialog::AnyFile );

 		fd->setDir (outDir);

 		fd->show();

 		if ( fd->exec() == QDialog::Accepted )

 		{

 			if (QFile (fd->selectedFile()).exists() )

 			{

 				QMessageBox mb( vymName,

 					QObject::tr("The file %1 exists already.\nDo you want to overwrite it?").arg(fd->selectedFile()), 

 				QMessageBox::Warning,

 				QMessageBox::Yes | QMessageBox::Default,

 				QMessageBox::Cancel | QMessageBox::Escape,

 				Qt::NoButton );

 				mb.setButtonText( QMessageBox::Yes, QObject::tr("Overwrite") );

 				mb.setButtonText( QMessageBox::No, QObject::tr("Cancel"));

 				ExportBase ex;

 				switch( mb.exec() ) 

 				{

 					case QMessageBox::Yes:

 						// save 

 						break;;

 					case QMessageBox::Cancel:

 						cancelFlag=true;

 						return false;

 						break;

 				}

 			}

 			outputFile=fd->selectedFile();

 			cancelFlag=false;

 			return true;

 		}

 	}

 	return false;

 }

 bool ExportBase::canceled()

 {

 	return cancelFlag;

 }

 QString ExportBase::getSectionString(TreeItem *start)

 {

 	// Make prefix like "2.5.3" for "bo:2,bo:5,bo:3"

 	QString r;

 	TreeItem *ti=start;

 	int depth=ti->depth();

 	while (depth>0)

 	{

 		r=QString("%1").arg(1+ti->num(),0,10)+"." + r;

 		ti=ti->parent(); 

 		depth=ti->depth();

 	}	

 	if (r.isEmpty())

 		return r;

 	else	

 		return r + " ";

 }

 ////////////////////////////////////////////////////////////////////////

 ExportAO::ExportAO()

 {

 	filter="TXT (*.txt)";

 	caption=vymName+ " -" +QObject::tr("Export as ASCII")+" "+QObject::tr("(still experimental)");

 }

 void ExportAO::doExport()	

 {

 	QFile file (outputFile);

 	if ( !file.open( QIODevice::WriteOnly ) )

 	{

 		qWarning ("ExportAO::doExport couldn't open "+outputFile);

 		return;

 	}

 	QTextStream ts( &file );	// use LANG decoding here...

 	// Main loop over all branches

 	QString s;

 	QString curIndent;

 	int i;

 	BranchItem *cur=NULL;

 	BranchItem *prev=NULL;

 	QString colString;

 	QColor col;

 	cur=model->nextBranch (cur,prev);

 	while (cur) 

 	{

 		if (cur->getType()==TreeItem::Branch || cur->getType()==TreeItem::MapCenter)

 		{

 			// Make indentstring

 			curIndent="";

 			for (i=0;i<cur->depth()-1;i++) curIndent+= indentPerDepth;

 			if (!cur->hasHiddenExportParent() )

 			{

 				col=cur->getHeadingColor();

 				if (col==QColor (255,0,0))

 					colString="[R]";

 				else if (col==QColor (217,81,0))

 					colString="[O]";

 				else if (col==QColor (0,85,0))

 					colString="[G]";

 				else  	

 					colString="[?]";

 				switch (cur->depth())

 				{

 					case 0:

 						//ts << underline (cur->getHeading(),QString("="));

 						//ts << "\n";

 						break;

 					case 1:

 						//ts << "\n";

 						//ts << (underline ( cur->getHeading(), QString("-") ) );

 						//ts << "\n";

 						break;

 					case 2: // Main heading

 						ts << "\n";

 						ts << underline ( cur->getHeading(), QString("=") );

 						ts << "\n\n";

 						break;

 					case 3: // Achievement, Bonus, Objective ...

 						ts << underline ( cur->getHeading(), "-");

 						ts << "\n\n";

 						break;

 					case 4:	// That's the item we need to know

 						//ts << (curIndent + "* " + colString+" "+ cur->getHeading());

 						ts << colString+" "+ cur->getHeading();

 						if (cur->isActiveStandardFlag ("hook-green"))

 							ts << " [DONE] ";

 						else	if (cur->isActiveStandardFlag ("wip"))

 							ts << " [WIP] ";

 						else	if (cur->isActiveStandardFlag ("cross-red"))

 							ts << " [NOT STARTED] ";

 						ts << "\n";

 					default:

 						break;

 						ts << (curIndent + "- " + cur->getHeading());

 						ts << "\n";

 						break;

 				}

 				// If necessary, write note

 				if (!cur->getNoteObj().isEmpty())

 				{

 					curIndent +="  | ";

 					s=cur->getNoteASCII( curIndent, 80);

 					ts << s;

 				}

 			}

 		}

 		cur=model->nextBranch(cur,prev);

 	}

 	file.close();

 }

 QString ExportAO::underline (const QString &text, const QString &line)

 {

 	QString r=text + "\n";

 	for (int j=0;j<text.length();j++) r+=line;

 	return r;

 }

 ////////////////////////////////////////////////////////////////////////

 ExportASCII::ExportASCII()

 {

 	filter="TXT (*.txt)";

 	caption=vymName+ " -" +QObject::tr("Export as ASCII")+" "+QObject::tr("(still experimental)");

 }

 void ExportASCII::doExport()	

 {

 	QFile file (outputFile);

 	if ( !file.open( QIODevice::WriteOnly ) )

 	{

 		qWarning ("ExportASCII::doExport couldn't open "+outputFile);

 		return;

 	}

 	QTextStream ts( &file );	// use LANG decoding here...

 	// Main loop over all branches

 	QString s;

 	QString curIndent;

 	int i;

 	BranchItem *cur=NULL;

 	BranchItem *prev=NULL;

 	cur=model->nextBranch (cur,prev);

 	while (cur) 

 	{

 		if (cur->getType()==TreeItem::Branch || cur->getType()==TreeItem::MapCenter)

 		{

 			// Make indentstring

 			curIndent="";

 			for (i=0;i<cur->depth()-1;i++) curIndent+= indentPerDepth;

 			if (!cur->hasHiddenExportParent() )

 			{

 				//std::cout << "ExportASCII::  "<<curIndent.toStdString()<<cur->getHeading().toStdString()<<std::endl;

 				switch (cur->depth())

 				{

 					case 0:

 						ts << underline (cur->getHeading(),QString("="));

 						ts << "\n";

 						break;

 					case 1:

 						ts << "\n";

 						ts << (underline (getSectionString(cur) + cur->getHeading(), QString("-") ) );

 						ts << "\n";

 						break;

 					case 2:

 						ts << "\n";

 						ts << (curIndent + "* " + cur->getHeading());

 						ts << "\n";

 						break;

 					case 3:

 						ts << (curIndent + "- " + cur->getHeading());

 						ts << "\n";

 						break;

 					default:

 						ts << (curIndent + "- " + cur->getHeading());

 						ts << "\n";

 						break;

 				}

 				// If necessary, write note

 				if (!cur->getNoteObj().isEmpty())

 				{

 					curIndent +="  | ";

 					s=cur->getNoteASCII( curIndent, 80);

 					ts << s;

 				}

 			}

 		}

 		cur=model->nextBranch(cur,prev);

 	}

 	file.close();

 }

 QString ExportASCII::underline (const QString &text, const QString &line)

 {

 	QString r=text + "\n";

 	for (int j=0;j<text.length();j++) r+=line;

 	return r;

 }

 ////////////////////////////////////////////////////////////////////////

 void ExportCSV::doExport()

 {

 	QFile file (outputFile);

 	if ( !file.open( QIODevice::WriteOnly ) )

 	{

 		qWarning ("ExportBase::exportXML  couldn't open "+outputFile);

 		return;

 	}

 	QTextStream ts( &file );	// use LANG decoding here...

 	// Write header

 	ts << "\"Note\""  <<endl;

 	// Main loop over all branches

 	QString s;

 	QString curIndent("");

 	int i;

 	BranchObj *bo;  //FIXME-3 still needed?

 	BranchItem *cur=NULL;

 	BranchItem *prev=NULL;

 	cur=model->nextBranch (cur,prev);

 	while (cur) 

 	{

 		bo=(BranchObj*)(cur->getLMO());

 		if (!cur->hasHiddenExportParent() )

 		{

 			// If necessary, write note

 			if (!cur->getNoteObj().isEmpty())

 			{

 				s =cur->getNoteASCII();

 				s=s.replace ("\n","\n"+curIndent);

 				ts << ("\""+s+"\",");

 			} else

 				ts <<"\"\",";

 			// Make indentstring

 			for (i=0;i<cur->depth();i++) curIndent+= "\"\",";

 			// Write heading

 			ts << curIndent << "\"" << cur->getHeading()<<"\""<<endl;

 		}

 		cur=model->nextBranch(cur,prev);

 		curIndent="";

 	}

 	file.close();

 }

 ////////////////////////////////////////////////////////////////////////

 void ExportKDE3Bookmarks::doExport() 

 {

 	WarningDialog dia;

 	dia.showCancelButton (true);

 	dia.setText(QObject::tr("Exporting the %1 bookmarks will overwrite\nyour existing bookmarks file.").arg("KDE"));

 	dia.setCaption(QObject::tr("Warning: Overwriting %1 bookmarks").arg("KDE 3"));

 	dia.setShowAgainName("/exports/KDE/overwriteKDEBookmarks");

 	if (dia.exec()==QDialog::Accepted)

 	{

 		model->exportXML(tmpDir.path(),false);

 		XSLTProc p;

 		p.setInputFile (tmpDir.path()+"/"+model->getMapName()+".xml");

 		p.setOutputFile (tmpDir.home().path()+"/.kde/share/apps/konqueror/bookmarks.xml");

 		p.setXSLFile (vymBaseDir.path()+"/styles/vym2kdebookmarks.xsl");

 		p.process();

 		QString ub=vymBaseDir.path()+"/scripts/update-bookmarks";

 		QProcess *proc= new QProcess ;

 		proc->start( ub);

 		if (!proc->waitForStarted())

 		{

 			QMessageBox::warning(0, 

 				QObject::tr("Warning"),

 				QObject::tr("Couldn't find script %1\nto notifiy Browsers of changed bookmarks.").arg(ub));

 		}	

 	}

 }

 ////////////////////////////////////////////////////////////////////////

 void ExportKDE4Bookmarks::doExport() 

 {

 	WarningDialog dia;

 	dia.showCancelButton (true);

 	dia.setText(QObject::tr("Exporting the %1 bookmarks will overwrite\nyour existing bookmarks file.").arg("KDE 4"));

 	dia.setCaption(QObject::tr("Warning: Overwriting %1 bookmarks").arg("KDE"));

 	dia.setShowAgainName("/exports/KDE/overwriteKDEBookmarks");

 	if (dia.exec()==QDialog::Accepted)

 	{

 		model->exportXML(tmpDir.path(),false);

 		XSLTProc p;

 		p.setInputFile (tmpDir.path()+"/"+model->getMapName()+".xml");

 		p.setOutputFile (tmpDir.home().path()+"/.kde4/share/apps/konqueror/bookmarks.xml");

 		p.setXSLFile (vymBaseDir.path()+"/styles/vym2kdebookmarks.xsl");

 		p.process();

 		QString ub=vymBaseDir.path()+"/scripts/update-bookmarks";

 		QProcess *proc= new QProcess ;

 		proc->start( ub);

 		if (!proc->waitForStarted())

 		{

 			QMessageBox::warning(0, 

 				QObject::tr("Warning"),

 				QObject::tr("Couldn't find script %1\nto notifiy Browsers of changed bookmarks.").arg(ub));

 		}	

 	}

 }

 ////////////////////////////////////////////////////////////////////////

 void ExportFirefoxBookmarks::doExport() 

 {

 	WarningDialog dia;

 	dia.showCancelButton (true);

 	dia.setText(QObject::tr("Exporting the %1 bookmarks will overwrite\nyour existing bookmarks file.").arg("Firefox"));

 	dia.setCaption(QObject::tr("Warning: Overwriting %1 bookmarks").arg("Firefox"));

 	dia.setShowAgainName("/vym/warnings/overwriteImportBookmarks");

 	if (dia.exec()==QDialog::Accepted)

 	{

 		model->exportXML(tmpDir.path(),false);

 /*

 		XSLTProc p;

 		p.setInputFile (tmpDir.path()+"/"+me->getMapName()+".xml");

 		p.setOutputFile (tmpDir.home().path()+"/.kde/share/apps/konqueror/bookmarks.xml");

 		p.setXSLFile (vymBaseDir.path()+"/styles/vym2kdebookmarks.xsl");

 		p.process();

 		QString ub=vymBaseDir.path()+"/scripts/update-bookmarks";

 		QProcess *proc = new QProcess( );

 		proc->addArgument(ub);

 		if ( !proc->start() ) 

 		{

 			QMessageBox::warning(0, 

 				QObject::tr("Warning"),

 				QObject::tr("Couldn't find script %1\nto notifiy Browsers of changed bookmarks.").arg(ub));

 		}	

 */

 	}

 }

 ////////////////////////////////////////////////////////////////////////

 void ExportTaskjuggler::doExport() 

 {

 	model->exportXML(tmpDir.path(),false);

 	XSLTProc p;

 	p.setInputFile (tmpDir.path()+"/"+model->getMapName()+".xml");

 	p.setOutputFile (outputFile);

 	p.setXSLFile (vymBaseDir.path()+"/styles/vym2taskjuggler.xsl");

 	p.process();

 }

 ////////////////////////////////////////////////////////////////////////

 void ExportLaTeX::doExport() 

 {

 	// Exports a map to a LaTex file.  

 	// This file needs to be included 

 	// or inported into a LaTex document

 	// it will not add a preamble, or anything 

 	// that makes a full LaTex document.

   QFile file (outputFile);

   if ( !file.open( QIODevice::WriteOnly ) ) {

 	QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("Could not write %1").arg(outputFile));

 	mainWindow->statusMessage(QString(QObject::tr("Export failed.")));

     return;

   }

   QTextStream ts( &file );	// use LANG decoding here...

   ts.setEncoding (QTextStream::UnicodeUTF8); // Force UTF8

   // Main loop over all branches

   QString s;

   // QString curIndent("");

   // int i;

   BranchObj *bo;

   BranchItem *cur=NULL;

   BranchItem *prev=NULL;

   model->nextBranch(cur,prev);

   while (cur) 

   {

 	bo=(BranchObj*)(cur->getLMO());

 	if (!cur->hasHiddenExportParent() )

 	{

 		switch (cur->depth() ) 

 		{

 			case 0: break;

 			case 1: 

 			  ts << ("\\chapter{" + cur->getHeading()+ "}\n");

 			  break;

 			case 2: 

 			  ts << ("\\section{" + cur->getHeading()+ "}\n");

 			  break;

 			case 3: 

 			  ts << ("\\subsection{" + cur->getHeading()+ "}\n");

 			  break;

 			case 4: 

 			  ts << ("\\subsubsection{" + cur->getHeading()+ "}\n");

 			  break;

 			default:

 			  ts << ("\\paragraph*{" + cur->getHeading()+ "}\n");

 		}

 		// If necessary, write note

 		if (!cur->getNoteObj().isEmpty()) {

 		  ts << (cur->getNoteASCII());

 		  ts << ("\n");

 		}

 	}

     cur=model->nextBranch(cur,prev);

    }

   file.close();

 }

 ////////////////////////////////////////////////////////////////////////

 ExportOO::ExportOO()

 {

 	useSections=false;

 }

 ExportOO::~ExportOO()

 {

 }	

 QString ExportOO::buildList (TreeItem *current)

 {

     QString r;

     uint i=0;

 	BranchItem *bi=current->getFirstBranch();

 	if (bi)

     {

 		if (!bi->hasHiddenExportParent() )	// FIXME-3 use BranchItem...

 		{

 			// Start list

 			r+="<text:list text:style-name=\"vym-list\">\n";

 			while (bi)

 			{

 				r+="<text:list-item><text:p >";

 				r+=quotemeta(bi->getHeading());

 				// If necessary, write note

 				if (!bi->getNoteObj().isEmpty())

 					r+=bi->getNoteOpenDoc();

 				r+="</text:p>";

 				r+=buildList (bi);	// recursivly add deeper branches

 				r+="</text:list-item>\n";

 				i++;

 				bi=current->getBranchNum(i);

 			}

 			r+="</text:list>\n";

 		}

     }

     return r;

 }

 void ExportOO::exportPresentation()

 {

 	QString allPages;

 	BranchItem *firstMCO=(BranchItem*)(model->getRootItem()->getFirstBranch());

 	if (!firstMCO) 

 	{

 		QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("No objects in map!"));

 		return;

 	}

 	// Insert new content

 	// FIXME add extra title in mapinfo for vym 1.13.x

 	content.replace ("<!-- INSERT TITLE -->",quotemeta(firstMCO->getHeading()));

 	content.replace ("<!-- INSERT AUTHOR -->",quotemeta(model->getAuthor()));

 	QString	onePage;

 	QString list;

 	BranchItem *sectionBI;	

     int i=0;

 	BranchItem *pagesBI;

     int j=0;

 	int mapcenters=model->getRootItem()->branchCount();

 	// useSections already has been set in setConfigFile 

 	if (mapcenters>1)	

 		sectionBI=firstMCO;

 	else

 		sectionBI=firstMCO->getFirstBranch();

 	// Walk sections

 	while (sectionBI && !sectionBI->hasHiddenExportParent() )

 	{

 		if (useSections)

 		{

 			// Add page with section title

 			onePage=sectionTemplate;

 			onePage.replace ("<!-- INSERT PAGE HEADING -->", quotemeta(sectionBI->getHeading() ) );

 			allPages+=onePage;

 			pagesBI=sectionBI->getFirstBranch();

 		} else

 		{

 			//i=-2;	// only use inner loop to 

 			        // turn mainbranches into pages

 			//sectionBI=firstMCO;

 			pagesBI=sectionBI;

 		}

 		j=0;

 		while (pagesBI && !pagesBI->hasHiddenExportParent() )

 		{

 			// Add page with list of items

 			onePage=pageTemplate;

 			onePage.replace ("<!-- INSERT PAGE HEADING -->", quotemeta (pagesBI->getHeading() ) );

 			list=buildList (pagesBI);

 			onePage.replace ("<!-- INSERT LIST -->", list);

 			allPages+=onePage;

 			if (pagesBI!=sectionBI)

 			{

 				j++;

 				pagesBI=((BranchItem*)pagesBI->parent())->getBranchNum(j);

 			} else

 				pagesBI=NULL;	// We are already iterating over the sectionBIs

 		}

 		i++;

 		if (mapcenters>1 )

 			sectionBI=model->getRootItem()->getBranchNum (i);

 		else

 			sectionBI=firstMCO->getBranchNum (i);

 	}

 	content.replace ("<!-- INSERT PAGES -->",allPages);

 	// Write modified content

 	QFile f (contentFile);

     if ( !f.open( QIODevice::WriteOnly ) ) 

 	{

 		QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("Could not write %1").arg(contentFile));

 		mainWindow->statusMessage(QString(QObject::tr("Export failed.")));

 		return;

     }

     QTextStream t( &f );

     t << content;

     f.close();

 	// zip tmpdir to destination

 	zipDir (tmpDir,outputFile);	

 }

 bool ExportOO::setConfigFile (const QString &cf)

 {

 	configFile=cf;

 	int i=cf.findRev ("/");

 	if (i>=0) configDir=cf.left(i);

 	SimpleSettings set;

 	set.readSettings(configFile);

 	// set paths

 	templateDir=configDir+"/"+set.readEntry ("Template");

 	QDir d (templateDir);

 	if (!d.exists())

 	{

 		QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("Check \"%1\" in\n%2").arg("Template="+set.readEntry ("Template")).arg(configFile));

 		return false;

 	}

 	contentTemplateFile=templateDir+"content-template.xml";

 	contentFile=tmpDir.path()+"/content.xml";

 	pageTemplateFile=templateDir+"page-template.xml";

 	sectionTemplateFile=templateDir+"section-template.xml";

 	if (set.readEntry("useSections").contains("yes"))

 		useSections=true;

 	// Copy template to tmpdir

 	system ("cp -r "+templateDir+"* "+tmpDir.path());

 	// Read content-template

 	if (!loadStringFromDisk (contentTemplateFile,content))

 	{

 		QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("Could not read %1").arg(contentTemplateFile));

 		return false;

 	}

 	// Read page-template

 	if (!loadStringFromDisk (pageTemplateFile,pageTemplate))

 	{

 		QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("Could not read %1").arg(pageTemplateFile));

 		return false;

 	}

 	// Read section-template

 	if (useSections && !loadStringFromDisk (sectionTemplateFile,sectionTemplate))

 	{

 		QMessageBox::critical (0,QObject::tr("Critical Export Error"),QObject::tr("Could not read %1").arg(sectionTemplateFile));

 		return false;

 	}

 	return true;

 }