org.argeo.cms.jcr/src/org/argeo/cms/jcr/acr/JcrContentProvider.java

   1 package org.argeo.cms.jcr.acr;
   2
   3 import java.util.Arrays;
   4 import java.util.Collections;
   5 import java.util.HashMap;
   6 import java.util.Iterator;
   7 import java.util.Map;
   8 import java.util.Objects;
   9 import java.util.Spliterator;
  10 import java.util.function.Consumer;
  11 import java.util.function.Function;
  12
  13 import javax.jcr.Node;
  14 import javax.jcr.NodeIterator;
  15 import javax.jcr.Repository;
  16 import javax.jcr.RepositoryException;
  17 import javax.jcr.Session;
  18 import javax.jcr.query.Query;
  19 import javax.jcr.query.QueryResult;
  20 import javax.xml.namespace.NamespaceContext;
  21
  22 import org.argeo.api.acr.Content;
  23 import org.argeo.api.acr.search.BasicSearch;
  24 import org.argeo.api.acr.spi.ContentProvider;
  25 import org.argeo.api.acr.spi.ProvidedContent;
  26 import org.argeo.api.acr.spi.ProvidedSession;
  27 import org.argeo.api.cms.CmsConstants;
  28 import org.argeo.cms.acr.ContentUtils;
  29 import org.argeo.cms.jcr.CmsJcrUtils;
  30 import org.argeo.jcr.JcrException;
  31 import org.argeo.jcr.JcrUtils;
  32
  33 /** A JCR workspace accessed as an {@link ContentProvider}. */
  34 public class JcrContentProvider implements ContentProvider, NamespaceContext {
  35
  36         private Repository jcrRepository;
  37         private Session adminSession;
  38
  39         private String mountPath;
  40
  41         // cache
  42         private String jcrWorkspace;
  43
  44         private Map<ProvidedSession, JcrSessionAdapter> sessionAdapters = Collections.synchronizedMap(new HashMap<>());
  45
  46         public void start(Map<String, String> properties) {
  47                 mountPath = properties.get(CmsConstants.ACR_MOUNT_PATH);
  48                 if ("/".equals(mountPath))
  49                         throw new IllegalArgumentException("JCR content provider cannot be root /");
  50                 Objects.requireNonNull(mountPath);
  51                 jcrWorkspace = ContentUtils.getParentPath(mountPath)[1];
  52                 adminSession = CmsJcrUtils.openDataAdminSession(jcrRepository, jcrWorkspace);
  53         }
  54
  55         public void stop() {
  56                 if (adminSession.isLive())
  57                         JcrUtils.logoutQuietly(adminSession);
  58         }
  59
  60         public void setJcrRepository(Repository jcrRepository) {
  61                 this.jcrRepository = jcrRepository;
  62         }
  63
  64         @Override
  65         public ProvidedContent get(ProvidedSession contentSession, String relativePath) {
  66                 String jcrPath = "/" + relativePath;
  67                 return new JcrContent(contentSession, this, jcrWorkspace, jcrPath);
  68         }
  69
  70         @Override
  71         public boolean exists(ProvidedSession contentSession, String relativePath) {
  72                 String jcrPath = ContentUtils.SLASH + relativePath;
  73                 return new JcrContent(contentSession, this, jcrWorkspace, jcrPath).exists();
  74         }
  75
  76         public Session getJcrSession(ProvidedSession contentSession, String jcrWorkspace) {
  77                 JcrSessionAdapter sessionAdapter = sessionAdapters.get(contentSession);
  78                 if (sessionAdapter == null) {
  79                         final JcrSessionAdapter newSessionAdapter = new JcrSessionAdapter(jcrRepository, contentSession,
  80                                         contentSession.getSubject());
  81                         sessionAdapters.put(contentSession, newSessionAdapter);
  82                         contentSession.onClose().thenAccept((s) -> newSessionAdapter.close());
  83                         sessionAdapter = newSessionAdapter;
  84                 }
  85
  86                 Session jcrSession = sessionAdapter.getSession(jcrWorkspace);
  87                 return jcrSession;
  88         }
  89
  90         public Session getJcrSession(Content content, String jcrWorkspace) {
  91                 return getJcrSession(((ProvidedContent) content).getSession(), jcrWorkspace);
  92         }
  93
  94         @Override
  95         public String getMountPath() {
  96                 return mountPath;
  97         }
  98
  99         public synchronized <T> T doInAdminSession(Function<Session, T> toDo) {
 100                 try {
 101                         return toDo.apply(adminSession);
 102                 } finally {
 103                         try {
 104                                 if (adminSession.hasPendingChanges())
 105                                         adminSession.save();
 106                         } catch (RepositoryException e) {
 107                                 throw new JcrException("Cannot save admin session", e);
 108                         }
 109                 }
 110         }
 111
 112         /*
 113          * NAMESPACE CONTEXT
 114          */
 115         @Override
 116         public synchronized String getNamespaceURI(String prefix) {
 117                 try {
 118                         return adminSession.getNamespaceURI(prefix);
 119                 } catch (RepositoryException e) {
 120                         throw new JcrException(e);
 121                 }
 122         }
 123
 124         @Override
 125         public synchronized String getPrefix(String namespaceURI) {
 126                 try {
 127                         return adminSession.getNamespacePrefix(namespaceURI);
 128                 } catch (RepositoryException e) {
 129                         throw new JcrException(e);
 130                 }
 131         }
 132
 133         @Override
 134         public synchronized Iterator<String> getPrefixes(String namespaceURI) {
 135                 try {
 136                         return Arrays.asList(adminSession.getNamespacePrefix(namespaceURI)).iterator();
 137                 } catch (RepositoryException e) {
 138                         throw new JcrException(e);
 139                 }
 140         }
 141
 142         /*
 143          * SEARCH
 144          */
 145
 146         @Override
 147         public Spliterator<Content> search(ProvidedSession session, BasicSearch search, String relPath) {
 148                 try {
 149                         Session jcrSession = getJcrSession(session, jcrWorkspace);
 150                         BasicSearchToQom jcrBasicSearch = new BasicSearchToQom(jcrSession, search, relPath);
 151                         Query query = jcrBasicSearch.createQuery();
 152                         QueryResult queryResult = query.execute();
 153                         return new QueryResultSpliterator(session, queryResult.getNodes());
 154                 } catch (RepositoryException e) {
 155                         throw new JcrException(e);
 156                 }
 157         }
 158
 159         class QueryResultSpliterator implements Spliterator<Content> {
 160                 private ProvidedSession providedSession;
 161                 private NodeIterator nodeIterator;
 162
 163                 public QueryResultSpliterator(ProvidedSession providedSession, NodeIterator nodeIterator) {
 164                         super();
 165                         this.providedSession = providedSession;
 166                         this.nodeIterator = nodeIterator;
 167                 }
 168
 169                 @Override
 170                 public boolean tryAdvance(Consumer<? super Content> action) {
 171                         if (!nodeIterator.hasNext())
 172                                 return false;
 173                         try {
 174                                 Node node = nodeIterator.nextNode();
 175                                 // TODO optimise by reusing the Node
 176                                 JcrContent jcrContent = new JcrContent(providedSession, JcrContentProvider.this, jcrWorkspace,
 177                                                 node.getPath());
 178                                 action.accept(jcrContent);
 179                                 return true;
 180                         } catch (RepositoryException e) {
 181                                 throw new JcrException(e);
 182                         }
 183                 }
 184
 185                 @Override
 186                 public Spliterator<Content> trySplit() {
 187                         return null;
 188                 }
 189
 190                 @Override
 191                 public long estimateSize() {
 192                         return nodeIterator.getSize();
 193                 }
 194
 195                 @Override
 196                 public int characteristics() {
 197                         return NONNULL | SIZED;
 198                 }
 199
 200         }
 201 }