org.bndtools.templating.engine.mustache.AccumulateNamesObjectHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.bndtools.templating Show documentation

org.bndtools.templating

The newest version!

package org.bndtools.templating.engine.mustache;

import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

import com.github.mustachejava.Binding;
import com.github.mustachejava.Code;
import com.github.mustachejava.ObjectHandler;
import com.github.mustachejava.TemplateContext;
import com.github.mustachejava.reflect.BaseObjectHandler;
import com.github.mustachejava.util.Wrapper;

/**
 * A Mustache ObjectHandler that accumulates the names of requested parameters
 */
class AccumulateNamesObjectHandler extends BaseObjectHandler {

	private final ObjectHandler	delegateHandler;
	private final Set	names	= new HashSet<>();

	AccumulateNamesObjectHandler(ObjectHandler delegateHandler) {
		this.delegateHandler = delegateHandler;
	}

	@Override
	public Wrapper find(final String name, Object[] scopes) {
		final Wrapper delegateWrapper = delegateHandler.find(name, scopes);
		return scopes1 -> {
			names.add(name);
			return delegateWrapper.call(scopes1);
		};
	}

	@Override
	public Binding createBinding(final String name, TemplateContext context, Code code) {
		final Binding delegateBinding = delegateHandler.createBinding(name, context, code);
		return scopes -> {
			names.add(name);
			return delegateBinding.get(scopes);
		};
	}

	public Set getNames() {
		return Collections.unmodifiableSet(names);
	}

}