org.fuwjax.oss.util.io.Files2 Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2015 fuwjax.org ([email protected])
 *
 * 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 org.fuwjax.oss.util.io;

import java.io.IOException;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;

/**
 * File utility class for test-centric operations.
 * @author fuwjax
 */
public class Files2 {
	/**
	 * Copies a source to a destination, works recursively on directories.
	 * @param source the source path
	 * @param dest the destination path
	 * @throws IOException if the source cannot be copied to the destination
	 */
	public static void copy(final Path source, final Path dest) throws IOException {
		if(Files.exists(source)) {
			Files.walkFileTree(source, new SimpleFileVisitor() {
				@Override
				public FileVisitResult visitFile(final Path file, final BasicFileAttributes attrs) throws IOException {
					Files.copy(file, resolve(dest, relativize(source, file)));
					return super.visitFile(file, attrs);
				}

				@Override
				public FileVisitResult preVisitDirectory(final Path dir, final BasicFileAttributes attrs) throws IOException {
					Files.createDirectories(resolve(dest, relativize(source, dir)));
					return super.preVisitDirectory(dir, attrs);
				}
			});
		}
	}

	public static Path relativize(final Path self, final Path other) {
		try {
			return self.relativize(other);
		} catch(final IllegalArgumentException e) {
			return self.relativize(convert(self, other));
		}
	}

	public static Path resolve(final Path self, final Path other) {
		try {
			return self.resolve(other);
		} catch(final IllegalArgumentException e) {
			return self.resolve(convert(self, other));
		}
	}

	private static Path convert(final Path self, final Path other) {
		if(other.getNameCount() == 0) {
			return self.relativize(self);
		}
		final String first = other.getName(0).toString();
		final String[] more = new String[other.getNameCount() - 1];
		for(int i = 0; i < more.length; i++) {
			more[i] = other.getName(i + 1).toString();
		}
		final Path proxy = self.getFileSystem().getPath(first, more);
		return other.isAbsolute() ? self.getRoot().resolve(proxy) : proxy;
	}

	/**
	 * Deletes a path, including directories.
	 * @param target the path to delete
	 * @throws IOException if the path cannot be deleted
	 */
	public static void delete(final Path target) throws IOException {
		if(target != null && Files.exists(target)) {
			Files.walkFileTree(target, new SimpleFileVisitor() {
				@Override
				public FileVisitResult visitFile(final Path file, final BasicFileAttributes attrs) throws IOException {
					Files.delete(file);
					return super.visitFile(file, attrs);
				}

				@Override
				public FileVisitResult postVisitDirectory(final Path dir, final IOException exc) throws IOException {
					Files.delete(dir);
					return super.postVisitDirectory(dir, exc);
				}
			});
		}
	}
}