#ifndef XML_PROMPT_GENERATOR_H
#define XML_PROMPT_GENERATOR_H

#include <expat.h>
#include <map>
#include <set>
#include "wrapexpat.h"
#include "parserdata.h"

struct PromptGeneratorData : public ParserData
{
  std::map<std::string, std::map<std::string, std::set<std::string> > >
    attributeMap;
  std::map<std::string, std::set<std::string> > elementMap;
  std::map<std::string, std::set<std::string> > requiredAttributeMap;
  std::set<std::string> entitySet;
  std::string catalogPath, auxPath;
  int elementDeclRecurseLevel;
  bool rootElement, dtdFound;
  XML_Parser p;
};

class XmlPromptGenerator : public WrapExpat
{
  public:
    XmlPromptGenerator(
      const std::string& catalogPath = "",
      const std::string& basePath = "",
      const std::string& auxPath = "");
    virtual ~XmlPromptGenerator();
    void getAttributeMap(
      std::map<std::string, std::map<std::string, std::set<std::string> > >
        &attributeMap);
    void getRequiredAttributeMap(
      std::map<std::string, std::set<std::string> > &requiredAttributeMap);
    void getElementMap(
      std::map<std::string, std::set<std::string> > &elementMap);
    void getEntitySet(
      std::set<std::string> &entitySet);
    bool getDtdFound();
  private:
    std::auto_ptr<PromptGeneratorData> d;
    static void XMLCALL starthandler(
      void *data,
      const XML_Char *el,
      const XML_Char **attr);
    static void XMLCALL endhandler(
      void *data,
      const XML_Char *el);
    static void XMLCALL doctypedeclstarthandler(
      void *userData,
      const XML_Char *doctypeName,
      const XML_Char *sysid,
      const XML_Char *pubid,
      int has_internal_subset);
    static void XMLCALL doctypedeclendhandler(void *userData);
    static void XMLCALL elementdeclhandler(
      void *userData,
      const XML_Char *name,
      XML_Content *model);
    static void XMLCALL attlistdeclhandler(
      void *userData,
      const XML_Char *elname,
      const XML_Char *attname,
      const XML_Char *att_type,
      const XML_Char *dflt,
      int isrequired);
    static int XMLCALL externalentityrefhandler(
      XML_Parser p,
      const XML_Char *context,
      const XML_Char *base,
      const XML_Char *systemId,
      const XML_Char *publicId);
    static void XMLCALL entitydeclhandler(
      void *userData,
      const XML_Char *entityName,
      int is_parameter_entity,
      const XML_Char *value,
      int value_length, 
      const XML_Char *base,
      const XML_Char *systemId,
      const XML_Char *publicId,
      const XML_Char *notationName);
};

#endif