#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());

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

							ts << " [DONE] ";

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

							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;

}