import org.apache.cxf.tools.common.model.JavaAnnotation;
import org.apache.cxf.tools.java2wsdl.generator.wsdl11.model.WrapperBeanClass;
public class WrapperBeanAnnotator implements Annotator {
public void annotate(final JavaAnnotatable clz) {
WrapperBeanClass wrapperBeanClass = null;
if (clz instanceof WrapperBeanClass) {
wrapperBeanClass = (WrapperBeanClass) clz;
} else {
throw new RuntimeException("WrapperBeanAnnotator expect JavaClass as input");
}
JavaAnnotation xmlRootElement = new JavaAnnotation("XmlRootElement");
xmlRootElement.addArgument("name", wrapperBeanClass.getElementName().getLocalPart());
xmlRootElement.addArgument("namespace", wrapperBeanClass.getElementName().getNamespaceURI());
JavaAnnotation xmlAccessorType = new JavaAnnotation("XmlAccessorType");
xmlAccessorType.addArgument("XmlAccessType.FIELD", "null", "");
JavaAnnotation xmlType = new JavaAnnotation("XmlType");
//xmlType.addArgument("name", wrapperBeanClass.getElementName().getLocalPart());
xmlType.addArgument("name", wrapperBeanClass.getElementName().getLocalPart());
xmlType.addArgument("namespace", wrapperBeanClass.getElementName().getNamespaceURI());
// Revisit: why annotation is string?
wrapperBeanClass.addAnnotation(xmlRootElement.toString());
wrapperBeanClass.addAnnotation(xmlAccessorType.toString());
wrapperBeanClass.addAnnotation(xmlType.toString());
wrapperBeanClass.addImport("javax.xml.bind.annotation.XmlAccessType");
wrapperBeanClass.addImport("javax.xml.bind.annotation.XmlAccessorType");
wrapperBeanClass.addImport("javax.xml.bind.annotation.XmlRootElement");
wrapperBeanClass.addImport("javax.xml.bind.annotation.XmlType");
}