Added support for datastream properties.

fcrepo-exts · Apr 14, 2015 · 445a679 · 445a679
1 parent cbf42f8
commit 445a679
Show file tree

Hide file tree

Showing 3 changed files with 347 additions and 28 deletions.
diff --git a/src/main/java/org/fcrepo/migration/DatastreamVersion.java b/src/main/java/org/fcrepo/migration/DatastreamVersion.java
@@ -72,4 +72,18 @@ public interface DatastreamVersion {
      * redirect datastream.
      */
     public String getExternalOrRedirectURL();
+
+    /**
+     * Determines if this is the first version of a datastream.
+     *
+     * @return  True if this is the first version, false otherwise.
+     */
+    public boolean isFirstVersionIn(ObjectReference obj);
+
+    /**
+     * Determines if this is the last version of a datastream.
+     *
+     * @return  True if this is the last version, false otherwise.
+     */
+    public boolean isLastVersionIn(ObjectReference obj);
 }
diff --git a/src/main/java/org/fcrepo/migration/foxml11/Foxml11InputStreamFedoraObjectProcessor.java b/src/main/java/org/fcrepo/migration/foxml11/Foxml11InputStreamFedoraObjectProcessor.java
@@ -5,11 +5,11 @@
 import org.fcrepo.migration.DatastreamVersion;
 import org.fcrepo.migration.DefaultContentDigest;
 import org.fcrepo.migration.DefaultObjectInfo;
+import org.fcrepo.migration.ObjectReference;
 import org.fcrepo.migration.StreamingFedoraObjectHandler;
 import org.fcrepo.migration.FedoraObjectProcessor;
 import org.fcrepo.migration.ObjectInfo;
 import org.fcrepo.migration.ObjectProperties;
-
 import org.apache.commons.codec.binary.Base64OutputStream;
 
 import javax.xml.bind.JAXBContext;
@@ -24,6 +24,7 @@
 import javax.xml.stream.XMLStreamException;
 import javax.xml.stream.XMLStreamReader;
 import javax.xml.stream.events.XMLEvent;
+
 import java.io.ByteArrayOutputStream;
 import java.io.File;
 import java.io.FileOutputStream;
@@ -378,6 +379,17 @@ public String getExternalOrRedirectURL() {
             }
         }
 
+        @Override
+        public boolean isFirstVersionIn(ObjectReference obj) {
+            List<DatastreamVersion> datastreams = obj.getDatastreamVersions(getDatastreamInfo().getDatastreamId());
+            return datastreams.indexOf(this) == 0;
+        }
+
+        @Override
+        public boolean isLastVersionIn(ObjectReference obj) {
+            List<DatastreamVersion> datastreams = obj.getDatastreamVersions(getDatastreamInfo().getDatastreamId());
+            return datastreams.indexOf(this) == datastreams.size() - 1;
+        }
     }
 
     private static Map<String, String> getAttributes(XMLStreamReader r, String ... allowedNames) {