diff options
Diffstat (limited to 'mailnews/addrbook/src/nsAbLDAPAttributeMap.js')
| -rw-r--r-- | mailnews/addrbook/src/nsAbLDAPAttributeMap.js | 247 |
1 files changed, 247 insertions, 0 deletions
diff --git a/mailnews/addrbook/src/nsAbLDAPAttributeMap.js b/mailnews/addrbook/src/nsAbLDAPAttributeMap.js new file mode 100644 index 000000000..8ff2406d2 --- /dev/null +++ b/mailnews/addrbook/src/nsAbLDAPAttributeMap.js @@ -0,0 +1,247 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +Components.utils.import("resource://gre/modules/Services.jsm"); +Components.utils.import("resource://gre/modules/XPCOMUtils.jsm"); + +var NS_ABLDAPATTRIBUTEMAP_CID = Components.ID( + "{127b341a-bdda-4270-85e1-edff569a9b85}"); +var NS_ABLDAPATTRIBUTEMAPSERVICE_CID = Components.ID( + "{4ed7d5e1-8800-40da-9e78-c4f509d7ac5e}"); + +function nsAbLDAPAttributeMap() { + this.mPropertyMap = {}; + this.mAttrMap = {}; +} + +nsAbLDAPAttributeMap.prototype = { + classID: NS_ABLDAPATTRIBUTEMAP_CID, + + getAttributeList: function getAttributeList(aProperty) { + + if (!(aProperty in this.mPropertyMap)) { + return null; + } + + // return the joined list + return this.mPropertyMap[aProperty].join(","); + }, + + getAttributes: function getAttributes(aProperty, aCount, aAttrs) { + + // fail if no entry for this + if (!(aProperty in this.mPropertyMap)) { + throw Components.results.NS_ERROR_FAILURE; + } + + aAttrs = this.mPropertyMap[aProperty]; + aCount = aAttrs.length; + return aAttrs; + }, + + getFirstAttribute: function getFirstAttribute(aProperty) { + + // fail if no entry for this + if (!(aProperty in this.mPropertyMap)) { + return null; + } + + return this.mPropertyMap[aProperty][0]; + }, + + setAttributeList: function setAttributeList(aProperty, aAttributeList, + aAllowInconsistencies) { + + var attrs = aAttributeList.split(","); + + // check to make sure this call won't allow multiple mappings to be + // created, if requested + if (!aAllowInconsistencies) { + for (var attr of attrs) { + if (attr in this.mAttrMap && this.mAttrMap[attr] != aProperty) { + throw Components.results.NS_ERROR_FAILURE; + } + } + } + + // delete any attr mappings created by the existing property map entry + if (aProperty in this.mPropertyMap) { + for (attr of this.mPropertyMap[aProperty]) { + delete this.mAttrMap[attr]; + } + } + + // add these attrs to the attrmap + for (attr of attrs) { + this.mAttrMap[attr] = aProperty; + } + + // add them to the property map + this.mPropertyMap[aProperty] = attrs; + }, + + getProperty: function getProperty(aAttribute) { + + if (!(aAttribute in this.mAttrMap)) { + return null; + } + + return this.mAttrMap[aAttribute]; + }, + + getAllCardAttributes: function getAllCardAttributes() { + var attrs = []; + for (var prop in this.mPropertyMap) { + let attrArray = this.mPropertyMap[prop]; + attrs = attrs.concat(attrArray); + } + + if (!attrs.length) { + throw Components.results.NS_ERROR_FAILURE; + } + + return attrs.join(","); + }, + + getAllCardProperties: function getAllCardProperties(aCount) { + + var props = []; + for (var prop in this.mPropertyMap) { + props.push(prop); + } + + aCount.value = props.length; + return props; + }, + + setFromPrefs: function setFromPrefs(aPrefBranchName) { + // get the right pref branch + let branch = Services.prefs.getBranch(aPrefBranchName + "."); + + // get the list of children + var childCount = {}; + var children = branch.getChildList("", childCount); + + // do the actual sets + for (var child of children) { + this.setAttributeList(child, branch.getCharPref(child), true); + } + + // ensure that everything is kosher + this.checkState(); + }, + + setCardPropertiesFromLDAPMessage: function + setCardPropertiesFromLDAPMessage(aMessage, aCard) { + + var cardValueWasSet = false; + + var msgAttrCount = {}; + var msgAttrs = aMessage.getAttributes(msgAttrCount); + + // downcase the array for comparison + function toLower(a) { return a.toLowerCase(); } + msgAttrs = msgAttrs.map(toLower); + + // deal with each addressbook property + for (var prop in this.mPropertyMap) { + + // go through the list of possible attrs in precedence order + for (var attr of this.mPropertyMap[prop]) { + + attr = attr.toLowerCase(); + + // find the first attr that exists in this message + if (msgAttrs.indexOf(attr) != -1) { + + try { + var values = aMessage.getValues(attr, {}); + // strip out the optional label from the labeledURI + if (attr == "labeleduri" && values[0]) { + var index = values[0].indexOf(" "); + if (index != -1) + values[0] = values[0].substring(0, index); + } + aCard.setProperty(prop, values[0]); + + cardValueWasSet = true; + break; + } catch (ex) { + // ignore any errors getting message values or setting card values + } + } + } + } + + if (!cardValueWasSet) { + throw Components.results.NS_ERROR_FAILURE; + } + + return; + }, + + checkState: function checkState() { + + var attrsSeen = []; + + for (var prop in this.mPropertyMap) { + let attrArray = this.mPropertyMap[prop]; + for (var attr of attrArray) { + + // multiple attributes that mapped to the empty string are permitted + if (!attr.length) { + continue; + } + + // if we've seen this before, there's a problem + if (attrsSeen.indexOf(attr) != -1) { + throw Components.results.NS_ERROR_FAILURE; + } + + // remember that we've seen it now + attrsSeen.push(attr); + } + } + + return; + }, + + QueryInterface: XPCOMUtils + .generateQI([Components.interfaces.nsIAbLDAPAttributeMap]) +} + +function nsAbLDAPAttributeMapService() { +} + +nsAbLDAPAttributeMapService.prototype = { + + classID: NS_ABLDAPATTRIBUTEMAPSERVICE_CID, + + mAttrMaps: {}, + + getMapForPrefBranch: function getMapForPrefBranch(aPrefBranchName) { + + // if we've already got this map, return it + if (aPrefBranchName in this.mAttrMaps) { + return this.mAttrMaps[aPrefBranchName]; + } + + // otherwise, try and create it + var attrMap = new nsAbLDAPAttributeMap(); + attrMap.setFromPrefs("ldap_2.servers.default.attrmap"); + attrMap.setFromPrefs(aPrefBranchName + ".attrmap"); + + // cache + this.mAttrMaps[aPrefBranchName] = attrMap; + + // and return + return attrMap; + }, + + QueryInterface: XPCOMUtils + .generateQI([Components.interfaces.nsIAbLDAPAttributeMapService]) +} + +var NSGetFactory = XPCOMUtils.generateNSGetFactory([nsAbLDAPAttributeMap, nsAbLDAPAttributeMapService]); + |