Skip to content
This repository has been archived by the owner on Jan 27, 2025. It is now read-only.

Add support for Metamorph wildcards. #97

Merged
merged 5 commits into from
Dec 22, 2021
Merged

Add support for Metamorph wildcards. #97

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
930e45f
Add support for Metamorph wildcards. (#89)
blackwinter Dec 20, 2021
432ef88
Turn lookup trie into a member variable. (#97)
blackwinter Dec 21, 2021
2904ca7
Slightly refactor field matching. (#97)
blackwinter Dec 22, 2021
8e9b7d4
Add unit tests for Metamorph wildcards. (#97)
blackwinter Dec 22, 2021
3d8168f
Add unit tests from functional review. (#97)
blackwinter Dec 22, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion metafix/src/main/java/org/metafacture/metafix/Metafix.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,6 @@ public class Metafix implements StreamPipe<StreamReceiver>, Maps { // checkstyle
private final Map<String, Map<String, String>> maps = new HashMap<>();
private final StreamFlattener flattener = new StreamFlattener();

// TODO: Use SimpleRegexTrie / WildcardTrie for wildcard, alternation and character class support
private Record currentRecord = new Record();
private Fix fix;
private Map<String, String> vars = new HashMap<>();
Expand Down
120 changes: 107 additions & 13 deletions metafix/src/main/java/org/metafacture/metafix/Value.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,20 @@

package org.metafacture.metafix;

import org.metafacture.commons.tries.SimpleRegexTrie;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.function.BiConsumer;
import java.util.function.BiFunction;
import java.util.function.Consumer;
import java.util.function.Predicate;
import java.util.function.UnaryOperator;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
Expand All @@ -37,6 +42,8 @@ public class Value {
private static final String LAST_FIELD = "$last";
private static final String ASTERISK = "*";

private static final String UNEXPECTED = "expected array or hash, got ";

private final Array array;
private final Hash hash;
private final String string;
Expand Down Expand Up @@ -236,6 +243,19 @@ private static String[] tail(final String[] fields) {
return Arrays.copyOfRange(fields, 1, fields.length);
}

private void transformFields(final String[] fields, final UnaryOperator<String> operator) {
switch (type) {
case Array:
asArray().transformFields(fields, operator);
break;
case Hash:
asHash().transformFields(fields, operator);
break;
default:
throw new IllegalStateException(UNEXPECTED + type);
}
}

enum Type {
Array,
Hash,
Expand Down Expand Up @@ -396,6 +416,45 @@ private Value findInValue(final String[] path, final Value value) {
return result;
}

private void transformFields(final String[] fields, final UnaryOperator<String> operator) {
final String field = fields[0];
final String[] remainingFields = tail(fields);
final int size = size();

if (fields.length == 0 || field.equals(ASTERISK)) {
for (int i = 0; i < size; ++i) {
transformFields(i, remainingFields, operator);
}
}
else if (isNumber(field)) {
final int index = Integer.parseInt(field) - 1; // TODO: 0-based Catmandu vs. 1-based Metafacture
if (index >= 0 && index < size) {
transformFields(index, remainingFields, operator);
}
}
// TODO: WDCD? copy_field('your.name','author[].name'), where name is an array
else {
for (int i = 0; i < size; ++i) {
transformFields(i, fields, operator);
}
}

list.removeIf(v -> v == null);
}

private void transformFields(final int index, final String[] fields, final UnaryOperator<String> operator) {
final Value value = get(index);

if (value != null) {
if (value.isString()) {
set(index, operator != null ? new Value(operator.apply(value.asString())) : null);
}
else {
value.transformFields(fields, operator);
}
}
}

private void insert(final InsertMode mode, final String[] fields, final String newValue) {
switch (fields[0]) {
case ASTERISK:
Expand Down Expand Up @@ -448,9 +507,8 @@ public static class Hash extends AbstractValueType {

private static final String FIELD_PATH_SEPARATOR = "\\.";

private static final String UNEXPECTED = "expected array or hash, got ";

private final Map<String, Value> map = new LinkedHashMap<>();
private final SimpleRegexTrie<String> trie = new SimpleRegexTrie<>();

/**
* Creates an empty instance of {@link Hash}.
Expand All @@ -465,7 +523,7 @@ protected Hash() {
* @return true if this hash contains the metadata field, false otherwise
*/
public boolean containsField(final String field) {
return map.containsKey(field);
return matchFields(field, Stream::anyMatch);
}

/**
Expand Down Expand Up @@ -526,7 +584,9 @@ public Value append(final String fieldPath, final String newValue) {
* @return the metadata value
*/
public Value get(final String field) {
return map.get(field);
// TODO: special treatment (only) for exact matches?
final List<Value> list = findFields(field).map(map::get).collect(Collectors.toList());
return list.isEmpty() ? null : list.size() == 1 ? list.get(0) : new Value(list);
}

public Value find(final String fieldPath) {
Expand Down Expand Up @@ -642,7 +702,7 @@ private Value insert(final InsertMode mode, final String[] fields, final String
* @param field the field name
*/
public void remove(final String field) {
map.remove(field);
modifyFields(field, map::remove);
}

public void removeNested(final String fieldPath) {
Expand Down Expand Up @@ -706,14 +766,35 @@ private void appendValue(final String[] newName, final Value v) {
}

public void transformFields(final List<String> params, final UnaryOperator<String> operator) {
final String field = params.get(0);
final Value value = find(field);
if (value != null) {
removeNested(field);
if (operator != null) {
value.asList(a -> a.forEach(v -> append(field, operator.apply(v.toString()))));
}
transformFields(split(params.get(0)), operator);
}

private void transformFields(final String[] fields, final UnaryOperator<String> operator) {
final String field = fields[0];
final String[] remainingFields = tail(fields);

if (field.equals(ASTERISK)) {
// TODO: search in all elements of value.asHash()?
transformFields(remainingFields, operator);
return;
}

modifyFields(field, f -> {
final Value value = map.get(f);

if (value != null) {
if (remainingFields.length == 0) {
map.remove(f);

if (operator != null) {
value.asList(a -> a.forEach(v -> append(f, operator.apply(v.toString()))));
}
}
else {
value.transformFields(remainingFields, operator);
}
}
});
}

/**
Expand All @@ -722,7 +803,7 @@ public void transformFields(final List<String> params, final UnaryOperator<Strin
* @param fields the field names
*/
public void retainFields(final Collection<String> fields) {
map.keySet().retainAll(fields);
map.keySet().retainAll(fields.stream().flatMap(this::findFields).collect(Collectors.toSet()));
}

/**
Expand Down Expand Up @@ -752,6 +833,19 @@ public String asString() {
return map.toString();
}

private void modifyFields(final String pattern, final Consumer<String> consumer) {
findFields(pattern).collect(Collectors.toSet()).forEach(consumer);
}

private Stream<String> findFields(final String pattern) {
return matchFields(pattern, Stream::filter);
}

private <T> T matchFields(final String pattern, final BiFunction<Stream<String>, Predicate<String>, T> function) {
trie.put(pattern, pattern);
return function.apply(map.keySet().stream(), f -> trie.get(f).contains(pattern));
}

}

}
Loading