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.
tellico/src/fetch/bibsonomyfetcher.cpp

210 lines
6.1 KiB

/***************************************************************************
copyright : (C) 2007 by Robby Stephenson
email : robby@periapsis.org
***************************************************************************/
/***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of version 2 of the GNU General Public License as *
* published by the Free Software Foundation; *
* *
***************************************************************************/
#include "bibsonomyfetcher.h"
#include "messagehandler.h"
#include "../translators/bibteximporter.h"
#include "../tellico_kernel.h"
#include "../tellico_utils.h"
#include "../collection.h"
#include "../entry.h"
#include "../core/netaccess.h"
#include "../filehandler.h"
#include "../tellico_debug.h"
#include <tdelocale.h>
#include <tqlabel.h>
#include <tqlayout.h>
namespace {
// always bibtex
static const char* BIBSONOMY_BASE_URL = "http://bibsonomy.org";
static const int BIBSONOMY_MAX_RESULTS = 20;
}
using Tellico::Fetch::BibsonomyFetcher;
BibsonomyFetcher::BibsonomyFetcher(TQObject* parent_)
: Fetcher(parent_), m_job(0), m_started(false) {
}
BibsonomyFetcher::~BibsonomyFetcher() {
}
TQString BibsonomyFetcher::defaultName() {
return TQString::fromLatin1("Bibsonomy");
}
TQString BibsonomyFetcher::source() const {
return m_name.isEmpty() ? defaultName() : m_name;
}
bool BibsonomyFetcher::canFetch(int type) const {
return type == Data::Collection::Bibtex;
}
void BibsonomyFetcher::readConfigHook(const TDEConfigGroup&) {
}
void BibsonomyFetcher::search(FetchKey key_, const TQString& value_) {
m_key = key_;
m_value = value_.stripWhiteSpace();
m_started = true;
if(!canFetch(Kernel::self()->collectionType())) {
message(i18n("%1 does not allow searching for this collection type.").arg(source()), MessageHandler::Warning);
stop();
return;
}
m_data.truncate(0);
// myDebug() << "BibsonomyFetcher::search() - value = " << value_ << endl;
KURL u = searchURL(m_key, m_value);
if(u.isEmpty()) {
stop();
return;
}
m_job = TDEIO::get(u, false, false);
connect(m_job, TQT_SIGNAL(data(TDEIO::Job*, const TQByteArray&)),
TQT_SLOT(slotData(TDEIO::Job*, const TQByteArray&)));
connect(m_job, TQT_SIGNAL(result(TDEIO::Job*)),
TQT_SLOT(slotComplete(TDEIO::Job*)));
}
void BibsonomyFetcher::stop() {
if(!m_started) {
return;
}
// myDebug() << "BibsonomyFetcher::stop()" << endl;
if(m_job) {
m_job->kill();
m_job = 0;
}
m_data.truncate(0);
m_started = false;
emit signalDone(this);
}
void BibsonomyFetcher::slotData(TDEIO::Job*, const TQByteArray& data_) {
TQDataStream stream(m_data, IO_WriteOnly | IO_Append);
stream.writeRawBytes(data_.data(), data_.size());
}
void BibsonomyFetcher::slotComplete(TDEIO::Job* job_) {
// myDebug() << "BibsonomyFetcher::slotComplete()" << endl;
// since the fetch is done, don't worry about holding the job pointer
m_job = 0;
if(job_->error()) {
job_->showErrorDialog(Kernel::self()->widget());
stop();
return;
}
if(m_data.isEmpty()) {
myDebug() << "BibsonomyFetcher::slotComplete() - no data" << endl;
stop();
return;
}
Import::BibtexImporter imp(TQString::fromUtf8(m_data, m_data.size()));
Data::CollPtr coll = imp.collection();
if(!coll) {
myDebug() << "BibsonomyFetcher::slotComplete() - no valid result" << endl;
stop();
return;
}
Data::EntryVec entries = coll->entries();
for(Data::EntryVec::Iterator entry = entries.begin(); entry != entries.end(); ++entry) {
if(!m_started) {
// might get aborted
break;
}
TQString desc = entry->field(TQString::fromLatin1("author"))
+ TQChar('/') + entry->field(TQString::fromLatin1("publisher"));
if(!entry->field(TQString::fromLatin1("year")).isEmpty()) {
desc += TQChar('/') + entry->field(TQString::fromLatin1("year"));
}
SearchResult* r = new SearchResult(this, entry->title(), desc, entry->field(TQString::fromLatin1("isbn")));
m_entries.insert(r->uid, Data::EntryPtr(entry));
emit signalResultFound(r);
}
stop(); // required
}
Tellico::Data::EntryPtr BibsonomyFetcher::fetchEntry(uint uid_) {
return m_entries[uid_];
}
KURL BibsonomyFetcher::searchURL(FetchKey key_, const TQString& value_) const {
KURL u(TQString::fromLatin1(BIBSONOMY_BASE_URL));
u.setPath(TQString::fromLatin1("/bib/"));
switch(key_) {
case Person:
u.addPath(TQString::fromLatin1("author/%1").arg(value_));
break;
case Keyword:
u.addPath(TQString::fromLatin1("search/%1").arg(value_));
break;
default:
kdWarning() << "BibsonomyFetcher::search() - key not recognized: " << m_key << endl;
return KURL();
}
u.addQueryItem(TQString::fromLatin1("items"), TQString::number(BIBSONOMY_MAX_RESULTS));
myDebug() << "BibsonomyFetcher::search() - url: " << u.url() << endl;
return u;
}
void BibsonomyFetcher::updateEntry(Data::EntryPtr entry_) {
TQString title = entry_->field(TQString::fromLatin1("title"));
if(!title.isEmpty()) {
search(Fetch::Keyword, title);
return;
}
myDebug() << "BibsonomyFetcher::updateEntry() - insufficient info to search" << endl;
emit signalDone(this); // always need to emit this if not continuing with the search
}
Tellico::Fetch::ConfigWidget* BibsonomyFetcher::configWidget(TQWidget* parent_) const {
return new BibsonomyFetcher::ConfigWidget(parent_, this);
}
BibsonomyFetcher::ConfigWidget::ConfigWidget(TQWidget* parent_, const BibsonomyFetcher*)
: Fetch::ConfigWidget(parent_) {
TQVBoxLayout* l = new TQVBoxLayout(optionsWidget());
l->addWidget(new TQLabel(i18n("This source has no options."), optionsWidget()));
l->addStretch();
}
void BibsonomyFetcher::ConfigWidget::saveConfig(TDEConfigGroup&) {
}
TQString BibsonomyFetcher::ConfigWidget::preferredName() const {
return BibsonomyFetcher::defaultName();
}
#include "bibsonomyfetcher.moc"