org.argeo.cms/src/org/argeo/cms/internal/kernel/DeployConfig.java

   1 package org.argeo.cms.internal.kernel;
   2
   3 import java.io.IOException;
   4 import java.io.InputStream;
   5 import java.io.Writer;
   6 import java.nio.file.Files;
   7 import java.nio.file.Path;
   8 import java.util.ArrayList;
   9 import java.util.Dictionary;
  10 import java.util.List;
  11 import java.util.SortedMap;
  12 import java.util.TreeMap;
  13
  14 import javax.naming.InvalidNameException;
  15 import javax.naming.directory.Attributes;
  16 import javax.naming.directory.BasicAttributes;
  17 import javax.naming.ldap.LdapName;
  18 import javax.naming.ldap.Rdn;
  19
  20 import org.apache.commons.logging.Log;
  21 import org.apache.commons.logging.LogFactory;
  22 import org.argeo.cms.CmsException;
  23 import org.argeo.naming.AttributesDictionary;
  24 import org.argeo.naming.LdifParser;
  25 import org.argeo.naming.LdifWriter;
  26 import org.argeo.node.NodeConstants;
  27 import org.argeo.osgi.useradmin.UserAdminConf;
  28 import org.osgi.framework.BundleContext;
  29 import org.osgi.framework.FrameworkUtil;
  30 import org.osgi.service.cm.Configuration;
  31 import org.osgi.service.cm.ConfigurationAdmin;
  32 import org.osgi.service.cm.ConfigurationEvent;
  33 import org.osgi.service.cm.ConfigurationListener;
  34
  35 class DeployConfig implements ConfigurationListener {
  36         private final Log log = LogFactory.getLog(getClass());
  37         private final BundleContext bc = FrameworkUtil.getBundle(getClass()).getBundleContext();
  38
  39         private static Path deployConfigPath = KernelUtils.getOsgiInstancePath(KernelConstants.DEPLOY_CONFIG_PATH);
  40         private SortedMap<LdapName, Attributes> deployConfigs = new TreeMap<>();
  41         private final DataModels dataModels;
  42
  43         public DeployConfig(ConfigurationAdmin configurationAdmin, DataModels dataModels, boolean isClean) {
  44                 this.dataModels = dataModels;
  45                 // ConfigurationAdmin configurationAdmin =
  46                 // bc.getService(bc.getServiceReference(ConfigurationAdmin.class));
  47                 try {
  48                         boolean isFirstInit = false;
  49                         if (!isInitialized()) { // first init
  50                                 isFirstInit = true;
  51                                 firstInit();
  52                         }
  53                         init(configurationAdmin, isClean, isFirstInit);
  54                 } catch (IOException e) {
  55                         throw new CmsException("Could not init deploy configs", e);
  56                 }
  57                 // FIXME check race conditions during initialization
  58                 // bc.registerService(ConfigurationListener.class, this, null);
  59         }
  60
  61         private void firstInit() throws IOException {
  62                 log.info("## FIRST INIT ##");
  63                 Files.createDirectories(deployConfigPath.getParent());
  64
  65                 // FirstInit firstInit = new FirstInit();
  66                 InitUtils.prepareFirstInitInstanceArea();
  67
  68                 if (!Files.exists(deployConfigPath))
  69                         deployConfigs = new TreeMap<>();
  70                 else// config file could have juste been copied by preparation
  71                         try (InputStream in = Files.newInputStream(deployConfigPath)) {
  72                                 deployConfigs = new LdifParser().read(in);
  73                         }
  74                 save();
  75         }
  76
  77         private void setFromFrameworkProperties(boolean isFirstInit) {
  78                 // node repository
  79                 Dictionary<String, Object> nodeConfig = InitUtils
  80                                 .getNodeRepositoryConfig(getProps(NodeConstants.NODE_REPOS_FACTORY_PID, NodeConstants.NODE));
  81                 // node repository is mandatory
  82                 putFactoryDeployConfig(NodeConstants.NODE_REPOS_FACTORY_PID, nodeConfig);
  83
  84                 // additional repositories
  85                 dataModels: for (DataModels.DataModel dataModel : dataModels.getNonAbstractDataModels()) {
  86                         if (NodeConstants.NODE.equals(dataModel.getName()))
  87                                 continue dataModels;
  88                         Dictionary<String, Object> config = InitUtils.getRepositoryConfig(dataModel.getName(),
  89                                         getProps(NodeConstants.NODE_REPOS_FACTORY_PID, dataModel.getName()));
  90                         if (config.size() != 0)
  91                                 putFactoryDeployConfig(NodeConstants.NODE_REPOS_FACTORY_PID, config);
  92                 }
  93
  94                 // user admin
  95                 List<Dictionary<String, Object>> userDirectoryConfigs = InitUtils.getUserDirectoryConfigs();
  96                 if (userDirectoryConfigs.size() != 0) {
  97                         List<String> activeCns = new ArrayList<>();
  98                         for (int i = 0; i < userDirectoryConfigs.size(); i++) {
  99                                 Dictionary<String, Object> userDirectoryConfig = userDirectoryConfigs.get(i);
 100                                 String cn = UserAdminConf.baseDnHash(userDirectoryConfig);
 101                                 activeCns.add(cn);
 102                                 userDirectoryConfig.put(NodeConstants.CN, cn);
 103                                 putFactoryDeployConfig(NodeConstants.NODE_USER_ADMIN_PID, userDirectoryConfig);
 104                         }
 105                         // disable others
 106                         LdapName userAdminFactoryName = serviceFactoryDn(NodeConstants.NODE_USER_ADMIN_PID);
 107                         for (LdapName name : deployConfigs.keySet()) {
 108                                 if (name.startsWith(userAdminFactoryName) && !name.equals(userAdminFactoryName)) {
 109                                         try {
 110                                                 Attributes attrs = deployConfigs.get(name);
 111                                                 String cn = name.getRdn(name.size() - 1).getValue().toString();
 112                                                 if (!activeCns.contains(cn)) {
 113                                                         attrs.put(UserAdminConf.disabled.name(), "true");
 114                                                 }
 115                                         } catch (Exception e) {
 116                                                 throw new CmsException("Cannot disable user directory " + name, e);
 117                                         }
 118                                 }
 119                         }
 120                 }
 121
 122                 // http server
 123                 Dictionary<String, Object> webServerConfig = InitUtils
 124                                 .getHttpServerConfig(getProps(KernelConstants.JETTY_FACTORY_PID, NodeConstants.DEFAULT));
 125                 if (!webServerConfig.isEmpty())
 126                         putFactoryDeployConfig(KernelConstants.JETTY_FACTORY_PID, webServerConfig);
 127
 128                 save();
 129         }
 130
 131         private void init(ConfigurationAdmin configurationAdmin, boolean isClean, boolean isFirstInit) throws IOException {
 132
 133                 try (InputStream in = Files.newInputStream(deployConfigPath)) {
 134                         deployConfigs = new LdifParser().read(in);
 135                 }
 136                 if (isClean) {
 137                         setFromFrameworkProperties(isFirstInit);
 138                         for (LdapName dn : deployConfigs.keySet()) {
 139                                 Rdn lastRdn = dn.getRdn(dn.size() - 1);
 140                                 LdapName prefix = (LdapName) dn.getPrefix(dn.size() - 1);
 141                                 if (prefix.toString().equals(NodeConstants.DEPLOY_BASEDN)) {
 142                                         if (lastRdn.getType().equals(NodeConstants.CN)) {
 143                                                 // service
 144                                                 String pid = lastRdn.getValue().toString();
 145                                                 Configuration conf = configurationAdmin.getConfiguration(pid);
 146                                                 AttributesDictionary dico = new AttributesDictionary(deployConfigs.get(dn));
 147                                                 conf.update(dico);
 148                                         } else {
 149                                                 // service factory definition
 150                                         }
 151                                 } else {
 152                                         // service factory service
 153                                         Rdn beforeLastRdn = dn.getRdn(dn.size() - 2);
 154                                         assert beforeLastRdn.getType().equals(NodeConstants.OU);
 155                                         String factoryPid = beforeLastRdn.getValue().toString();
 156                                         Configuration conf = configurationAdmin.createFactoryConfiguration(factoryPid.toString(), null);
 157                                         AttributesDictionary dico = new AttributesDictionary(deployConfigs.get(dn));
 158                                         conf.update(dico);
 159                                 }
 160                         }
 161                 }
 162                 // TODO check consistency if not clean
 163         }
 164
 165         @Override
 166         public void configurationEvent(ConfigurationEvent event) {
 167                 try {
 168                         if (ConfigurationEvent.CM_UPDATED == event.getType()) {
 169                                 ConfigurationAdmin configurationAdmin = bc.getService(event.getReference());
 170                                 Configuration conf = configurationAdmin.getConfiguration(event.getPid(), null);
 171                                 LdapName serviceDn = null;
 172                                 String factoryPid = conf.getFactoryPid();
 173                                 if (factoryPid != null) {
 174                                         LdapName serviceFactoryDn = serviceFactoryDn(factoryPid);
 175                                         if (deployConfigs.containsKey(serviceFactoryDn)) {
 176                                                 for (LdapName dn : deployConfigs.keySet()) {
 177                                                         if (dn.startsWith(serviceFactoryDn)) {
 178                                                                 Rdn lastRdn = dn.getRdn(dn.size() - 1);
 179                                                                 assert lastRdn.getType().equals(NodeConstants.CN);
 180                                                                 Object value = conf.getProperties().get(lastRdn.getType());
 181                                                                 assert value != null;
 182                                                                 if (value.equals(lastRdn.getValue())) {
 183                                                                         serviceDn = dn;
 184                                                                         break;
 185                                                                 }
 186                                                         }
 187                                                 }
 188
 189                                                 Object cn = conf.getProperties().get(NodeConstants.CN);
 190                                                 if (cn == null)
 191                                                         throw new IllegalArgumentException("Properties must contain cn");
 192                                                 if (serviceDn == null) {
 193                                                         putFactoryDeployConfig(factoryPid, conf.getProperties());
 194                                                 } else {
 195                                                         Attributes attrs = deployConfigs.get(serviceDn);
 196                                                         assert attrs != null;
 197                                                         AttributesDictionary.copy(conf.getProperties(), attrs);
 198                                                 }
 199                                                 save();
 200                                                 if (log.isDebugEnabled())
 201                                                         log.debug("Updated deploy config " + serviceDn(factoryPid, cn.toString()));
 202                                         } else {
 203                                                 // ignore non config-registered service factories
 204                                         }
 205                                 } else {
 206                                         serviceDn = serviceDn(event.getPid());
 207                                         if (deployConfigs.containsKey(serviceDn)) {
 208                                                 Attributes attrs = deployConfigs.get(serviceDn);
 209                                                 assert attrs != null;
 210                                                 AttributesDictionary.copy(conf.getProperties(), attrs);
 211                                                 save();
 212                                                 if (log.isDebugEnabled())
 213                                                         log.debug("Updated deploy config " + serviceDn);
 214                                         } else {
 215                                                 // ignore non config-registered services
 216                                         }
 217                                 }
 218                         }
 219                 } catch (Exception e) {
 220                         log.error("Could not handle configuration event", e);
 221                 }
 222         }
 223
 224         void putFactoryDeployConfig(String factoryPid, Dictionary<String, Object> props) {
 225                 Object cn = props.get(NodeConstants.CN);
 226                 if (cn == null)
 227                         throw new IllegalArgumentException("cn must be set in properties");
 228                 LdapName serviceFactoryDn = serviceFactoryDn(factoryPid);
 229                 if (!deployConfigs.containsKey(serviceFactoryDn))
 230                         deployConfigs.put(serviceFactoryDn, new BasicAttributes(NodeConstants.OU, factoryPid));
 231                 LdapName serviceDn = serviceDn(factoryPid, cn.toString());
 232                 Attributes attrs = new BasicAttributes();
 233                 AttributesDictionary.copy(props, attrs);
 234                 deployConfigs.put(serviceDn, attrs);
 235         }
 236
 237         void putDeployConfig(String servicePid, Dictionary<String, Object> props) {
 238                 LdapName serviceDn = serviceDn(servicePid);
 239                 Attributes attrs = new BasicAttributes(NodeConstants.CN, servicePid);
 240                 AttributesDictionary.copy(props, attrs);
 241                 deployConfigs.put(serviceDn, attrs);
 242         }
 243
 244         void save() {
 245                 try (Writer writer = Files.newBufferedWriter(deployConfigPath)) {
 246                         new LdifWriter(writer).write(deployConfigs);
 247                 } catch (IOException e) {
 248                         throw new CmsException("Cannot save deploy configs", e);
 249                 }
 250         }
 251
 252         boolean isStandalone(String dataModelName) {
 253                 return getProps(NodeConstants.NODE_REPOS_FACTORY_PID, dataModelName) != null;
 254         }
 255
 256         /*
 257          * UTILITIES
 258          */
 259         private LdapName serviceFactoryDn(String factoryPid) {
 260                 try {
 261                         return new LdapName(NodeConstants.OU + "=" + factoryPid + "," + NodeConstants.DEPLOY_BASEDN);
 262                 } catch (InvalidNameException e) {
 263                         throw new IllegalArgumentException("Cannot generate DN from " + factoryPid, e);
 264                 }
 265         }
 266
 267         private LdapName serviceDn(String servicePid) {
 268                 try {
 269                         return new LdapName(NodeConstants.CN + "=" + servicePid + "," + NodeConstants.DEPLOY_BASEDN);
 270                 } catch (InvalidNameException e) {
 271                         throw new IllegalArgumentException("Cannot generate DN from " + servicePid, e);
 272                 }
 273         }
 274
 275         private LdapName serviceDn(String factoryPid, String cn) {
 276                 try {
 277                         return (LdapName) serviceFactoryDn(factoryPid).add(new Rdn(NodeConstants.CN, cn));
 278                 } catch (InvalidNameException e) {
 279                         throw new IllegalArgumentException("Cannot generate DN from " + factoryPid + " and " + cn, e);
 280                 }
 281         }
 282
 283         Dictionary<String, Object> getProps(String factoryPid, String cn) {
 284                 Attributes attrs = deployConfigs.get(serviceDn(factoryPid, cn));
 285                 if (attrs != null)
 286                         return new AttributesDictionary(attrs);
 287                 else
 288                         return null;
 289         }
 290
 291         static boolean isInitialized() {
 292                 return Files.exists(deployConfigPath);
 293         }
 294
 295 }