sbt/paths/src/main/java/coursier/CachePath.java

150 lines
4.7 KiB
Java
Raw Normal View History

package coursier;
import java.io.File;
import java.io.FileOutputStream;
import java.net.MalformedURLException;
import java.nio.channels.FileLock;
import java.util.concurrent.Callable;
import java.util.concurrent.ConcurrentHashMap;
/**
* Cache paths logic, shared by the cache and bootstrap modules
*/
public class CachePath {
// based on https://stackoverflow.com/questions/4571346/how-to-encode-url-to-avoid-special-characters-in-java/4605848#4605848
// '/' was removed from the unsafe list
private static String escape(String input) {
StringBuilder resultStr = new StringBuilder();
for (char ch : input.toCharArray()) {
if (isUnsafe(ch)) {
resultStr.append('%');
resultStr.append(toHex(ch / 16));
resultStr.append(toHex(ch % 16));
} else {
resultStr.append(ch);
}
}
return resultStr.toString();
}
private static char toHex(int ch) {
return (char) (ch < 10 ? '0' + ch : 'A' + ch - 10);
}
private static boolean isUnsafe(char ch) {
return ch > 128 || " %$&+,:;=?@<>#%".indexOf(ch) >= 0;
}
public static File localFile(String url, File cache, String user) throws MalformedURLException {
// use the File constructor accepting a URI in case of problem with the two cases below?
if (url.startsWith("file:///"))
return new File(url.substring("file://".length()));
if (url.startsWith("file:/"))
return new File(url.substring("file:".length()));
String[] split = url.split(":", 2);
if (split.length != 2)
throw new MalformedURLException("No protocol found in URL " + url);
String protocol = split[0];
String remaining = split[1];
if (remaining.startsWith("///"))
remaining = remaining.substring("///".length());
else if (remaining.startsWith("/"))
remaining = remaining.substring("/".length());
else
throw new MalformedURLException("URL " + url + " doesn't contain an absolute path");
if (remaining.endsWith("/"))
// keeping directory content in .directory files
remaining = remaining + ".directory";
while (remaining.startsWith("/"))
remaining = remaining.substring(1);
String userPart = "";
if (user != null)
userPart = user + "@";
return new File(cache, escape(protocol + "/" + userPart + remaining));
}
public static File temporaryFile(File file) {
File dir = file.getParentFile();
String name = file.getName();
return new File(dir, name + ".part");
}
public static File lockFile(File file) {
return new File(file.getParentFile(), file.getName() + ".lock");
}
public static File defaultCacheDirectory() {
// cache this method result so that it's only ever computed once?
String path = System.getenv("COURSIER_CACHE");
if (path == null)
path = System.getProperty("coursier.cache");
if (path == null)
path = System.getProperty("user.home") + "/.coursier/cache/v1"; // shouldn't have put a "v1" component here...
return new File(path).getAbsoluteFile();
}
private static ConcurrentHashMap<File, Object> processStructureLocks = new ConcurrentHashMap<File, Object>();
public static <V> V withStructureLock(File cache, Callable<V> callable) throws Exception {
Object intraProcessLock = processStructureLocks.get(cache);
if (intraProcessLock == null) {
Object lock = new Object();
Object prev = processStructureLocks.putIfAbsent(cache, lock);
if (prev == null)
intraProcessLock = lock;
else
intraProcessLock = prev;
}
synchronized (intraProcessLock) {
File lockFile = new File(cache, ".structure.lock");
lockFile.getParentFile().mkdirs();
FileOutputStream out = null;
try {
out = new FileOutputStream(lockFile);
FileLock lock = null;
try {
lock = out.getChannel().lock();
try {
return callable.call();
}
finally {
lock.release();
lock = null;
out.close();
out = null;
lockFile.delete();
}
}
finally {
if (lock != null) lock.release();
}
} finally {
if (out != null) out.close();
}
}
}
}