Add grooveshark track parsing, and add a job notification for shortened url unwrapping

resources.qrc

@@ -133,6 +133,7 @@
         <file>data/images/headphones-bigger.png</file>
         <file>data/images/no-album-no-case.png</file>
         <file>data/images/rdio.png</file>
+        <file>data/images/grooveshark.png</file>
         <file>data/sql/dbmigrate-27_to_28.sql</file>
         <file>data/images/process-stop.png</file>
     </qresource>

src/libtomahawk/utils/groovesharkparser.cpp

@@ -39,6 +39,10 @@
 #include <QtNetwork/QNetworkAccessManager>
 #include <QtNetwork/QNetworkReply>
 
+#include <QWebPage>
+#include <QWebFrame>
+#include <QWebElement>
+
 using namespace Tomahawk;
 
 QPixmap* GroovesharkParser::s_pixmap = 0;
@@ -81,6 +85,8 @@ GroovesharkParser::lookupUrl( const QString& link )
 
         lookupGroovesharkPlaylist( link );
     }
+    else if ( link.contains( "grooveshark.com/s/" ) || link.contains( "grooveshark.com/#/s/" ) )
+        lookupGroovesharkTrack( link );
     else
         return;
 
@@ -153,6 +159,53 @@ GroovesharkParser::lookupGroovesharkPlaylist( const QString& linkRaw )
     m_queries.insert( reply );
 }
 
+
+void
+GroovesharkParser::lookupGroovesharkTrack( const QString& track )
+{
+    tLog() << "Parsing Grooveshark Track Page:" << track;
+
+    QNetworkReply* reply = TomahawkUtils::nam()->get( QNetworkRequest( QUrl( track ) ) );
+    connect( reply, SIGNAL( finished() ), this, SLOT( trackPageFetchFinished() ) );
+
+    m_browseJob = new DropJobNotifier( pixmap(), "Grooveshark", DropJob::Track, reply );
+    JobStatusView::instance()->model()->addJob( m_browseJob );
+
+    m_queries << reply;
+}
+
+
+void
+GroovesharkParser::trackPageFetchFinished()
+{
+    QNetworkReply* r = qobject_cast< QNetworkReply* >( sender() );
+    Q_ASSERT( r );
+
+    m_queries.remove( r );
+    r->deleteLater();
+
+    QWebPage page;
+    page.settings()->setAttribute( QWebSettings::JavascriptEnabled, false );
+    page.settings()->setAttribute( QWebSettings::PluginsEnabled, false );
+    page.settings()->setAttribute( QWebSettings::JavaEnabled, false );
+    page.settings()->setAttribute( QWebSettings::AutoLoadImages, false );
+    page.mainFrame()->setHtml( QString::fromUtf8( r->readAll() ) );
+    QWebElement title = page.mainFrame()->findFirstElement("span[itemprop='name']");
+    QWebElement artist = page.mainFrame()->findFirstElement("noscript span[itemprop='byArtist']");
+    QWebElement album = page.mainFrame()->findFirstElement("noscript span[itemprop='inAlbum']");
+
+    if ( !title.toPlainText().isEmpty() && !artist.toPlainText().isEmpty() )
+    {
+        tDebug() << "Got track info from grooveshark, enough to create a query:" << title.toPlainText() << artist.toPlainText() << album.toPlainText();
+
+        Tomahawk::query_ptr q = Tomahawk::Query::get( artist.toPlainText(), title.toPlainText(), album.toPlainText(), uuid(), true );
+        m_tracks << q;
+    }
+
+    checkTrackFinished();
+}
+
+
 void
 GroovesharkParser::groovesharkLookupFinished()
 {

src/libtomahawk/utils/groovesharkparser.h

@@ -58,6 +58,7 @@ signals:
 
 private slots:
     void groovesharkLookupFinished();
+    void trackPageFetchFinished();
 
     void playlistCreated();
 private:
@@ -65,6 +66,8 @@ private:
 
     void lookupUrl( const QString& url );
     void lookupGroovesharkPlaylist( const QString& playlist );
+    void lookupGroovesharkTrack( const QString& track );
+
     void checkTrackFinished();
     void checkPlaylistFinished();
     int  m_limit;

src/libtomahawk/utils/shortenedlinkparser.cpp

@@ -21,6 +21,7 @@
 
 #include "utils/logger.h"
 #include "utils/tomahawkutils.h"
+#include "dropjobnotifier.h"
 #include "query.h"
 #include "jobview/ErrorStatusMessage.h"
 #include "jobview/JobStatusModel.h"
@@ -33,6 +34,7 @@
 
 using namespace Tomahawk;
 
+QPixmap* ShortenedLinkParser::s_pixmap = 0;
 
 ShortenedLinkParser::ShortenedLinkParser ( const QStringList& urls, QObject* parent )
     : QObject( parent )
@@ -61,6 +63,8 @@ ShortenedLinkParser::handlesUrl( const QString& url )
              url.contains( "itun.es" ) ||
              url.contains( "tinyurl.com" ) ||
              url.contains( "tinysong.com" ) ||
+             url.contains( "grooveshark.com/s/~/" ) || // These redirect to the 'real' grooveshark track url
+             url.contains( "grooveshark.com/#/s/~/" ) ||
              url.contains( "rd.io" ) );
 }
 
@@ -68,11 +72,18 @@ void
 ShortenedLinkParser::lookupUrl ( const QString& url )
 {
     tDebug() << "Looking up..." << url;
+    QString cleaned = url;
+    if ( cleaned.contains( "/#/s/" ) )
+        cleaned.replace( "/#", "" );
 
-    QNetworkReply* reply = TomahawkUtils::nam()->get( QNetworkRequest( QUrl( url ) ) );
+    QNetworkReply* reply = TomahawkUtils::nam()->get( QNetworkRequest( QUrl( cleaned ) ) );
     connect( reply, SIGNAL( finished() ), this, SLOT( lookupFinished() ) );
 
     m_queries.insert( reply );
+
+    m_expandJob = new DropJobNotifier( pixmap(), "shortened", DropJob::Track, reply );
+    JobStatusView::instance()->model()->addJob( m_expandJob );
+
 }
 
 void
@@ -114,3 +125,13 @@ ShortenedLinkParser::checkFinished()
         deleteLater();
     }
 }
+
+
+QPixmap
+ShortenedLinkParser::pixmap()
+{
+    if ( !s_pixmap )
+        s_pixmap = new QPixmap( RESPATH "images/add.png" );
+
+    return *s_pixmap;
+}

src/libtomahawk/utils/shortenedlinkparser.h

@@ -26,12 +26,15 @@
 #include <QObject>
 #include <QSet>
 #include <QStringList>
+#include <QPixmap>
 
 class QNetworkReply;
 
 namespace Tomahawk
 {
 
+class DropJobNotifier;
+
 /**
  * Small class to parse whitelisted shortened links into the redirected urls
  *
@@ -58,8 +61,13 @@ private:
     void lookupUrl( const QString& url );
     void checkFinished();
 
+    static QPixmap pixmap();
+
     QStringList m_links;
     QSet< QNetworkReply* > m_queries;
+    DropJobNotifier* m_expandJob;
+
+    static QPixmap* s_pixmap;
 };
 
 }