Fix amp urls which do not have origin (they are path only)

Author: dinoboy197

amplifier.js

@@ -44,6 +44,12 @@ chrome.runtime.sendMessage({
   method: "clear"
 });
 
+chrome.runtime.sendMessage({
+  sentinel: "__SIMPLIFYMESSAGE__",
+  method: "origin",
+  data: location.origin
+})
+
 // look for amp header
 chrome.runtime.sendMessage({
   sentinel: "__SIMPLIFYMESSAGE__",

background.js

@@ -174,7 +174,7 @@ function processTabState(tabId, senderUrl) {
       console.log("setting previous url to " + senderUrl);
       setTabStatus(tabStatus, function() {
         working = false;
-        chrome.tabs.update(tabId, { url : status.ampUrl });
+        chrome.tabs.update(tabId, { url : status.ampUrl.startsWith('/') ? (status['origin'] + status.ampUrl) : status.ampUrl });
       });
     } else {
       updatePageActionIcon(tabId, senderUrl, status);
@@ -236,6 +236,18 @@ function handleClear(sender) {
   });
 }
 
+function handleOrigin(tabId, origin) {
+  checkAndWork(function() {
+    getTabStatus(function(tabStatus) {
+      var status = tabId in tabStatus ? tabStatus[tabId] : {};
+      status['origin'] = origin;
+      console.log('got new origin ' + origin);
+      tabStatus[tabId] = status;
+      setTabStatus(tabStatus, function() { working = false });
+    });
+  });
+}
+
 function handleGetBack(sender, callback) {
   var theurl = sender.url;
   checkAndWork(function() {
@@ -295,6 +307,9 @@ chrome.runtime.onMessage.addListener(function(message, sender, sendResponse) {
     case "clear":
       handleClear(sender);
       break;
+    case "origin":
+      handleOrigin(sender.tab.id, message.data);
+      break;
     case "onAmpPage":
       handleOnAmpPage(sender, message.data);
       break;