Add an initCommand hook

[GM_direct_download_links.git] / direct_download_links.user.js

diff --git a/direct_download_links.user.js b/direct_download_links.user.js
index 418b3a5..fdf6ece 100644 (file)
--- a/direct_download_links.user.js
+++ b/direct_download_links.user.js
@@ -53,6 +53,12 @@
  *
  *
  * Optional fields:
+ *
+ *   initCommand: a function called before the regExp is matched, this can
+ *                be useful in cases when some action needs to be done in
+ *                order to make the element containing the regExp be actually
+ *                rendered. It must accept  a 'site' parameter.
+ *
  *   onEvent: used to delay the urlRegexp matching to a certain event like
  *            'DOMNodeInserted' useful when the URL is added by some javascript
  *            library. It has two fields:
@@ -62,6 +68,11 @@
  *              targetElement: the element in the event handler we want the
  *                urlRegexp is performed on.
  *
+ *  processURL: a function to process the URL before adding the Direct
+ *              Downdload Link to the page, it must accept  a 'site' and a
+ *              'URL' parameters and dispatch the UrlFetched to pass the
+ *              modified URL to _add_link().
+ *
  */
 var supported_sites = [
   {
@@ -93,6 +104,7 @@ var supported_sites = [
     urlContainer: 'Player',
     urlRegexp: /mediaUri=(http:\/\/[^,]*)/,
     onEvent: { evt: 'DOMNodeInserted', targetElement: 'object' },
+    processURL: _rai_get_actual_url,
     linkDest: 'Player',
   },
 ];
@@ -103,6 +115,9 @@ for (i = 0; i < supported_sites.length; i++) {
 
   var result = window.location.href.match(site.locationRegexp);
   if (result) {
+    if (site.initCommand) {
+      site.initCommand(site);
+    }
     direct_download_link_add(window.location.href, site);
   }
 }
@@ -134,6 +149,7 @@ function direct_download_link_add(pageURL, site) {
       }
      _get_URL(site, element);
     }, false);
+    return;
   }
 
   _get_URL(site, element);
@@ -157,6 +173,11 @@ function _get_URL(site, element) {
     return;
   }
 
+  if (site.processURL) {
+    site.processURL(site, URL);
+    return;
+  }
+
   var evt = document.createEvent('Event');  
   evt.initEvent('UrlFetched', true, true);  
   evt.site = site;
@@ -168,22 +189,30 @@ function _add_link(e) {
   var site = e.site;
   var URL = e.URL;;
 
-  var links = document.getElementById(site.linkDest);
-  if (!links) {
+  var destination = document.getElementById(site.linkDest);
+  if (!destination) {
     DDL_log('DirectDl (' + site.pageURL + '): Cannot add the direct download link.');
     return;
   }
 
-  var download_link = document.createElement('a');
-  download_link.textContent = 'Direct Link';
-  download_link.setAttribute('href', URL);
-  var style = 'background-color: white; color: blue;';
-  style += ' border: 2px solid red;'
-  style += ' float: right; font-size: large;';
-  style += ' padding: .5em; margin: 1em;'
-  download_link.setAttribute('style', style);
-
-  links.insertBefore(download_link, links.firstChild);
+  // Check if we added the link already, if so just update the href attribute.
+  // This is useful when _get_URL() is called on async events.
+  var download_link = document.getElementById('GM_direct_downaload_link');
+  if (download_link) {
+    download_link.setAttribute('href', URL);
+  } else {
+    download_link = document.createElement('a');
+    download_link.textContent = 'Direct Link';
+    download_link.setAttribute('id', 'GM_direct_downaload_link');
+    download_link.setAttribute('href', URL);
+    var style = 'background-color: white; color: blue;';
+    style += ' border: 2px solid red;'
+    style += ' float: right; font-size: large;';
+    style += ' padding: .5em; margin: 1em;'
+    download_link.setAttribute('style', style);
+
+    destination.insertBefore(download_link, destination.firstChild);
+  }
 }
 
 function DDL_log(message) {
@@ -194,3 +223,33 @@ function DDL_log(message) {
     GM_log(message);
   }
 }
+
+function _rai_get_actual_url(site, URL) {
+  // http://www.neaveru.com/wordpress/index.php/2008/05/09/greasemonkey-bug-domnodeinserted-event-doesnt-allow-gm_xmlhttprequest/
+  setTimeout( function() {
+    GM_xmlhttpRequest({
+      method: "GET",
+      // XXX A custom header. This is the "clever" trick Rai uses to ensure
+      // the content is accessed by www.rai.tv only...
+      headers: {'viaurl': 'www.rai.tv'},
+      url: URL,
+      onload: function(response) {
+        text = response.responseText;
+        text = text.replace(/&/g, '&amp;')
+        parser = new DOMParser();
+        xmlDoc = parser.parseFromString(text, "text/xml");
+
+        ref = xmlDoc.getElementsByTagName('REF');
+        if (ref.length > 0) {
+          href = ref[0].getAttribute('HREF');;
+
+          var evt = document.createEvent('Event');  
+          evt.initEvent('UrlFetched', true, true);  
+          evt.site = site;
+          evt.URL = href;
+          document.dispatchEvent(evt);
+        }
+      }
+    });
+  }, 0);
+}