/**
* 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.jaxws.interceptors;
import java.awt.Component;
import java.awt.Graphics;
import java.awt.Image;
import java.awt.MediaTracker;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.StringWriter;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.UUID;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.activation.DataHandler;
import javax.activation.DataSource;
import javax.activation.URLDataSource;
import javax.imageio.ImageIO;
import javax.imageio.ImageWriter;
import javax.mail.util.ByteArrayDataSource;
import javax.xml.bind.JAXBContext;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.transform.Source;
import javax.xml.transform.stream.StreamSource;
import com.sun.xml.bind.v2.runtime.JAXBContextImpl;
import com.sun.xml.bind.v2.util.DataSourceSource;
import org.apache.cxf.attachment.AttachmentImpl;
import org.apache.cxf.binding.soap.SoapMessage;
import org.apache.cxf.binding.soap.interceptor.AbstractSoapInterceptor;
import org.apache.cxf.binding.soap.model.SoapBodyInfo;
import org.apache.cxf.common.logging.LogUtils;
import org.apache.cxf.databinding.DataBinding;
import org.apache.cxf.helpers.CastUtils;
import org.apache.cxf.helpers.IOUtils;
import org.apache.cxf.interceptor.AttachmentOutInterceptor;
import org.apache.cxf.interceptor.Fault;
import org.apache.cxf.jaxb.JAXBDataBinding;
import org.apache.cxf.message.Attachment;
import org.apache.cxf.message.Exchange;
import org.apache.cxf.message.Message;
import org.apache.cxf.phase.Phase;
import org.apache.cxf.service.Service;
import org.apache.cxf.service.model.BindingMessageInfo;
import org.apache.cxf.service.model.BindingOperationInfo;
import org.apache.cxf.service.model.MessagePartInfo;
import org.apache.cxf.staxutils.StaxUtils;
public class SwAOutInterceptor extends AbstractSoapInterceptor {
private static final Logger LOG = LogUtils.getL7dLogger(SwAOutInterceptor.class);
AttachmentOutInterceptor attachOut = new AttachmentOutInterceptor();
public SwAOutInterceptor() {
super(Phase.PRE_LOGICAL);
addAfter(HolderOutInterceptor.class.getName());
addBefore(WrapperClassOutInterceptor.class.getName());
}
public void handleMessage(SoapMessage message) throws Fault {
Exchange ex = message.getExchange();
BindingOperationInfo bop = ex.get(BindingOperationInfo.class);
if (bop == null) {
return;
}
if (bop.isUnwrapped()) {
bop = bop.getWrappedOperation();
}
boolean client = isRequestor(message);
BindingMessageInfo bmi = client ? bop.getInput() : bop.getOutput();
if (bmi == null) {
return;
}
SoapBodyInfo sbi = bmi.getExtensor(SoapBodyInfo.class);
if (sbi == null || sbi.getAttachments() == null || sbi.getAttachments().size() == 0) {
Service s = ex.get(Service.class);
DataBinding db = s.getDataBinding();
if (db instanceof JAXBDataBinding
&& hasSwaRef((JAXBDataBinding) db)) {
setupAttachmentOutput(message);
}
return;
}
Collection<Attachment> atts = setupAttachmentOutput(message);
List<Object> outObjects = CastUtils.cast(message.getContent(List.class));
for (MessagePartInfo mpi : sbi.getAttachments()) {
String partName = mpi.getConcreteName().getLocalPart();
String ct = (String) mpi.getProperty(Message.CONTENT_TYPE);
String id = new StringBuilder().append(partName)
.append("=")
.append(UUID.randomUUID())
.append("@apache.org").toString();
// this assumes things are in order...
int idx = mpi.getMessageInfo().getMessagePartIndex(mpi);
Object o = outObjects.get(idx);
if (o == null) {
continue;
}
outObjects.set(idx, null);
DataHandler dh = null;
// This code could probably be refactored out somewhere...
if (o instanceof Source) {
dh = new DataHandler(createDataSource((Source)o, ct));
} else if (o instanceof Image) {
// TODO: make this streamable. This is one of my pet
// peeves in JAXB RI as well, so if you fix this, submit the
// code to the JAXB RI as well (see RuntimeBuiltinLeafInfoImpl)! - DD
ByteArrayOutputStream bos = new ByteArrayOutputStream();
Iterator<ImageWriter> writers = ImageIO.getImageWritersByMIMEType(ct);
if (writers.hasNext()) {
ImageWriter writer = writers.next();
try {
BufferedImage bimg = convertToBufferedImage((Image) o);
writer.setOutput(ImageIO.createImageOutputStream(bos));
writer.write(bimg);
writer.dispose();
bos.close();
} catch (IOException e) {
throw new Fault(e);
}
} else {
throw new Fault(new org.apache.cxf.common.i18n.Message("ATTACHMENT_NOT_SUPPORTED",
LOG, ct));
}
dh = new DataHandler(new ByteArrayDataSource(bos.toByteArray(), ct));
} else if (o instanceof DataHandler) {
dh = (DataHandler) o;
ct = dh.getContentType();
try {
if ("text/xml".equals(ct)
&& dh.getContent() instanceof Source) {
dh = new DataHandler(createDataSource((Source)dh.getContent(), ct));
}
} catch (IOException e) {
//ignore, use same dh
}
} else if (dh == null) {
throw new Fault(new org.apache.cxf.common.i18n.Message("ATTACHMENT_NOT_SUPPORTED",
LOG, o.getClass()));
} else if (dh.getDataSource() instanceof URLDataSource) {
URLDataSource ds = (URLDataSource)dh.getDataSource();
dh = new DataHandler(ds.getURL());
ct = ds.getContentType();
}
AttachmentImpl att = new AttachmentImpl(id);
att.setDataHandler(dh);
att.setHeader("Content-Type", ct);
atts.add(att);
}
}
private boolean hasSwaRef(JAXBDataBinding db) {
JAXBContext context = db.getContext();
if (context instanceof JAXBContextImpl) {
JAXBContextImpl riCtx = (JAXBContextImpl) context;
try {
// We're using reflection here because this won't work on
// JAXB < 2.0.5 and Java 6 ships with 2.0.2. This means
// SwA won't work correctly unless you're using at least
// JAXB 2.0.5.
Method m = riCtx.getClass().getMethod("hasSwaRef", new Class[0]);
return (Boolean) m.invoke(riCtx, new Object[0]);
} catch (Exception e) {
LOG.log(Level.FINER, "Could not check hasSwaRef", e);
}
}
return false;
}
private DataSource createDataSource(Source o, String ct) {
DataSource ds = null;
if (o instanceof DataSourceSource) {
ds = (DataSource) o;
} else if (o instanceof StreamSource) {
StreamSource src = (StreamSource)o;
ByteArrayOutputStream bos = new ByteArrayOutputStream();
try {
if (src.getInputStream() != null) {
IOUtils.copy(src.getInputStream(), bos, 1024);
ds = new ByteArrayDataSource(bos.toByteArray(), ct);
} else {
ds = new ByteArrayDataSource(IOUtils.toString(src.getReader()),
ct);
}
} catch (IOException e) {
throw new Fault(e);
}
} else {
XMLStreamReader reader = StaxUtils.createXMLStreamReader((Source)o);
StringWriter stringWriter = new StringWriter();
XMLStreamWriter writer = StaxUtils.createXMLStreamWriter(stringWriter);
try {
StaxUtils.copy(reader, writer);
writer.flush();
ds = new ByteArrayDataSource(stringWriter.toString(), ct);
} catch (XMLStreamException e1) {
throw new Fault(e1);
} catch (IOException e) {
throw new Fault(e);
}
}
return ds;
}
private BufferedImage convertToBufferedImage(Image image) throws IOException {
if (image instanceof BufferedImage) {
return (BufferedImage)image;
}
// Wait until the image is completely loaded
MediaTracker tracker = new MediaTracker(new Component() { });
tracker.addImage(image, 0);
try {
tracker.waitForAll();
} catch (InterruptedException e) {
throw new Fault(e);
}
// Create a BufferedImage so we can write it out later
BufferedImage bufImage = new BufferedImage(
image.getWidth(null),
image.getHeight(null),
BufferedImage.TYPE_INT_ARGB);
Graphics g = bufImage.createGraphics();
g.drawImage(image, 0, 0, null);
return bufImage;
}
private Collection<Attachment> setupAttachmentOutput(SoapMessage message) {
// We have attachments, so add the interceptor
message.getInterceptorChain().add(attachOut);
// We should probably come up with another property for this
message.put(AttachmentOutInterceptor.WRITE_ATTACHMENTS, Boolean.TRUE);
Collection<Attachment> atts = message.getAttachments();
if (atts == null) {
atts = new ArrayList<Attachment>();
message.setAttachments(atts);
}
return atts;
}
}