/*

  * Copyright (c) 2006-2007 David Holroyd

  */

 package uk.co.badgersinfoil.asxsd;

 import java.util.HashMap;

 import java.util.Map;

 import javax.xml.namespace.QName;

 import org.eclipse.xsd.XSDComplexTypeDefinition;

 import org.eclipse.xsd.XSDConcreteComponent;

 import org.eclipse.xsd.XSDElementDeclaration;

 import org.eclipse.xsd.XSDSchema;

 import org.eclipse.xsd.XSDTypeDefinition;

 public class TypeNameGenerator {

         private Map names = new HashMap();

         public String typeName(XSDTypeDefinition type) {

                 if (type instanceof XSDComplexTypeDefinition) {

                         if (type.getName() == null) {

                                 throw new IllegalArgumentException("Can't get type name for anonymous complex type: "+type+", at "+XSDUtils.path(type));

                         }

                         return getNameFor(new QName(type.getTargetNamespace(), type.getName()));

                 }

                 throw new IllegalArgumentException("unhandled type: "+type.getClass().getName());

         }

         private String getNameFor(QName name) {

                 String existing = (String)names.get(name);

                 if (existing != null) {

                         return existing;

                 }

                 String pkg = StringUtils.toPackageName(name.getNamespaceURI());

                 String clazz = classNameFor(name.getLocalPart());

                 String baseName = pkg + "." + clazz;

                 int count = 1;

                 while (names.containsValue(baseName)) {

                         String mangled = baseName + "_" + (count++);

                         if (!names.containsValue(baseName)) {

                                 names.put(name, baseName);

                                 return mangled;

                         }

                         if (count > 5000) {

                                 // assume there's a bug,

                                 throw new Error("Couldn't assign a unique name after 5000 iterations");

                         }

                 }

                 names.put(name, baseName);

                 return baseName;

         }

         private String classNameFor(String localPart) {

                 return StringUtils.sanitize(localPart);

         }

         public String getNameFor(XSDElementDeclaration elementDecl) {

                 elementDecl = elementDecl.getResolvedElementDeclaration();

                 if (elementDecl.getName() == null) {

                         throw new IllegalArgumentException("Element declaration has no name "+XSDUtils.path(elementDecl));

                 }

                 StringBuffer result = new StringBuffer(elementDecl.getName());

                 XSDConcreteComponent container = elementDecl;

                 String namespace = elementDecl.getTargetNamespace();

                 while ((container = container.getContainer()) != null) {

                         if (container instanceof XSDSchema) {

                                 // we're at the top level

                                 break;

                         }

                         if (container instanceof XSDElementDeclaration) {

                                 XSDElementDeclaration elem = (XSDElementDeclaration)container;

                                 result.insert(0, elem.getName());

                                 namespace = elem.getTargetNamespace();

                         }

                         if (container instanceof XSDComplexTypeDefinition) {

                                 XSDComplexTypeDefinition complexType = (XSDComplexTypeDefinition)container;

                                 if (complexType.getName() != null) {

                                         result.insert(0, complexType.getName());

                                         // we're at the top level

                                         break;

                                 }

                         }

                 }

                 result.insert(0, "Element_");

                 if (namespace != null) {

                         result.insert(0, ".");

                         result.insert(0, StringUtils.toPackageName(namespace));

                 }

                 return result.toString();

         }

 }