browser/actors/PageStyleChild.jsm

   1 /* This Source Code Form is subject to the terms of the Mozilla Public
   2  * License, v. 2.0. If a copy of the MPL was not distributed with this
   3  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   4
   5 ChromeUtils.import("resource://gre/modules/Services.jsm");
   6
   7 var EXPORTED_SYMBOLS = ["PageStyleChild"];
   8
   9 ChromeUtils.import("resource://gre/modules/ActorChild.jsm");
  10
  11 class PageStyleChild extends ActorChild {
  12   getViewer(content) {
  13     return content.docShell.contentViewer;
  14   }
  15
  16   sendStyleSheetInfo(mm) {
  17     let content = mm.content;
  18     let filteredStyleSheets = this._filterStyleSheets(this.getAllStyleSheets(content), content);
  19
  20     mm.sendAsyncMessage("PageStyle:StyleSheets", {
  21       filteredStyleSheets,
  22       authorStyleDisabled: this.getViewer(content).authorStyleDisabled,
  23       preferredStyleSheetSet: content.document.preferredStyleSheetSet,
  24     });
  25   }
  26
  27   getAllStyleSheets(frameset) {
  28     let selfSheets = Array.slice(frameset.document.styleSheets);
  29     let subSheets = Array.map(frameset.frames, frame => this.getAllStyleSheets(frame));
  30     return selfSheets.concat(...subSheets);
  31   }
  32
  33   receiveMessage(msg) {
  34     let content = msg.target.content;
  35     switch (msg.name) {
  36       case "PageStyle:Switch":
  37         this.getViewer(content).authorStyleDisabled = false;
  38         this._stylesheetSwitchAll(content, msg.data.title);
  39         break;
  40
  41       case "PageStyle:Disable":
  42         this.getViewer(content).authorStyleDisabled = true;
  43         break;
  44     }
  45
  46     this.sendStyleSheetInfo(msg.target);
  47   }
  48
  49   handleEvent(event) {
  50     let win = event.target.ownerGlobal;
  51     if (win != win.top) {
  52       return;
  53     }
  54
  55     let mm = win.docShell.messageManager;
  56     this.sendStyleSheetInfo(mm);
  57   }
  58
  59   _stylesheetSwitchAll(frameset, title) {
  60     if (!title || this._stylesheetInFrame(frameset, title)) {
  61       this._stylesheetSwitchFrame(frameset, title);
  62     }
  63
  64     for (let i = 0; i < frameset.frames.length; i++) {
  65       // Recurse into sub-frames.
  66       this._stylesheetSwitchAll(frameset.frames[i], title);
  67     }
  68   }
  69
  70   _stylesheetSwitchFrame(frame, title) {
  71     var docStyleSheets = frame.document.styleSheets;
  72
  73     for (let i = 0; i < docStyleSheets.length; ++i) {
  74       let docStyleSheet = docStyleSheets[i];
  75       if (docStyleSheet.title) {
  76         docStyleSheet.disabled = (docStyleSheet.title != title);
  77       } else if (docStyleSheet.disabled) {
  78         docStyleSheet.disabled = false;
  79       }
  80     }
  81   }
  82
  83   _stylesheetInFrame(frame, title) {
  84     return Array.some(frame.document.styleSheets, (styleSheet) => styleSheet.title == title);
  85   }
  86
  87   _filterStyleSheets(styleSheets, content) {
  88     let result = [];
  89
  90     for (let currentStyleSheet of styleSheets) {
  91       if (!currentStyleSheet.title)
  92         continue;
  93
  94       // Skip any stylesheets that don't match the screen media type.
  95       if (currentStyleSheet.media.length > 0) {
  96         let mediaQueryList = currentStyleSheet.media.mediaText;
  97         if (!content.matchMedia(mediaQueryList).matches) {
  98           continue;
  99         }
 100       }
 101
 102       let URI;
 103       try {
 104         if (!currentStyleSheet.ownerNode ||
 105             // special-case style nodes, which have no href
 106             currentStyleSheet.ownerNode.nodeName.toLowerCase() != "style") {
 107           URI = Services.io.newURI(currentStyleSheet.href);
 108         }
 109       } catch (e) {
 110         if (e.result != Cr.NS_ERROR_MALFORMED_URI) {
 111           throw e;
 112         }
 113         continue;
 114       }
 115
 116       // We won't send data URIs all of the way up to the parent, as these
 117       // can be arbitrarily large.
 118       let sentURI = (!URI || URI.scheme == "data") ? null : URI.spec;
 119
 120       result.push({
 121         title: currentStyleSheet.title,
 122         disabled: currentStyleSheet.disabled,
 123         href: sentURI,
 124       });
 125     }
 126
 127     return result;
 128   }
 129 }