001/*
002 * Licensed to DuraSpace under one or more contributor license agreements.
003 * See the NOTICE file distributed with this work for additional information
004 * regarding copyright ownership.
005 *
006 * DuraSpace licenses this file to you under the Apache License,
007 * Version 2.0 (the "License"); you may not use this file except in
008 * compliance with the License.  You may obtain a copy of the License at
009 *
010 *     http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.fcrepo.http.commons.exceptionhandlers;
019
020import org.slf4j.Logger;
021
022import javax.ws.rs.WebApplicationException;
023import javax.ws.rs.core.Response;
024import javax.ws.rs.ext.ExceptionMapper;
025import javax.ws.rs.ext.Provider;
026
027import static javax.ws.rs.core.Response.fromResponse;
028import static org.fcrepo.http.commons.domain.RDFMediaType.TEXT_PLAIN_WITH_CHARSET;
029import static org.slf4j.LoggerFactory.getLogger;
030
031/**
032 * Handle Jersey WebApplicationException
033 *
034 * @author lsitu
035 */
036@Provider
037public class WebApplicationExceptionMapper implements
038        ExceptionMapper<WebApplicationException>, ExceptionDebugLogging {
039
040    private static final Logger LOGGER =
041        getLogger(WebApplicationExceptionMapper.class);
042
043    @Override
044    public Response toResponse(final WebApplicationException e) {
045        LOGGER.warn("Web application error", e);
046        final String msg = null == e.getCause() ? e.getMessage() : e.getCause().getMessage();
047        // 204, 205, 304 MUST NOT contain an entity body - RFC2616
048        switch (e.getResponse().getStatus()) {
049            case 204:
050            case 205:
051            case 304:
052                return fromResponse(e.getResponse()).entity(null).build();
053            default:
054                return fromResponse(e.getResponse()).entity(msg).type(TEXT_PLAIN_WITH_CHARSET).build();
055        }
056    }
057}