Fix: implement styleViaAPI

background/style-manager.js

@@ -26,7 +26,7 @@ const styleManager = (() => {
     maybeMatch: Set<styleId>,
     sections: Object<styleId => {
       id: styleId,
-      code: String
+      code: Array<String>
     }>
   } */
   const cachedStyleForUrl = createCache({
@@ -353,13 +353,13 @@ const styleManager = (() => {
     if (urlMatchStyle(url, data) !== true) {
       return;
     }
-    let code = '';
+    const code = [];
     for (const section of data.sections) {
       if (urlMatchSection(url, section) === true && !styleCodeEmpty(section.code)) {
-        code += section.code;
+        code.push(section.code);
       }
     }
-    return code;
+    return code.length && code;
   }
 
   function prepare() {

background/style-via-api.js

@@ -23,14 +23,14 @@ API_METHODS.styleViaAPI = !CHROME && (() => {
   let observingTabs = false;
 
   return function (request) {
-    const action = ACTIONS[request.action];
+    const action = ACTIONS[request.method];
     return !action ? NOP :
       action(request, this.sender)
         .catch(onError)
         .then(maybeToggleObserver);
   };
 
-  function styleApply({id = null, ignoreUrlCheck}, {tab, frameId, url}) {
+  function styleApply({id = null, ignoreUrlCheck = false}, {tab, frameId, url}) {
     if (prefs.get('disableAll')) {
       return NOP;
     }
@@ -42,15 +42,11 @@ API_METHODS.styleViaAPI = !CHROME && (() => {
       const tasks = [];
       for (const section of Object.values(sections)) {
         const styleId = section.id;
-        const code = section.code;
+        const code = section.code.join('\n');
-        if (!code) {
-          delete frameStyles[styleId];
-          continue;
-        }
         if (code === (frameStyles[styleId] || []).join('\n')) {
           continue;
         }
-        frameStyles[styleId] = [code];
+        frameStyles[styleId] = section.code;
         tasks.push(
           browser.tabs.insertCSS(tab.id, {
             code,

content/apply.js

@@ -7,6 +7,7 @@
 // define a constant so it throws when redefined
 const APPLY = (() => {
   const CHROME = chrome.app ? parseInt(navigator.userAgent.match(/Chrom\w+\/(?:\d+\.){2}(\d+)|$/)[1]) : NaN;
+  const STYLE_VIA_API = !chrome.app && document instanceof XMLDocument;
   var ID_PREFIX = 'stylus-';
   var ROOT;
   var isOwnPage = location.protocol.endsWith('-extension:');
@@ -29,26 +30,23 @@ const APPLY = (() => {
 
   let parentDomain;
 
-  // FIXME: does it work with styleViaAPI?
   prefs.subscribe(['disableAll'], (key, value) => doDisableAll(value));
   if (window !== parent) {
     prefs.subscribe(['exposeIframes'], updateExposeIframes);
   }
 
   function init() {
-    // FIXME: styleViaAPI
     // FIXME: getStylesFallback?
-    if (!chrome.app && document instanceof XMLDocument) {
+    if (STYLE_VIA_API) {
-      return API.styleViaAPI({action: 'styleApply'});
+      return API.styleViaAPI({method: 'styleApply'});
     }
     return API.getSectionsByUrl(getMatchUrl())
       .then(result => {
         ROOT = document.documentElement;
-        const styles = Object.values(result);
+        applyStyles(result, () => {
           // CSS transition bug workaround: since we insert styles asynchronously,
           // the browsers, especially Firefox, may apply all transitions on page load
-        applyStyles(styles, () => {
+          if ([...styleElements.values()].some(n => n.textContent.includes('transition'))) {
-          if (styles.some(s => s.code.includes('transition'))) {
             applyTransitionPatch();
           }
         });
@@ -136,10 +134,6 @@ const APPLY = (() => {
     return matchUrl;
   }
 
-  function buildSections(cache) {
-    return Object.values(cache);
-  }
-
   /**
    * TODO: remove when FF fixes the bug.
    * Firefox borks sendMessage in same-origin iframes that have 'src' with a real path on the site.
@@ -163,12 +157,12 @@ const APPLY = (() => {
   // }
 
   function applyOnMessage(request) {
-    if (!chrome.app && document instanceof XMLDocument && request.method !== 'ping') {
+    if (request.method === 'ping') {
-      request.action = request.method;
+      return true;
-      request.method = null;
+    }
-      request.styles = null;
+    if (STYLE_VIA_API) {
-      if (request.style) {
+      if (request.method === 'urlChanged') {
-        request.style.sections = null;
+        request.method = 'styleReplaceAll';
       }
       API.styleViaAPI(request);
       return;
@@ -188,7 +182,7 @@ const APPLY = (() => {
               if (!sections[request.style.id]) {
                 removeStyle(request.style);
               } else {
-                applyStyles(buildSections(sections));
+                applyStyles(sections);
               }
             });
         } else {
@@ -199,20 +193,15 @@ const APPLY = (() => {
       case 'styleAdded':
         if (request.style.enabled) {
           API.getSectionsByUrl(getMatchUrl(), request.style.id)
-            .then(buildSections)
             .then(applyStyles);
         }
         break;
 
       case 'urlChanged':
         API.getSectionsByUrl(getMatchUrl())
-          .then(buildSections)
           .then(replaceAll);
         break;
 
-      case 'ping':
-        return true;
-
       case 'backgroundReady':
         initializing.catch(err => {
           if (msg.RX_NO_RECEIVER.test(err.message)) {
@@ -232,6 +221,9 @@ const APPLY = (() => {
       return;
     }
     disableAll = disable;
+    if (STYLE_VIA_API) {
+      API.styleViaAPI({method: 'prefChanged', prefs: {disableAll}});
+    } else {
       Array.prototype.forEach.call(document.styleSheets, stylesheet => {
         if (stylesheet.ownerNode.matches(`style.stylus[id^="${ID_PREFIX}"]`)
         && stylesheet.disabled !== disable) {
@@ -239,6 +231,7 @@ const APPLY = (() => {
         }
       });
     }
+  }
 
   function fetchParentDomain() {
     if (parentDomain) {
@@ -290,7 +283,6 @@ const APPLY = (() => {
         disabledElements.delete(id);
       } else {
         return API.getSectionsByUrl(getMatchUrl(), id)
-          .then(buildSections)
           .then(applyStyles);
       }
     } else {
@@ -313,8 +305,8 @@ const APPLY = (() => {
     }
   }
 
-  function applyStyles(styles, done) {
+  function applyStyles(sections, done) {
-    if (!styles.length) {
+    if (!Object.keys(sections).length) {
       if (done) {
         done();
       }
@@ -325,7 +317,7 @@ const APPLY = (() => {
       new MutationObserver((mutations, observer) => {
         if (document.documentElement) {
           observer.disconnect();
-          applyStyles(styles, done);
+          applyStyles(sections, done);
         }
       }).observe(document, {childList: true});
       return;
@@ -336,8 +328,8 @@ const APPLY = (() => {
     } else {
       initDocRootObserver();
     }
-    for (const section of styles) {
+    for (const section of Object.values(sections)) {
-      applySections(section.id, section.code);
+      applySections(section.id, section.code.join(''));
     }
     docRootObserver.firstStart();