"use strict";
const {classes: Cc, interfaces: Ci, results: Cr, utils: Cu} = Components;

Cu.import("resource://gre/modules/Services.jsm");
const {XPCOMUtils} = Cu.import("resource://gre/modules/XPCOMUtils.jsm");
const CONTAINER_STORE = "firefox-container-";

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;
}

function getCookieStoreIdForContainer(containerId) {
  return CONTAINER_STORE + containerId;
}

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;
  }
}