com.github.davidmoten.oas3.puml.ConverterMain Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of openapi-to-plantuml Show documentation

Generates PlantUML file from an OpenAPI 3.0 Definition

There is a newer version: 0.1.30

package com.github.davidmoten.oas3.puml;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.Arrays;
import java.util.stream.Collectors;

import net.sourceforge.plantuml.FileFormat;
import net.sourceforge.plantuml.FileFormatOption;
import net.sourceforge.plantuml.SourceStringReader;
import net.sourceforge.plantuml.core.DiagramDescription;

public final class ConverterMain {

    private ConverterMain() {
        // prevent instantiation
    }

    static DiagramDescription writeFileFormatFromPuml(String puml, String filename,
            FileFormat fileFormat) throws IOException {
        File file = new File(filename);
        SourceStringReader reader = new SourceStringReader(puml);
        try (OutputStream os = new BufferedOutputStream(new FileOutputStream(file))) {
            // Write the first image to "os"
            return reader.outputImage(os, new FileFormatOption(fileFormat));
        }
    }

    public static void main(String[] args) throws IOException {
        String usage = "Usage: java -jar openapi-to-plantuml-all.jar   "
                + "\n  File formats are:\n    PUML\n" + Arrays.stream(FileFormat.values())
                        .map(x -> "    " + x + "\n").collect(Collectors.joining());
        if (args.length != 3) {
            System.out.println(usage);
            throw new IllegalArgumentException("must pass 3 arguments");
        } else {
            String puml = Converter.openApiToPuml(new File(args[0]));
            String format = args[1];
            File out = new File(args[2]);
            if (format.equals("PUML")) {
                Files.write(out.toPath(), puml.getBytes(StandardCharsets.UTF_8));
            } else {
                FileFormat ff = FileFormat.valueOf(format);
                writeFileFormatFromPuml(puml, out.getPath(), ff);
            }
        }
    }

}