Initial import

1 files changed, 259 insertions, 0 deletions

diff --git a/src/fileexporterbibutils.cpp b/src/fileexporterbibutils.cpp
new file mode 100644
index 0000000..4de7aeb
--- /dev/null
+++ b/src/fileexporterbibutils.cpp
@@ -0,0 +1,259 @@
+/***************************************************************************
+ *   Copyright (C) 2004-2009 by Thomas Fischer                             *
+ *   [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.                                   *
+ *                                                                         *
+ *   This program is distributed in the hope that it will be useful,       *
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+ *   GNU General Public License for more details.                          *
+ *                                                                         *
+ *   You should have received a copy of the GNU General Public License     *
+ *   along with this program; if not, write to the                         *
+ *   Free Software Foundation, Inc.,                                       *
+ *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+ ***************************************************************************/
+#include <qfile.h>
+#include <qdir.h>
+#include <qstringlist.h>
+#include <qapplication.h>
+#include <qbuffer.h>
+#include <qprocess.h>
+#include <qwaitcondition.h>
+
+#include <fileexporterbibtex.h>
+#include "fileexporterbibutils.h"
+
+namespace BibTeX
+{
+
+    FileExporterBibUtils::FileExporterBibUtils( BibTeX::File::FileFormat outputFormat )
+            : FileExporter(), m_outputFormat( outputFormat ), m_bibTeXExporter( new FileExporterBibTeX() )
+    {
+        m_bibTeXExporter->setEncoding( "utf-8" );
+        m_processBuffer = new QBuffer();
+    }
+
+    FileExporterBibUtils::~FileExporterBibUtils()
+    {
+        delete m_processBuffer;
+        delete m_bibTeXExporter;
+    }
+
+    bool FileExporterBibUtils::save( QIODevice* iodevice, const Element* element, QStringList* errorLog )
+    {
+        m_cancelFlag = false;
+        QBuffer bibBuffer;
+        if ( !toBuffer( element, &bibBuffer, errorLog ) )
+            return false;
+        if ( !bufferToXMLbuffer( &bibBuffer ) || m_cancelFlag )
+            return false;
+        return !m_cancelFlag && xmlBufferToIOdevice( iodevice );
+    }
+
+    bool FileExporterBibUtils::save( QIODevice* iodevice, const File* bibtexfile, QStringList* errorLog )
+    {
+        emit progress( 0, 3 );
+        m_cancelFlag = false;
+        QBuffer bibBuffer;
+        if ( !toBuffer( bibtexfile, &bibBuffer, errorLog ) )
+            return false;
+        emit progress( 1, 3 );
+        if ( m_cancelFlag || !bufferToXMLbuffer( &bibBuffer ) )
+            return false;
+        emit progress( 2, 3 );
+        if ( m_cancelFlag || !xmlBufferToIOdevice( iodevice ) )
+            return false;
+        emit progress( 3, 3 );
+        return !m_cancelFlag;
+    }
+
+    void FileExporterBibUtils::cancel()
+    {
+        m_bibTeXExporter->cancel();
+        m_cancelFlag = true;
+    }
+
+    bool FileExporterBibUtils::toBuffer( const File *bibFile, QBuffer *buffer, QStringList* errorLog )
+    {
+        buffer->open( IO_WriteOnly );
+        bool result = m_bibTeXExporter->save( buffer, bibFile, errorLog );
+        buffer->close();
+        return result;
+    }
+
+    bool FileExporterBibUtils::toBuffer( const Element *bibElement, QBuffer *buffer, QStringList* errorLog )
+    {
+        buffer->open( IO_WriteOnly );
+        bool result = m_bibTeXExporter->save( buffer, bibElement, errorLog );
+        buffer->close();
+        return result;
+    }
+
+    bool FileExporterBibUtils::bufferToXMLbuffer( QBuffer *bibBuffer )
+    {
+        QWaitCondition wc;
+
+        m_processBuffer->open( IO_WriteOnly );
+        m_waiting = true;
+        m_process = new QProcess( QStringList::split( ' ', "bib2xml -i utf8" ) );
+        connect( m_process, SIGNAL( processExited() ), this, SLOT( wakeUp() ) );
+        connect( m_process, SIGNAL( readyReadStdout() ), this, SLOT( slotReadyStdout() ) );
+        connect( m_process, SIGNAL( readyReadStderr() ), this, SLOT( slotReadyStderr() ) );
+
+        m_process->start();
+        if ( m_process->isRunning() )
+        {
+            bibBuffer->open( IO_ReadOnly );
+            m_process->writeToStdin( bibBuffer->readAll() );
+            qApp->processEvents();
+            m_process->closeStdin();
+            bibBuffer->close();
+
+            int nothingHappens = 20;
+            while ( m_waiting )
+            {
+                wc.wait( 250 );
+                qApp->processEvents();
+                --nothingHappens;
+            }
+
+            if ( nothingHappens <= 0 )
+                m_process->kill();
+
+            if ( !m_process->normalExit() )
+            {
+                qDebug( "%s did not exit in a clean fashion", m_process->arguments()[0].latin1() );
+                delete m_process;
+                return false;
+            }
+        }
+        else
+        {
+            qDebug( "%s did not start", m_process->arguments()[0].latin1() );
+            delete m_process;
+            return false;
+        }
+
+        m_processBuffer->close();
+
+        delete m_process;
+        return true;
+    }
+
+    bool FileExporterBibUtils::xmlBufferToIOdevice( QIODevice *iodevice )
+    {
+        QWaitCondition wc;
+        m_waiting = true;
+
+        m_process = NULL;
+        switch ( m_outputFormat )
+        {
+        case BibTeX::File::formatISI:
+            m_process = new QProcess( QStringList::split( ' ', "xml2isi" ) );
+            break;
+        case BibTeX::File::formatWordBib:
+            m_process = new QProcess( QStringList::split( ' ', "xml2wordbib" ) );
+            break;
+        case BibTeX::File::formatAds:
+            m_process = new QProcess( QStringList::split( ' ', "xml2ads" ) );
+            break;
+        case BibTeX::File::formatEndNote:
+            m_process = new QProcess( QStringList::split( ' ', "xml2end" ) );
+            break;
+        case BibTeX::File::formatRIS:
+            m_process = new QProcess( QStringList::split( ' ', "xml2ris" ) );
+            break;
+        case BibTeX::File::formatMODS:
+            /* m_process = NULL; */
+            break;
+        default:
+            qDebug( "Cannot handle output format %i", m_outputFormat );
+            return false;
+        }
+
+        if ( m_process != NULL )
+        {
+            connect( m_process, SIGNAL( processExited() ), this, SLOT( wakeUp() ) );
+            connect( m_process, SIGNAL( readyReadStdout() ), this, SLOT( slotReadyStdout() ) );
+            connect( m_process, SIGNAL( readyReadStderr() ), this, SLOT( slotReadyStderr() ) );
+
+            if ( m_process->start() )
+            {
+                QBuffer *tempBuffer = m_processBuffer;
+                m_processBuffer = new QBuffer();
+
+                tempBuffer->open( IO_ReadOnly );
+                m_process->writeToStdin( tempBuffer->readAll() );
+                qApp->processEvents();
+                m_process->closeStdin();
+                tempBuffer->close();
+
+                m_processBuffer->open( IO_WriteOnly );
+                int nothingHappens = 20;
+                while ( m_waiting )
+                {
+                    wc.wait( 250 );
+                    qApp->processEvents();
+                    --nothingHappens;
+                }
+                m_processBuffer->close();
+
+                delete tempBuffer;
+
+                if ( nothingHappens <= 0 )
+                    m_process->kill();
+
+                if ( ! m_process->normalExit() )
+                {
+                    delete m_process;
+                    return false;
+                }
+
+                m_processBuffer->open( IO_ReadOnly );
+                iodevice->writeBlock( m_processBuffer->buffer() );
+                m_processBuffer->close();
+
+                delete m_process;
+            }
+            else
+            {
+                qDebug( "%s did not start", m_process->arguments()[0].latin1() );
+                delete m_process;
+                return false;
+            }
+        }
+        else
+        {
+            m_processBuffer->open( IO_ReadOnly );
+            iodevice->writeBlock( m_processBuffer->buffer() );
+            m_processBuffer->close();
+        }
+
+        return true;
+    }
+
+    void FileExporterBibUtils::wakeUp()
+    {
+        m_waiting = false;
+    }
+
+    void FileExporterBibUtils::slotReadyStdout()
+    {
+        m_processBuffer->writeBlock( m_process->readStdout() );
+    }
+
+    void FileExporterBibUtils::slotReadyStderr()
+    {
+        QByteArray ba = m_process->readStderr();
+        QTextStream bats( ba, IO_ReadOnly );
+        bats.setEncoding( QTextStream::UnicodeUTF8 );
+        qDebug( "%s", bats.read().latin1() );
+    }
+
+}
+#include "fileexporterbibutils.moc"