/* ***** BEGIN LICENSE BLOCK *****

  * Version: MPL 1.1

  *

  * The contents of this file are subject to the Mozilla Public License Version

  * 1.1 (the "License"); you may not use this file except in compliance with

  * the License. You may obtain a copy of the License at

  * http://www.mozilla.org/MPL/

  *

  * Software distributed under the License is distributed on an "AS IS" basis,

  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License

  * for the specific language governing rights and limitations under the

  * License.

  *

  * The Original Code is Weave Sync for Adblock Plus.

  *

  * The Initial Developer of the Original Code is

  * Wladimir Palant.

  * Portions created by the Initial Developer are Copyright (C) 2009

  * the Initial Developer. All Rights Reserved.

  *

  * Contributor(s):

  *

  * ***** END LICENSE BLOCK ***** */

 /*

  * This file is loaded by Startup.js component.

  */

 var abp = Components.classes["@mozilla.org/adblockplus;1"].createInstance().wrappedJSObject;

 Components.utils.import("resource://weave/engines.js");

 Components.utils.import("resource://weave/stores.js");

 Components.utils.import("resource://weave/trackers.js");

 Components.utils.import("resource://weave/base_records/crypto.js");

 Components.utils.import("resource://weave/async.js");

 Components.utils.import("resource://weave/util.js");

 Function.prototype.async = Async.sugar;

 function ABPRecord(uri)

 {

   CryptoWrapper.call(this, uri);

   this.cleartext = {};

 }

 ABPRecord.prototype =

 {

   __proto__: CryptoWrapper.prototype,

   _logname: "Record.AdblockPlus",

   _source: null,

   get source() this._source,

   set source(source)

   {

     this._source = source;

     if (source instanceof abp.Filter)

     {

       this.cleartext.type = "filter";

       this.cleartext.text = source.text;

       if (source instanceof abp.ActiveFilter)

         this.cleartext.disabled = source.disabled;

     }

     else if (source instanceof abp.Subscription)

     {

       this.cleartext.type = "subscription";

       this.cleartext.url = source.url;

       this.cleartext.disabled = source.disabled;

       if (source instanceof abp.RegularSubscription)

         this.cleartext.title = source.title;

       if (source instanceof abp.DownloadableSubscription)

         this.cleartext.autoDownload = source.autoDownload;

     }

     else if (typeof source == "string")

     {

       this.cleartext.type = "pref";

       this.cleartext.name = source;

       this.cleartext.value = abp.prefs[source];

     }

     else

     {

       this.deleted = true;

       this.payload = null;  // Weave 0.3 compatibility

     }

   },

   _saveBack: function()

   {

     switch (this.cleartext.type)

     {

       case "filter":

         let filter = abp.Filter.fromText(this.cleartext.text);

         if (!filter)

           break;

         if (filter instanceof abp.ActiveFilter && this.cleartext.disabled != filter.disabled)

         {

           filter.disabled = this.cleartext.disabled;

           abp.filterStorage.triggerFilterObservers(filter.disabled ? "disable" : "enable", [filter]);

         }

         this._source = filter;

         break;

       case "subscription":

         let subscription = abp.Subscription.fromURL(this.cleartext.url);

         if (!subscription)

           break;

         if (this.cleartext.disabled != subscription.disabled)

         {

           subscription.disabled = this.cleartext.disabled;

           abp.filterStorage.triggerSubscriptionObservers(subscription.disabled ? "disable" : "enable", [subscription]);

         }

         let changed = false;

         if (subscription instanceof abp.RegularSubscription && this.cleartext.title != subscription.title)

         {

           subscription.title = this.cleartext.title;

           changed = true;

         }

         if (subscription instanceof abp.DownloadableSubscription && this.cleartext.autoDownload != subscription.autoDownload)

         {

           subscription.autoDownload = this.cleartext.autoDownload;

           changed = true;

         }

         if (changed)

           abp.filterStorage.triggerSubscriptionObservers("updateinfo", [subscription]);

         this._source = subscription;

         break;

       case "pref":

         let prefName = this.cleartext.name;

         if (this.cleartext.value != abp.prefs[prefName])

         {

           abp.prefs[prefName] = this.cleartext.value;

           abp.prefs.save();

         }

         this._source = prefName;

         break;

     }

   }

 };

 function ABPStore()

 {

   Store.call(this);

 }

 ABPStore.prototype =

 {

   __proto__: Store.prototype,

   _logName: "AdblockPlus",

   _abpItems: null,

   createRecord: function(id, cryptoMetaURL)

   {

     let record = this.cache.get(id);

     if (record)

       return record;

     record = new ABPRecord();

     record.id = id;

     record.encryption = cryptoMetaURL;

     if (id in this._abpItems)

       record.source = this._abpItems[id];

     else

     {

       record.deleted = true;

       record.payload = null;  // Weave 0.3 compatibility

     }

     this.cache.put(id, record);

     return record;

   },

   itemExists: function(id)

   {

     return (id in this._abpItems);

   },

   // This method shouldn't be called because Engine._recordLike() always returns false

   changeItemID: function(oldId, newId) {},

   getAllIDs: function()

   {

     let result = {};

     for each (let subscription in abp.filterStorage.subscriptions)

     {

       if (!(subscription instanceof abp.ExternalSubscription))

         result[Utils.sha1("subscription " + subscription.url)] = subscription;

       if (subscription instanceof abp.SpecialSubscription)

       {

         for each (let filter in subscription.filters)

           result[Utils.sha1("filter " + filter.text)] = filter;

       }

     }

     for each (let [prefName, prefType, defaultValue] in abp.prefs.prefList)

       if (prefName != "currentVersion")

         result[Utils.sha1("pref " + prefName)] = prefName;

     return result;

   },

   wipe: function()

   {

     this._log.debug("Wiping all client data");

     let subscriptions = abp.filterStorage.subscriptions.slice();

     for each (let subscription in subscriptions)

     {

       if (subscription instanceof abp.SpecialSubscription)

         abp.filterStorage.updateSubscriptionFilters(subscription, []);

       else if (!(subscription instanceof abp.ExternalSubscription))

         abp.filterStorage.removeSubscription(subscription);

     }

     for each (let [prefName, prefType, defaultValue] in abp.prefs.prefList)

     {

       if (prefName != "currentVersion")

         abp.prefs[prefName] = defaultValue;

     }

     abp.prefs.save();

   },

   create: function(record)

   {

     this._log.debug("Got create command for " + record.id);

     record._saveBack();

     if (record.source instanceof abp.Filter)

       abp.filterStorage.addFilter(record.source);

     else if (record.source instanceof abp.DownloadableSubscription)

     {

       abp.filterStorage.addSubscription(record.source);

       if (!record.source.lastDownload)

         synchronizer.execute(record.source);

     }

   },

   update: function(record)

   {

     this._log.debug("Got update command for " + record.id);

     record._saveBack();

   },

   remove: function(record)

   {

     this._log.debug("Got remove command for " + record.id);

     if (record.id in this._abpItems)

     {

       let item = this._abpItems[record.id];

       if (item instanceof abp.Filter)

         abp.filterStorage.removeFilter(item);

       else if (item instanceof abp.DownloadableSubscription)

         abp.filterStorage.removeSubscription(item);

     }

   },

   cacheABPItems: function()

   {

     this._log.debug("Caching all ABP items");

     this._abpItems = this.getAllIDs();

   },

   clearABPCache: function()

   {

     this._log.debug("Clearing ABP item cache");

     this._abpItems = null;

   }

 };

 function ABPTracker()

 {

   Tracker.call(this);

   let me = this;

   abp.prefs.addListener(function() { me.onPrefChange.apply(me, arguments); });

   abp.filterStorage.addSubscriptionObserver(function() { me.onSubscriptionChange.apply(me, arguments); });

   abp.filterStorage.addFilterObserver(function() { me.onFilterChange.apply(me, arguments); });

 }

 ABPTracker.prototype =

 {

   __proto__: Tracker.prototype,

   _logName: "Tracker.AdblockPlus",

   file: "abpdata",

   onPrefChange: function()

   {

     for each (let [prefName, prefType, defaultValue] in abp.prefs.prefList)

       if (prefName != "currentVersion")

         this.addChangedID(Utils.sha1("pref " + prefName));

     this._score += 20;

   },

   onSubscriptionChange: function(action, subscriptions)

   {

     for each (let subscription in subscriptions)

     {

       if (!(subscription instanceof abp.ExternalSubscription))

         this.addChangedID(Utils.sha1("subscription " + subscription.url));

       if (subscription instanceof abp.SpecialSubscription && action == "update")

       {

         let oldFilters = {__proto__: null};

         for each (let filter in subscription.oldFilters)

           oldFilters[filter.text] = true;

         let newFilters = {__proto__: null};

         for each (let filter in subscription.filters)

           newFilters[filter.text] = true;

         for (let text in oldFilters)

           if (!(text in newFilters))

             this.addChangedID(Utils.sha1("filter " + text));

         for (let text in newFilters)

           if (!(text in oldFilters))

             this.addChangedID(Utils.sha1("filter " + text));

       }

     }

     this._score += 5;

   },

   onFilterChange: function(action, filters)

   {

     if (action == "hit")

       return;

     for each (let filter in filters)

       this.addChangedID(Utils.sha1("filter " + filter.text));

     this._score += 10;

   }

 };

 function ABPEngine()

 {

   SyncEngine.call(this);

 }

 ABPEngine.prototype =

 {

   __proto__: SyncEngine.prototype,

   name: "abpdata",

   displayName: Cc["@mozilla.org/intl/stringbundle;1"]

                 .getService(Ci.nsIStringBundleService)

                 .createBundle("chrome://abpweavesync/locale/global.properties")

                 .GetStringFromName("dataProvider_name"),

   logName: "AdblockPlus",

   _storeObj: ABPStore,

   _trackerObj: ABPTracker,

   _recordObj: ABPRecord,

   _recordLike: function() false,

   // Work-around for https://bugzilla.mozilla.org/show_bug.cgi?id=493256

   _isEqual: function(item)

   {

     if (item.deleted || item.payload === null /* Weave 0.3 compatibility */)

       return false;

     return this.__proto__.__proto__._isEqual.apply(this, arguments);

   },

   _syncStartup: function()

   {

     let self = yield;

     this._store.cacheABPItems();

     yield SyncEngine.prototype._syncStartup.async(this, self.cb);

   },

   _syncFinish: function()

   {

     let self = yield;

     this._store.clearABPCache();

     yield SyncEngine.prototype._syncFinish.async(this, self.cb);

   }

 };

 if ("isArray" in Utils)

   Engines.register(ABPEngine);

 else

   Engines.register(new ABPEngine());  /* Weave 0.3 compatibility */