WfsResponseExceptionMapper.java

/**
 * Copyright (c) Codice Foundation
 *
 * This is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser
 * General Public License as published by the Free Software Foundation, either version 3 of the
 * License, or any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 * even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details. A copy of the GNU Lesser General Public License
 * is distributed along with this program and can be found at
 * <http://www.gnu.org/licenses/lgpl.html>.
 *
 **/
package org.codice.ddf.spatial.ogc.wfs.v1_0_0.catalog.source;

import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import javax.ws.rs.core.Response;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;

import org.apache.cxf.helpers.IOUtils;
import org.apache.cxf.jaxrs.client.ResponseExceptionMapper;
import org.apache.cxf.jaxrs.provider.JAXBElementProvider;
import org.codice.ddf.spatial.ogc.wfs.catalog.common.WfsException;

import ogc.schema.opengis.wfs.exception.v_1_0_0.ServiceExceptionReport;
import ogc.schema.opengis.wfs.exception.v_1_0_0.ServiceExceptionType;

public class WfsResponseExceptionMapper implements ResponseExceptionMapper<WfsException> {

    public WfsException fromResponse(Response response) {

        WfsException wfsEx = null;

        if (response != null) {
            if (response.getEntity() instanceof InputStream) {
                String msg = null;
                try {
                    InputStream is = (InputStream) response.getEntity();
                    is.reset();
                    msg = IOUtils.toString(is);
                } catch (IOException e) {
                    wfsEx = new WfsException(
                            "Error reading Response" + (msg != null ? ": " + msg : ""), e);
                }
                if (msg != null) {
                    try {
                        JAXBElementProvider<ServiceExceptionReport> provider = new JAXBElementProvider<ServiceExceptionReport>();
                        Unmarshaller um = provider.getJAXBContext(ServiceExceptionReport.class,
                                ServiceExceptionReport.class).createUnmarshaller();
                        ServiceExceptionReport report = (ServiceExceptionReport) um
                                .unmarshal(new StringReader(msg));
                        wfsEx = convertToWfsException(report);
                    } catch (JAXBException e) {
                        wfsEx = new WfsException("Error parsing Response: " + msg, e);
                    }
                }
            } else {
                wfsEx = new WfsException(
                        "Error reading response, entity type not understood: " + response
                                .getEntity().getClass().getName());
            }
            if (wfsEx != null) {
                wfsEx.setHttpStatus(response.getStatus());
            }
        } else {
            wfsEx = new WfsException("Error handling response, response is null");
        }
        return wfsEx;
    }

    private WfsException convertToWfsException(ServiceExceptionReport report) {

        WfsException wfsException = null;
        List<ServiceExceptionType> list = new ArrayList<ServiceExceptionType>(
                report.getServiceException());

        if (list.size() > 0) {
            Collections.reverse(list);
            StringBuilder exceptionMsg = new StringBuilder();
            for (ServiceExceptionType serviceException : list) {
                exceptionMsg.append(serviceException.getValue());
            }
            wfsException = new WfsException(exceptionMsg.toString());
        }

        if (null == wfsException) {
            wfsException = new WfsException(
                    "Empty Service Exception Report (version = " + report.getVersion() + ")");
        }
        return wfsException;
    }
}