net.truelicense.jax.rs.ConsumerLicenseManagementServiceExceptionMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of truelicense-jax-rs Show documentation

The TrueLicense JAX-RS module provides a RESTful web service interface for consuming license keys.

There is a newer version: 3.2.0

/*
 * Copyright (C) 2005-2017 Schlichtherle IT Services.
 * All rights reserved. Use is subject to license terms.
 */

package net.truelicense.jax.rs;

import net.truelicense.dto.ErrorDTO;
import net.truelicense.obfuscate.Obfuscate;

import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.ResponseBuilder;
import javax.ws.rs.ext.ExceptionMapper;
import javax.ws.rs.ext.Provider;
import javax.xml.bind.JAXBElement;
import javax.xml.namespace.QName;
import java.util.Objects;

import static javax.ws.rs.core.MediaType.*;

/**
 * Maps a consumer license management service exception to an HTTP response.
 * This class is immutable.
 *
 * @author Christian Schlichtherle
 * @since TrueLicense 2.3
 */
@Provider
@Produces({APPLICATION_JSON, APPLICATION_XML, TEXT_XML, TEXT_PLAIN})
public final class ConsumerLicenseManagementServiceExceptionMapper
        implements ExceptionMapper {

    @Obfuscate
    private static final String ERROR = "error";

    private static final QName error = new QName(ERROR);

    private final HttpHeaders headers;

    public ConsumerLicenseManagementServiceExceptionMapper(final @Context HttpHeaders headers) {
        this.headers = Objects.requireNonNull(headers);
    }

    @Override
    public Response toResponse(final ConsumerLicenseManagementServiceException ex) {
        final String message = ex.getMessage();
        final ResponseBuilder rb = Response.status(ex.getStatus()).type(TEXT_PLAIN_TYPE).entity(message);
        for (final MediaType mt : headers.getAcceptableMediaTypes()) {
            if (APPLICATION_JSON_TYPE.equals(mt)) {
                rb.type(mt).entity(new ErrorDTO(message));
                break;
            } else if (APPLICATION_XML_TYPE.equals(mt) || TEXT_XML_TYPE.equals(mt)) {
                rb.type(mt).entity(new JAXBElement<>(error, String.class, message));
                break;
            }
        }
        return rb.build();
    }
}