Utils: Made escapeHtml()/unescapeHtml() more efficient.

ubiquity/modules/utils.js

@@ -826,26 +826,30 @@ function signHMAC(algo, key, str) {
 // Useful when you just want to concatenate a bunch of strings into
 // an HTML fragment and ensure that everything's escaped properly.
 
-function escapeHtml(string) (
+function escapeHtml(s) String(s).replace(escapeHtml.re, escapeHtml.fn);
-  String(string)
+escapeHtml.re = /[&<>\"\']/g;
-  .replace(/&/g, "&amp;")
+escapeHtml.fn = function escapeHtml_sub($) {
-  .replace(/</g, "&lt;")
+  switch ($) {
-  .replace(/>/g, "&gt;")
+    case "&": return "&amp;";
-  .replace(/\"/g, "&quot;")
+    case "<": return "&lt;";
-  .replace(/\'/g, "&#39;"));
+    case ">": return "&gt;";
+    case '"': return "&quot;";
+    case "'": return "&#39;";
+  }
+};
 
 // === {{{ Utils.unescapeHtml(string) }}} ===
 // Returns a version of the {{{string}}} with all occurrences of HTML character
-// references (&spades; &#x2665; &#9827; etc.) in it decoded.
+// references (e.g. &spades; &#x2665; &#9827; etc.) in it decoded.
 
-function unescapeHtml(string) (
+function unescapeHtml(s) String(s).replace(unescapeHtml.re, unescapeHtml.fn);
-  String.replace(string, /&#?\w+;/g, unescapeHtml.parse));
+unescapeHtml.re = /(?:&#?\w+;)+/g;
-unescapeHtml.parse = function unescapeHtml_parse(ref) {
+unescapeHtml.fn = function unescapeHtml_parse(ref) {
   var {div} = unescapeHtml_parse;
   div.innerHTML = ref;
   return div.textContent;
 };
-defineLazyProperty(unescapeHtml.parse, function div() (
+defineLazyProperty(unescapeHtml.fn, function div() (
   Utils.hiddenWindow.document.createElementNS(
     "http://www.w3.org/1999/xhtml", "div")));
 

ubiquity/tests/test_all.js

@@ -611,6 +611,16 @@ function testUtilsPrefs() {
   }
 }
 
+function testUtilsEscapeUnescapeHtml() {
+  this.skipIfXPCShell();
+
+  var {escapeHtml, unescapeHtml} = Utils;
+  var html = Utils.hiddenWindow.document.documentElement.innerHTML;
+  this.assertEquals(
+    html += "♠'♥♣",
+    unescapeHtml(unescapeHtml(escapeHtml(escapeHtml(html)))));
+}
+
 function testL10nUtilsPropertySelector() {
   var ps = LocalizationUtils.propertySelector("data:," + encodeURI(<![CDATA[
     foo=%S %S