From 7fefaa9688ceba876306aba25426f9958ab06d9b Mon Sep 17 00:00:00 2001
From: Gerben <gerben@treora.com>
Date: Tue, 21 Mar 2017 14:49:37 +0100
Subject: [PATCH] Add pageAction

---
 background.js | 79 ++++++++++++++++++++++++++++++++++++++++++---------
 manifest.json | 11 +++++++
 2 files changed, 76 insertions(+), 14 deletions(-)

diff --git a/background.js b/background.js
index 948510c..b556b6a 100644
--- a/background.js
+++ b/background.js
@@ -3,26 +3,77 @@ if (typeof browser === 'undefined') {
   this.browser = chrome
 };
 
-var urlPattern = /^(https?):\/\/(?:.+\.)youtube\.com\/watch\?.*v=([^&#]+)/;
+var cinemaModeEnabled = true;
 
-function onBeforeRequestListener(details) {
-  var match = details.url.match(urlPattern);
-  var scheme = match[1];
-  var videoId = match[2];
-  if (scheme && videoId) {
-    // Watch the embedded version instead. And without related video suggestions!
-    var newUrl = scheme + '://www.youtube.com/embed/' + videoId + '?rel=0&autoplay=1';
-
-    // From the embed, one should be able to follow the "watch on YouTube" link
-    if (newUrl === details.originUrl)
-      return
-
-    return {redirectUrl: newUrl};
+var cruftedUrlPattern = /^(https?):\/\/(?:.+\.)youtube\.com\/watch\?.*v=([^&#]+)/;
+var embeddableUrlPattern = /^(https?):\/\/(?:.+\.)youtube\.com\/embed\/([^&?#]+)/;
+
+// Turn a video url into its embeddable (full-window) version (and vice versa, if bothDirections==true).
+function makeNewUrl(url, bothDirections) {
+  if (cinemaModeEnabled) {
+    var match = url.match(cruftedUrlPattern);
+    if (!match)
+      return;
+    var scheme = match[1];
+    var videoId = match[2];
+    return scheme + '://www.youtube.com/embed/' + videoId + '?rel=0&autoplay=1';
+  }
+  else if (bothDirections) {
+    var match = url.match(embeddableUrlPattern);
+    if (!match)
+      return;
+    var scheme = match[1];
+    var videoId = match[2];
+    return scheme + '://www.youtube.com/watch?v=' + videoId;
   }
 }
 
+
+// Redirect crufted videos to their full-window version.
+function onBeforeRequestListener(details) {
+  var newUrl = makeNewUrl(details.url, false)
+
+  // From the embed, enable one to follow the "watch on YouTube" link (on browsers that pass originUrl)
+  if (newUrl === details.originUrl)
+    return
+
+  return {redirectUrl: newUrl};
+}
+
 browser.webRequest.onBeforeRequest.addListener(
   onBeforeRequestListener,
   {urls: ["*://*.youtube.com/watch*"]},
   ['blocking']
 );
+
+
+// Show the pageAction button if looking at a video (either with or without cruft).
+browser.webNavigation.onCommitted.addListener(function (details) {maybeShowPageAction(details.tabId, details.url);});
+browser.webNavigation.onHistoryStateUpdated.addListener(function (details) {maybeShowPageAction(details.tabId, details.url);});
+
+function maybeShowPageAction(tabId, url) {
+  var matchEmbeddable = url.match(embeddableUrlPattern);
+  var matchCruft = url.match(cruftedUrlPattern);
+  if (matchEmbeddable || matchCruft) {
+    browser.pageAction.show(tabId);
+    // In Chrome|ium, listeners stay across page changes, in Firefox they don't. So check first.
+    if (!browser.pageAction.onClicked.hasListener(handlePageAction))
+      browser.pageAction.onClicked.addListener(handlePageAction);
+  }
+}
+
+
+// Enable/Disable cinema mode when the pageAction button is clicked.
+function handlePageAction(tab) {
+  var matchEmbeddable = tab.url.match(embeddableUrlPattern);
+  if (matchEmbeddable)
+    cinemaModeEnabled = false;
+  else
+    cinemaModeEnabled = true;
+
+  // Relocate this page to reflect the new mode.
+  var newUrl = makeNewUrl(tab.url, true)
+  if (newUrl)
+    browser.tabs.update(tab.id, {url: newUrl});
+}
+
diff --git a/manifest.json b/manifest.json
index 8c1b5b6..95c7f9c 100644
--- a/manifest.json
+++ b/manifest.json
@@ -4,8 +4,19 @@
     "background": {
         "scripts": ["background.js"]
     },
+    "page_action": {
+        "browser_style": true,
+        "default_icon": {
+            "19": "img/icon19.png",
+            "38": "img/icon38.png",
+            "48": "img/icon48.png",
+            "96": "img/icon96.png"
+        }
+    },
     "permissions": [
         "*://*.youtube.com/*",
+        "tabs",
+        "webNavigation",
         "webRequest",
         "webRequestBlocking"
     ],