insilmaril@81: #include <QDir>
insilmaril@81: #include <QMessageBox>
insilmaril@81: #include <QPixmap>
insilmaril@81: #include <QLabel>
insilmaril@81: #include <QTextStream>
insilmaril@81: #include <iostream>
insilmaril@81: #include <cstdlib>
insilmaril@81: 
insilmaril@81: #include "file.h"
insilmaril@81: #include "process.h"
insilmaril@81: 
insilmaril@81: #if defined(Q_OS_WIN32)
insilmaril@81: #include "mkdtemp.h"
insilmaril@81: #include <windows.h>
insilmaril@81: #endif
insilmaril@81: 
insilmaril@81: QString maskPath(QString p)
insilmaril@81: {
insilmaril@81: 	// Change " " to "\ " to enable blanks in filenames
insilmaril@81: 	p=p.replace(QChar('&'),"\\&");
insilmaril@81: 	return p.replace(QChar(' '),"\\ ");
insilmaril@81: }
insilmaril@81: 
insilmaril@81: QString convertToRel (const QString &src, const QString &dst)
insilmaril@81: {
insilmaril@81: 	QString s=src;
insilmaril@81: 	QString d=dst;
insilmaril@81: 	int i;
insilmaril@81: 
insilmaril@81: 	if (s==d) 
insilmaril@81: 	{
insilmaril@81: 		// Special case, we just need the name of the file,
insilmaril@81: 		// not the complete path
insilmaril@81: 		i=d.findRev ("/");
insilmaril@81: 		d=d.right (d.length()-i-1);
insilmaril@81: 	} else
insilmaril@81: 	{
insilmaril@81: 		// Find relative path from src to dst
insilmaril@81: 
insilmaril@81: 		// Remove the first "/"
insilmaril@81: 		if (s.section ("/",0,0).isEmpty()) 
insilmaril@81: 		{
insilmaril@81: 			s=s.right (s.length()-1);
insilmaril@81: 			d=d.right (d.length()-1);
insilmaril@81: 		}
insilmaril@81: 		
insilmaril@81: 		// remove identical left parts
insilmaril@81: 		while (s.section("/",0,0) == d.section("/",0,0) ) 
insilmaril@81: 		{
insilmaril@81: 			i=s.find ("/");
insilmaril@81: 			s=s.right (s.length()-i-1);
insilmaril@81: 			d=d.right (d.length()-i-1);
insilmaril@81: 		}
insilmaril@81: 
insilmaril@81: 		// Now take care of paths where we have to go back first
insilmaril@81: 		int srcsep=s.count("/");
insilmaril@81: 		int dstsep=d.count("/");
insilmaril@81: 		if (srcsep <=  dstsep )
insilmaril@81: 		{
insilmaril@81: 			// find path to go up first and then back to dst
insilmaril@81: 			i=1;
insilmaril@81: 			while (i<=srcsep) 
insilmaril@81: 			{
insilmaril@81: 				d="../"+d;
insilmaril@81: 				i++;
insilmaril@81: 			}	
insilmaril@81: 		}
insilmaril@81: 	}	
insilmaril@81: 	return d;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: #include <QFileDialog>
insilmaril@81: extern QString vymName;
insilmaril@81: extern QDir lastFileDir;
insilmaril@81: 
insilmaril@81: QString browseDirectory (QWidget *parent,const QString &caption)
insilmaril@81: {
insilmaril@81: 	QFileDialog fd(parent,caption);
insilmaril@81: 	fd.setMode (QFileDialog::DirectoryOnly);
insilmaril@81: 	fd.setCaption(vymName+ " - "+caption);
insilmaril@81: 	fd.setDir (lastFileDir);
insilmaril@81: 	fd.show();
insilmaril@81: 	
insilmaril@81: 	if ( fd.exec() == QDialog::Accepted )
insilmaril@81: 		return fd.selectedFile();
insilmaril@81: 	else
insilmaril@81: 		return "";
insilmaril@81: }
insilmaril@81: 
insilmaril@81: 
insilmaril@81: 
insilmaril@81: bool reallyWriteDirectory(const QString &dir)
insilmaril@81: {
insilmaril@81: 	QStringList eList = QDir(dir).entryList();
insilmaril@81: 	if (eList.first() ==".")  eList.pop_front();	// remove "."
insilmaril@81: 	if (eList.first() =="..") eList.pop_front();	// remove "."
insilmaril@81: 	if (!eList.isEmpty())
insilmaril@81: 	{
insilmaril@81: 		QMessageBox mb( vymName,
insilmaril@81: 			QObject::tr("The directory %1 is not empty.\nDo you risk to overwrite its contents?","write directory").arg(dir),
insilmaril@81: 		QMessageBox::Warning,
insilmaril@81: 		QMessageBox::Yes ,
insilmaril@81: 		QMessageBox::Cancel | QMessageBox::Default,
insilmaril@81: 		QMessageBox::NoButton );
insilmaril@81: 
insilmaril@81: 		mb.setButtonText( QMessageBox::Yes, QObject::tr("Overwrite") );
insilmaril@81: 		mb.setButtonText( QMessageBox::No, QObject::tr("Cancel"));
insilmaril@81: 		switch( mb.exec() ) 
insilmaril@81: 		{
insilmaril@81: 			case QMessageBox::Yes:
insilmaril@81: 				// save 
insilmaril@81: 				return true;
insilmaril@81: 			case QMessageBox::Cancel:
insilmaril@81: 				// do nothing
insilmaril@81: 				return false;
insilmaril@81: 		}
insilmaril@81: 	}
insilmaril@81: 	return true;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: QString makeTmpDir (bool &ok, QString prefix)
insilmaril@81: {
insilmaril@81: 	bool b;
insilmaril@81: 	QString path=makeUniqueDir (b,QDir::tempPath()+"/"+prefix+"-XXXXXX");
insilmaril@81: 	ok=b;
insilmaril@81: 	return path;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: bool isInTmpDir(QString fn)
insilmaril@81: {
insilmaril@81: 	QString temp=QDir::tempPath();
insilmaril@81: 	int l=temp.length();
insilmaril@81: 	return fn.left(l)==temp;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: QString makeUniqueDir (bool &ok,QString s)
insilmaril@81: {
insilmaril@81: 	// Create unique directory e.g. for s="/tmp/vym-XXXXXX"
insilmaril@81: 
insilmaril@81: 	// Convert Separators
insilmaril@81: 	s=QDir::convertSeparators(s);
insilmaril@81: 
insilmaril@81: 	// Convert QString to string 
insilmaril@81: 	ok=true;
insilmaril@81: 	char *p;
insilmaril@81: 	int bytes=s.length();
insilmaril@81: 	p=(char*) malloc (bytes+1);
insilmaril@81: 	int i;
insilmaril@81: 	for (i=0;i<bytes;i++)
insilmaril@81: 		p[i]=s.at(i).latin1();
insilmaril@81: 	p[bytes]=0;	
insilmaril@81: 
insilmaril@81: 	QString r=mkdtemp (p);
insilmaril@81: 	if (r.isEmpty()) ok=false;
insilmaril@81: 	free (p);
insilmaril@81: 	return r;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: void removeDir(QDir d)
insilmaril@81: {
insilmaril@81: 	// This check should_ not be necessary, but proved to be useful ;-)
insilmaril@81: 	if (!isInTmpDir(d.path()))
insilmaril@81: 	{
insilmaril@81: 		qWarning ("file.cpp::removeDir should remove "+d.path()+" - aborted.");
insilmaril@81: 		return;
insilmaril@81: 	}
insilmaril@81: 
insilmaril@81: 	// Traverse directories
insilmaril@81: 	d.setFilter( QDir::Dirs| QDir::Hidden | QDir::NoSymLinks );
insilmaril@81: 	QFileInfoList list = d.entryInfoList();
insilmaril@81: 	QFileInfo fi;
insilmaril@81: 
insilmaril@81: 	for (int i = 0; i < list.size(); ++i) 
insilmaril@81: 	{
insilmaril@81: 		fi=list.at(i);
insilmaril@81: 		if (fi.fileName() != "." && fi.fileName() != ".." )
insilmaril@81: 		{
insilmaril@81: 			if ( !d.cd(fi.fileName()) ) 
insilmaril@81: 				qWarning ("removeDir() cannot find the directory "+fi.fileName());
insilmaril@81: 			else 
insilmaril@81: 			{
insilmaril@81: 				// Recursively remove subdirs
insilmaril@81: 				removeDir (d);
insilmaril@81: 				d.cdUp();
insilmaril@81: 			}
insilmaril@81: 		}	
insilmaril@81: 	}
insilmaril@81: 
insilmaril@81: 	// Traverse files
insilmaril@81: 	d.setFilter( QDir::Files| QDir::Hidden | QDir::NoSymLinks );
insilmaril@81: 	list = d.entryInfoList();
insilmaril@81: 
insilmaril@81: 	for (int i = 0; i < list.size(); ++i) 
insilmaril@81: 	{
insilmaril@81: 		fi=list.at(i);
insilmaril@81: 		QFile (fi.filePath()).remove(); 
insilmaril@81: 	}	
insilmaril@81: 
insilmaril@81: 	if (!d.rmdir(d.path()))
insilmaril@81: 		qWarning ("removeDir("+d.path()+") failed!");
insilmaril@81: }		
insilmaril@81: 
insilmaril@81: void copyDir (QDir src, QDir dst)
insilmaril@81: {
insilmaril@81: 	system ("cp -r "+src.path()+"/* "+dst.path());
insilmaril@81: 
insilmaril@81: 	/*
insilmaril@81: 	ErrorCode err=success;
insilmaril@81: 
insilmaril@81: 	Process *cpProc=new Process ();
insilmaril@81: 	QStringList args;
insilmaril@81: 	cpProc->setWorkingDirectory (src.path());
insilmaril@81: 	args <<"-r";
insilmaril@81: 	args <<src.path();
insilmaril@81: 	args <<dst.path();
insilmaril@81: 
insilmaril@81: 	cpProc->start ("cp",args);
insilmaril@81: 	if (!cpProc->waitForStarted() )
insilmaril@81: 	{	
insilmaril@81: 		// zip could not be started
insilmaril@81: 		QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 					   QObject::tr("Couldn't start zip to compress data."));
insilmaril@81: 		err=aborted;
insilmaril@81: 	} else
insilmaril@81: 	{
insilmaril@81: 		// zip could be started
insilmaril@81: 		cpProc->waitForFinished();
insilmaril@81: 		if (cpProc->exitStatus()!=QProcess::NormalExit )
insilmaril@81: 		{
insilmaril@81: 			QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 						   QObject::tr("cp didn't exit normally")+
insilmaril@81: 						   "\n" + cpProc->getErrout());
insilmaril@81: 			err=aborted;
insilmaril@81: 		} else
insilmaril@81: 		{
insilmaril@81: 			if (cpProc->exitCode()>0)
insilmaril@81: 			{
insilmaril@81: 				QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 						   QString("cp exit code:  %1").arg(cpProc->exitCode() )+
insilmaril@81: 						   "\n" + cpProc->getErrout() );
insilmaril@81: 				err=aborted;
insilmaril@81: 			}
insilmaril@81: 		}
insilmaril@81: 	}	// cp could be started
insilmaril@81: 	*/
insilmaril@81: }
insilmaril@81: 
insilmaril@81: void makeSubDirs (const QString &s)
insilmaril@81: {
insilmaril@81: 	QDir d(s);
insilmaril@81: 	d.mkdir(s);
insilmaril@81: 	d.mkdir ("images");	
insilmaril@81: 	d.mkdir ("flags");	
insilmaril@81: }
insilmaril@81: 
insilmaril@81: ErrorCode zipDir (const QDir &zipDir, const QString &zipName)
insilmaril@81: {
insilmaril@81: 	ErrorCode err=success;
insilmaril@81: 	
insilmaril@81: 	// zip the temporary directory
insilmaril@81: 	QStringList args;
insilmaril@81: 	Process *zipProc=new Process ();
insilmaril@81: 	zipProc->setWorkingDirectory (zipDir.path());
insilmaril@81: 	args <<"-r";
insilmaril@81: 	args <<zipName;
insilmaril@81: 	args <<".";
insilmaril@81: 
insilmaril@81: 	zipProc->start ("zip",args);
insilmaril@81: 	if (!zipProc->waitForStarted() )
insilmaril@81: 	{	
insilmaril@81: 		// zip could not be started
insilmaril@81: 		QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 					   QObject::tr("Couldn't start zip to compress data."));
insilmaril@81: 		err=aborted;
insilmaril@81: 	} else
insilmaril@81: 	{
insilmaril@81: 		// zip could be started
insilmaril@81: 		zipProc->waitForFinished();
insilmaril@81: 		if (zipProc->exitStatus()!=QProcess::NormalExit )
insilmaril@81: 		{
insilmaril@81: 			QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 						   QObject::tr("zip didn't exit normally")+
insilmaril@81: 						   "\n" + zipProc->getErrout());
insilmaril@81: 			err=aborted;
insilmaril@81: 		} else
insilmaril@81: 		{
insilmaril@81: 			if (zipProc->exitCode()>0)
insilmaril@81: 			{
insilmaril@81: 				QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 						   QString("zip exit code:  %1").arg(zipProc->exitCode() )+
insilmaril@81: 						   "\n" + zipProc->getErrout() );
insilmaril@81: 				err=aborted;
insilmaril@81: 			}
insilmaril@81: 		}
insilmaril@81: 	}	// zip could be started
insilmaril@81: 	return err;	
insilmaril@81: }
insilmaril@81: 
insilmaril@81: ErrorCode unzipDir (const QDir &zipDir, const QString &zipName)
insilmaril@81: {
insilmaril@81: 	ErrorCode err=success;
insilmaril@81: 
insilmaril@81: 	// Try to unzip file
insilmaril@81: #if !defined(Q_OS_WIN32)
insilmaril@81: 	QStringList args;
insilmaril@81: 	Process *zipProc=new Process ();
insilmaril@81: 	zipProc->setWorkingDirectory (zipDir.path());
insilmaril@81: 	args << "-o";	// overwrite existing files!
insilmaril@81: 	args << zipName ;
insilmaril@81: 	args << "-d";
insilmaril@81: 	args << zipDir.path();
insilmaril@81: 
insilmaril@81: 	zipProc->start ("unzip",args);
insilmaril@81: 	if (!zipProc->waitForStarted() )
insilmaril@81: 	{
insilmaril@81: 		QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 					   QObject::tr("Couldn't start unzip to decompress data."));
insilmaril@81: 		err=aborted;
insilmaril@81: 		
insilmaril@81: 	} else
insilmaril@81: 	{
insilmaril@81: 		zipProc->waitForFinished();
insilmaril@81: 		if (zipProc->exitStatus()!=QProcess::NormalExit )
insilmaril@81: 		{
insilmaril@81: 			QMessageBox::critical( 0,QObject::tr( "Critical Error" ),
insilmaril@81: 						   QObject::tr("unzip didn't exit normally") +
insilmaril@81: 						   zipProc->getErrout() );
insilmaril@81: 			err=aborted;
insilmaril@81: 		} else
insilmaril@81: 		{
insilmaril@81: 			if (zipProc->exitCode()>0)
insilmaril@81: 			{
insilmaril@81: 				if (zipProc->exitCode()==9)
insilmaril@81: 					// no zipped file, but maybe .xml or old version? Try again.
insilmaril@81: 					err=nozip;
insilmaril@81: 				else	
insilmaril@81: 				{
insilmaril@81: 					QMessageBox::critical( 0, QObject::tr( "Critical Error" ),
insilmaril@81: 								   QString("unzip exit code:  %1").arg(zipProc->exitCode() ) +
insilmaril@81: 								   zipProc->getErrout() );
insilmaril@81: 					err=aborted;
insilmaril@81: 				}
insilmaril@81: 			} 
insilmaril@81: 		}
insilmaril@81: 	}
insilmaril@81: #else
insilmaril@81:     // Do this process creation using Win32 API.
insilmaril@81:     //! Create process.
insilmaril@81:     PROCESS_INFORMATION piProcInfo;
insilmaril@81:     STARTUPINFO siStartInfo;
insilmaril@81: 
insilmaril@81:     // Initialize members of the PROCESS_INFORMATION structure.
insilmaril@81:     ::ZeroMemory( &piProcInfo, sizeof(PROCESS_INFORMATION) );
insilmaril@81: 
insilmaril@81:     // Set up members of the STARTUPINFO structure.
insilmaril@81:     ::ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
insilmaril@81:     siStartInfo.cb = sizeof(STARTUPINFO);
insilmaril@81: 
insilmaril@81:     // Create command line.
insilmaril@81:     QString argv("unzip -o ");
insilmaril@81:     argv.append(QDir::convertSeparators(zipName));
insilmaril@81:     argv.append(" -d ");
insilmaril@81:     argv.append(QDir::convertSeparators(zipDir.path()));
insilmaril@81: 
insilmaril@81:     // Create the child process.
insilmaril@81:     if( !::CreateProcess(NULL, 
insilmaril@81:         (LPWSTR)argv.unicode(), // command line
insilmaril@81:         NULL, // process security attributes
insilmaril@81:         NULL, // primary thread security attributes
insilmaril@81:         TRUE, // handles are inherited
insilmaril@81:         0, // creation flags
insilmaril@81:         NULL, // use parent's environment
insilmaril@81:         NULL, // use parent's current directory
insilmaril@81:         &siStartInfo, // STARTUPINFO pointer
insilmaril@81:         &piProcInfo) ) // receives PROCESS_INFORMATION
insilmaril@81:     {
insilmaril@81:         err = aborted;
insilmaril@81:     }
insilmaril@81:     else
insilmaril@81:     {
insilmaril@81:         // Wait for it to finish.
insilmaril@81:         ::WaitForSingleObject( piProcInfo.hProcess, 10000 );
insilmaril@81:     }
insilmaril@81: #endif
insilmaril@81: 	return err;	
insilmaril@81: }
insilmaril@81: 
insilmaril@81: bool loadStringFromDisk (const QString &fname, QString &s)
insilmaril@81: {
insilmaril@81: 	s="";
insilmaril@81: 	QFile file ( fname);
insilmaril@81: 	if ( !file.open( QIODevice::ReadOnly ) ) return false;
insilmaril@81: 
insilmaril@81: 	QTextStream ts( &file );
insilmaril@81: 	ts.setEncoding (QTextStream::UnicodeUTF8);
insilmaril@81: 	while ( !ts.atEnd() ) 
insilmaril@81: 		s+=ts.readLine()+"\n"; 
insilmaril@81: 	file.close();
insilmaril@81: 	return true;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: bool saveStringToDisk (const QString &fname, const QString &s)
insilmaril@81: {
insilmaril@81: 	QFile file( fname);
insilmaril@81: 
insilmaril@81: 	file.setName ( fname);
insilmaril@81: 	if ( !file.open( QIODevice::WriteOnly ) ) 
insilmaril@81: 	{
insilmaril@81: 		file.close();
insilmaril@81: 		return false;
insilmaril@81: 	}	
insilmaril@81: 
insilmaril@81: 	// Write it finally, and write in UTF8, no matter what 
insilmaril@81: 	QTextStream ts( &file );
insilmaril@81: 	ts.setEncoding (QTextStream::UnicodeUTF8);
insilmaril@81: 	ts << s;
insilmaril@81: 	file.close();
insilmaril@81: 	return true;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: 
insilmaril@81: ImagePreview::ImagePreview (QWidget *par=0): QLabel (par)
insilmaril@81: {
insilmaril@81: 	fdia=(Q3FileDialog*)par;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: void ImagePreview::previewUrl( const Q3Url &u )
insilmaril@81: {
insilmaril@81:     QString path = u.path();
insilmaril@81:     QPixmap pix( path );
insilmaril@81:     if ( pix.isNull() )
insilmaril@81: 	{
insilmaril@81: 		// Strange: If we have fd->setMode (QFileDialog::ExistingFiles)
insilmaril@81: 		// in the filedialog, then there are 3 calls to previewURL 
insilmaril@81: 		// for each selection. And only the first is the actual selected file
insilmaril@81: 		// while the following 2 point to the directory above the current one.
insilmaril@81: 		// So here's my workaround:
insilmaril@81: 		
insilmaril@81: 		if (fdia && fdia->selectedFiles().count()==0)
insilmaril@81: 			setText( QObject::tr("This is not an image.") );
insilmaril@81: 		if (fdia &&fdia->selectedFiles().count()>1)
insilmaril@81: 			setText( QObject::tr("Sorry, no preview for\nmultiple selected files.") );
insilmaril@81: 	}	
insilmaril@81:     else
insilmaril@81: 	{
insilmaril@81: 		float max_w=300;
insilmaril@81: 		float max_h=300;
insilmaril@81: 		float r;
insilmaril@81: 		if (pix.width()>max_w)
insilmaril@81: 		{
insilmaril@81: 			r=max_w / pix.width();
insilmaril@81: 			pix.resize(qRound(pix.width()*r), qRound(pix.height()*r));
insilmaril@81: 			// TODO not a resize, but a shrink/enlarge is needed here...
insilmaril@81: 		}
insilmaril@81: 		if (pix.height()>max_h)
insilmaril@81: 		{
insilmaril@81: 			r=max_h / pix.height();
insilmaril@81: 			pix.resize(qRound(pix.width()*r), qRound(pix.height()*r));
insilmaril@81: 			// TODO not a resize, but a shrink/enlarge is needed here...
insilmaril@81: 		}
insilmaril@81:         setPixmap( pix );
insilmaril@81: 	}	
insilmaril@81: }
insilmaril@81: 
insilmaril@81: ImageIO::ImageIO ()
insilmaril@81: {
insilmaril@81: 	// Create list with supported image types
insilmaril@81: 	// foreach (QByteArray format, QImageWriter::supportedImageFormats()) 
insilmaril@81: 	// imageTypes.append( tr("%1...").arg(QString(format).toUpper()));
insilmaril@81: 	imageFilters.append ("Images (*.png *.jpg *.jpeg *.bmp *.bmp *.ppm *.xpm *.xbm)");
insilmaril@81: 	imageTypes.append ("PNG");
insilmaril@81: 	imageFilters.append ("Portable Network Graphics (*.png)");
insilmaril@81: 	imageTypes.append ("PNG");
insilmaril@81: 	imageFilters.append ("Joint Photographic Experts Group (*.jpg)");
insilmaril@81: 	imageTypes.append ("JPG");
insilmaril@81: 	imageFilters.append ("Joint Photographic Experts Group (*.jpeg)");
insilmaril@81: 	imageTypes.append ("JPG");
insilmaril@81: 	imageFilters.append ("Windows Bitmap (*.bmp)");
insilmaril@81: 	imageTypes.append ("BMP");
insilmaril@81: 	imageFilters.append ("Portable Pixmap (*.ppm)");
insilmaril@81: 	imageTypes.append ("PPM");
insilmaril@81: 	imageFilters.append ("X11 Bitmap (*.xpm)");
insilmaril@81: 	imageTypes.append ("XPM");
insilmaril@81: 	imageFilters.append ("X11 Bitmap (*.xbm)");
insilmaril@81: 	imageTypes.append ("XBM");
insilmaril@81: }
insilmaril@81: 
insilmaril@81: QStringList ImageIO::getFilters()
insilmaril@81: {
insilmaril@81: 	return imageFilters;
insilmaril@81: }
insilmaril@81: 
insilmaril@81: QString ImageIO::getType(QString filter)
insilmaril@81: {
insilmaril@81: 	for (int i=0;i<imageFilters.count()+1;i++)
insilmaril@81: 		if (imageFilters.at(i)==filter) return imageTypes.at(i);
insilmaril@81: 	return QString();	
insilmaril@81: }
insilmaril@81: 
insilmaril@81: