io.polyapi.plugin.service.PolyCodeWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of polyapi-maven-plugin Show documentation

Maven plugin to run handle Poly API functions.

There is a newer version: 0.15.3

package io.polyapi.plugin.service;

import static java.nio.charset.Charset.defaultCharset;
import static java.util.stream.Collectors.toMap;

import java.io.ByteArrayOutputStream;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;

import com.sun.codemodel.CodeWriter;
import com.sun.codemodel.JPackage;

import lombok.extern.slf4j.Slf4j;

@Slf4j
public class PolyCodeWriter extends CodeWriter implements AutoCloseable {
    private final Map outputStreams = new HashMap<>();

    @Override
    public OutputStream openBinary(JPackage pkg, String fileName) throws IOException {
        log.debug("Opening binary on package {} for file {}", pkg.name(), fileName);
        var outputStream = new ByteArrayOutputStream();
        outputStreams.put(fileName.replace(".java", ""), outputStream);
        return new FilterOutputStream(outputStream) {
            @Override
            public void close() {
                // don't let this stream close
            }
        };
    }

    public Map getClasses() {
        return outputStreams.entrySet().stream()
                .collect(toMap(Map.Entry::getKey, entry -> entry.getValue().toString(defaultCharset())));
    }

    @Override
    public void close() throws IOException {
        for (OutputStream outputStream : outputStreams.values()) {
            outputStream.close();
        }
    }
}