name.remal.gradle_plugins.merge_resources.mergers.SpringFactoriesMerger Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of merge-resources Show documentation
Remal Gradle plugins: merge-resources
The newest version!
package name.remal.gradle_plugins.merge_resources.mergers;

import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.Collections.singletonList;
import static java.util.function.Function.identity;
import static java.util.stream.Collectors.toList;
import static java.util.stream.Collectors.toMap;
import static name.remal.gradle_plugins.toolkit.ObjectUtils.defaultValue;
import static name.remal.gradle_plugins.toolkit.PropertiesUtils.loadProperties;
import static org.apache.commons.text.translate.EntityArrays.JAVA_CTRL_CHARS_ESCAPE;

import com.google.common.annotations.VisibleForTesting;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.InputStream;
import java.nio.file.Path;
import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.Set;
import java.util.TreeMap;
import java.util.regex.Pattern;
import java.util.stream.Stream;
import lombok.val;
import name.remal.gradle_plugins.merge_resources.ResourceMerger;
import name.remal.gradle_plugins.toolkit.ObjectUtils;
import org.apache.commons.text.translate.AggregateTranslator;
import org.apache.commons.text.translate.CharSequenceTranslator;
import org.apache.commons.text.translate.LookupTranslator;
import org.apache.commons.text.translate.UnicodeEscaper;
import org.gradle.api.file.RelativePath;

public abstract class SpringFactoriesMerger extends ResourceMerger {

    private static final Pattern IMPL_DELIMITER = Pattern.compile(",");

    private static final CharSequenceTranslator ESCAPER = new AggregateTranslator(
        new LookupTranslator(Stream.of("\\", " ", "=", ":", "#", "!")
            .collect(toMap(identity(), it -> "\\" + it))
        ),
        new LookupTranslator(JAVA_CTRL_CHARS_ESCAPE),
        UnicodeEscaper.outsideOf(0x0020, 0x007e)
    );


    @Override
    protected Collection getIncludes() {
        return singletonList("META-INF/spring.factories");
    }

    @Override
    protected InputStream merge(RelativePath relativePath, Collection files) throws Throwable {
        return mergeSpringFactories(files.stream()
            .map(File::toPath)
            .collect(toList())
        );
    }

    @VisibleForTesting
    static InputStream mergeSpringFactories(Collection paths) {
        val allServices = new TreeMap>();
        for (val path : paths) {
            val properties = loadProperties(path);
            for (val service : properties.stringPropertyNames()) {
                val impls = allServices.computeIfAbsent(service, __ -> new LinkedHashSet<>());
                IMPL_DELIMITER.splitAsStream(defaultValue(properties.getProperty(service)))
                    .map(String::trim)
                    .filter(ObjectUtils::isNotEmpty)
                    .forEach(impls::add);
            }
        }

        allServices.values().removeIf(ObjectUtils::isEmpty);

        val content = new StringBuilder();
        allServices.forEach((service, impls) -> {
            if (content.length() > 0) {
                content.append("\n\n");
            }

            content.append(ESCAPER.translate(service)).append(" = \\\n");
            boolean isFirstImpl = true;
            for (val impl : impls) {
                if (isFirstImpl) {
                    isFirstImpl = false;
                } else {
                    content.append(",\\\n");
                }

                content.append(ESCAPER.translate(impl));
            }
        });
        val bytes = content.toString().getBytes(UTF_8);
        return new ByteArrayInputStream(bytes);
    }

}