Metacat

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import org.dataone.client.formats.ObjectFormatInfo;

import gov.loc.repository.bagit.Bag;

import gov.loc.repository.bagit.BagFactory;

import gov.loc.repository.bagit.writer.Writer;

import gov.loc.repository.bagit.writer.impl.ZipWriter;

	/**

	 * Packages the given package in a Bagit collection for download

	 * @param pid

	 * @throws IOException 

	 * @throws NotImplemented 

	 * @throws NotFound 

	 * @throws NotAuthorized 

	 * @throws ServiceFailure 

	 * @throws InvalidToken 

	 * @throws OREParserException 

	 * @throws URISyntaxException 

	 * @throws OREException 

	 */

	public InputStream getPackage(Session session, Identifier pid) throws IOException, InvalidToken, ServiceFailure, NotAuthorized, NotFound, NotImplemented, OREException, URISyntaxException, OREParserException {

		InputStream bagInputStream = null;

		BagFactory bagFactory = new BagFactory();

		Bag bag = bagFactory.createBag();

		// track the temp files we use so we can delete them when finished

		List<File> tempFiles = new ArrayList<File>();

		// the pids to include in the package

		List<Identifier> packagePids = new ArrayList<Identifier>();

		// find the package contents

		SystemMetadata sysMeta = this.getSystemMetadata(session, pid);

		if (ObjectFormatService.getInstance().getFormat(sysMeta.getFormatId()).getFormatType().equals("RESOURCE")) {

			InputStream oreInputStream = this.get(session, pid);

			Map<Identifier, Map<Identifier, List<Identifier>>> resourceMapStructure = ResourceMapFactory.getInstance().parseResourceMap(oreInputStream);

			packagePids.addAll(resourceMapStructure.keySet());

			for (Map<Identifier, List<Identifier>> entries: resourceMapStructure.values()) {

				packagePids.addAll(entries.keySet());

				for (List<Identifier> dataPids: entries.values()) {

					packagePids.addAll(dataPids);

				}

			}

		} else {

			// just the lone pid in this package

			packagePids.add(pid);

		}

		// loop through the package contents

		for (Identifier entryPid: packagePids) {

			SystemMetadata entrySysMeta = this.getSystemMetadata(session, entryPid);

			String extension = ObjectFormatInfo.instance().getExtension(entrySysMeta.getFormatId().getValue());

	        String prefix = entryPid.getValue();

			File tempFile = File.createTempFile(prefix + "_bagit", extension);

			tempFiles.add(tempFile);

			InputStream entryInputStream = this.get(session, entryPid);

			IOUtils.copy(entryInputStream, new FileOutputStream(tempFile));

			bag.addFileToPayload(tempFile);

		}

		bag = bag.makeComplete();

		File bagFile = File.createTempFile("bagit", ".zip");

		// TODO: delete more confidently

		bagFile.deleteOnExit();

		bag.setFile(bagFile);

		Writer zipWriter = new ZipWriter(bagFactory);

		bag.write(zipWriter, bagFile);

		bagFile = bag.getFile();

		bagInputStream = new FileInputStream(bagFile);

		// clean up temp entry files

		for (File tf: tempFiles) {

			tf.delete();

		}

		return bagInputStream;

	}