/* The contents of this file are subject to the license and copyright terms
* detailed in the license directory at the root of the source tree (also
* available online at http://fedora-commons.org/license/).
*/
package org.fcrepo.client.utility.validate.remote;
import static org.fcrepo.client.utility.validate.types.ContentModelInfo.DS_COMPOSITE_MODEL;
import static org.fcrepo.client.utility.validate.types.ContentModelInfo.DS_COMPOSITE_MODEL_FORMAT;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import javax.xml.rpc.ServiceException;
import org.fcrepo.client.FedoraClient;
import org.fcrepo.client.utility.validate.InvalidContentModelException;
import org.fcrepo.client.utility.validate.ObjectSource;
import org.fcrepo.client.utility.validate.ObjectSourceException;
import org.fcrepo.client.utility.validate.types.BasicContentModelInfo;
import org.fcrepo.client.utility.validate.types.BasicObjectInfo;
import org.fcrepo.client.utility.validate.types.ContentModelInfo;
import org.fcrepo.client.utility.validate.types.DatastreamInfo;
import org.fcrepo.client.utility.validate.types.DsCompositeModelDoc;
import org.fcrepo.client.utility.validate.types.ObjectInfo;
import org.fcrepo.client.utility.validate.types.RelationshipInfo;
import org.fcrepo.server.access.FedoraAPIAMTOM;
import org.fcrepo.server.management.FedoraAPIMMTOM;
import org.fcrepo.server.search.FieldSearchQuery;
import org.fcrepo.server.types.gen.Datastream;
import org.fcrepo.server.types.gen.RelationshipTuple;
import org.fcrepo.server.types.mtom.gen.MIMETypedStream;
/**
* An {@link ObjectSource} that is based on a {@link FedoraClient} link to a
* remote server.
*
* @author Jim Blake
*/
public class RemoteObjectSource
implements ObjectSource {
private final FedoraAPIAMTOM apia;
private final FedoraAPIMMTOM apim;
public RemoteObjectSource(ServiceInfo serviceInfo)
throws ServiceException, IOException {
FedoraClient fc =
new FedoraClient(serviceInfo.getBaseUrlString(),
serviceInfo.getUsername(),
serviceInfo.getPassword());
apia = fc.getAPIAMTOM();
apim = fc.getAPIMMTOM();
fc.shutdown();
}
/**
* Get a series of PIDs, representing all digital objects in the repository
* that satisfy the specified query.
*/
public Iterator<String> findObjectPids(FieldSearchQuery query)
throws ObjectSourceException {
return new RemotePidIterator(apia, query);
}
/**
* {@inheritDoc}
*/
@Override
public ObjectInfo getValidationObject(String pid)
throws ObjectSourceException {
List<RelationshipInfo> relations = getRelationships(pid);
Set<DatastreamInfo> dsDefs = getDatastreams(pid);
return new BasicObjectInfo(pid, relations, dsDefs);
}
private Set<DatastreamInfo> getDatastreams(String pid)
throws ObjectSourceException {
try {
List<Datastream> datastreams = apim.getDatastreams(pid, null, null);
return TypeUtility
.convertGenDatastreamArrayToDatastreamInfoSet(datastreams.toArray(new Datastream[]{}));
} catch (Exception e) {
throw new ObjectSourceException(e);
}
}
private List<RelationshipInfo> getRelationships(String pid)
throws ObjectSourceException {
try {
RelationshipTuple[] tuples =
apim.getRelationships(pid, null)
.toArray(new RelationshipTuple[0]);
return TypeUtility
.convertGenRelsTupleArrayToRelationshipInfoList(tuples);
} catch (Exception e) {
throw new ObjectSourceException(e);
}
}
/**
* A content model must exist as an object. If must have a (@link
* DS_COMPOSITE_MODEL} datastream.
*/
@Override
public ContentModelInfo getContentModelInfo(String pid)
throws ObjectSourceException, InvalidContentModelException {
try {
ObjectInfo object = getValidationObject(pid);
if (object == null) {
return null;
}
DatastreamInfo dsInfo =
object.getDatastreamInfo(DS_COMPOSITE_MODEL);
if (dsInfo == null) {
throw new InvalidContentModelException(pid,
"Content model has no '"
+ DS_COMPOSITE_MODEL
+ "' datastream.");
}
if (!DS_COMPOSITE_MODEL_FORMAT.equals(dsInfo.getFormatUri())) {
throw new InvalidContentModelException(pid, "Datastream '"
+ DS_COMPOSITE_MODEL + "' has incorrect format URI: '"
+ dsInfo.getFormatUri() + "'.");
}
MIMETypedStream ds =
apia.getDatastreamDissemination(pid,
DS_COMPOSITE_MODEL,
null);
DsCompositeModelDoc model =
new DsCompositeModelDoc(pid,
org.fcrepo.server.utilities.TypeUtility
.convertDataHandlerToBytes(ds
.getStream()));
return new BasicContentModelInfo(object, model.getTypeModels());
} catch (Exception e) {
throw new ObjectSourceException("Problem fetching '"
+ DS_COMPOSITE_MODEL
+ "' datastream for pid='"
+ pid + "'",
e);
}
}
}