mirror of https://github.com/jacekkow/keycloak-protocol-cas

Jacek Kowalski
2023-11-24 fcbc7c31990c50d1dde8c676a901d00421d73d08
commit | author | age
513246 1 package org.keycloak.protocol.cas.utils;
MP 2
fdb9f6 3 import jakarta.ws.rs.BadRequestException;
JK 4 import jakarta.ws.rs.core.*;
513246 5 import org.jboss.resteasy.spi.HttpRequest;
MP 6 import org.keycloak.protocol.cas.CASLoginProtocol;
7
8 public class ContentTypeHelper {
9     private final HttpRequest request;
10     private final Request restRequest;
11     private final UriInfo uriInfo;
12
13     public ContentTypeHelper(HttpRequest request, Request restRequest, UriInfo uriInfo) {
14         this.request = request;
15         this.restRequest = restRequest;
16         this.uriInfo = uriInfo;
17     }
18
19     public MediaType selectResponseType() {
20         String format = uriInfo.getQueryParameters().getFirst(CASLoginProtocol.FORMAT_PARAM);
21         if (format != null && !format.isEmpty()) {
5ba0b0 22             //if parameter is set, it overrides all header values (see spec section 2.5.1)
MP 23             request.getMutableHeaders().putSingle(HttpHeaders.ACCEPT, "application/" + format.toLowerCase());
513246 24         }
de7d25 25         try {
MP 26             Variant variant = restRequest.selectVariant(Variant.mediaTypes(MediaType.APPLICATION_XML_TYPE, MediaType.APPLICATION_JSON_TYPE).build());
27             return variant == null ? MediaType.APPLICATION_XML_TYPE : variant.getMediaType();
28         } catch (BadRequestException e) {
29             //the default Accept header set by java.net.HttpURLConnection is invalid (cf. RESTEASY-960)
30             return MediaType.APPLICATION_XML_TYPE;
31         }
513246 32     }
MP 33 }