fr.sii.ogham.template.thymeleaf.buider.AbstractThymeleafMultiContentBuilder Maven / Gradle / Ivy

Go to download
package fr.sii.ogham.template.thymeleaf.buider;

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

import fr.sii.ogham.core.builder.env.EnvironmentBuilder;
import fr.sii.ogham.core.builder.template.VariantBuilder;
import fr.sii.ogham.core.builder.template.VariantWithExt;
import fr.sii.ogham.core.message.content.Variant;
import fr.sii.ogham.template.common.adapter.ExtensionMappingVariantResolver;
import fr.sii.ogham.template.common.adapter.VariantResolver;

public abstract class AbstractThymeleafMultiContentBuilder, P> extends AbstractThymeleafBuilder implements VariantBuilder {
	private List variants;

	protected AbstractThymeleafMultiContentBuilder(Class selfType, P parent, EnvironmentBuilder environmentBuilder) {
		super(selfType, parent, environmentBuilder);
		variants = new ArrayList<>();
	}

	@Override
	public MYSELF variant(Variant variant, String extension) {
		variants.add(new VariantWithExt(variant, extension));
		return myself;
	}

	@Override
	public VariantResolver buildVariant() {
		return buildExtMappingVariantResolver();
	}

	private ExtensionMappingVariantResolver buildExtMappingVariantResolver() {
		ExtensionMappingVariantResolver resolver = new ExtensionMappingVariantResolver(buildResolver());
		for(VariantWithExt v : variants) {
			resolver.register(v.getVariant(), v.getExtension());
		}
		return resolver;
	}
}