diff options
Diffstat (limited to 'libkcddb/asynccddbplookup.cpp')
-rw-r--r-- | libkcddb/asynccddbplookup.cpp | 352 |
1 files changed, 352 insertions, 0 deletions
diff --git a/libkcddb/asynccddbplookup.cpp b/libkcddb/asynccddbplookup.cpp new file mode 100644 index 00000000..f1ebe528 --- /dev/null +++ b/libkcddb/asynccddbplookup.cpp @@ -0,0 +1,352 @@ +/* + Copyright (C) 2002 Rik Hemsley (rikkus) <rik@kde.org> + Copyright (C) 2002 Benjamin Meyer <ben-devel@meyerhome.net> + Copyright (C) 2005 Richard Lärkäng <nouseforaname@home.se> + + This library is free software; you can redistribute it and/or + modify it under the terms of the GNU Library General Public + License as published by the Free Software Foundation; either + version 2 of the License, or (at your option) any later version. + + This library 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 + Library General Public License for more details. + + You should have received a copy of the GNU Library General Public License + along with this library; see the file COPYING.LIB. If not, write to + the Free Software Foundation, Inc., 59 Temple Place - Suite 330, + Boston, MA 02111-1307, USA. +*/ + +#include <kdebug.h> + +#include "asynccddbplookup.h" + +namespace KCDDB +{ + AsyncCDDBPLookup::AsyncCDDBPLookup() + : CDDBPLookup(), + state_(Idle) + { + + } + + AsyncCDDBPLookup::~AsyncCDDBPLookup() + { + } + + CDDB::Result + AsyncCDDBPLookup::lookup + ( + const QString & hostname, + uint port, + const TrackOffsetList & trackOffsetList + ) + { + socket_ = new KNetwork::KBufferedSocket(hostname,QString::number(port)); + + socket_->setBlocking( false ); + + connect (socket_, SIGNAL(gotError(int)), SLOT(slotGotError(int))); + + connect (socket_, SIGNAL( connected(const KResolverEntry &) ), + SLOT( slotConnectionSuccess() ) ); + + connect (socket_, SIGNAL( readyRead() ), SLOT( slotReadyRead() ) ); + + if ( trackOffsetList.count() < 3 ) + return UnknownError; + + trackOffsetList_ = trackOffsetList; + + state_ = WaitingForConnection; + + if ( !socket_->connect(hostname, QString::number(port)) ) + { + state_ = Idle; + emit finished( NoResponse ); + return NoResponse; + } + + return Success; + } + + void + AsyncCDDBPLookup::slotGotError(int error) + { + state_ = Idle; + + if ( error == KNetwork::KSocketBase::LookupFailure ) + emit finished( HostNotFound ); + else if ( error == KNetwork::KSocketBase::ConnectionTimedOut || + error == KNetwork::KSocketBase::NetFailure ) + emit finished( NoResponse ); + else + emit finished( UnknownError ); + } + + void + AsyncCDDBPLookup::slotConnectionSuccess() + { + kdDebug(60010) << "Connection successful" << endl; + state_ = WaitingForGreeting; + } + + void + AsyncCDDBPLookup::slotReadyRead() + { + kdDebug(60010) << "Ready to read. State: " << stateToString() << endl; + + while ( Idle != state_ && isConnected() && socket_->canReadLine() ) + read(); + } + + void + AsyncCDDBPLookup::read() + { + switch ( state_ ) + { + case WaitingForGreeting: + + if ( !parseGreeting( readLine() ) ) + { + result_ = ServerError; + doQuit(); + return; + } + + doHandshake(); + + break; + + case WaitingForHandshake: + + if ( !parseHandshake( readLine() ) ) + { + result_ = ServerError; + doQuit(); + return; + } + + doProto(); + + break; + + case WaitingForProtoResponse: + + // Ignore the response for now + readLine(); + + doQuery(); + + break; + + case WaitingForQueryResponse: + result_ = parseQuery( readLine() ); + + switch ( result_ ) + { + case Success: + requestCDInfoForMatch(); + break; + + case MultipleRecordFound: + state_ = WaitingForMoreMatches; + break; + + default: // Error :( + doQuit(); + return; + } + + break; + + case WaitingForMoreMatches: + { + QString line = readLine(); + + if (line.startsWith(".")) + requestCDInfoForMatch(); + else + parseExtraMatch( line ); + } + + break; + + case WaitingForCDInfoResponse: + { + Result result = parseRead( readLine() ); + + if ( Success != result ) + { + result_ = result; + doQuit(); + return; + } + + state_ = WaitingForCDInfoData; + } + + break; + + case WaitingForCDInfoData: + { + QString line = readLine(); + + if (line.startsWith(".")) + { + parseCDInfoData(); + requestCDInfoForMatch(); + } + else + cdInfoBuffer_ << line; + } + + break; + + case WaitingForQuitResponse: + + state_ = Idle; + + while ( socket_->bytesAvailable() ) + socket_->getch(); + + close(); + + emit finished( result_ ); + + break; + + default: + + break; + } + } + + QString + AsyncCDDBPLookup::readLine() + { + return QString::fromUtf8(socket_->readLine()); + } + + void + AsyncCDDBPLookup::doHandshake() + { + sendHandshake(); + + state_ = WaitingForHandshake; + } + + void + AsyncCDDBPLookup::doProto() + { + sendProto(); + + state_ = WaitingForProtoResponse; + } + + void + AsyncCDDBPLookup::doQuery() + { + sendQuery(); + + state_ = WaitingForQueryResponse; + } + + void + AsyncCDDBPLookup::requestCDInfoForMatch() + { + if (matchList_.isEmpty()) + { + result_ = cdInfoList_.isEmpty()? NoRecordFound : Success; + doQuit(); + return; + } + + CDDBMatch match = matchList_.first(); + matchList_.remove( match ); + + sendRead( match ); + + state_ = WaitingForCDInfoResponse; + } + + void + AsyncCDDBPLookup::parseCDInfoData() + { + CDInfo info; + + if (info.load( cdInfoBuffer_ )) + { + info.category = category_; + cdInfoList_.append( info ); + } + + cdInfoBuffer_.clear(); + } + + void + AsyncCDDBPLookup::doQuit() + { + state_ = WaitingForQuitResponse; + + sendQuit(); + } + + QString + AsyncCDDBPLookup::stateToString() const + { + switch (state_) + { + case Idle: + return "Idle"; + break; + + case WaitingForConnection: + return "WaitingForConnection"; + break; + + case WaitingForGreeting: + return "WaitingForGreeting"; + break; + + case WaitingForProtoResponse: + return "WaitingForProtoResponse"; + break; + + case WaitingForHandshake: + return "WaitingForHandshake"; + break; + + case WaitingForQueryResponse: + return "WaitingForQueryResponse"; + break; + + case WaitingForMoreMatches: + return "WaitingForMoreMatches"; + break; + + case WaitingForCDInfoResponse: + return "WaitingForCDInfoResponse"; + break; + + case WaitingForCDInfoData: + return "WaitingForCDInfoData"; + break; + + case WaitingForQuitResponse: + return "WaitingForQuitResponse"; + break; + + default: + return "Unknown"; + break; + } + } +} + + +#include "asynccddbplookup.moc" + +// vim:tabstop=2:shiftwidth=2:expandtab:cinoptions=(s,U1,m1 |