Added ad counting functionality

lib/stats.js

 /*
  * This file is part of Adblock Plus <http://adblockplus.org/>,
  * Copyright (C) 2006-2013 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 /**
  * @fileOverview Provides usage stats
  */
 
 let {Prefs} = require("prefs");
 let {BlockingFilter} = require("filterClasses");
 let {FilterNotifier} = require("filterNotifier");
 
 /**
- * Base class for individual stats
- * @param {Object} stats simple object representation of stats
+ * Get statistics for specified tab
+ * @param  {String} key   field key
+ * @param  {Number} tabId tab ID (leave undefined for total stats)
+ * @return {Number}       field value
  */
-let StatsObject = exports.StatsObject = function(stats)
+let getStats = exports.getStats = function getStats(key, tabId)
 {
-  if (typeof stats == "undefined")
-    return;
-  
-  this.blocked = stats.blocked || this.blocked;
-}
-StatsObject.prototype = {
-  /**
-   * Number of blocked resources
-   * @type Number
-   */
-  blocked: 0
+  if (tabId)
+  {
+    let frameData = getFrameData(tabId, 0);
+    return (frameData && key in frameData ? frameData[key] : 0);
+  }
+  else
+    return (key in Prefs.stats_total ? Prefs.stats_total[key] : 0);
 };
 
-let total = new StatsObject(Prefs["stats_total"]);
-
-let Stats = exports.Stats = {
-  /**
-   * Total counts
-   */
-  get total()
-  {
-    return total;
-  }
-}
-
-FilterNotifier.addListener(function(action, item, tabId)
+FilterNotifier.addListener(function(action, item, newValue, oldValue, tabId)
 {
-  if (action != "document.stats")
+  if (action != "filter.hitCount")
     return;
   
   var blocked = item instanceof BlockingFilter;
   
-  // increment counts
+  // Increment counts
   if (blocked)
-    total.blocked++;
-  
-  ChromeCompat.tabs.sendMessage(tabId, {
-    reqtype: "update-stats",
-    blocked: blocked
-  }, null);
-  
-  Prefs["stats_total"] = total;
+  {
+    if ("blocked" in Prefs.stats_total)
+      Prefs.stats_total.blocked++;
+    else
+      Prefs.stats_total.blocked = 1;
+    
+    let frameData = getFrameData(tabId, 0);
+    if (frameData)
+    {
+      if ("blocked" in frameData)
+        frameData.blocked++;
+      else
+        frameData.blocked = 1;
+    }
+  }
 });