Adding in extension experiment for API changes

2016-12-20 17:54:30 +00:00
parent e286c2a3d4
commit 9092bd5fdf
6 changed files with 297 additions and 0 deletions
@@ -0,0 +1,111 @@
+"use strict";
+const {classes: Cc, interfaces: Ci, results: Cr, utils: Cu} = Components;
+
+Cu.import("resource://gre/modules/Services.jsm");
+XPCOMUtils.defineLazyModuleGetter(this, "ContextualIdentityService",
+                                  "resource://gre/modules/ContextualIdentityService.jsm");
+
+function convert(identity) {
+  let result = {
+    name: ContextualIdentityService.getUserContextLabel(identity.userContextId),
+    icon: identity.icon,
+    color: identity.color,
+    cookieStoreId: getCookieStoreIdForContainer(identity.userContextId),
+  };
+
+  return result;
+}
+
+class API extends ExtensionAPI {
+  getAPI(context) {
+    let self = {
+      contextualIdentities: {
+        get(cookieStoreId) {
+          let containerId = getContainerForCookieStoreId(cookieStoreId);
+          if (!containerId) {
+            return Promise.resolve(null);
+          }
+  
+          let identity = ContextualIdentityService.getIdentityFromId(containerId);
+          return Promise.resolve(convert(identity));
+        },
+  
+        query(details) {
+          let identities = [];
+          ContextualIdentityService.getIdentities().forEach(identity => {
+            if (details.name &&
+                ContextualIdentityService.getUserContextLabel(identity.userContextId) != details.name) {
+              return;
+            }
+  
+            identities.push(convert(identity));
+          });
+  
+          return Promise.resolve(identities);
+        },
+  
+        create(details) {
+          let identity = ContextualIdentityService.create(details.name,
+                                                          details.icon,
+                                                          details.color);
+          return Promise.resolve(convert(identity));
+        },
+  
+        update(cookieStoreId, details) {
+          let containerId = getContainerForCookieStoreId(cookieStoreId);
+          if (!containerId) {
+            return Promise.resolve(null);
+          }
+  
+          let identity = ContextualIdentityService.getIdentityFromId(containerId);
+          if (!identity) {
+            return Promise.resolve(null);
+          }
+  
+          if (details.name !== null) {
+            identity.name = details.name;
+          }
+  
+          if (details.color !== null) {
+            identity.color = details.color;
+          }
+  
+          if (details.icon !== null) {
+            identity.icon = details.icon;
+          }
+  
+          if (!ContextualIdentityService.update(identity.userContextId,
+                                                identity.name, identity.icon,
+                                                identity.color)) {
+            return Promise.resolve(null);
+          }
+  
+          return Promise.resolve(convert(identity));
+        },
+  
+        remove(cookieStoreId) {
+          let containerId = getContainerForCookieStoreId(cookieStoreId);
+          if (!containerId) {
+            return Promise.resolve(null);
+          }
+  
+          let identity = ContextualIdentityService.getIdentityFromId(containerId);
+          if (!identity) {
+            return Promise.resolve(null);
+          }
+  
+          // We have to create the identity object before removing it.
+          let convertedIdentity = convert(identity);
+  
+          if (!ContextualIdentityService.remove(identity.userContextId)) {
+            return Promise.resolve(null);
+          }
+  
+          return Promise.resolve(convertedIdentity);
+        },
+      },
+    };
+  
+    return self;
+  }
+}