110 lines
2.2 KiB
Java
110 lines
2.2 KiB
Java
package net.filebot.format;
|
|
|
|
import groovy.lang.Closure;
|
|
|
|
import java.io.IOException;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Paths;
|
|
import java.util.ArrayList;
|
|
import java.util.LinkedHashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Global functions available in the {@link ExpressionFormat}
|
|
*/
|
|
public class ExpressionFormatFunctions {
|
|
|
|
/**
|
|
* General helpers and utilities
|
|
*/
|
|
public static Object c(Closure<?> c) {
|
|
try {
|
|
return c.call();
|
|
} catch (Exception e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static Object any(Object a0, Object a1, Object... args) {
|
|
for (Object it : new Object[] { a0, a1 }) {
|
|
try {
|
|
Object result = callIfCallable(it);
|
|
if (result != null) {
|
|
return result;
|
|
}
|
|
} catch (Exception e) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
for (Object it : args) {
|
|
try {
|
|
Object result = callIfCallable(it);
|
|
if (result != null) {
|
|
return result;
|
|
}
|
|
} catch (Exception e) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public static List<Object> allOf(Object a0, Object a1, Object... args) {
|
|
List<Object> values = new ArrayList<Object>();
|
|
|
|
for (Object it : new Object[] { a0, a1 }) {
|
|
try {
|
|
Object result = callIfCallable(it);
|
|
if (result != null) {
|
|
values.add(result);
|
|
}
|
|
} catch (Exception e) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
for (Object it : args) {
|
|
try {
|
|
Object result = callIfCallable(it);
|
|
if (result != null) {
|
|
values.add(result);
|
|
}
|
|
} catch (Exception e) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
return values;
|
|
}
|
|
|
|
private static Object callIfCallable(Object obj) throws Exception {
|
|
if (obj instanceof Closure<?>) {
|
|
return ((Closure<?>) obj).call();
|
|
}
|
|
return obj;
|
|
}
|
|
|
|
public static Map<String, String> csv(String path) throws IOException {
|
|
Map<String, String> map = new LinkedHashMap<String, String>();
|
|
for (String line : Files.readAllLines(Paths.get(path), StandardCharsets.UTF_8)) {
|
|
for (String delim : new String[] { "\t", ";" }) {
|
|
String[] field = line.split(delim, 2);
|
|
if (field.length >= 2) {
|
|
map.put(field[0], field[1]);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return map;
|
|
}
|
|
|
|
public static List<String> readLines(String path) throws IOException {
|
|
return Files.readAllLines(Paths.get(path), StandardCharsets.UTF_8);
|
|
}
|
|
|
|
}
|