summaryrefslogtreecommitdiffstats
path: root/src/webqueryieeexplore.cpp
diff options
context:
space:
mode:
authorSlávek Banko <[email protected]>2013-06-24 02:08:15 +0200
committerSlávek Banko <[email protected]>2013-07-04 02:44:37 +0200
commit998f21e02a725cd553d7c278819f67cd81295af4 (patch)
tree4bd158018e9302c31367b00c01cd2b41eb228414 /src/webqueryieeexplore.cpp
downloadkbibtex-998f21e02a725cd553d7c278819f67cd81295af4.tar.gz
kbibtex-998f21e02a725cd553d7c278819f67cd81295af4.zip
Initial import
Diffstat (limited to 'src/webqueryieeexplore.cpp')
-rw-r--r--src/webqueryieeexplore.cpp267
1 files changed, 267 insertions, 0 deletions
diff --git a/src/webqueryieeexplore.cpp b/src/webqueryieeexplore.cpp
new file mode 100644
index 0000000..070f8ca
--- /dev/null
+++ b/src/webqueryieeexplore.cpp
@@ -0,0 +1,267 @@
+/***************************************************************************
+ * Copyright (C) 2004-2009 by Thomas Fischer *
+ * *
+ * 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 <qspinbox.h>
+#include <qfile.h>
+
+#include <kio/jobclasses.h>
+#include <kio/job.h>
+#include <kio/netaccess.h>
+#include <klineedit.h>
+#include <kurl.h>
+#include <klocale.h>
+#include <kmessagebox.h>
+
+#include <settings.h>
+#include <entryfield.h>
+#include <file.h>
+#include <fileimporterbibtex.h>
+#include "webqueryieeexplore.h"
+
+#define min(a,b) ((a)>(b)?(b):(a))
+
+namespace KBibTeX
+{
+ WebQueryIEEExploreWidget::WebQueryIEEExploreWidget( QWidget *parent, const char *name )
+ : WebQueryWidget( parent, name )
+ {
+ init();
+
+ Settings *settings = Settings::self();
+ QString value = settings->getWebQueryDefault( "IEEE" );
+ value = value == QString::null ? "" : value;
+ lineEditQuery->setText( value );
+ slotTextChanged( value, true );
+ }
+
+ WebQueryIEEExplore::WebQueryIEEExplore( QWidget* parent ): WebQuery( parent ), m_numberOfMatches( "Your search matched <strong>(\\d+)</strong> of" ), m_findArNumber( "srchabstract.jsp\\?arnumber=(\\d+)" ), m_date( "(?:([A-Z][a-z]{2,3})(?:(?:/|-\\d+\\s+)([A-Z][a-z]{2,3}))?)?\\.?\\s*(\\d{4})$" ), m_risURL( "http://ieeexplore.ieee.org/xpls/citationAct" )
+ {
+ m_widget = new WebQueryIEEExploreWidget( parent );
+ m_bibtexImporter = new BibTeX::FileImporterBibTeX( false );
+ }
+
+ WebQueryIEEExplore::~WebQueryIEEExplore()
+ {
+ delete m_bibtexImporter;
+ delete m_widget;
+ }
+
+ QString WebQueryIEEExplore::title()
+ {
+ return i18n( "IEEExplore" );
+ }
+
+ QString WebQueryIEEExplore::disclaimer()
+ {
+ return i18n( "What is IEEE Xplore?" );
+ }
+
+ QString WebQueryIEEExplore::disclaimerURL()
+ {
+ return "http://ieeexplore.ieee.org/guide/g_oview_faq.jsp";
+ }
+
+ WebQueryWidget *WebQueryIEEExplore::widget()
+ {
+ return m_widget;
+ }
+
+ void WebQueryIEEExplore::query()
+ {
+ WebQuery::query();
+ Settings *settings = Settings::self();
+ settings->setWebQueryDefault( "IEEE", m_widget->lineEditQuery->text() );
+
+ m_arnumList.clear();
+ m_numMaxHits = min( 50, m_widget->spinBoxMaxHits->value() );
+ setNumStages( m_numMaxHits + 1 );
+ QString searchTerm = m_widget->lineEditQuery->text().stripWhiteSpace().replace( '$', "" );
+ if ( searchTerm.isEmpty() )
+ {
+ setEndSearch( WebQuery::statusInvalidQuery );
+ return;
+ }
+
+ QString category = "metadata";
+ KURL url = KURL( QString( "http://ieeexplore.ieee.org/search/freesearchresult.jsp?queryText=" ).append( searchTerm.replace( "%", "%25" ).replace( "+", "%2B" ).replace( " ", "%20" ).replace( "#", "%23" ).replace( "&", "%26" ).replace( "?", "%3F" ) ).append( "+%3Cin%3E+" ).append( category ).append( "&ResultCount=" ).append( QString::number( m_numMaxHits ) ).append( "&ResultStart=0" ) );
+
+ QString completeText = downloadHTML( url );
+ if ( completeText != QString::null && !m_aborted )
+ {
+ if ( completeText.find( "You have entered an invalid search" ) > -1 )
+ {
+ KMessageBox::information( m_widget, i18n( "You have entered an invalid search." ), i18n( "Search Error" ) );
+ setEndSearch( WebQuery::statusInvalidQuery );
+ return;
+ }
+ else if ( completeText.find( "No results" ) > -1 )
+ {
+ setEndSearch( WebQuery::statusSuccess );
+ return;
+ }
+
+ m_numberOfMatches.search( completeText );
+ bool ok;
+ m_numMatches = m_numberOfMatches.cap( 1 ).toInt( &ok );
+ if ( !ok ) m_numMatches = 0;
+
+ if ( m_numMatches <= 0 )
+ {
+ setEndSearch( WebQuery::statusSuccess );
+ return;
+ }
+
+ m_findArNumber.search( completeText );
+ for ( int i = 0; i < m_numMatches; ++i )
+ {
+ if ( m_findArNumber.cap( 1 ).isEmpty() ) break;
+ int arnum = m_findArNumber.cap( 1 ).toInt( &ok );
+ if ( !ok || arnum <= 0 ) break;
+ m_arnumList.append( arnum );
+
+ m_findArNumber.search( completeText, m_findArNumber.pos( 1 ) + 1 );
+ }
+
+ if ( !m_arnumList.isEmpty() )
+ {
+ m_hitCounter = 1;
+ fetchNext();
+ }
+ else
+ {
+ setEndSearch( WebQuery::statusSuccess );
+ }
+ }
+ else if ( !m_aborted )
+ {
+ QString message = KIO::NetAccess::lastErrorString();
+ if ( message.isEmpty() )
+ message.prepend( '\n' );
+ message.prepend( QString( i18n( "Querying database '%1' failed." ) ).arg( title() ) );
+ KMessageBox::error( m_parent, message );
+ setEndSearch( WebQuery::statusError );
+ }
+ else
+ {
+ setEndSearch( WebQuery::statusSuccess );
+ }
+ }
+
+ void WebQueryIEEExplore::fetchNext()
+ {
+ if ( m_arnumList.isEmpty() ) return;
+
+ int arnum = m_arnumList.first();
+ m_arnumList.remove( m_arnumList.begin() );
+
+ m_incomingData = "";
+ QString data = "dlSelect=cite&fileFormate=BibTex&arnumber=%3Carnumber%3E" + QString::number( arnum ) + "%3C%2Farnumber%3E&Submit=Download";
+
+ KIO::TransferJob *job = KIO::http_post( m_risURL, data.utf8(), false );
+ job->addMetaData( "content-type", "Content-Type: application/x-www-form-urlencoded" );
+ connect( job, SIGNAL( data( KIO::Job *, const QByteArray & ) ), this, SLOT( slotData( KIO::Job *, const QByteArray & ) ) );
+ connect( job, SIGNAL( result( KIO::Job * ) ), this, SLOT( slotResult( KIO::Job * ) ) );
+ }
+
+ void WebQueryIEEExplore::slotData( KIO::Job *, const QByteArray &data )
+ {
+ if ( data.size() > 0 )
+ m_incomingData.append( QCString( data, data.size() + 1 ) );
+ }
+
+ void WebQueryIEEExplore::slotResult( KIO::Job *job )
+ {
+ QRegExp m_date( "^(((\\d{1,2}(-\\d{1,2}))\\s+)?(([A-Z][a-z]{2,3})(/([A-Z][a-z]{2,3}))?)\\.?\\s+)?(\\d{4})$" );
+ if ( job->error() )
+ {
+ job->showErrorDialog();
+ return;
+ }
+
+ enterNextStage();
+ m_incomingData.replace( "<br>", "" );
+
+ BibTeX::File *bibtexFile = m_bibtexImporter->load( m_incomingData );
+ if ( bibtexFile != NULL )
+ {
+ for ( BibTeX::File::ElementList::iterator it = bibtexFile->begin(); it != bibtexFile->end(); ++it )
+ {
+ BibTeX::Entry *entry = dynamic_cast<BibTeX::Entry*>( *it );
+ if ( entry != NULL )
+ {
+ fixDate( entry );
+ emit foundEntry( new BibTeX::Entry( entry ), false );
+ }
+ }
+ delete bibtexFile;
+ }
+
+ if ( !m_arnumList.isEmpty() )
+ fetchNext();
+ else
+ {
+ setEndSearch( WebQuery::statusSuccess );
+ }
+ }
+
+ void WebQueryIEEExplore::fixDate( BibTeX::Entry *entry )
+ {
+ m_date.search( entry->getField( BibTeX::EntryField::ftYear )->value()->text() );
+
+ if ( !m_date.cap( 3 ).isNull() && !m_date.cap( 3 ).isEmpty() )
+ {
+ BibTeX::EntryField * field = entry->getField( BibTeX::EntryField::ftYear );
+ if ( field == NULL )
+ {
+ field = new BibTeX::EntryField( BibTeX::EntryField::ftJournal );
+ entry->addField( field );
+ }
+ field->setValue( new BibTeX::Value( m_date.cap( 3 ), true ) );
+ }
+
+ QString newMonth = QString::null;
+ if ( !m_date.cap( 1 ).isNull() && !m_date.cap( 1 ).isEmpty() && ( newMonth = parseMonth( m_date.cap( 1 ) ) ) != QString::null )
+ {
+ BibTeX::EntryField * field = entry->getField( BibTeX::EntryField::ftMonth );
+ if ( field == NULL )
+ {
+ field = new BibTeX::EntryField( BibTeX::EntryField::ftJournal );
+ entry->addField( field );
+ }
+
+ field->setValue( new BibTeX::Value( newMonth, true ) );
+
+ newMonth = QString::null;
+ if ( !m_date.cap( 2 ).isNull() && !m_date.cap( 2 ).isEmpty() && ( newMonth = parseMonth( m_date.cap( 2 ) ) ) != QString::null )
+ {
+ field->value()->items.append( new BibTeX::PlainText( "/" ) );
+ field->value()->items.append( new BibTeX::MacroKey( newMonth ) );
+ }
+ }
+ }
+
+ QString WebQueryIEEExplore::parseMonth( const QString &month )
+ {
+ for ( unsigned int i = 0; i < sizeof( BibTeX::MonthsTriple ) / sizeof( BibTeX::MonthsTriple[0] ); ++i )
+ if ( month.startsWith( BibTeX::MonthsTriple[i], false ) ) return BibTeX::MonthsTriple[i];
+ return QString::null;
+ }
+
+}
+#include "webqueryieeexplore.moc"