103 lines
3.8 KiB
Java
103 lines
3.8 KiB
Java
/*
|
|
* Copyright (C) 2010 Ryszard Wiśniewski <brut.alll@gmail.com>
|
|
* Copyright (C) 2010 Connor Tumbleson <connor.tumbleson@gmail.com>
|
|
*
|
|
* 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
|
|
*
|
|
* https://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 brut.directory;
|
|
|
|
import brut.common.BrutException;
|
|
import brut.common.InvalidUnknownFileException;
|
|
import brut.common.RootUnknownFileException;
|
|
import brut.common.TraversalUnknownFileException;
|
|
import brut.util.BrutIO;
|
|
import brut.util.OS;
|
|
import java.io.*;
|
|
import java.nio.file.Files;
|
|
import java.util.logging.Logger;
|
|
|
|
public class DirUtil {
|
|
private static final Logger LOGGER = Logger.getLogger("");
|
|
|
|
private DirUtil() {
|
|
// Private constructor for utility class
|
|
}
|
|
|
|
public static void copyToDir(Directory in, Directory out)
|
|
throws DirectoryException {
|
|
for (String fileName : in.getFiles(true)) {
|
|
copyToDir(in, out, fileName);
|
|
}
|
|
}
|
|
|
|
public static void copyToDir(Directory in, Directory out,
|
|
String[] fileNames) throws DirectoryException {
|
|
for (String fileName : fileNames) {
|
|
copyToDir(in, out, fileName);
|
|
}
|
|
}
|
|
|
|
public static void copyToDir(Directory in, Directory out, String fileName)
|
|
throws DirectoryException {
|
|
copyToDir(in, out, fileName, fileName);
|
|
}
|
|
|
|
public static void copyToDir(Directory in, Directory out, String inFile, String outFile)
|
|
throws DirectoryException {
|
|
try {
|
|
if (in.containsDir(inFile)) {
|
|
in.getDir(inFile).copyToDir(out.createDir(outFile));
|
|
} else {
|
|
BrutIO.copyAndClose(in.getFileInput(inFile), out.getFileOutput(outFile));
|
|
}
|
|
} catch (IOException ex) {
|
|
throw new DirectoryException("Error copying file: " + inFile, ex);
|
|
}
|
|
}
|
|
|
|
public static void copyToDir(Directory in, File out)
|
|
throws DirectoryException {
|
|
for (String fileName : in.getFiles(true)) {
|
|
copyToDir(in, out, fileName);
|
|
}
|
|
}
|
|
|
|
public static void copyToDir(Directory in, File out, String[] fileNames)
|
|
throws DirectoryException {
|
|
for (String fileName : fileNames) {
|
|
copyToDir(in, out, fileName);
|
|
}
|
|
}
|
|
|
|
public static void copyToDir(Directory in, File out, String fileName)
|
|
throws DirectoryException {
|
|
try {
|
|
if (in.containsDir(fileName)) {
|
|
OS.rmdir(new File(out, fileName));
|
|
in.getDir(fileName).copyToDir(new File(out, fileName));
|
|
} else if (!in.containsDir(fileName) && !in.containsFile(fileName)) {
|
|
// Skip copies of directories/files not found.
|
|
} else {
|
|
String cleanedFilename = BrutIO.sanitizeUnknownFile(out, fileName);
|
|
File outFile = new File(out, cleanedFilename);
|
|
outFile.getParentFile().mkdirs();
|
|
BrutIO.copyAndClose(in.getFileInput(fileName), Files.newOutputStream(outFile.toPath()));
|
|
}
|
|
} catch (RootUnknownFileException | InvalidUnknownFileException | TraversalUnknownFileException exception) {
|
|
LOGGER.warning(String.format("Skipping file %s (%s)", fileName, exception.getMessage()));
|
|
} catch (IOException | BrutException ex) {
|
|
throw new DirectoryException("Error copying file: " + fileName, ex);
|
|
}
|
|
}
|
|
}
|