]> git.argeo.org Git - lgpl/argeo-commons.git/blobdiff - server/runtime/org.argeo.server.jackrabbit/src/main/java/org/argeo/server/jackrabbit/JackrabbitContainer.java
Remote JCR client working
[lgpl/argeo-commons.git] / server / runtime / org.argeo.server.jackrabbit / src / main / java / org / argeo / server / jackrabbit / JackrabbitContainer.java
index f06163f34cfaaccc878cb4ecda9fbea3bca14fab..c828d703a7daa6bd2a9ae320cf474e4f07dd1187 100644 (file)
 
 package org.argeo.server.jackrabbit;
 
+import java.io.ByteArrayInputStream;
 import java.io.File;
 import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
+import java.util.Properties;
 
 import javax.jcr.Credentials;
 import javax.jcr.LoginException;
@@ -33,62 +38,115 @@ import org.apache.commons.io.FileUtils;
 import org.apache.commons.io.IOUtils;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
+import org.apache.jackrabbit.api.JackrabbitRepository;
+import org.apache.jackrabbit.commons.JcrUtils;
 import org.apache.jackrabbit.commons.NamespaceHelper;
+import org.apache.jackrabbit.commons.cnd.CndImporter;
+import org.apache.jackrabbit.core.RepositoryFactoryImpl;
 import org.apache.jackrabbit.core.RepositoryImpl;
 import org.apache.jackrabbit.core.TransientRepository;
 import org.apache.jackrabbit.core.config.RepositoryConfig;
+import org.apache.jackrabbit.core.config.RepositoryConfigurationParser;
+import org.apache.jackrabbit.jcr2dav.Jcr2davRepositoryFactory;
 import org.argeo.ArgeoException;
 import org.springframework.beans.factory.DisposableBean;
 import org.springframework.beans.factory.InitializingBean;
+import org.springframework.context.ResourceLoaderAware;
 import org.springframework.core.io.Resource;
+import org.springframework.core.io.ResourceLoader;
+import org.xml.sax.InputSource;
 
 /**
  * Wrapper around a Jackrabbit repository which allows to configure it in Spring
  * and expose it as a {@link Repository}.
  */
 public class JackrabbitContainer implements InitializingBean, DisposableBean,
-               Repository {
+               Repository, ResourceLoaderAware {
        private Log log = LogFactory.getLog(JackrabbitContainer.class);
 
        private Resource configuration;
        private File homeDirectory;
+       private Resource variables;
 
        private Boolean inMemory = false;
+       private String uri = null;
 
        private Repository repository;
 
+       private ResourceLoader resourceLoader;
+
+       /** Node type definitions in CND format */
+       private List<byte[]> cnds = new ArrayList<byte[]>();
+       private List<String> cndFiles = new ArrayList<String>();
+
        /** Namespaces to register: key is prefix, value namespace */
        private Map<String, String> namespaces = new HashMap<String, String>();
 
+       private Boolean autocreateWorkspaces = false;
+
        public void afterPropertiesSet() throws Exception {
-               if (inMemory && homeDirectory.exists()) {
-                       FileUtils.deleteDirectory(homeDirectory);
-                       log.warn("Deleted Jackrabbit home directory " + homeDirectory);
+               // Load cnds as resources
+               for (String resUrl : cndFiles) {
+                       Resource res = resourceLoader.getResource(resUrl);
+                       byte[] arr = IOUtils.toByteArray(res.getInputStream());
+                       cnds.add(arr);
                }
 
-               RepositoryConfig config;
-               InputStream in = configuration.getInputStream();
-               try {
-                       config = RepositoryConfig.create(in,
-                                       homeDirectory.getCanonicalPath());
-               } catch (Exception e) {
-                       throw new RuntimeException("Cannot read configuration", e);
-               } finally {
-                       IOUtils.closeQuietly(in);
-               }
+               if (uri != null) {
+                       // Thread.currentThread().setContextClassLoader(
+                       // getClass().getClassLoader());
+                       // repository = JcrUtils.getRepository(uri);
+                       Map<String, String> params = new HashMap<String, String>();
+                       params.put(JcrUtils.REPOSITORY_URI, uri);
+                       repository = new Jcr2davRepositoryFactory().getRepository(params);
+                       if (repository == null)
+                               throw new ArgeoException("Remote Davex repository " + uri
+                                               + " not found");
+                       log.info("Initialized Jackrabbit repository " + repository
+                                       + " from uri " + uri);
+               } else {
+                       if (inMemory && homeDirectory.exists()) {
+                               FileUtils.deleteDirectory(homeDirectory);
+                               log.warn("Deleted Jackrabbit home directory " + homeDirectory);
+                       }
 
-               if (inMemory)
-                       repository = new TransientRepository(config);
-               else
-                       repository = RepositoryImpl.create(config);
+                       RepositoryConfig config;
+                       InputStream in = configuration.getInputStream();
+                       InputStream propsIn = null;
+                       try {
+                               Properties vars = new Properties();
+                               if (variables != null) {
+                                       propsIn = variables.getInputStream();
+                                       vars.load(propsIn);
+                               }
+                               // override with system properties
+                               vars.putAll(System.getProperties());
+                               vars.put(
+                                               RepositoryConfigurationParser.REPOSITORY_HOME_VARIABLE,
+                                               homeDirectory.getCanonicalPath());
+                               config = RepositoryConfig.create(new InputSource(in), vars);
+                       } catch (Exception e) {
+                               throw new RuntimeException("Cannot read configuration", e);
+                       } finally {
+                               IOUtils.closeQuietly(in);
+                               IOUtils.closeQuietly(propsIn);
+                       }
+
+                       if (inMemory)
+                               repository = new TransientRepository(config);
+                       else
+                               repository = RepositoryImpl.create(config);
 
-               log.info("Initialized Jackrabbit repository " + repository + " in "
-                               + homeDirectory + " with config " + configuration);
+                       log.info("Initialized Jackrabbit repository " + repository + " in "
+                                       + homeDirectory + " with config " + configuration);
+               }
        }
 
        public void destroy() throws Exception {
                if (repository != null) {
-                       if (repository instanceof RepositoryImpl)
+                       if (repository instanceof JackrabbitRepository)
+                               ((JackrabbitRepository) repository).shutdown();
+                       else if (repository instanceof RepositoryImpl)
                                ((RepositoryImpl) repository).shutdown();
                        else if (repository instanceof TransientRepository)
                                ((TransientRepository) repository).shutdown();
@@ -101,8 +159,12 @@ public class JackrabbitContainer implements InitializingBean, DisposableBean,
                                        log.debug("Deleted Jackrabbit home directory "
                                                        + homeDirectory);
                        }
-               log.info("Destroyed Jackrabbit repository " + repository + " in "
-                               + homeDirectory + " with config " + configuration);
+
+               if (uri != null)
+                       log.info("Destroyed Jackrabbit repository with uri " + uri);
+               else
+                       log.info("Destroyed Jackrabbit repository " + repository + " in "
+                                       + homeDirectory + " with config " + configuration);
        }
 
        // JCR REPOSITORY (delegated)
@@ -123,7 +185,15 @@ public class JackrabbitContainer implements InitializingBean, DisposableBean,
        public Session login(Credentials credentials, String workspaceName)
                        throws LoginException, NoSuchWorkspaceException,
                        RepositoryException {
-               Session session = repository.login(credentials, workspaceName);
+               Session session;
+               try {
+                       session = repository.login(credentials, workspaceName);
+               } catch (NoSuchWorkspaceException e) {
+                       if (autocreateWorkspaces)
+                               session = createWorkspaceAndLogsIn(credentials, workspaceName);
+                       else
+                               throw e;
+               }
                processNewSession(session);
                return session;
        }
@@ -137,7 +207,15 @@ public class JackrabbitContainer implements InitializingBean, DisposableBean,
 
        public Session login(String workspaceName) throws LoginException,
                        NoSuchWorkspaceException, RepositoryException {
-               Session session = repository.login(workspaceName);
+               Session session;
+               try {
+                       session = repository.login(workspaceName);
+               } catch (NoSuchWorkspaceException e) {
+                       if (autocreateWorkspaces)
+                               session = createWorkspaceAndLogsIn(null, workspaceName);
+                       else
+                               throw e;
+               }
                processNewSession(session);
                return session;
        }
@@ -146,11 +224,33 @@ public class JackrabbitContainer implements InitializingBean, DisposableBean,
                try {
                        NamespaceHelper namespaceHelper = new NamespaceHelper(session);
                        namespaceHelper.registerNamespaces(namespaces);
-               } catch (RepositoryException e) {
+
+                       for (byte[] arr : cnds)
+                               CndImporter.registerNodeTypes(new InputStreamReader(
+                                               new ByteArrayInputStream(arr)), session, true);
+               } catch (Exception e) {
                        throw new ArgeoException("Cannot process new session", e);
                }
        }
 
+       /**
+        * Logs in to the default workspace, creates the required workspace, logs
+        * out, logs in to the required workspace.
+        */
+       protected Session createWorkspaceAndLogsIn(Credentials credentials,
+                       String workspaceName) throws RepositoryException {
+               if (workspaceName == null)
+                       throw new ArgeoException("No workspace specified.");
+               Session session = repository.login(credentials);
+               session.getWorkspace().createWorkspace(workspaceName);
+               session.logout();
+               return repository.login(credentials, workspaceName);
+       }
+
+       public void setResourceLoader(ResourceLoader resourceLoader) {
+               this.resourceLoader = resourceLoader;
+       }
+
        public boolean isStandardDescriptor(String key) {
                return repository.isStandardDescriptor(key);
        }
@@ -184,4 +284,16 @@ public class JackrabbitContainer implements InitializingBean, DisposableBean,
                this.namespaces = namespaces;
        }
 
+       public void setCndFiles(List<String> cndFiles) {
+               this.cndFiles = cndFiles;
+       }
+
+       public void setVariables(Resource variables) {
+               this.variables = variables;
+       }
+
+       public void setUri(String uri) {
+               this.uri = uri;
+       }
+
 }