io.swagger.jaxrs2.integration.OpenApiServlet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swagger-jaxrs2 Show documentation
There is a newer version: 2.0.0-rc2
package io.swagger.jaxrs2.integration;

import io.swagger.oas.integration.OpenApiConfigurationException;
import io.swagger.oas.integration.api.OpenApiContext;
import io.swagger.oas.integration.OpenApiContextLocator;
import io.swagger.oas.models.OpenAPI;
import io.swagger.util.Json;
import io.swagger.util.Yaml;
import org.apache.commons.lang3.StringUtils;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;

import static io.swagger.jaxrs2.integration.ServletConfigContextUtils.getContextIdFromServletConfig;

public class OpenApiServlet extends HttpServlet {

  public static final String APPLICATION_JSON = "application/json";
  public static final String APPLICATION_YAML = "application/yaml";
  public static final String ACCEPT_HEADER = "Accept";


  @Override
  public void init(ServletConfig config) throws ServletException {

    super.init(config);
    String ctxId = getContextIdFromServletConfig(config);
    try {
      new ServletOpenApiContextBuilder()
              .servletConfig(config)
              .ctxId(ctxId)
              .buildContext(true);
    } catch (OpenApiConfigurationException e) {
      e.printStackTrace();
    }
  }


  // TODO move to own servlet non jaxrs project and reference from there
  // TODO cleanup and errors
  @Override
  protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

    String ctxId = getContextIdFromServletConfig(getServletConfig());
    OpenApiContext ctx = OpenApiContextLocator.getInstance().getOpenApiContext(ctxId);
    OpenAPI oas = ctx.read();

    String type = "json";

    String acceptHeader = req.getHeader(ACCEPT_HEADER);
    if (!StringUtils.isBlank(acceptHeader) && acceptHeader.toLowerCase().contains(APPLICATION_YAML)) {
      type = "yaml";
    } else {
        // check URL:
        if (req.getRequestURL().toString().toLowerCase().endsWith("yaml")) {
          type = "yaml";
        }
    }

    boolean pretty = false;
    if (ctx.getOpenApiConfiguration() != null && Boolean.TRUE.equals(ctx.getOpenApiConfiguration().isPrettyPrint())) {
      pretty = true;
    }

    resp.setStatus(200);

    if (type.equalsIgnoreCase("yaml")) {
      resp.setContentType(APPLICATION_YAML);
      PrintWriter pw=resp.getWriter();
      pw.write(pretty ? Yaml.pretty(oas) : Yaml.mapper().writeValueAsString(oas));
      pw.close();
    } else {
      resp.setContentType(APPLICATION_JSON);
      PrintWriter pw=resp.getWriter();
      pw.write(pretty ? Json.pretty(oas) : Json.mapper().writeValueAsString(oas));
      pw.close();
    }

  }
}