You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
kbibtex/src/webqueryieeexplore.cpp

268 lines
10 KiB

/***************************************************************************
* Copyright (C) 2004-2009 by Thomas Fischer *
* fischer@unix-ag.uni-kl.de *
* *
* 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 <tqspinbox.h>
#include <tqfile.h>
#include <tdeio/jobclasses.h>
#include <tdeio/job.h>
#include <tdeio/netaccess.h>
#include <klineedit.h>
#include <kurl.h>
#include <tdelocale.h>
#include <tdemessagebox.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( TQWidget *parent, const char *name )
: WebQueryWidget( parent, name )
{
init();
Settings *settings = Settings::self();
TQString value = settings->getWebQueryDefault( "IEEE" );
value = value == TQString::null ? "" : value;
lineEditQuery->setText( value );
slotTextChanged( value, true );
}
WebQueryIEEExplore::WebQueryIEEExplore( TQWidget* 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;
}
TQString WebQueryIEEExplore::title()
{
return i18n( "IEEExplore" );
}
TQString WebQueryIEEExplore::disclaimer()
{
return i18n( "What is IEEE Xplore?" );
}
TQString 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 );
TQString searchTerm = m_widget->lineEditQuery->text().stripWhiteSpace().replace( '$', "" );
if ( searchTerm.isEmpty() )
{
setEndSearch( WebQuery::statusInvalidQuery );
return;
}
TQString category = "metadata";
KURL url = KURL( TQString( "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( TQString::number( m_numMaxHits ) ).append( "&ResultStart=0" ) );
TQString completeText = downloadHTML( url );
if ( completeText != TQString::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 )
{
TQString message = TDEIO::NetAccess::lastErrorString();
if ( message.isEmpty() )
message.prepend( '\n' );
message.prepend( TQString( 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 = "";
TQString data = "dlSelect=cite&fileFormate=BibTex&arnumber=%3Carnumber%3E" + TQString::number( arnum ) + "%3C%2Farnumber%3E&Submit=Download";
TDEIO::TransferJob *job = TDEIO::http_post( m_risURL, data.utf8(), false );
job->addMetaData( "content-type", "Content-Type: application/x-www-form-urlencoded" );
connect( job, SIGNAL( data( TDEIO::Job *, const TQByteArray & ) ), this, SLOT( slotData( TDEIO::Job *, const TQByteArray & ) ) );
connect( job, SIGNAL( result( TDEIO::Job * ) ), this, SLOT( slotResult( TDEIO::Job * ) ) );
}
void WebQueryIEEExplore::slotData( TDEIO::Job *, const TQByteArray &data )
{
if ( data.size() > 0 )
m_incomingData.append( TQCString( data, data.size() + 1 ) );
}
void WebQueryIEEExplore::slotResult( TDEIO::Job *job )
{
TQRegExp 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 ) );
}
TQString newMonth = TQString::null;
if ( !m_date.cap( 1 ).isNull() && !m_date.cap( 1 ).isEmpty() && ( newMonth = parseMonth( m_date.cap( 1 ) ) ) != TQString::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 = TQString::null;
if ( !m_date.cap( 2 ).isNull() && !m_date.cap( 2 ).isEmpty() && ( newMonth = parseMonth( m_date.cap( 2 ) ) ) != TQString::null )
{
field->value()->items.append( new BibTeX::PlainText( "/" ) );
field->value()->items.append( new BibTeX::MacroKey( newMonth ) );
}
}
}
TQString WebQueryIEEExplore::parseMonth( const TQString &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 TQString::null;
}
}
#include "webqueryieeexplore.moc"