io.swagger.v3.jaxrs2.SwaggerSerializers Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swagger-jaxrs2-jakarta Show documentation
swagger-jaxrs2-jakarta
There is a newer version: 2.2.24
Show newest version
package io.swagger.v3.jaxrs2;

import io.swagger.v3.core.util.Json;
import io.swagger.v3.core.util.Json31;
import io.swagger.v3.core.util.Yaml;
import io.swagger.v3.core.util.Yaml31;
import io.swagger.v3.oas.models.OpenAPI;

import jakarta.ws.rs.Produces;
import jakarta.ws.rs.core.MediaType;
import jakarta.ws.rs.core.MultivaluedMap;
import jakarta.ws.rs.ext.MessageBodyWriter;
import jakarta.ws.rs.ext.Provider;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;

@Provider
@Produces({MediaType.APPLICATION_JSON, MediaType.APPLICATION_XML, "application/yaml"})
public class SwaggerSerializers implements MessageBodyWriter {
    static boolean prettyPrint = false;
    static boolean openapi31 = false;

    public static void setPrettyPrint(boolean shouldPrettyPrint) {
        SwaggerSerializers.prettyPrint = shouldPrettyPrint;
    }

    public static void setOpenapi31(boolean openapi31) {
        SwaggerSerializers.openapi31 = openapi31;
    }

    @Override
    public boolean isWriteable(Class type, Type genericType, Annotation[] annotations,
                               MediaType mediaType) {
        return OpenAPI.class.isAssignableFrom(type);
    }

    @Override
    public long getSize(OpenAPI data, Class type, Type genericType, Annotation[] annotations, MediaType mediaType) {
        return -1L;
    }

    @Override
    public void writeTo(OpenAPI data,
                        Class type,
                        Type genericType,
                        Annotation[] annotations,
                        MediaType mediaType,
                        MultivaluedMap headers,
                        OutputStream out) throws IOException {
        if (mediaType.isCompatible(MediaType.APPLICATION_JSON_TYPE)) {
            if (prettyPrint) {
                if (openapi31) {
                    out.write(Json31.pretty().writeValueAsBytes(data));
                } else {
                    out.write(Json.pretty().writeValueAsBytes(data));
                }
            } else {
                if (openapi31) {
                    out.write(Json31.mapper().writeValueAsBytes(data));
                } else {
                    out.write(Json.mapper().writeValueAsBytes(data));
                }
            }
        } else if (mediaType.toString().startsWith("application/yaml")) {
            headers.remove("Content-Type");
            headers.add("Content-Type", "application/yaml");
            if (prettyPrint) {
                if (openapi31) {
                    out.write(Yaml31.pretty().writeValueAsBytes(data));
                } else {
                    out.write(Yaml.pretty().writeValueAsBytes(data));
                }
            } else {
                if (openapi31) {
                    out.write(Yaml31.mapper().writeValueAsBytes(data));
                } else {
                    out.write(Yaml.mapper().writeValueAsBytes(data));
                }
            }
        } else if (mediaType.isCompatible(MediaType.APPLICATION_XML_TYPE)) {
            headers.remove("Content-Type");
            headers.add("Content-Type", MediaType.APPLICATION_JSON);
            if (prettyPrint) {
                if (openapi31) {
                    out.write(Json31.pretty().writeValueAsBytes(data));
                } else {
                    out.write(Json.pretty().writeValueAsBytes(data));
                }
            } else {
                if (openapi31) {
                    out.write(Json31.mapper().writeValueAsBytes(data));
                } else {
                    out.write(Json.mapper().writeValueAsBytes(data));
                }
            }
        }
    }
}