package nl.beesting.beangenerator.generator;

 import java.lang.reflect.Constructor;

 import java.net.URL;

 import nl.beesting.beangenerator.TypeNotSupportedException;

 import nl.beesting.beangenerator.util.ParameterDescriptor;

 import nl.beesting.beangenerator.util.ParameterPair;

 import nl.beesting.beangenerator.util.ParameterPairUtil;

 import nl.beesting.beangenerator.util.RandomUtil;

 import org.apache.commons.logging.Log;

 import org.apache.commons.logging.LogFactory;

 import org.apache.xerces.parsers.DOMParser;

 import org.w3c.dom.Document;

 import org.w3c.dom.Element;

 import org.w3c.dom.Node;

 import org.w3c.dom.NodeList;

 import org.xml.sax.InputSource;

 public class PhoneticalNameGenerator extends AbstractInstanceGenerator {

    /** The <code>logger</code> for this class. */

    private static final Log logger = LogFactory.getLog(PhoneticalNameGenerator.class);

    public static final String DEFAULT_NAME_DATA = "NameData.xml";

    public static final String NAME_DATA_PARAMETER = "nameDataLocation";

    public static final String DEFAULT_NAME_TEMPLATE = "Sawyer";

    public static final String NAME_TEMPLATE_PARAMETER = "nameTemplate";

    private String nameTemplate;

    private NodeList phonemes;

    private NodeList nameTemplates;

    public PhoneticalNameGenerator() {

       addSupportedTypes(new Class[]{java.lang.String.class, java.lang.StringBuffer.class});

       setSupportedParameterDescriptors(new ParameterDescriptor[]{

          new ParameterDescriptor(NAME_DATA_PARAMETER, String.class, "NameData location. Points to the location of where the NameData.xml should be."),

          new ParameterDescriptor(NAME_TEMPLATE_PARAMETER, String.class, "Name template specified in the namedate file used to generated the names.")

       });

    }

    /**

     * Initializes the instnce generator with the given array of objects.

     */

    public void init(ParameterPair[] initParams) {

       ClassLoader loader = this.getClass().getClassLoader();

       String nameDateLocation = null;

       if (initParams != null) {

          nameDateLocation = (String) ParameterPairUtil.getFirstParameterValue(initParams, NAME_DATA_PARAMETER, false, String.class);

          nameTemplate = (String) ParameterPairUtil.getFirstParameterValue(initParams, NAME_TEMPLATE_PARAMETER, false, String.class);

          if (nameDateLocation == null) {

             nameDateLocation = DEFAULT_NAME_DATA;

          }

          if (nameTemplate == null) {

             nameTemplate = DEFAULT_NAME_TEMPLATE;

          }

       } else {

          nameDateLocation = DEFAULT_NAME_DATA;

          nameTemplate = DEFAULT_NAME_TEMPLATE;

       }

       URL nameData = loader.getResource(nameDateLocation);

       if (nameData == null) {

          nameData = loader.getResource(DEFAULT_NAME_DATA);

       }

       DOMParser parser = new DOMParser();

       try {

          parser.parse(new InputSource(nameData.openStream()));

       } catch (Exception e) {

          // TODO read default

          logger.error("unable to read namedata.xml", e);

       }

       Document document = parser.getDocument();

       phonemes = document.getElementsByTagName("Phoneme");

       nameTemplates = document.getElementsByTagName("NameTemplate");

    }

    public Object generateInstance(Class<?> type) throws TypeNotSupportedException {

       if (type == null) {

          throw new IllegalArgumentException("Input parameter type cannot be null.");

       }

       if (!checkSupport(type)) {

          throw new TypeNotSupportedException(type.getName(), "generateInstance");

       }

       StringBuffer name = new StringBuffer();

       Element template = null;

       for (int i = 0; i < nameTemplates.getLength(); i++) {

          Element element = (Element) nameTemplates.item(i);

          if (nameTemplate.equals(element.getAttribute("Name"))) {

             template = element;

             break;

          }

       }

       if (template == null) {

          throw new IllegalArgumentException("Specified template name: " + nameTemplate + " does not exsist");

       }

       for (int i = 0; i < template.getChildNodes().getLength(); i++) {

          Node node = template.getChildNodes().item(i);

          String phonemeName = node.getFirstChild() != null ? node.getFirstChild().getNodeValue() : null;

          for (int j = 0; j < phonemes.getLength(); j++) {

             Element phonemeElement = (Element) phonemes.item(j);

             if (phonemeElement.getAttribute("Name").equals(phonemeName)) {

                NodeList nodeList = phonemeElement.getElementsByTagName("Part");

                int random = RandomUtil.initRandomIntBetween(0, nodeList.getLength());

                Node n = nodeList.item(random);

                String namePart = n.getFirstChild().getNodeValue();

                if (node.getAttributes().item(0).getNodeValue().equals("Cap")) {

                   name.append(namePart.substring(0, 1).toUpperCase()).append(namePart.substring(1));

                } else {

                   name.append(namePart);

                }

             }

          }

       }

       try {

          Constructor<?> c = type.getConstructor(String.class);

          return c.newInstance(name.toString());

       } catch (Exception e) {

          logger.warn("Could not construct Object of type " + type + " . Returning null.", e);

       }

       return null;

    }

    public Object generateInstance(Class<?> type, int modus) throws TypeNotSupportedException {

       return generateInstance(type);

    }

    protected Object getDefaultFromValue() {

       // TODO Auto-generated method stub

       return null;

    }

    protected Object getDefaultUntilValue() {

       // TODO Auto-generated method stub

       return null;

    }

    protected Object getFromValue() {

       // TODO Auto-generated method stub

       return null;

    }

    protected Object getUntilValue() {

       // TODO Auto-generated method stub

       return null;

    }

    public String getNameTemplate() {

       return nameTemplate;

    }

    public static void main(String[] args) throws TypeNotSupportedException {

       PhoneticalNameGenerator generator = new PhoneticalNameGenerator();

       generator.init(null);

       String test = (String) generator.generateInstance(String.class);

       System.out.println(test);

    }

 }