Merge pull request #12 from kkszysiu/master

Added Vkontakte resolver for Tomahawk.

vkontakte/vkontakte-resolver.js

@@ -0,0 +1,368 @@
+/*
+ * (c) 2011 Krzysztof Klinikowski <kkszysiu@gmail.com>
+ */
+
+debugMode = false;
+
+function xhrRequest(url, method, data, callback){
+    var xhr = new XMLHttpRequest()
+
+    if (debugMode == true) { Tomahawk.log('Sending request:', url+'?'+data); }
+
+    if(method == "POST"){
+        xhr.open(method, url, true)
+        xhr.setRequestHeader("Content-type", "application/x-www-form-urlencoded")
+        xhr.send(data)
+    } else {
+        if(url.match(/\?/))
+            xhr.open(method, url+'&'+data, true)
+        else
+            xhr.open(method, url+'?'+data, true)
+
+        xhr.send()
+    }
+
+    xhr.onreadystatechange = function(){
+        if(xhr.readyState == 4){
+            //if (debugMode == true) { console.log("Response on "+url+" :", xhr); }
+            callback(xhr)
+        }
+    }
+}
+
+
+/**
+    VK
+
+    Module for working with vk.com
+**/
+var VK = {
+    determineSearchMethod: function(callback){
+        if (debugMode == true) { Tomahawk.log("Trying to determine search method"); }
+
+        xhrRequest("http://vkontakte.ru", "GET", null, function(xhr){
+            //if (debugMode == true) { console.log(xhr.responseText.match(/logout/)); }
+
+            if(!xhr.responseText.match(/logout/)){
+                xhrRequest("http://vk.com", "GET", null, function(xhr_vk){
+                    if(!xhr_vk.responseText.match(/logout/)) {
+                        callback({search_method:'test_mode'})
+                    } else {
+                        callback({search_method:'vk.com'})
+                    }
+                })
+            } else {
+                callback({search_method:'vkontakte.ru'})
+            }
+        })
+    },
+
+    /**
+        VK.#_rawSearch(artist, song, callback)
+
+        Searching by parsing page. Callback function must return hash with url.        
+    **/
+    _rawSearch: function(artist, song, duration, callback){    
+        var track = artist + " " + song
+
+        var url  = "http://"+VK.search_method+"/al_search.php"
+        var data = "c%5Bq%5D="+encodeURIComponent(track)+"&al=1&c%5Bsection%5D=audio&c%5Bsort%5D=2"
+
+        xhrRequest(url, "POST", data, function(xhr){
+
+            // User logged off from vkontakte
+            if(xhr.responseText.match(/progress\.gif/)){
+                delete VK.search_method
+
+                VK.search(artist, song, duration, callback)
+
+                return
+            }
+
+            var container = document.getElementById('vk_search');
+            response = xhr.responseText.replace(/^<!--/, '').replace(/->->/g, '-->');
+
+            container.innerHTML = response.split('<!>').join('');
+
+            var audio_data = []
+            var audio_rows = container.querySelectorAll('div.audio') 
+
+            for(var i=0; i<audio_rows.length; i++){
+                if(i>10) break
+
+                var url_data = audio_rows[i].querySelector('input[type=hidden]').value.split(',');
+                var url = url_data[0];
+                var duration = url_data[1];
+
+                var title = audio_rows[i].querySelector('td.info span');
+
+                audio_data.push({
+                  artist: audio_rows[i].querySelector('td.info a').innerHTML,
+                  title: title.textContent.trim(),
+                  duration: duration,
+                  url: url
+                })
+            }
+
+            if (debugMode == true) { Tomahawk.log("Tracks: "+JSON.stringify(audio_data)); }
+
+            if(audio_data.length > 0){
+
+                audio_data.lastIndex = 0
+
+                for(var i=0;i<audio_data.length; i++){
+                    //if (debugMode == true) { console.log(audio_data[i], Math.abs(parseInt(audio_data[i].duration) - duration), artist, song, audio_data[i].artist.toLowerCase(), artist.toLowerCase() == audio_data[i].artist.toLowerCase()); }
+
+                    if(audio_data[i].artist.toLowerCase() == artist && audio_data[i].title.toLowerCase() == song){
+                        if(!duration || Math.abs(parseInt(audio_data[i].duration) - duration) <= 2){
+                            audio_data.lastIndex = i
+                            break
+                        }
+                    } else if(!audio_data[i].title.toLowerCase().match(/(remix|mix)/) &&
+                               audio_data[i].artist.toLowerCase() == artist &&
+                               audio_data[i].title.toLowerCase() == song){
+                        audio_data.lastIndex = i
+                    }
+                }                
+
+                //Caching for 3 hours
+                //CACHE.set(track, audio_data, 1000*60*60*2)
+
+                callback(audio_data)
+            } else {
+                callback({error:'not_found'})
+            }
+        })
+
+        return false
+    },
+
+    /**
+        VK Applications for using in test_mode. [user_id, app_id, app_key]
+    **/
+    apps: [       
+        [327488, 525159, 'g5vuj9EWFO'],
+        [2118012, 1882836,'xYsD1Dtsng'],
+        [19730188, 1881945, 'rcj0HPk4Wk'],
+        [85838504, 1887506, 'nTCyM7WEBo'],
+        [9142393, 1891705, 'MlO3y0UXFV'],
+        [86844124, 1891718, '8NgTW7tjWm'],
+        [4824199, 1915951, 'pvHpN0V001'],
+        [5573107, 1914989, 'CChij669jU'],
+
+        [6240007, 1972474, 'DMXPeITyti'],
+        [102802046, 1985438, 'sPZCZS5YxJ'],
+        [102813567, 1985491, 'UjgLqfMPgc'],
+        [102815638, 1985494, 'vI7EmQNdS9'],
+        [102819736, 1985507, '3V0H9Y7zo9']
+
+    ],
+
+    getApiData: function(){
+        return VK.apps[Math.floor(Math.random()*VK.apps.length)]    
+    },    
+
+    /**
+        VK#_testmodeSearch(artist, song, callback)
+
+        Searching vkontakte with api in test_mode
+    **/
+    _testmodeSearch: function(artist, song, duration, callback){
+        var track = artist + " " + song
+
+        var api = VK.getApiData()
+        var url = "http://api.vk.com/api.php"
+
+        md5hash = Tomahawk.md5(api[0]+'api_id='+api[1]+'count=10format=jsonmethod=audio.searchq='+track+'sort=2test_mode=1'+api[2])
+        var data = 'api_id='+api[1]+'&method=audio.search&format=json&sig='+md5hash+'&sort=2&test_mode=1&count=10&q='+encodeURIComponent(track)
+
+        if (debugMode == true) { Tomahawk.log("Search url: "+url+'?'+data); }
+
+        xhrRequest("http://chromusapp.appspot.com/sign_data", "POST", "track="+encodeURIComponent(artist+song), function(xhr){
+            if (debugMode == true) { Tomahawk.log(xhr.responseText); }
+        }) 
+
+        xhrRequest(url, "GET", data, function(xhr){
+            // Too many requests and now we banned for some time
+            if(xhr.responseText.match(/\:false/)){
+                // Checking if user logged into vkontakte
+                VK.determineSearchMethod(function(response){
+                    if(response.search_method == "test_mode"){
+                        callback({error:'overload'})
+                    } else {
+                        VK.search_method = response.search_method
+                        VK.search(artist, song, duration, callback)
+                    }
+                })
+
+                return
+            }
+
+            if (debugMode == true) { Tomahawk.log(xhr.responseText); }
+            try {
+                var response_text = xhr.responseText.replace(/\u001D/g,'').replaceEntities()
+            } catch(err) {
+                var response_text = xhr.responseText
+            }
+
+            var results = JSON.parse(response_text);
+
+            if(results.response){
+                var vk_tracks = []
+
+                if(results.response[1]){
+                    vk_tracks.lastIndex = 0
+                    for(var i=1; i<results.response.length; i++){
+                        var audio = results.response[i].audio
+
+                        if (debugMode == true) { Tomahawk.log(JSON.stringify(audio)); }
+                        vk_tracks.push(audio)
+
+                        if(audio.artist.toLowerCase() == artist && audio.title.toLowerCase() == song){
+                            if(!duration || Math.abs((parseInt(audio.duration) - duration) <= 2)){
+                                vk_tracks.lastIndex = vk_tracks.length - 1
+
+                                break
+                            }
+                        } else if(!audio.title.toLowerCase().match(/(remix|mix)/) && audio.artist.toLowerCase() == artist){
+                            vk_tracks.lastIndex = vk_tracks.length - 1
+                        }
+                    }
+                    if (debugMode == true) { Tomahawk.log("Selected track: "+JSON.stringify(vk_tracks[vk_tracks.lastIndex])); }
+                }
+
+                if(vk_tracks && vk_tracks.length > 0){
+                    //vk_track.duration = parseInt(vk_track.duration)
+
+                    //Caching for 3 hours
+                    //CACHE.set(track, vk_tracks, 1000*60*60*3)
+
+                    callback(vk_tracks)
+                } else {
+                    callback({error:'not_found'})
+                }
+            } else {            
+                if(results.error)
+                    callback({error:results.error})
+                else{                    
+                    if (debugMode == true) { Tomahawk.log("ERROR!: Unknown error while searching track"); }
+                    callback({error:'Unknown error while searching track'})
+                }
+            }
+        })
+    },
+
+
+    /**
+        VK#search(artist, song, callback)
+        - artist (String): Artist
+        - song (String): Song
+        - callback (Function): Function to be called when search compete, to obtain results. 
+    **/    
+    search: function(artist, song, duration, callback){
+        if (debugMode == true) { Tomahawk.log("Seaching: "+artist+" - "+song); }
+
+        if(this.search_method == undefined){
+            this.determineSearchMethod(function(response){
+                if (debugMode == true) { Tomahawk.log("Search method: "+response.search_method); }
+
+                VK.search_method = response.search_method
+
+                VK.search(artist, song, duration, callback)
+            })
+
+            return 
+        }
+
+        artist = artist.toLowerCase()
+        song = song.toLowerCase()
+
+        if(duration != undefined)
+            duration = parseInt(duration)
+
+        var track = artist + " " + song
+
+        //if(CACHE.get(track))
+        //    return callback(CACHE.get(track))
+
+        if(this.search_method == "test_mode")
+            this._testmodeSearch(artist, song, duration, callback)
+        else
+            this._rawSearch(artist, song, duration, callback)
+
+        if (debugMode == true) { Tomahawk.log("Setting search method to null"); }
+        this.search_method = undefined;
+    }
+}
+
+
+var VKResolver = Tomahawk.extend(TomahawkResolver,
+{
+    settings:
+    {
+        name: 'Vkontakte',
+        weight: 75,
+        timeout: 5
+    },
+    resolve: function( qid, artist, album, title )
+    {
+        var data = {
+            qid: qid,
+            results: [],
+        }
+
+        VK.search(artist, title, null, function (response) {
+            if (response.length != 0) {
+                var elem = response[0];
+                var songinfo = {
+                    artist:     elem.artist,
+                    track:      elem.title,
+                    duration:   elem.duration,
+                    source:     "Vkontakte",
+                    extension:  "mp3",
+                    mimetype:   "audio/mp3",
+                    url:        elem.url,
+                };
+                data.results.push(songinfo);
+            }
+            return Tomahawk.addTrackResults( data );
+        });
+    },
+    search: function( qid, searchString )
+    {
+        var data = {
+            qid: qid,
+            results: [],
+        }
+        b = this;
+
+        VK.search(searchString, "", null, function (response) {
+            for (i = 0; i < response.length; i++) {
+                var elem = response[i];
+
+                //console.log(elem.artist+" - "+elem.title);
+                var songinfo = {
+                    artist:     elem.artist,
+                    track:      elem.title,
+                    duration:   elem.duration,
+                    source:     "Vkontakte",
+                    extension:  "mp3",
+                    mimetype:   "audio/mp3",
+                    url:        elem.url,
+                };
+
+                data.results.push(songinfo);
+            }
+            Tomahawk.addTrackResults( data );
+        });
+    },
+    internalSearch: function( qid )
+    {
+        return {
+            qid: qid,
+            results: []
+        };
+    }
+});
+
+Tomahawk.resolver.instance = VKResolver;