Working FreeMarker/JCR integration

[lgpl/argeo-commons.git] / org.argeo.core / src / org / argeo / fm / jcr / JcrModel.java

diff --git a/org.argeo.core/src/org/argeo/fm/jcr/JcrModel.java b/org.argeo.core/src/org/argeo/fm/jcr/JcrModel.java
index e6e8efad1ed3bb8fe08a4edbbcd748076e5dc080..5cae5c1a86394737dd9bc6bff2128266fed6af91 100644 (file)
--- a/org.argeo.core/src/org/argeo/fm/jcr/JcrModel.java
+++ b/org.argeo.core/src/org/argeo/fm/jcr/JcrModel.java
@@ -2,12 +2,17 @@ package org.argeo.fm.jcr;
 
 import javax.jcr.Node;
 import javax.jcr.Property;
+import javax.jcr.PropertyIterator;
 import javax.jcr.RepositoryException;
+import javax.jcr.Value;
 
 import freemarker.template.SimpleScalar;
+import freemarker.template.TemplateCollectionModel;
 import freemarker.template.TemplateHashModel;
+import freemarker.template.TemplateHashModelEx2;
 import freemarker.template.TemplateModel;
 import freemarker.template.TemplateModelException;
+import freemarker.template.TemplateModelIterator;
 import freemarker.template.TemplateNodeModel;
 import freemarker.template.TemplateSequenceModel;
 
@@ -63,9 +68,20 @@ public class JcrModel implements TemplateNodeModel, TemplateHashModel {
                }
        }
 
+       // HASH
+
        @Override
        public TemplateModel get(String key) throws TemplateModelException {
                try {
+                       if ("jcr:path".equals(key))
+                               return new SimpleScalar(node.getPath());
+                       if ("jcr:name".equals(key))
+                               return new SimpleScalar(node.getName());
+                       if ("jcr:properties".equals(key))
+                               return new PropertiesModel();
+                       if ("jcr:parent".equals(key))
+                               return node.getParent() != null ? new JcrModel(node.getParent()) : null;
+
                        Property property = node.getProperty(key);
                        if (property == null)
                                return null;
@@ -77,10 +93,147 @@ public class JcrModel implements TemplateNodeModel, TemplateHashModel {
 
        @Override
        public boolean isEmpty() throws TemplateModelException {
-               // JCR default properties always accessible
                return false;
        }
 
-       // HASH
+       public Node getNode() {
+               return node;
+       }
+
+       protected TemplateModel propertyValues(Property property) throws RepositoryException {
+               if (!property.isMultiple())
+                       return new SimpleScalar(property.getString());
+               Value[] values = property.getValues();
+               StringBuilder sb = new StringBuilder();
+               for (Value value : values) {
+                       sb.append(value.getString()).append('\n');
+               }
+               return new SimpleScalar(sb.toString());
+       }
+
+       class PropertiesModel implements TemplateHashModelEx2 {
+               @Override
+               public TemplateModel get(String key) throws TemplateModelException {
+                       return JcrModel.this.get(key);
+               }
+
+               @Override
+               public boolean isEmpty() throws TemplateModelException {
+                       return false;
+               }
+
+               @Override
+               public TemplateCollectionModel keys() throws TemplateModelException {
+                       try {
+                               PropertyIterator pit = node.getProperties();
+                               return new TemplateCollectionModel() {
+
+                                       @Override
+                                       public TemplateModelIterator iterator() throws TemplateModelException {
+                                               return new TemplateModelIterator() {
+
+                                                       @Override
+                                                       public TemplateModel next() throws TemplateModelException {
+                                                               try {
+                                                                       return new SimpleScalar(pit.nextProperty().getName());
+                                                               } catch (RepositoryException e) {
+                                                                       throw new TemplateModelException("Cannot list properties of " + node, e);
+                                                               }
+                                                       }
 
+                                                       @Override
+                                                       public boolean hasNext() throws TemplateModelException {
+                                                               return pit.hasNext();
+                                                       }
+                                               };
+                                       }
+                               };
+                       } catch (RepositoryException e) {
+                               throw new TemplateModelException("Cannot list properties of " + node, e);
+                       }
+               }
+
+               @Override
+               public int size() throws TemplateModelException {
+                       try {
+                               PropertyIterator pit = node.getProperties();
+                               return (int) pit.getSize();
+                       } catch (RepositoryException e) {
+                               throw new TemplateModelException("Cannot list properties of " + node, e);
+                       }
+               }
+
+               @Override
+               public TemplateCollectionModel values() throws TemplateModelException {
+                       try {
+                               PropertyIterator pit = node.getProperties();
+                               return new TemplateCollectionModel() {
+
+                                       @Override
+                                       public TemplateModelIterator iterator() throws TemplateModelException {
+                                               return new TemplateModelIterator() {
+
+                                                       @Override
+                                                       public TemplateModel next() throws TemplateModelException {
+                                                               try {
+                                                                       return propertyValues(pit.nextProperty());
+                                                               } catch (RepositoryException e) {
+                                                                       throw new TemplateModelException("Cannot list properties of " + node, e);
+                                                               }
+                                                       }
+
+                                                       @Override
+                                                       public boolean hasNext() throws TemplateModelException {
+                                                               return pit.hasNext();
+                                                       }
+                                               };
+                                       }
+                               };
+                       } catch (RepositoryException e) {
+                               throw new TemplateModelException("Cannot list properties of " + node, e);
+                       }
+               }
+
+               @Override
+               public KeyValuePairIterator keyValuePairIterator() throws TemplateModelException {
+                       try {
+                               PropertyIterator pit = node.getProperties();
+                               return new KeyValuePairIterator() {
+
+                                       @Override
+                                       public boolean hasNext() throws TemplateModelException {
+                                               return pit.hasNext();
+                                       }
+
+                                       @Override
+                                       public KeyValuePair next() throws TemplateModelException {
+                                               Property property = pit.nextProperty();
+                                               return new KeyValuePair() {
+
+                                                       @Override
+                                                       public TemplateModel getValue() throws TemplateModelException {
+                                                               try {
+                                                                       return propertyValues(property);
+                                                               } catch (RepositoryException e) {
+                                                                       throw new TemplateModelException("Cannot list properties of " + node, e);
+                                                               }
+                                                       }
+
+                                                       @Override
+                                                       public TemplateModel getKey() throws TemplateModelException {
+                                                               try {
+                                                                       return new SimpleScalar(property.getName());
+                                                               } catch (RepositoryException e) {
+                                                                       throw new TemplateModelException("Cannot list properties of " + node, e);
+                                                               }
+                                                       }
+                                               };
+                                       }
+                               };
+                       } catch (RepositoryException e) {
+                               throw new TemplateModelException("Cannot list properties of " + node, e);
+                       }
+               }
+
+       }
 }