package org.geotools.maven.xmlcodegen.templates;
import java.io.IOException;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.sax.SAXTransformerFactory;
import javax.xml.transform.sax.TransformerHandler;
import javax.xml.transform.stream.StreamResult;
import org.eclipse.xsd.XSDTypeDefinition;
import org.geotools.maven.xmlcodegen.SchemaGenerator;
import org.opengis.feature.type.AttributeDescriptor;
import org.opengis.feature.type.AttributeType;
import org.opengis.feature.type.ComplexType;
import org.opengis.feature.type.Name;
import org.opengis.feature.type.PropertyDescriptor;
import org.opengis.feature.type.PropertyType;
import org.opengis.feature.type.Schema;
public class CycleSchemaClassTemplate
{
protected static String nl;
public static synchronized CycleSchemaClassTemplate create(String lineSeparator)
{
nl = lineSeparator;
CycleSchemaClassTemplate result = new CycleSchemaClassTemplate();
nl = null;
return result;
}
public final String NL = nl == null ? (System.getProperties().getProperty("line.separator")) : nl;
protected final String TEXT_1 = "";
protected final String TEXT_2 = NL + NL + "import java.util.ArrayList;" + NL + "import java.util.Collection;" + NL + "import java.util.List;" + NL + "import java.util.TreeMap;" + NL + "" + NL + "import org.geotools.feature.NameImpl;" + NL + "import org.geotools.feature.type.AbstractLazyAttributeTypeImpl;" + NL + "import org.geotools.feature.type.AbstractLazyComplexTypeImpl;" + NL + "import org.geotools.feature.type.AttributeDescriptorImpl;" + NL + "import org.geotools.feature.type.SchemaImpl;" + NL + "import org.opengis.feature.type.AttributeType;" + NL + "import org.opengis.feature.type.ComplexType;" + NL + "import org.opengis.feature.type.Name;" + NL + "import org.opengis.feature.type.PropertyDescriptor;" + NL + "import org.opengis.feature.type.Schema;";
protected final String TEXT_3 = NL + "import ";
protected final String TEXT_4 = ";";
protected final String TEXT_5 = NL + NL + "public class ";
protected final String TEXT_6 = "Schema extends SchemaImpl {" + NL;
protected final String TEXT_7 = NL + " /**" + NL + " * <p>" + NL + " * <pre>" + NL + " * <code>";
protected final String TEXT_8 = NL + " * ";
protected final String TEXT_9 = NL + " *" + NL + " * </code>" + NL + " * </pre>" + NL + " * </p>" + NL + " *" + NL + " * @generated" + NL + " */";
protected final String TEXT_10 = NL + " public static final ComplexType ";
protected final String TEXT_11 = "_TYPE = build_";
protected final String TEXT_12 = "_TYPE();" + NL + " " + NL + " private static ComplexType build_";
protected final String TEXT_13 = "_TYPE() {" + NL + " ComplexType builtType = new AbstractLazyComplexTypeImpl(" + NL + " new NameImpl(\"";
protected final String TEXT_14 = "\",\"";
protected final String TEXT_15 = "\"),";
protected final String TEXT_16 = NL + " ";
protected final String TEXT_17 = ", ";
protected final String TEXT_18 = ", ";
protected final String TEXT_19 = ", ";
protected final String TEXT_20 = ") {";
protected final String TEXT_21 = NL + " public static final AttributeType ";
protected final String TEXT_22 = "_TYPE = build_";
protected final String TEXT_23 = "_TYPE();" + NL + " " + NL + " private static AttributeType build_";
protected final String TEXT_24 = "_TYPE() {" + NL + " AttributeType builtType = new AbstractLazyAttributeTypeImpl(" + NL + " new NameImpl(\"";
protected final String TEXT_25 = "\",\"";
protected final String TEXT_26 = "\"),";
protected final String TEXT_27 = NL + " ";
protected final String TEXT_28 = ", ";
protected final String TEXT_29 = ", ";
protected final String TEXT_30 = ", ";
protected final String TEXT_31 = ", ";
protected final String TEXT_32 = ") {";
protected final String TEXT_33 = NL + " @Override" + NL + " public AttributeType buildSuper() {" + NL + " return ";
protected final String TEXT_34 = ";" + NL + " }";
protected final String TEXT_35 = NL + " @Override" + NL + " public Collection<PropertyDescriptor> buildDescriptors() {";
protected final String TEXT_36 = NL + " return null;";
protected final String TEXT_37 = NL + " List<PropertyDescriptor> descriptors = new ArrayList<PropertyDescriptor>();";
protected final String TEXT_38 = NL + " descriptors.add(" + NL + " new AttributeDescriptorImpl(";
protected final String TEXT_39 = NL + " ";
protected final String TEXT_40 = ",";
protected final String TEXT_41 = NL + " ";
protected final String TEXT_42 = ",";
protected final String TEXT_43 = NL + " ";
protected final String TEXT_44 = ", ";
protected final String TEXT_45 = ", ";
protected final String TEXT_46 = ", null));";
protected final String TEXT_47 = NL + " return descriptors;";
protected final String TEXT_48 = NL + " }";
protected final String TEXT_49 = NL + " };";
protected final String TEXT_50 = NL + " builtType.put(";
protected final String TEXT_51 = ",";
protected final String TEXT_52 = ");";
protected final String TEXT_53 = NL + " return builtType;" + NL + " }" + NL;
protected final String TEXT_54 = NL + NL + " public ";
protected final String TEXT_55 = "Schema() {" + NL + " super(\"";
protected final String TEXT_56 = "\");";
protected final String TEXT_57 = NL + " put(";
protected final String TEXT_58 = "_TYPE);";
protected final String TEXT_59 = NL + " }" + NL + "" + NL + " /**" + NL + " * Complete the definition of a type and store it in the schema." + NL + " * " + NL + " * <p>" + NL + " * " + NL + " * This method calls {@link AttributeType#getSuper()} (and {@link ComplexType#getDescriptors()}" + NL + " * where applicable) to ensure the construction of the type (a concrete" + NL + " * {@link AbstractLazyAttributeTypeImpl} or {@link AbstractLazyComplexTypeImpl} sublass) is" + NL + " * complete. This should be sufficient to avoid any nasty thread-safety surprises in code using" + NL + " * this schema." + NL + " * " + NL + " * @param type" + NL + " * the type to complete and store" + NL + " */" + NL + " private void put(AttributeType type) {" + NL + " type.getSuper();" + NL + " if (type instanceof ComplexType) {" + NL + " ((ComplexType) type).getDescriptors();" + NL + " }" + NL + " put(type.getName(), type);" + NL + " }" + NL + "" + NL + " /**" + NL + " * Test that this class can be loaded." + NL + " */" + NL + " public static void main(String[] args) {" + NL + " Schema schema = new ";
protected final String TEXT_60 = "Schema();" + NL + " for (Entry<Name, AttributeType> entry : new TreeMap<Name, AttributeType>(schema).entrySet()) {" + NL + " System.out.println(\"Type: \" + entry.getValue().getName());" + NL + " System.out.println(\" Super type: \" + entry.getValue().getSuper().getName());" + NL + " if (entry.getValue() instanceof ComplexType) {" + NL + " for (PropertyDescriptor descriptor : ((ComplexType) entry.getValue())" + NL + " .getDescriptors()) {" + NL + " System.out.println(\" Property descriptor: \" + descriptor.getName());" + NL + " System.out.println(\" Property type: \" + descriptor.getType().getName());" + NL + " }" + NL + " }" + NL + " }" + NL + " }" + NL + "" + NL + "}";
public String generate(Object argument)
{
final StringBuffer stringBuffer = new StringBuffer();
stringBuffer.append(TEXT_1);
Object[] arguments = (Object[]) argument;
Schema schema = (Schema) arguments[0];
String prefix = (String) arguments[1];
prefix = prefix.toUpperCase();
SchemaGenerator sg = (SchemaGenerator) arguments[2];
List<AttributeType> types = sg.sort();
stringBuffer.append(TEXT_2);
Map<String, String> ns2import = new HashMap<String, String>();
for (Schema imported : sg.getImports()) {
String fullClassName = imported.getClass().getName();
String className = fullClassName.substring(fullClassName.lastIndexOf(".")+1);
ns2import.put(imported.getURI(), className);
stringBuffer.append(TEXT_3);
stringBuffer.append(fullClassName);
stringBuffer.append(TEXT_4);
}
stringBuffer.append(TEXT_5);
stringBuffer.append(prefix);
stringBuffer.append(TEXT_6);
Map<Name, String> typeBindings = sg.getTypeBindings();
for (AttributeType type : types) {
Name name = type.getName();
stringBuffer.append(TEXT_7);
XSDTypeDefinition xsdType = sg.getXSDType(type);
StringWriter writer = new StringWriter();
SAXTransformerFactory txFactory =
(SAXTransformerFactory) SAXTransformerFactory.newInstance();
TransformerHandler xmls;
try {
xmls = txFactory.newTransformerHandler();
} catch (TransformerConfigurationException e) {
throw new RuntimeException(e);
}
xmls.getTransformer().setOutputProperty(OutputKeys.METHOD, "XML");
xmls.getTransformer().setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "true");
xmls.getTransformer().setOutputProperty(OutputKeys.INDENT, "true");
try {
xmls.getTransformer().transform(new DOMSource(xsdType.getElement()), new StreamResult(writer));
}
catch (Exception e) {
e.printStackTrace();
return null;
}
String[] lines = writer.getBuffer().toString().split("\n");
for (int i = 0; i < lines.length; i++) {
stringBuffer.append(TEXT_8);
stringBuffer.append(lines[i].replaceAll("<","<").replaceAll(">",">"));
}
stringBuffer.append(TEXT_9);
String uri = name.getNamespaceURI();
String local = name.getLocalPart();
String binding;
if (typeBindings.containsKey(name)) {
binding = typeBindings.get(name) + ".class";
} else {
binding = type.getBinding().getName() + ".class";
}
String isIdentified = type.isIdentified() ? "true" : "false";
String isAbstract = type.isAbstract() ? "true" : "false";
String restrictions = "null";
String superType = "null";
if (type.getSuper() != null) {
superType = type.getSuper().getName()
.getLocalPart().toUpperCase() + "_TYPE";
String superURI = type.getSuper().getName().getNamespaceURI();
if (!uri.equals(superURI)) {
superType = ns2import.get(superURI) + "." + superType;
}
}
String description = "null";
if (type instanceof ComplexType && !typeBindings.containsKey(name)) {
stringBuffer.append(TEXT_10);
stringBuffer.append(name.getLocalPart().toUpperCase());
stringBuffer.append(TEXT_11);
stringBuffer.append(name.getLocalPart().toUpperCase());
stringBuffer.append(TEXT_12);
stringBuffer.append(name.getLocalPart().toUpperCase());
stringBuffer.append(TEXT_13);
stringBuffer.append(uri);
stringBuffer.append(TEXT_14);
stringBuffer.append(local);
stringBuffer.append(TEXT_15);
stringBuffer.append(TEXT_16);
stringBuffer.append(isIdentified);
stringBuffer.append(TEXT_17);
stringBuffer.append(isAbstract);
stringBuffer.append(TEXT_18);
stringBuffer.append(restrictions);
stringBuffer.append(TEXT_19);
stringBuffer.append(description);
stringBuffer.append(TEXT_20);
} else {
stringBuffer.append(TEXT_21);
stringBuffer.append(name.getLocalPart().toUpperCase());
stringBuffer.append(TEXT_22);
stringBuffer.append(name.getLocalPart().toUpperCase());
stringBuffer.append(TEXT_23);
stringBuffer.append(name.getLocalPart().toUpperCase());
stringBuffer.append(TEXT_24);
stringBuffer.append(uri);
stringBuffer.append(TEXT_25);
stringBuffer.append(local);
stringBuffer.append(TEXT_26);
stringBuffer.append(TEXT_27);
stringBuffer.append(binding);
stringBuffer.append(TEXT_28);
stringBuffer.append(isIdentified);
stringBuffer.append(TEXT_29);
stringBuffer.append(isAbstract);
stringBuffer.append(TEXT_30);
stringBuffer.append(restrictions);
stringBuffer.append(TEXT_31);
stringBuffer.append(description);
stringBuffer.append(TEXT_32);
}
stringBuffer.append(TEXT_33);
stringBuffer.append(superType);
stringBuffer.append(TEXT_34);
if (type instanceof ComplexType && !typeBindings.containsKey(name)) {
ComplexType cType = (ComplexType)type;
stringBuffer.append(TEXT_35);
if (cType.getDescriptors().isEmpty()) {
stringBuffer.append(TEXT_36);
} else {
stringBuffer.append(TEXT_37);
for (PropertyDescriptor pd : cType.getDescriptors()) {
if ( !(pd instanceof AttributeDescriptor) ) {
continue;
}
AttributeDescriptor ad = (AttributeDescriptor) pd;
AttributeType adType = ad.getType();
String adTypeName = adType.getName().getLocalPart().toUpperCase() +
"_TYPE";
String adTypeURI = adType.getName().getNamespaceURI();
if (!uri.equals(adTypeURI)) {
adTypeName = ns2import.get(adTypeURI) + "." + adTypeName;
}
String adName;
if (ad.getName().getNamespaceURI() == null) {
adName = "new NameImpl(\"" + ad.getName().getLocalPart() + "\")";
} else {
adName = "new NameImpl(\"" + ad.getName().getNamespaceURI() +
"\",\"" + ad.getName().getLocalPart() + "\")";
}
String min = ad.getMinOccurs() + "";
String max = ad.getMaxOccurs() + "";
String isNillable = ad.isNillable() ? "true" : "false";
stringBuffer.append(TEXT_38);
stringBuffer.append(TEXT_39);
stringBuffer.append(adTypeName);
stringBuffer.append(TEXT_40);
stringBuffer.append(TEXT_41);
stringBuffer.append(adName);
stringBuffer.append(TEXT_42);
stringBuffer.append(TEXT_43);
stringBuffer.append(min);
stringBuffer.append(TEXT_44);
stringBuffer.append(max);
stringBuffer.append(TEXT_45);
stringBuffer.append(isNillable);
stringBuffer.append(TEXT_46);
}
stringBuffer.append(TEXT_47);
}
stringBuffer.append(TEXT_48);
}
stringBuffer.append(TEXT_49);
if (!type.getUserData().isEmpty()) {
//attributes
for (Map.Entry<Object, Object> entry : type.getUserData().entrySet()) {
PropertyDescriptor pd = (PropertyDescriptor) entry.getValue();
PropertyType pdType = pd.getType();
String pdTypeName = pdType.getName().getLocalPart().toUpperCase() +
"_TYPE";
if (ns2import.containsKey(pdType.getName().getNamespaceURI())) {
String importClassName = (String) ns2import.get(pdType.getName().getNamespaceURI());
pdTypeName = importClassName + "." + pdTypeName;
}
String pdName = "new NameImpl(\"" + pd.getName().getNamespaceURI() +
"\",\"" + pd.getName().getLocalPart() + "\")";
stringBuffer.append(TEXT_50);
stringBuffer.append(pdName);
stringBuffer.append(TEXT_51);
stringBuffer.append(pdTypeName);
stringBuffer.append(TEXT_52);
}
}
stringBuffer.append(TEXT_53);
}
stringBuffer.append(TEXT_54);
stringBuffer.append(prefix);
stringBuffer.append(TEXT_55);
stringBuffer.append(schema.getURI());
stringBuffer.append(TEXT_56);
for (AttributeType type : types) {
stringBuffer.append(TEXT_57);
stringBuffer.append(type.getName().getLocalPart().toUpperCase());
stringBuffer.append(TEXT_58);
}
stringBuffer.append(TEXT_59);
stringBuffer.append(prefix);
stringBuffer.append(TEXT_60);
return stringBuffer.toString();
}
}