name.remal.gradle_plugins.merge_resources.mergers.MetaInfServicesMerger 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.lang.String.join;
import static java.nio.charset.StandardCharsets.UTF_8;
import static java.nio.file.Files.newInputStream;
import static java.util.Collections.singletonList;
import static java.util.stream.Collectors.toList;
import static name.remal.gradle_plugins.toolkit.FunctionUtils.toSubstringedBefore;
import static name.remal.gradle_plugins.toolkit.InputOutputStreamUtils.readStringFromStream;

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.regex.Pattern;
import lombok.val;
import name.remal.gradle_plugins.merge_resources.ResourceMerger;
import name.remal.gradle_plugins.toolkit.ObjectUtils;
import org.gradle.api.file.RelativePath;

public abstract class MetaInfServicesMerger extends ResourceMerger {

    private static final Pattern NEW_LINES = Pattern.compile("[\\n\\r]+");


    @Override
    protected Collection getIncludes() {
        return singletonList("META-INF/services/*");
    }

    @Override
    protected Collection getExcludes() {
        return singletonList("META-INF/services/org.codehaus.groovy.runtime.ExtensionModule");
    }

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

    @VisibleForTesting
    static InputStream mergeMetaInfServices(Collection paths) throws Throwable {
        val services = new LinkedHashSet();
        for (val path : paths) {
            final String content;
            try (val inputStream = newInputStream(path)) {
                content = readStringFromStream(inputStream, UTF_8);
            }

            NEW_LINES.splitAsStream(content)
                .map(toSubstringedBefore("#"))
                .map(String::trim)
                .filter(ObjectUtils::isNotEmpty)
                .forEach(services::add);
        }

        val mergedContent = join("\n", services);
        val bytes = mergedContent.getBytes(UTF_8);
        return new ByteArrayInputStream(bytes);
    }

}