/////////////////////////////////////////////////////////////////
//  xmlParser.cpp - prototype demonstration program            //
//                                                             //
//  Jim Fawcett, CSE687 - Object Oriented Design, Spring 2002  //
/////////////////////////////////////////////////////////////////
/*
  Note:
  - This demonstration uses a tokenizer to look for XML elements.
    You may find it just as easy, and simpler to directly scan
    the XML string, read from file.
  - This prototype has no error handling.

*/
#include <fstream>
#include <iostream>
#include "tok.h"
using namespace std;

///////////////////////////////////////////////////////////////
//  xmlParse class declaration

class xmlParse {

  public:
    xmlParse(istream &in) : toker(in), level(0) { };
    bool getElem(const std::string &parent);
    bool nextElem();
    void indent();

  private:
    token toker;
    int level;
};

//----< send spaces to output to indent display line >---------

void xmlParse::indent() {

  for(int i=0; i<level; i++)
    cout.put(' ');
}

//

//----< recursively display elements in document string >------

bool xmlParse::getElem(const std::string &parent) {

  const int delta = 3;

  // get element name and display it

  std::string tok, name;
  if(toker.getTok(tok)) {
    name = tok;
    cout << "\n  ";
    indent();
    cout << "element " << name;
  }
  if(parent.size() > 0)
    cout << ", child of " << parent;
  
  // ignoring attributes look for child elements or end
  
  while(toker.getTok(tok)) {
    if(tok == string("<")) {
      level += delta;
      getElem(name);
    }
    else if(tok == string("</")) {
      toker.getTok(tok);  // name
      toker.getTok(tok);  // '>'
      level -= delta;
      return toker.good();
    }
  }
  return false;
}

//----< find start of next element >---------------------------
//
//  skips over opening banner
//
bool xmlParse::nextElem() {

  std::string tok;
  while(toker.getTok(tok))
    if(tok == string("<"))
      return true;
  return false;
}

//

//----< test stub >--------------------------------------------

void main(int argc, char *argv[]) {

  if(argc==1) {
    cout << "\n  Please enter name of XML file to parse\n\n";
    return;
  }

  ifstream in;
  for(int i=1; i<argc; i++) {

    if(i==1) {

      cout << "\n  Show action of tokenizer on first file "
           << "\n ========================================\n";

      in.open(argv[i]);
      if(!in.good()) {
        cout << "\n  can't open file " << argv[i] << endl;
        continue;
      }

      token toker(in);
      string tok;
      while(toker.getTok(tok)) {
        cout << "\n  token: " << tok;
      }

      in.clear();
      in.seekg(0,ios::beg);
    }

    cout << "\n  Display XML string, read from file "
         << "\n ====================================\n";

    in.clear();
    in.seekg(0,ios::beg);

    string xmlString;
    while(in.good())
      xmlString += in.get();    // unformatted read
    
    cout << "\n  " << xmlString << endl;

//

    cout << "\n  Display xmlParse action "
         << "\n =========================\n";

    in.clear();
    in.seekg(0,ios::beg);
    xmlParse parse(in);
    
    // find first element

    parse.nextElem();

    // recursively find all the elements

    parse.getElem("root");
 
    in.close();
    in.clear();

    std::cout << "\n\n";
  }
}