/*

  * Copyright (c) 2006-2007 David Holroyd

  */

 package uk.co.badgersinfoil.asxsd;

 import java.util.HashSet;

 import java.util.Set;

 import org.eclipse.xsd.XSDConcreteComponent;

 import org.eclipse.xsd.XSDNamedComponent;

 public class VariableNameGenerator {

         private static final Set KEYWORDS;

         static {

                 KEYWORDS = new HashSet();

                 KEYWORDS.add("return");

                 KEYWORDS.add("if");

                 KEYWORDS.add("else");

                 KEYWORDS.add("public");

                 KEYWORDS.add("private");

                 KEYWORDS.add("protected");

         }

         public String fieldName(XSDConcreteComponent component) {

                 if (component instanceof XSDNamedComponent) {

                         return fieldName((XSDNamedComponent)component);

                 }

                 throw new IllegalArgumentException("Components of type " + component.getClass().getName() + " not supported");

         }

         /**

          * returns the name of the AS field which should hold values of the

          * given element declaration.

          */

         private String fieldName(XSDNamedComponent named) {

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

                         throw new IllegalArgumentException("Component has no name: " + XSDUtils.path(named));

                 }

                 return fieldName(named.getName());

         }

         public String fieldName(String name) {

                 // TODO: do we need to ensure unique names?

                 if (KEYWORDS.contains(name)) {

                         return "_" + name;

                 }

                 return StringUtils.sanitize(name);

         }

 //        /**

 //         * create a field name based on the given attribute declaration

 //         */

 //        public String fieldName(XSDAttributeDeclaration attrDecl) {

 //                // TODO: do we need to ensure unique names?

 //                if (KEYWORDS.contains(attrDecl.getName())) {

 //                        return "_" + attrDecl.getName();

 //                }

 //                return StringUtils.sanitize(attrDecl.getName());

 //        }

 }