Update headers in QtScriptResolver

lfranchi · Feb 18, 2013 · a5f2b0e · a5f2b0e
1 parent 3c95d57
commit a5f2b0e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/libtomahawk/resolvers/QtScriptResolver.cpp b/src/libtomahawk/resolvers/QtScriptResolver.cpp
@@ -36,7 +36,6 @@
 #include "jobview/ErrorStatusMessage.h"
 
 #include "utils/TomahawkUtilsGui.h"
-#include "utils/Logger.h"
 
 #include "config.h"
 

diff --git a/src/libtomahawk/resolvers/QtScriptResolver.h b/src/libtomahawk/resolvers/QtScriptResolver.h
@@ -25,6 +25,8 @@
 #include "Query.h"
 #include "utils/TomahawkUtils.h"
 #include "config.h"
+#include "TomahawkVersion.h"
+#include "utils/Logger.h"
 
 #include <QDir>
 #include <QFile>
@@ -103,6 +105,19 @@ Q_OBJECT
         settings()->setAttribute( QWebSettings::LocalStorageDatabaseEnabled, true );
         settings()->setAttribute( QWebSettings::LocalContentCanAccessFileUrls, true );
         settings()->setAttribute( QWebSettings::LocalContentCanAccessRemoteUrls, true );
+
+        // Tomahawk is not a user agent
+        m_header = QWebPage::userAgentForUrl( QUrl() ).replace( QString( "%1/%2" )
+                                                                .arg( TOMAHAWK_APPLICATION_NAME )
+                                                                .arg( TOMAHAWK_VERSION )
+                                                                ,"");
+        tLog() << "QtScriptResolver Using header" << m_header;
+    }
+
+    QString userAgentForUrl ( const QUrl & url ) const
+    {
+        Q_UNUSED(url);
+        return m_header;
     }
 
     void setScriptPath( const QString& scriptPath )
@@ -122,6 +137,7 @@ public slots:
 private:
     QtScriptResolver* m_parent;
     QString m_scriptPath;
+    QString m_header;
 };