diff --git a/firefox/chrome/content/firequery.js b/firefox/chrome/content/firequery.js index a47329c..540590f 100644 --- a/firefox/chrome/content/firequery.js +++ b/firefox/chrome/content/firequery.js @@ -275,11 +275,20 @@ FBL.ns(function() { jQuery.data = function(elem, name, data) { var res = origDataFn.call(jQuery, elem, name, data); try { - if (name && data!=undefined) { + if (name && data!==undefined) { mutateData.call(context.getPanel('html'), elem, MODIFICATION, name, data); } - if (name && data===null) { - mutateData.call(context.getPanel('html'), elem, REMOVAL, name, data); + } catch (ex) { + // html panel may not exist yet (also want to be safe, when our highlighter throws for any reason) + } + return res; + }; + var origRemoveDataFn = jQuery.removeData; + jQuery.removeData = function(elem, name) { + var res = origRemoveDataFn.call(jQuery, elem, name); + try { + if (name) { + mutateData.call(context.getPanel('html'), elem, REMOVAL, name); } } catch (ex) { // html panel may not exist yet (also want to be safe, when our highlighter throws for any reason) diff --git a/test/index.html b/test/index.html index 3cc80d2..a87402e 100644 --- a/test/index.html +++ b/test/index.html @@ -6,11 +6,12 @@ +

FireQuery test page