aQute.lib.io.FileTree Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.apache.aries.spifly.dynamic.framework.extension Show documentation
This framework extension fragment contains an extender that facilitates the use of JRE SPI providers (components typically plugged in to the JRE through META-INF/services resources).
There is a newer version: 1.3.7
Show newest version
package aQute.lib.io;

import static java.util.stream.Collectors.toList;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.regex.Pattern;
import java.util.stream.Stream;

import aQute.libg.glob.AntGlob;

public class FileTree {
	private List		files		= new ArrayList<>();
	private List	includes	= new ArrayList<>();
	private List	excludes	= new ArrayList<>();

	public FileTree() {}

	/**
	 * Can be used by subclasses to add specific files to the return value of
	 * {@link #getFiles(File, String)}.
	 * 
	 * @param file A file to include in the return value of
	 *            {@link #getFiles(File, String)}.
	 */
	protected void addFile(File file) {
		if (file == null) {
			return;
		}
		if (!files.contains(file)) {
			files.add(file);
		}
	}

	/**
	 * Add an Ant-style glob to the include patterns.
	 * 
	 * @param includes Add an Ant-style glob
	 */
	public void addIncludes(List includes) {
		if (includes == null) {
			return;
		}
		for (String include : includes) {
			if (include == null) {
				continue;
			}
			this.includes.add(AntGlob.toPattern(include));
		}
	}

	/**
	 * Add an Ant-style glob to the include patterns.
	 * 
	 * @param includes Add an Ant-style glob
	 */
	public void addIncludes(String... includes) {
		if (includes == null) {
			return;
		}
		for (String include : includes) {
			if (include == null) {
				continue;
			}
			this.includes.add(AntGlob.toPattern(include));
		}
	}

	/**
	 * Add an Ant-style glob to the exclude patterns.
	 * 
	 * @param excludes Add an Ant-style glob
	 */
	public void addExcludes(String... excludes) {
		if (excludes == null) {
			return;
		}
		for (String exclude : excludes) {
			if (exclude == null) {
				continue;
			}
			this.excludes.add(AntGlob.toPattern(exclude));
		}
	}

	/**
	 * Add an Ant-style glob to the exclude patterns.
	 * 
	 * @param excludes Add an Ant-style glob
	 */
	public void addExcludes(List excludes) {
		if (excludes == null) {
			return;
		}
		for (String exclude : excludes) {
			if (exclude == null) {
				continue;
			}
			this.excludes.add(AntGlob.toPattern(exclude));
		}
	}

	/**
	 * Return a list of files using the specified baseDir and the configured
	 * include and exclude Ant-style glob expressions.
	 * 
	 * @param baseDir The base directory for locating files.
	 * @param defaultIncludes The default include patterns to use if no include
	 *            patterns were configured.
	 * @return A list of files.
	 * @throws IOException If an exception occurs.
	 */
	public List getFiles(File baseDir, String... defaultIncludes) throws IOException {
		if (includes.isEmpty() && files.isEmpty() && (defaultIncludes != null) && (defaultIncludes.length > 0)) {
			return getFiles(baseDir, toPatterns(Stream.of(defaultIncludes)), excludes);
		}
		return getFiles(baseDir, includes, excludes);
	}

	private List toPatterns(Stream globs) {
		return globs.filter(Objects::nonNull)
			.map(AntGlob::toPattern)
			.collect(toList());
	}

	/**
	 * Return a list of files using the specified baseDir and the configured
	 * include and exclude Ant-style glob expressions.
	 * 
	 * @param baseDir The base directory for locating files.
	 * @param defaultIncludes The default include patterns to use if no include
	 *            patterns were configured.
	 * @return A list of files.
	 * @throws IOException If an exception occurs.
	 */
	public List getFiles(File baseDir, List defaultIncludes) throws IOException {
		if (includes.isEmpty() && files.isEmpty() && (defaultIncludes != null) && !defaultIncludes.isEmpty()) {
			return getFiles(baseDir, toPatterns(defaultIncludes.stream()), excludes);
		}
		return getFiles(baseDir, includes, excludes);
	}

	private List getFiles(File baseDir, List includePatterns, List excludePatterns)
		throws IOException {
		if (includePatterns.isEmpty()) {
			return new ArrayList<>(files);
		}
		Path basePath = baseDir.toPath();
		try (Stream walker = Files.walk(basePath)
			.skip(1)) {
			List result = Stream.concat(files.stream(), //
				walker.filter(p -> {
					String path = basePath.relativize(p)
						.toString();
					return includePatterns.stream()
						.anyMatch(i -> i.matcher(path)
							.matches())
						&& !excludePatterns.stream()
							.anyMatch(e -> e.matcher(path)
								.matches());
				})
					.sorted()
					.map(Path::toFile))
				.distinct()
				.collect(toList());
			return result;
		}
	}
}