net.thisptr.jackson.jq.internal.functions.IndicesFunction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jackson-jq Show documentation
There is a newer version: 1.2.0
package net.thisptr.jackson.jq.internal.functions;

import java.util.ArrayList;
import java.util.List;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.JsonNodeType;
import com.fasterxml.jackson.databind.node.NullNode;
import com.google.auto.service.AutoService;

import net.thisptr.jackson.jq.BuiltinFunction;
import net.thisptr.jackson.jq.Expression;
import net.thisptr.jackson.jq.Function;
import net.thisptr.jackson.jq.PathOutput;
import net.thisptr.jackson.jq.Scope;
import net.thisptr.jackson.jq.Version;
import net.thisptr.jackson.jq.exception.JsonQueryException;
import net.thisptr.jackson.jq.internal.misc.JsonNodeComparator;
import net.thisptr.jackson.jq.internal.misc.Preconditions;
import net.thisptr.jackson.jq.path.Path;

@AutoService(Function.class)
@BuiltinFunction("indices/1")
public class IndicesFunction implements Function {
	private static final JsonNodeComparator comparator = JsonNodeComparator.getInstance();

	@Override
	public void apply(final Scope scope, final List args, final JsonNode in, final Path ipath, final PathOutput output, final Version version) throws JsonQueryException {
		Preconditions.checkInputType("indices", in, JsonNodeType.STRING, JsonNodeType.ARRAY, JsonNodeType.NULL);

		if (in.isNull()) {
			output.emit(NullNode.getInstance(), null);
			return;
		}

		args.get(0).apply(scope, in, (needle) -> {
			final ArrayNode indices = scope.getObjectMapper().createArrayNode();
			for (final int index : indices(needle, in))
				indices.add(index);
			output.emit(indices, null);
		});
	}

	public static List indices(final JsonNode needle, final JsonNode haystack) throws JsonQueryException {
		final List result = new ArrayList<>();
		if (needle.isTextual() && haystack.isTextual()) {
			final String haystackText = haystack.asText();
			final String needleText = needle.asText();
			if (!needleText.isEmpty()) {
				for (int index = haystackText.indexOf(needleText); index >= 0; index = haystackText.indexOf(needleText, index + 1))
					result.add(index);
			}
		} else if (needle.isArray() && haystack.isArray()) {
			if (needle.size() != 0) {
				shift: for (int i = 0; i < haystack.size() - needle.size() + 1; ++i) {
					for (int j = 0; j < needle.size(); ++j)
						if (comparator.compare(haystack.get(i + j), needle.get(j)) != 0)
							continue shift;
					result.add(i);
				}
			}
		} else if (haystack.isArray()) {
			for (int i = 0; i < haystack.size(); ++i)
				if (comparator.compare(haystack.get(i), needle) == 0)
					result.add(i);
		} else {
			throw new JsonQueryException("indices() is not applicable to " + haystack.getNodeType());
		}
		return result;
	}
}