package org.alfresco.repo.preference.script; import java.io.Serializable; import java.util.HashMap; import java.util.Map; import org.alfresco.repo.jscript.BaseScopableProcessorExtension; import org.alfresco.service.ServiceRegistry; import org.alfresco.service.cmr.preference.PreferenceService; import org.alfresco.service.transaction.TransactionService; import org.mozilla.javascript.NativeObject; /** * @author Roy Wetherall */ public class ScriptPreferenceService extends BaseScopableProcessorExtension { @SuppressWarnings("unused") private ServiceRegistry services; /** Preference Service */ private PreferenceService preferenceService; private TransactionService transactionService; public void setTransactionService(TransactionService transactionService) { this.transactionService = transactionService; } public void setServiceRegistry(ServiceRegistry services) { this.services = services; } public void setPreferenceService(PreferenceService preferenceService) { this.preferenceService = preferenceService; } public boolean getAllowWrite() { return transactionService.getAllowWrite(); } public NativeObject getPreferences(String userName) { return getPreferences(userName, null); } public NativeObject getPreferences(String userName, String preferenceFilter) { // It's a tad unusual to return a NativeObject like this - at least within Alfresco. // But we can't change it to e.g. a ScriptableHashMap as the API is published. Map prefs = this.preferenceService.getPreferences(userName, preferenceFilter); NativeObject result = new NativeObjectDV(); for (Map.Entry entry : prefs.entrySet()) { String key = entry.getKey(); String[] keys; int colonIndex = key.indexOf(":"); if (colonIndex > -1) { keys = key.substring(0, colonIndex).replace(".", "+").split("\\+"); keys[keys.length - 1] = keys[keys.length - 1].concat(key.substring(colonIndex)); } else { keys = key.replace(".", "+").split("\\+"); } setPrefValue(keys, entry.getValue(), result); } return result; } /** * This extension of NativeObject adds a default value. See ALF-20023 for some background. */ private static class NativeObjectDV extends NativeObject { private static final long serialVersionUID = 1L; @Override public Object getDefaultValue(@SuppressWarnings("rawtypes") Class typeHint) { return toString(); } } private void setPrefValue(String[] keys, Serializable value, NativeObject object) { NativeObject currentObject = object; int index = 0; for (String key : keys) { if (index == keys.length-1) { currentObject.put(key, currentObject, value); } else { NativeObject newObject = null; Object temp = currentObject.get(key, currentObject); if (temp == null || temp instanceof NativeObject == false) { newObject = new NativeObjectDV(); currentObject.put(key, currentObject, newObject); } else { newObject = (NativeObject)temp; } currentObject = newObject; } index ++; } } public void setPreferences(String userName, NativeObject preferences) { Map values = new HashMap(10); getPrefValues(preferences, null, values); this.preferenceService.setPreferences(userName, values); } private void getPrefValues(NativeObject currentObject, String currentKey, Map values) { Object[] ids = currentObject.getIds(); for (Object id : ids) { String key = getAppendedKey(currentKey, id.toString()); Object value = currentObject.get(id.toString(), currentObject); if (value instanceof NativeObject) { getPrefValues((NativeObject)value, key, values); } else { values.put(key, (Serializable)value); } } } public void clearPreferences(String userName) { this.preferenceService.clearPreferences(userName, null); } /** * Clear the preference values * * @param userName String * @param preferenceFilter String */ public void clearPreferences(String userName, String preferenceFilter) { this.preferenceService.clearPreferences(userName, preferenceFilter); } private String getAppendedKey(String currentKey, String key) { StringBuffer buffer = new StringBuffer(64); if (currentKey != null && currentKey.length() != 0) { buffer.append(currentKey).append(".").append(key); } else { buffer.append(key); } return buffer.toString(); } }