/*
* SK's Minecraft Launcher
* Copyright (C) 2010-2014 Albert Pham <http://www.sk89q.com> and contributors
* Please see LICENSE.txt for license information.
*/
package com.skcraft.launcher.builder;
import com.google.common.hash.HashFunction;
import com.google.common.hash.Hashing;
import com.google.common.io.Files;
import com.skcraft.launcher.model.modpack.FileInstall;
import com.skcraft.launcher.model.modpack.Manifest;
import lombok.NonNull;
import lombok.extern.java.Log;
import org.apache.commons.io.FilenameUtils;
import java.io.File;
import java.io.IOException;
/**
* Walks a path and adds hashed path versions to the given
* {@link com.skcraft.launcher.model.modpack.Manifest}.
*/
@Log
public class ClientFileCollector extends DirectoryWalker {
private final Manifest manifest;
private final PropertiesApplicator applicator;
private final File destDir;
private HashFunction hf = Hashing.sha1();
/**
* Create a new collector.
*
* @param manifest the manifest
* @param applicator applies properties to manifest entries
* @param destDir the destination directory to copy the hashed objects
*/
public ClientFileCollector(@NonNull Manifest manifest, @NonNull PropertiesApplicator applicator,
@NonNull File destDir) {
this.manifest = manifest;
this.applicator = applicator;
this.destDir = destDir;
}
@Override
protected DirectoryBehavior getBehavior(@NonNull String name) {
return getDirectoryBehavior(name);
}
@Override
protected void onFile(File file, String relPath) throws IOException {
if (file.getName().endsWith(FileInfoScanner.FILE_SUFFIX)) {
return;
}
FileInstall entry = new FileInstall();
String hash = Files.hash(file, hf).toString();
String hashedPath = hash.substring(0, 2) + "/" + hash.substring(2, 4) + "/" + hash;
File destPath = new File(destDir, hashedPath);
entry.setHash(hash);
entry.setLocation(hashedPath);
entry.setTo(FilenameUtils.separatorsToUnix(FilenameUtils.normalize(relPath)));
entry.setSize(file.length());
applicator.apply(entry);
destPath.getParentFile().mkdirs();
ClientFileCollector.log.info(String.format("Adding %s from %s...", relPath, file.getAbsolutePath()));
Files.copy(file, destPath);
manifest.getTasks().add(entry);
}
public static DirectoryBehavior getDirectoryBehavior(@NonNull String name) {
if (name.startsWith(".")) {
return DirectoryBehavior.SKIP;
} else if (name.equals("_OPTIONAL")) {
return DirectoryBehavior.IGNORE;
} else if (name.equals("_SERVER")) {
return DirectoryBehavior.SKIP;
} else if (name.equals("_CLIENT")) {
return DirectoryBehavior.IGNORE;
} else {
return DirectoryBehavior.CONTINUE;
}
}
}