org.apache.cxf.jaxrs.ext.MessageContextImpl Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.cxf.jaxrs.ext;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import javax.mail.internet.InternetHeaders;
import javax.servlet.ServletConfig;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.Request;
import javax.ws.rs.core.SecurityContext;
import javax.ws.rs.core.UriInfo;
import javax.ws.rs.ext.ContextResolver;
import javax.ws.rs.ext.Providers;
import org.apache.cxf.attachment.AttachmentImpl;
import org.apache.cxf.attachment.AttachmentUtil;
import org.apache.cxf.endpoint.Endpoint;
import org.apache.cxf.interceptor.AttachmentOutInterceptor;
import org.apache.cxf.jaxrs.ext.multipart.Attachment;
import org.apache.cxf.jaxrs.ext.multipart.MultipartBody;
import org.apache.cxf.jaxrs.interceptor.AttachmentInputInterceptor;
import org.apache.cxf.jaxrs.interceptor.AttachmentOutputInterceptor;
import org.apache.cxf.jaxrs.utils.JAXRSUtils;
import org.apache.cxf.message.Message;
public class MessageContextImpl implements MessageContext {
private Message m;
public MessageContextImpl(Message m) {
this.m = m;
}
public Object get(Object key) {
if (MultipartBody.INBOUND_MESSAGE_ATTACHMENTS.equals(key.toString())) {
return createAttachments(MultipartBody.INBOUND_MESSAGE_ATTACHMENTS);
}
return m.get(key);
}
public T getContent(Class format) {
return m.getContent(format);
}
public Object getContextualProperty(Object key) {
return m.getContextualProperty(key.toString());
}
public T getContext(Class contextClass) {
return getContext(contextClass, contextClass);
}
protected T getContext(Type genericType, Class clazz) {
return JAXRSUtils.createContextValue(m, genericType, clazz);
}
public T getResolver(Class resolverClazz, Class resolveClazz) {
if (ContextResolver.class == resolverClazz) {
return resolverClazz.cast(getContext(resolveClazz, ContextResolver.class));
}
return null;
}
public Request getRequest() {
return getContext(Request.class);
}
public HttpHeaders getHttpHeaders() {
return getContext(HttpHeaders.class);
}
public Providers getProviders() {
return getContext(Providers.class);
}
public SecurityContext getSecurityContext() {
return getContext(SecurityContext.class);
}
public UriInfo getUriInfo() {
return getContext(UriInfo.class);
}
public HttpServletRequest getHttpServletRequest() {
return JAXRSUtils.createServletResourceValue(m, HttpServletRequest.class);
}
public HttpServletResponse getHttpServletResponse() {
return JAXRSUtils.createServletResourceValue(m, HttpServletResponse.class);
}
public ServletConfig getServletConfig() {
return JAXRSUtils.createServletResourceValue(m, ServletConfig.class);
}
public ServletContext getServletContext() {
return JAXRSUtils.createServletResourceValue(m, ServletContext.class);
}
public void put(Object key, Object value) {
if (MultipartBody.OUTBOUND_MESSAGE_ATTACHMENTS.equals(key.toString())) {
convertToAttachments(value);
}
m.put(key.toString(), value);
}
private void convertToAttachments(Object value) {
List handlers = (List)value;
List atts =
new ArrayList();
for (int i = 1; i < handlers.size(); i++) {
Attachment handler = (Attachment)handlers.get(i);
AttachmentImpl att = new AttachmentImpl(handler.getContentId(), handler.getDataHandler());
for (String key : handler.getHeaders().keySet()) {
att.setHeader(key, att.getHeader(key));
}
att.setXOP(false);
atts.add(att);
}
Message outMessage = getOutMessage();
outMessage.setAttachments(atts);
outMessage.put(AttachmentOutInterceptor.WRITE_ATTACHMENTS, "true");
Attachment root = (Attachment)handlers.get(0);
AttachmentOutputInterceptor attInterceptor =
new AttachmentOutputInterceptor(outMessage.get(Message.CONTENT_TYPE).toString(),
root.getHeaders());
outMessage.put(Message.CONTENT_TYPE, root.getContentType().toString());
attInterceptor.handleMessage(outMessage);
}
private Message getOutMessage() {
Message message = m.getExchange().getOutMessage();
if (message == null) {
Endpoint ep = m.getExchange().get(Endpoint.class);
message = ep.getBinding().createMessage();
m.getExchange().setOutMessage(message);
}
return message;
}
private MultipartBody createAttachments(String propertyName) {
Message inMessage = m.getExchange().getInMessage();
Object o = inMessage.get(propertyName);
if (o != null) {
return (MultipartBody)o;
}
new AttachmentInputInterceptor().handleMessage(inMessage);
List newAttachments = new LinkedList();
try {
Attachment first = new Attachment(AttachmentUtil.createAttachment(
inMessage.getContent(InputStream.class),
(InternetHeaders)inMessage.get(InternetHeaders.class.getName())));
newAttachments.add(first);
} catch (IOException ex) {
throw new WebApplicationException(500);
}
Collection childAttachments = inMessage.getAttachments();
if (childAttachments == null) {
childAttachments = Collections.emptyList();
}
childAttachments.size();
for (org.apache.cxf.message.Attachment a : childAttachments) {
newAttachments.add(new Attachment(a));
}
MultipartBody body = new MultipartBody(newAttachments, getHttpHeaders().getMediaType(), false);
inMessage.put(propertyName, body);
return body;
}
}