1
0
mirror of https://github.com/mitb-archive/filebot synced 2024-12-24 16:58:51 -05:00

Added new expression format methods:

* String.replace(Map)
* Collection.join(String, String, String)
* Object.check(Closure)
This commit is contained in:
Reinhard Pointner 2016-03-09 05:58:52 +00:00
parent 4e90c5d7af
commit 83e7f189d6

View File

@ -1,6 +1,7 @@
package net.filebot.format; package net.filebot.format;
import static java.util.regex.Pattern.*; import static java.util.regex.Pattern.*;
import groovy.lang.Closure;
import java.io.File; import java.io.File;
import java.io.IOException; import java.io.IOException;
@ -8,12 +9,20 @@ import java.nio.file.Files;
import java.nio.file.attribute.BasicFileAttributeView; import java.nio.file.attribute.BasicFileAttributeView;
import java.nio.file.attribute.BasicFileAttributes; import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collection;
import java.util.List; import java.util.List;
import java.util.Locale; import java.util.Locale;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Objects;
import java.util.regex.Matcher; import java.util.regex.Matcher;
import java.util.regex.Pattern;
import net.filebot.similarity.Normalization; import net.filebot.similarity.Normalization;
import net.filebot.util.FileUtilities; import net.filebot.util.FileUtilities;
import net.filebot.util.StringUtilities;
import org.codehaus.groovy.runtime.typehandling.DefaultTypeTransformation;
import com.ibm.icu.text.Transliterator; import com.ibm.icu.text.Transliterator;
@ -303,21 +312,49 @@ public class ExpressionFormatMethods {
/** /**
* Replace multiple replacement pairs * Replace multiple replacement pairs
* *
* e.g. replace('ä', 'ae', 'ö', 'oe', 'ü', 'ue') * e.g. replace(ä:'ae', ö:'oe', ü:'ue')
*/ */
public static String replace(String self, String tr0, String tr1, String... tr) { public static String replace(String self, Map<?, ?> replace) {
// the first two parameters are required, the rest of the parameter sequence is optional // the first two parameters are required, the rest of the parameter sequence is optional
self = self.replace(tr0, tr1); for (Entry<?, ?> it : replace.entrySet()) {
if (it.getKey() instanceof Pattern) {
for (int i = 0; i < tr.length - 1; i += 2) { self = ((Pattern) it.getKey()).matcher(self).replaceAll(it.getValue().toString());
String t = tr[i]; } else {
String r = tr[i + 1]; self = self.replace(it.getKey().toString(), it.getValue().toString());
self = self.replace(t, r); }
} }
return self; return self;
} }
/**
* Join non-empty String values and prepend prefix / append suffix values
*
* e.g. (1..3).join('-', '[', ']')
*
* Unwind if list is empty
*
* e.g. [].join('-', '[', ']') => Exception: List is empty
*/
public static String join(Collection<?> self, String delimiter, String prefix, String suffix) throws Exception {
String[] values = self.stream().map(StringUtilities::asNonEmptyString).filter(Objects::nonNull).toArray(String[]::new);
if (values.length > 0) {
return prefix + String.join(delimiter, values) + suffix;
}
throw new Exception("List is empty");
}
/**
* Unwind if an object does not satisfy the given predicate
*
* e.g. (0..9)*.check{it < 10}.sum()
*/
public static Object check(Object self, Closure<?> c) throws Exception {
if (DefaultTypeTransformation.castToBoolean(c.call(self))) {
return self;
}
throw new Exception("Check failed");
}
/** /**
* File utilities * File utilities
*/ */