ContentUIP.java

/**
 * Copyright 2008 The University of North Carolina at Chapel Hill
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package edu.unc.lib.dl.update;

import java.io.File;
import java.io.FileOutputStream;
import java.util.Map;

import edu.unc.lib.dl.fedora.AccessClient;
import edu.unc.lib.dl.fedora.PID;
import edu.unc.lib.dl.fedora.types.MIMETypedStream;

public class ContentUIP extends FedoraObjectUIP {

	public ContentUIP(PID pid, String user, UpdateOperation operation) {
		super(pid, user, operation);
	}

	@SuppressWarnings("unchecked")
	@Override
	public Map<String, File> getIncomingData() {
		return (Map<String, File>) incomingData;
	}

	@SuppressWarnings("unchecked")
	@Override
	public Map<String, File> getOriginalData() {
		return (Map<String, File>) originalData;
	}

	@SuppressWarnings("unchecked")
	@Override
	public Map<String, File> getModifiedData() {
		return (Map<String, File>) modifiedData;
	}

	@Override
	public void storeOriginalDatastreams(AccessClient accessClient)
			throws UIPException {
		// For efficiency, only pulling down the original if it is being
		// modified, not replaced/deleted
		if (!(this.operation.equals(UpdateOperation.ADD) || this.operation
				.equals(UpdateOperation.UPDATE)))
			return;

		for (String datastream : modifiedData.keySet()) {
			try {
				File dsFile = File.createTempFile(datastream, ".tmp");
				MIMETypedStream dsStream = accessClient
						.getDatastreamDissemination(pid, datastream, null);
				try (FileOutputStream outputStream = new FileOutputStream(
						dsFile)) {
					outputStream.write(dsStream.getStream());
				}
				this.getOriginalData().put(datastream, dsFile);
			} catch (Exception e) {
				throw new UIPException(
						"Exception occurred while attempting to store datastream "
								+ datastream + " for " + pid.getPid(), e);
			}
		}

	}
}