summaryrefslogtreecommitdiffstats
path: root/libk3b/cddb/k3bcddbquery.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'libk3b/cddb/k3bcddbquery.cpp')
-rw-r--r--libk3b/cddb/k3bcddbquery.cpp275
1 files changed, 275 insertions, 0 deletions
diff --git a/libk3b/cddb/k3bcddbquery.cpp b/libk3b/cddb/k3bcddbquery.cpp
new file mode 100644
index 0000000..783f9a4
--- /dev/null
+++ b/libk3b/cddb/k3bcddbquery.cpp
@@ -0,0 +1,275 @@
+/*
+ *
+ * $Id: k3bcddbquery.cpp 619556 2007-01-03 17:38:12Z trueg $
+ * Copyright (C) 2003 Sebastian Trueg <trueg@k3b.org>
+ *
+ * This file is part of the K3b project.
+ * Copyright (C) 1998-2007 Sebastian Trueg <trueg@k3b.org>
+ *
+ * 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.
+ * See the file "COPYING" for the exact licensing terms.
+ */
+
+
+
+#include "k3bcddbquery.h"
+
+#include "k3bcddbresult.h"
+
+#include <kdebug.h>
+#include <kapplication.h>
+#include <kaboutdata.h>
+#include <klocale.h>
+
+
+#include <qtextstream.h>
+#include <qstringlist.h>
+#include <qregexp.h>
+#include <qtimer.h>
+
+#include <stdlib.h>
+
+
+
+
+K3bCddbQuery::K3bCddbQuery( QObject* parent, const char* name )
+ : QObject(parent, name)
+{
+ m_bQueryFinishedEmited = false;
+}
+
+
+K3bCddbQuery::~K3bCddbQuery()
+{
+}
+
+
+void K3bCddbQuery::query( const K3bDevice::Toc& toc )
+{
+ m_bQueryFinishedEmited = false;
+ m_toc = toc;
+ m_inexactMatches.clear();
+
+ QTimer::singleShot( 0, this, SLOT(doQuery()) );
+}
+
+
+void K3bCddbQuery::queryMatch( const K3bCddbResultHeader& header )
+{
+ m_header = header;
+ m_result = K3bCddbResultEntry();
+ m_result.category = header.category;
+ m_result.discid = header.discid;
+
+ QTimer::singleShot( 0, this, SLOT(doMatchQuery()) );
+}
+
+
+const QStringList& K3bCddbQuery::categories()
+{
+ static QStringList s_cat = QStringList::split( ",", "rock,blues,misc,classical,"
+ "country,data,folk,jazz,newage,reggae,soundtrack" );
+ return s_cat;
+}
+
+
+bool K3bCddbQuery::parseEntry( QTextStream& stream, K3bCddbResultEntry& entry )
+{
+ entry.rawData = "";
+
+ stream.setEncoding( QTextStream::UnicodeUTF8 );
+
+ // parse data
+ QString line;
+ while( !(line = stream.readLine()).isNull() ) {
+ entry.rawData.append(line + "\n");
+
+ // !all fields may be splitted into several lines!
+
+ if( line.startsWith( "DISCID" ) ) {
+ // TODO: this could be several discids separated by comma!
+ }
+
+ else if( line.startsWith( "DYEAR" ) ) {
+ QString year = line.mid( 6 );
+ if( year.length() == 4 )
+ entry.year = year.toInt();
+ }
+
+ else if( line.startsWith( "DGENRE" ) ) {
+ entry.genre = line.mid( 7 );
+ }
+
+ else if( line.startsWith( "DTITLE" ) ) {
+ entry.cdTitle += line.mid( 7 );
+ }
+
+ else if( line.startsWith( "TTITLE" ) ) {
+ int eqSgnPos = line.find( "=" );
+ bool ok;
+ uint trackNum = (uint)line.mid( 6, eqSgnPos - 6 ).toInt( &ok );
+ if( !ok )
+ kdDebug() << "(K3bCddbQuery) !!! PARSE ERROR: " << line << endl;
+ else {
+ // kdDebug() << "(K3bCddbQuery) Track title for track " << trackNum << endl;
+
+ // make sure the list is big enough
+ while( entry.titles.count() <= trackNum )
+ entry.titles.append( "" );
+
+ entry.titles[trackNum] += line.mid( eqSgnPos+1 );
+ }
+ }
+
+ else if( line.startsWith( "EXTD" ) ) {
+ entry.cdExtInfo += line.mid( 5 );
+ }
+
+ else if( line.startsWith( "EXTT" ) ) {
+ int eqSgnPos = line.find( "=" );
+ bool ok;
+ uint trackNum = (uint)line.mid( 4, eqSgnPos - 4 ).toInt( &ok );
+ if( !ok )
+ kdDebug() << "(K3bCddbQuery) !!! PARSE ERROR: " << line << endl;
+ else {
+ // kdDebug() << "(K3bCddbQuery) Track extr track " << trackNum << endl;
+
+ // make sure the list is big enough
+ while( entry.extInfos.count() <= trackNum )
+ entry.extInfos.append( "" );
+
+ entry.extInfos[trackNum] += line.mid( eqSgnPos+1 );
+ }
+ }
+
+ else if( line.startsWith( "#" ) ) {
+ // kdDebug() << "(K3bCddbQuery) comment: " << line << endl;
+ }
+
+ else {
+ kdDebug() << "(K3bCddbQuery) Unknown field: " << line << endl;
+ }
+ }
+
+ // now split the titles in the last added match
+ // if no " / " delimiter is present title and artist are the same
+ // -------------------------------------------------------------------
+ QString fullTitle = entry.cdTitle;
+ int splitPos = fullTitle.find( " / " );
+ if( splitPos < 0 )
+ entry.cdArtist = fullTitle;
+ else {
+ // split
+ entry.cdTitle = fullTitle.mid( splitPos + 3 );
+ entry.cdArtist = fullTitle.left( splitPos );
+ }
+
+
+ for( QStringList::iterator it = entry.titles.begin();
+ it != entry.titles.end(); ++it ) {
+ QString fullTitle = *it;
+ int splitPos = fullTitle.find( " / " );
+ if( splitPos < 0 )
+ entry.artists.append( entry.cdArtist );
+ else {
+ // split
+ *it = fullTitle.mid( splitPos + 3 );
+ entry.artists.append( fullTitle.left( splitPos ) );
+ }
+ }
+
+
+ // replace all "\\n" with "\n"
+ for( QStringList::iterator it = entry.titles.begin();
+ it != entry.titles.end(); ++it ) {
+ (*it).replace( "\\\\\\\\n", "\\n" );
+ }
+
+ for( QStringList::iterator it = entry.artists.begin();
+ it != entry.artists.end(); ++it ) {
+ (*it).replace( "\\\\\\\\n", "\\n" );
+ }
+
+ for( QStringList::iterator it = entry.extInfos.begin();
+ it != entry.extInfos.end(); ++it ) {
+ (*it).replace( "\\\\\\\\n", "\\n" );
+ }
+
+ entry.cdTitle.replace( "\\\\\\\\n", "\\n" );
+ entry.cdArtist.replace( "\\\\\\\\n", "\\n" );
+ entry.cdExtInfo.replace( "\\\\\\\\n", "\\n" );
+ entry.genre.replace( "\\\\\\\\n", "\\n" );
+
+ return true;
+}
+
+
+int K3bCddbQuery::getCode( const QString& line )
+{
+ bool ok;
+ int code = line.left( 3 ).toInt( &ok );
+ if( !ok )
+ code = -1;
+ return code;
+}
+
+
+QString K3bCddbQuery::handshakeString() const
+{
+ QString user( getenv("USER") );
+ QString host( getenv("HOST") );
+ if( user.isEmpty() )
+ user = "kde-user";
+ if( host.isEmpty() )
+ host = "kde-host";
+
+ return QString("%1 %2 K3b %3").arg(user).arg(host).arg(kapp->aboutData()->version());
+}
+
+
+QString K3bCddbQuery::queryString() const
+{
+ QString query = "cddb query "
+ + QString::number( m_toc.discId(), 16 ).rightJustify( 8, '0' )
+ + " "
+ + QString::number( (unsigned int)m_toc.count() );
+
+ for( K3bDevice::Toc::const_iterator it = m_toc.begin(); it != m_toc.end(); ++it ) {
+ query.append( QString( " %1" ).arg( (*it).firstSector().lba() ) );
+ }
+
+ query.append( QString( " %1" ).arg( m_toc.length().lba() / 75 ) );
+
+ return query;
+}
+
+
+bool K3bCddbQuery::parseMatchHeader( const QString& line, K3bCddbResultHeader& header )
+{
+ // format: category id title
+ // where title could be artist and title splitted with a /
+ header.category = line.section( ' ', 0, 0 );
+ header.discid = line.section( ' ', 1, 1 );
+ header.title = line.mid( header.category.length() + header.discid.length() + 2 );
+ int slashPos = header.title.find( "/" );
+ if( slashPos > 0 ) {
+ header.artist = header.title.left(slashPos).stripWhiteSpace();
+ header.title = header.title.mid( slashPos+1 ).stripWhiteSpace();
+ }
+ return true;
+}
+
+
+void K3bCddbQuery::emitQueryFinished()
+{
+ if( !m_bQueryFinishedEmited ) {
+ m_bQueryFinishedEmited = true;
+ emit queryFinished( this );
+ }
+}
+
+
+#include "k3bcddbquery.moc"