Added strawman getDocumentById()

fcrepo4-archive · Apr 3, 2013 · 1714808 · 1714808
1 parent 53fc41c
commit 1714808
Showing 1 changed file with 131 additions and 29 deletions.
diff --git a/src/main/java/org/fcrepo/federation/bagit/BagItConnector.java b/src/main/java/org/fcrepo/federation/bagit/BagItConnector.java
@@ -1,80 +1,184 @@
 
 package org.fcrepo.federation.bagit;
 
+import static org.fcrepo.utils.FedoraJcrTypes.FEDORA_OBJECT;
+import static org.modeshape.jcr.api.JcrConstants.JCR_CONTENT;
+import static org.modeshape.jcr.api.JcrConstants.JCR_DATA;
+import static org.modeshape.jcr.api.JcrConstants.NT_FOLDER;
+import static org.modeshape.jcr.api.JcrConstants.NT_RESOURCE;
+
 import java.io.File;
 import java.io.IOException;
-import java.util.Map;
+import java.nio.file.Files;
 
 import javax.jcr.NamespaceRegistry;
 import javax.jcr.RepositoryException;
 
+import org.fcrepo.utils.FedoraJcrTypes;
 import org.infinispan.schematic.document.Document;
 import org.modeshape.connector.filesystem.FileSystemConnector;
 import org.modeshape.jcr.JcrI18n;
+import org.modeshape.jcr.api.JcrConstants;
 import org.modeshape.jcr.api.nodetype.NodeTypeManager;
 import org.modeshape.jcr.federation.spi.DocumentChanges;
-import org.modeshape.jcr.federation.spi.ExtraPropertiesStore;
-import org.modeshape.jcr.value.Name;
-import org.modeshape.jcr.value.Property;
+import org.modeshape.jcr.federation.spi.DocumentWriter;
+import org.modeshape.jcr.value.BinaryValue;
 
 public class BagItConnector extends FileSystemConnector {
-    
+
     private static final String FILE_SEPARATOR = File.separator;
+
     private static final String DELIMITER = File.pathSeparator;
-
+
+    private static final String JCR_LAST_MODIFIED = "jcr:lastModified";
+
+    private static final String JCR_CREATED = "jcr:created";
+
+    private static final String JCR_CREATED_BY = "jcr:createdBy";
+
+    private static final String JCR_LAST_MODIFIED_BY = "jcr:lastModified";
+
+    private static final String MIX_MIME_TYPE = "mix:mimeType";
+
+    private static final String JCR_MIME_TYPE = "jcr:mimeType";
+
+    private static final String JCR_ENCODING = "jcr:encoding";
+
+    private static final String JCR_CONTENT_SUFFIX = DELIMITER + JCR_CONTENT;
+
+    /**
+     * A boolean flag that specifies whether this connector should add the 'mix:mimeType' mixin to the 'nt:resource' nodes to
+     * include the 'jcr:mimeType' property. If set to <code>true</code>, the MIME type is computed immediately when the
+     * 'nt:resource' node is accessed, which might be expensive for larger files. This is <code>false</code> by default.
+     */
+    private boolean addMimeTypeMixin = false;
+
     /**
      * The string path for a {@link File} object that represents the top-level directory accessed by this connector. This is set
      * via reflection and is required for this connector.
      */
     private String directoryPath;
+
     private File directory;
-    
+
     /**
      * A string that is created in the {@link #initialize(NamespaceRegistry, NodeTypeManager)} method that represents the absolute
      * path to the {@link #directory}. This path is removed from an absolute path of a file to obtain the ID of the node.
      */
     private String directoryAbsolutePath;
-    private int directoryAbsolutePathLength;
 
-    private NamespaceRegistry registry;
+    private int directoryAbsolutePathLength;
 
     @Override
-    public void initialize( NamespaceRegistry registry,
-                            NodeTypeManager nodeTypeManager ) throws RepositoryException, IOException {
-
+    public void initialize(NamespaceRegistry registry,
+            NodeTypeManager nodeTypeManager) throws RepositoryException,
+            IOException {
+
         super.initialize(registry, nodeTypeManager);
-        this.registry = registry;
 
         // Initialize the directory path field that has been set via reflection when this method is called...
         checkFieldNotNull(directoryPath, "directoryPath");
         directory = new File(directoryPath);
         if (!directory.exists() || !directory.isDirectory()) {
-            String msg = JcrI18n.fileConnectorTopLevelDirectoryMissingOrCannotBeRead.text(getSourceName(), "directoryPath");
+            String msg =
+                    JcrI18n.fileConnectorTopLevelDirectoryMissingOrCannotBeRead
+                            .text(getSourceName(), "directoryPath");
             throw new RepositoryException(msg);
         }
         if (!directory.canRead() && !directory.setReadable(true)) {
-            String msg = JcrI18n.fileConnectorTopLevelDirectoryMissingOrCannotBeRead.text(getSourceName(), "directoryPath");
+            String msg =
+                    JcrI18n.fileConnectorTopLevelDirectoryMissingOrCannotBeRead
+                            .text(getSourceName(), "directoryPath");
             throw new RepositoryException(msg);
         }
         directoryAbsolutePath = directory.getAbsolutePath();
-        if (!directoryAbsolutePath.endsWith(FILE_SEPARATOR)) directoryAbsolutePath = directoryAbsolutePath + FILE_SEPARATOR;
-        directoryAbsolutePathLength = directoryAbsolutePath.length() - FILE_SEPARATOR.length(); // does NOT include the separator
-
+        if (!directoryAbsolutePath.endsWith(FILE_SEPARATOR))
+            directoryAbsolutePath = directoryAbsolutePath + FILE_SEPARATOR;
+        directoryAbsolutePathLength =
+                directoryAbsolutePath.length() - FILE_SEPARATOR.length(); // does NOT include the separator
+
         setExtraPropertiesStore(new BagItExtraPropertiesStore());
     }
 
-
-
     @Override
     public Document getDocumentById(String id) {
-        // TODO Auto-generated method stub
-        return null;
-    }
+        final File file = fileFor(id);
+        if (isExcluded(file) || !file.exists()) return null;
+        final boolean isRoot = isRoot(id);
+        final boolean isResource = isContentNode(id);
+        final DocumentWriter writer = newDocument(id);
+        File parentFile = file.getParentFile();
+        if (isResource) {
+            BinaryValue binaryValue = binaryFor(file);
+            writer.setPrimaryType(NT_RESOURCE);
+            writer.addProperty(JCR_DATA, binaryValue);
+            if (addMimeTypeMixin) {
+                writer.addMixinType(MIX_MIME_TYPE);
+                String mimeType = null;
+                String encoding = null; // We don't really know this
+                try {
+                    mimeType = binaryValue.getMimeType();
+                } catch (Throwable e) {
+                    getLogger().error(e, JcrI18n.couldNotGetMimeType,
+                            getSourceName(), id, e.getMessage());
+                }
+                writer.addProperty(JCR_ENCODING, encoding);
+                writer.addProperty(JCR_MIME_TYPE, mimeType);
+            }
+            writer.addProperty(JCR_LAST_MODIFIED, factories().getDateFactory()
+                    .create(file.lastModified()));
+            writer.addProperty(JCR_LAST_MODIFIED_BY, null); // ignored
+
+            //make these binary not queryable. If we really want to query them, we need to switch to external binaries
+            writer.setNotQueryable();
+            parentFile = file;
+        } else if (file.isFile()) {
+            writer.setPrimaryType(JcrConstants.NT_FILE);
+            writer.addMixinType(FedoraJcrTypes.FEDORA_DATASTREAM);
+            writer.addProperty(JCR_CREATED, factories().getDateFactory()
+                    .create(file.lastModified()));
+            try {
+                writer.addProperty(JCR_CREATED_BY, Files
+                        .getOwner(file.toPath()));
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+            String childId =
+                    isRoot ? JCR_CONTENT_SUFFIX : id + JCR_CONTENT_SUFFIX;
+            writer.addChild(childId, JCR_CONTENT);
+        } else {
+            final File dataDir =
+                    new File(file.getAbsolutePath() + DELIMITER + "data");
+            writer.setPrimaryType(NT_FOLDER);
+            writer.addMixinType(FEDORA_OBJECT);
+            writer.addProperty(JCR_CREATED, factories().getDateFactory()
+                    .create(file.lastModified()));
+            writer.addProperty(JCR_CREATED_BY, null); // ignored
+            for (File child : dataDir.listFiles()) {
+                // Only include as a datastream if we can access and read the file. Permissions might prevent us from
+                // reading the file, and the file might not exist if it is a broken symlink (see MODE-1768 for details).
+                if (child.exists() && child.canRead() &&
+                        (child.isFile() || child.isDirectory())) {
+                    // We use identifiers that contain the file/directory name ...
+                    String childName = child.getName();
+                    String childId =
+                            isRoot ? DELIMITER + childName : id + DELIMITER +
+                                    childName;
+                    writer.addChild(childId, childName);
+                }
+            }
+        }
 
-    @Override
-    public boolean hasDocument(String id) {
-        // TODO Auto-generated method stub
-        return false;
+        if (!isRoot) {
+            // Set the reference to the parent ...
+            writer.setParents(idFor(parentFile));
+        }
+
+        // Add the extra properties (if there are any), overwriting any properties with the same names
+        // (e.g., jcr:primaryType, jcr:mixinTypes, jcr:mimeType, etc.) ...
+        writer.addProperties(extraPropertiesStore().getProperties(id));
+
+        return writer.document();
     }
 
     @Override
@@ -89,6 +193,4 @@ public void updateDocument(DocumentChanges documentChanges) {
 
     }
 
-
-
 }