CopyCommand.java

/**
 * Copyright (c) 2012, Thilo Planz. All rights reserved.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package v7db.files;

import static v7db.files.CatCommand.decodeSHAPrefix;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.codec.DecoderException;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;

import v7db.files.mongodb.MongoContentStorage;
import v7db.files.mongodb.V7File;
import v7db.files.mongodb.V7GridFS;
import v7db.files.spi.ContentPointer;
import v7db.files.spi.ContentSHA;

import com.mongodb.DB;
import com.mongodb.MongoException;

class CopyCommand {

	private static ContentSHA findContentByPrefix(MongoContentStorage storage,
			String shaPrefix) throws DecoderException, IOException {
		return storage.findContentPointerByPrefix(decodeSHAPrefix(shaPrefix));
	}

	static V7File getParent(V7GridFS fs, String[] childPath)
			throws FileNotFoundException {
		V7File parent = fs.getFile(ArrayUtils.subarray(childPath, 0,
				childPath.length - 1));
		if (parent == null)
			throw new FileNotFoundException("target path "
					+ ArrayUtils.subarray(childPath, 0, childPath.length - 1));
		return parent;
	}

	private static void createFile(V7GridFS fs, ContentPointer content,
			String[] path, String contentType) throws IOException {
		V7File existing = fs.getFile(path);

		if (existing != null) {
			if (contentType == null)
				contentType = existing.getContentType();
			existing.setContent(content, contentType);
		} else {
			V7File parent = getParent(fs, path);
			parent.createChild(content, path[path.length - 1], contentType);
		}
	}

	static String[] getPath(String root, String path) {
		List<String> t = new ArrayList<String>();
		t.add(root);
		t.addAll(Arrays.asList(StringUtils.split(path, "/")));
		return t.toArray(ArrayUtils.EMPTY_STRING_ARRAY);
	}

	public static void main(String[] args) throws MongoException, IOException {

		if (args.length != 5) {
			System.err.println("Copy files");
			System.err
					.println("  by name:   copy <sourceRoot> <sourcePath> <targetRoot> <targetPath>");
			System.err
					.println("  by hash:   copy -sha <shaHex> <targetRoot> <targetPath>");
			System.exit(1);
		}

		DB db = Configuration.getMongo().getDB(
				Configuration.getProperty("mongo.db"));

		V7GridFS fs = new V7GridFS(db);

		if ("-sha".equals(args[1])) {
			MongoContentStorage storage = new MongoContentStorage(db);
			String sha = args[2];
			try {
				ContentSHA file = findContentByPrefix(storage, sha);
				if (file == null)
					throw new FileNotFoundException("-sha " + sha);
				String[] path = getPath(args[3], args[4]);
				createFile(fs, file, path, null);
			} catch (DecoderException e) {
				throw new IllegalArgumentException("invalid parameter :" + sha
						+ " is not a hex-encoded SHA-1 prefix");
			}
		} else {
			String[] srcPath = getPath(args[1], args[2]);
			String[] targetPath = getPath(args[3], args[4]);
			V7File src = fs.getFile(srcPath);
			if (src == null) {
				throw new FileNotFoundException(args[1] + " " + args[2]);
			}
			if (src.hasContent()) {
				createFile(fs, src.getContentPointer(), targetPath, src
						.getContentType());
			} else {
				V7File existing = fs.getFile(targetPath);
				if (existing != null) {
					throw new IOException("copy target " + targetPath
							+ " already exists");
				}
				V7File parent = getParent(fs, targetPath);
				src.copyTo(parent.getId(), targetPath[targetPath.length - 1]);
			}

		}

	}

}