/*
* 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.tuscany.core.databinding.impl;
import java.util.List;
import javax.xml.namespace.QName;
import org.apache.tuscany.spi.annotation.Autowire;
import org.apache.tuscany.spi.databinding.DataBinding;
import org.apache.tuscany.spi.databinding.DataBindingRegistry;
import org.apache.tuscany.spi.databinding.Mediator;
import org.apache.tuscany.spi.databinding.PullTransformer;
import org.apache.tuscany.spi.databinding.TransformationContext;
import org.apache.tuscany.spi.databinding.TransformationException;
import org.apache.tuscany.spi.databinding.Transformer;
import org.apache.tuscany.spi.databinding.WrapperHandler;
import org.apache.tuscany.spi.databinding.extension.TransformerExtension;
import org.apache.tuscany.spi.idl.ElementInfo;
import org.apache.tuscany.spi.idl.WrapperInfo;
import org.apache.tuscany.spi.model.DataType;
import org.apache.tuscany.spi.model.Operation;
import org.osoa.sca.annotations.Service;
/**
* This is a special transformer to transform the output from one IDL to the
* other one
*/
@Service(Transformer.class)
public class Output2OutputTransformer extends TransformerExtension<Object, Object> implements
PullTransformer<Object, Object> {
private static final String IDL_OUTPUT = "idl:output";
protected DataBindingRegistry dataBindingRegistry;
protected Mediator mediator;
/**
* @param wrapperHandler
*/
public Output2OutputTransformer() {
super();
}
/**
* @param mediator the mediator to set
*/
@Autowire
public void setMediator(Mediator mediator) {
this.mediator = mediator;
}
/**
* @param dataBindingRegistry the dataBindingRegistry to set
*/
@Autowire
public void setDataBindingRegistry(DataBindingRegistry dataBindingRegistry) {
this.dataBindingRegistry = dataBindingRegistry;
}
@Override
public String getSourceDataBinding() {
return IDL_OUTPUT;
}
@Override
public String getTargetDataBinding() {
return IDL_OUTPUT;
}
/**
* @see org.apache.tuscany.spi.databinding.extension.TransformerExtension#getSourceType()
*/
@Override
protected Class getSourceType() {
return Object.class;
}
/**
* @see org.apache.tuscany.spi.databinding.extension.TransformerExtension#getTargetType()
*/
@Override
protected Class getTargetType() {
return Object.class;
}
/**
* @see org.apache.tuscany.spi.databinding.Transformer#getWeight()
*/
public int getWeight() {
return 10;
}
private WrapperHandler getWapperHandler(Operation<?> operation) {
String dataBindingId;
dataBindingId = operation.getDataBinding();
DataBinding dataBinding = dataBindingRegistry.getDataBinding(dataBindingId);
WrapperHandler wrapperHandler = dataBinding == null ? null : dataBinding.getWrapperHandler();
if (wrapperHandler == null) {
throw new TransformationException(
"No wrapper handler is provided for databinding: " + dataBindingId);
}
return wrapperHandler;
}
private WrapperHandler getWapperHandler(String dataBindingId) {
DataBinding dataBinding = dataBindingRegistry.getDataBinding(dataBindingId);
return dataBinding == null ? null : dataBinding.getWrapperHandler();
}
@SuppressWarnings("unchecked")
public Object transform(Object response, TransformationContext context) {
try {
DataType<DataType> sourceType = context.getSourceDataType();
Operation<?> sourceOp = (Operation<?>)sourceType.getMetadata(Operation.class.getName());
boolean sourceWrapped = sourceOp != null && sourceOp.isWrapperStyle();
WrapperHandler sourceWrapperHandler = null;
if (sourceWrapped) {
sourceWrapperHandler = getWapperHandler(sourceOp);
}
DataType<DataType> targetType = context.getTargetDataType();
Operation<?> targetOp = (Operation<?>)targetType.getMetadata(Operation.class.getName());
boolean targetWrapped = targetOp != null && targetOp.isWrapperStyle();
WrapperHandler targetWrapperHandler = null;
if (targetWrapped) {
targetWrapperHandler = getWapperHandler(targetOp);
}
if ((!sourceWrapped) && targetWrapped) {
// Unwrapped --> Wrapped
WrapperInfo wrapper = targetOp.getWrapper();
Object targetWrapper =
targetWrapperHandler.create(wrapper.getOutputWrapperElement(), context);
if (response == null) {
return targetWrapper;
}
ElementInfo argElement = wrapper.getOutputChildElements().get(0);
DataType<QName> argType = wrapper.getUnwrappedOutputType();
Object child = response;
child = mediator.mediate(response, sourceType.getLogical(), argType, context.getMetadata());
targetWrapperHandler.setChild(targetWrapper, 0, argElement, child);
return targetWrapper;
} else if (sourceWrapped && (!targetWrapped)) {
// Wrapped to Unwrapped
Object sourceWrapper = response;
List<ElementInfo> childElements = sourceOp.getWrapper().getOutputChildElements();
ElementInfo childElement = childElements.get(0);
targetWrapperHandler = getWapperHandler(targetType.getLogical().getDataBinding());
if (targetWrapperHandler != null) {
ElementInfo wrapperElement = sourceOp.getWrapper().getInputWrapperElement();
// Object targetWrapper =
// targetWrapperHandler.create(wrapperElement, context);
DataType<QName> targetWrapperType =
new DataType<QName>(targetType.getLogical().getDataBinding(), Object.class,
wrapperElement.getQName());
Object targetWrapper =
mediator.mediate(sourceWrapper, sourceType.getLogical(), targetWrapperType, context
.getMetadata());
return targetWrapperHandler.getChild(targetWrapper, 0, childElement);
} else {
Object child = sourceWrapperHandler.getChild(sourceWrapper, 0, childElement);
DataType<?> childType = sourceOp.getWrapper().getUnwrappedOutputType();
return mediator.mediate(child, childType, targetType.getLogical(), context.getMetadata());
}
} else {
// FIXME: Do we want to handle wrapped to wrapped?
return mediator.mediate(response, sourceType.getLogical(), targetType.getLogical(), context
.getMetadata());
}
} catch (Exception e) {
throw new TransformationException(e);
}
}
}