import java.io.IOException;
import java.util.*;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import org.xml.sax.Attributes;
import org.xml.sax.SAXException;

import org.xml.sax.helpers.DefaultHandler;

import java.sql.*;


public class ProjectParser extends DefaultHandler{
	List myDocs;

	private String tempVal;
	//to maintain context
	private Document tempDoc = new Document();
	private String currentDoc;

	static Connection conn = null;

	static PreparedStatement psInsertGenre = null;
	static String sqlInsertGenre = null;

	static PreparedStatement psInsertPeople = null;
	static String sqlInsertPeople = null;

	static PreparedStatement psInsertTitle = null;
	static String sqlInsertTitle = null;

	static PreparedStatement psInsertPublisher = null;
	static String sqlInsertPublisher = null;

	static Hashtable<String, String> hashtable = new Hashtable<String, String>();
	static int hashID = 0;


	public ProjectParser() {
		myDocs = new ArrayList();
	}


	public void runExample() {
		parseDocument();
	}

	private void parseDocument() {

		//get a factory
		SAXParserFactory spf = SAXParserFactory.newInstance();
		try {
			//get a new instance of parser
			SAXParser sp = spf.newSAXParser();

			//parse the file and also register this class for call backs
			sp.parse("/Users/bshiaw/desktop/bigData/dblp-data.xml", this);

		}catch(SAXException se) {
			se.printStackTrace();
		}catch(ParserConfigurationException pce) {
			pce.printStackTrace();
		}catch (IOException ie) {
			ie.printStackTrace();
		}
	}

	/**
	 * Iterate through the list and print
	 * the contents
	 */
	private void printData(){

		System.out.println("Number of Documents '" + myDocs.size() + "'.");

		Iterator it = myDocs.iterator();
		while(it.hasNext()) {
			System.out.println(it.next().toString());
		}
	}

	public static void main(String[] args) throws InstantiationException, IllegalAccessException, ClassNotFoundException {
		// Incorporate mySQL driver
		Class.forName("com.mysql.jdbc.Driver").newInstance();

		// Connect to the test database
		try {
			conn = DriverManager.getConnection("jdbc:mysql:///bookdb","root", "lakers");
		}catch (SQLException e) {
			e.printStackTrace();
		}

		int[] iNoRows = null;
		//For tbl_genres
		sqlInsertGenre = "insert into tbl_genres (genre_name) values(?)";
		try {
			psInsertGenre = conn.prepareStatement(sqlInsertGenre);
		}catch (SQLException e) {
			e.printStackTrace();
		}


		//For tbl_people
		sqlInsertPeople = "INSERT INTO tbl_people (name) values(?)";
		try {
			psInsertPeople = conn.prepareStatement(sqlInsertPeople);
		}catch (SQLException e) {
			e.printStackTrace();
		}
		//For tbl_booktitle
		sqlInsertTitle = "INSERT INTO tbl_booktitle (title) values(?)";
		try {
			psInsertTitle = conn.prepareStatement(sqlInsertTitle);
		}catch (SQLException e) {
			e.printStackTrace();
		}
		//For tbl_publisher
		sqlInsertPublisher = "INSERT INTO tbl_publisher(publisher_name) values(?)";
		try {
			psInsertPublisher = conn.prepareStatement(sqlInsertPublisher);
		}catch (SQLException e) {
			e.printStackTrace();
		}


		ProjectParser pp = new ProjectParser();
		pp.runExample();
		System.out.println("Parsing Complete.\n");

		try {
			iNoRows = psInsertGenre.executeBatch();
		}catch (SQLException e) {
			e.printStackTrace();
		}

		try {
			iNoRows = psInsertPeople.executeBatch();
		}catch (SQLException e) {
			e.printStackTrace();
		}

		try {
			iNoRows = psInsertTitle.executeBatch();
		}catch (SQLException e) {
			e.printStackTrace();
		}

		try {
			iNoRows = psInsertPublisher.executeBatch();
		}catch (SQLException e) {
			e.printStackTrace();
		}


		System.out.println("Execute Batch Complete.");

		try {
			if(psInsertGenre!=null) psInsertGenre.close();
			if(psInsertPeople!=null) psInsertPeople.close();
			if(psInsertTitle!=null) psInsertTitle.close();
			if(psInsertPublisher!=null) psInsertPublisher.close();
			if(conn!=null) conn.close();

		}catch (SQLException e) {
			e.printStackTrace();
		}
	}



	//Event Handlers
	public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
		//reset
		tempVal = "";

		if(qName.equalsIgnoreCase("book")) {
			//create a new instance of employee
			//			tempDoc = new Document();
			//			tempDoc.setmdate(attributes.getValue("mdate"));
			//			tempDoc.setKey(attributes.getValue("key"));
			//			tempDoc.setDocType("Book");
			currentDoc = "Book";
			if (! hashtable.containsValue(currentDoc)) {
				hashtable.put(Integer.toString(hashID), currentDoc);
				hashID++;
				try {
					psInsertGenre.setString(1, currentDoc);
					psInsertGenre.addBatch();
				}catch (Exception e) {
					e.printStackTrace();
				}
			}
		}

		else if (qName.equalsIgnoreCase("incollection")) {
			//			tempDoc = new Document();
			//			tempDoc.setmdate(attributes.getValue("mdate"));
			//			tempDoc.setKey(attributes.getValue("key"));
			//			tempDoc.setDocType("Incollection");
			currentDoc = "Incollection";
			if (! hashtable.containsValue(currentDoc)) {
				hashtable.put(Integer.toString(hashID), currentDoc);
				hashID++;
				try {
					psInsertGenre.setString(1, currentDoc);
					psInsertGenre.addBatch();
				}catch (Exception e) {
					e.printStackTrace();
				}
			}
		}


		else if (qName.equalsIgnoreCase("proceedings")) {
			//			tempDoc = new Document();
			//			tempDoc.setmdate(attributes.getValue("mdate"));
			//			tempDoc.setKey(attributes.getValue("key"));
			//			tempDoc.setDocType("Proceedings");
			currentDoc = "Proceedings";
			if (! hashtable.containsValue(currentDoc)) {
				hashtable.put(Integer.toString(hashID), currentDoc);
				hashID++;
				try {
					psInsertGenre.setString(1, currentDoc);
					psInsertGenre.addBatch();
				}catch (Exception e) {
					e.printStackTrace();
				}
			}

		}
		else if (qName.equalsIgnoreCase("inproceedings")) {
			//			tempDoc = new Document();
			//			tempDoc.setmdate(attributes.getValue("mdate"));
			//			tempDoc.setKey(attributes.getValue("key"));
			//			tempDoc.setDocType("Inproceedings");
			currentDoc = "Inproceedings";
			if (! hashtable.containsValue(currentDoc)) {
				hashtable.put(Integer.toString(hashID), currentDoc);
				hashID++;
				try {
					psInsertGenre.setString(1, currentDoc);
					psInsertGenre.addBatch();
				}catch (Exception e) {
					e.printStackTrace();
				}
			}
		}
	}



	public void characters(char[] ch, int start, int length) throws SAXException {
		tempVal = new String(ch,start,length);
	}

	public void endElement(String uri, String localName, String qName) throws SAXException {


		if (qName.equalsIgnoreCase(currentDoc)){
			//add it to the list
			return;
			}

		if (  (qName.equalsIgnoreCase("author")) || (qName.equalsIgnoreCase("editor")) ) {
			String author = tempVal;
			if (! hashtable.containsValue(author)) {
				hashtable.put(Integer.toString(hashID), author);
				hashID++;
				try {
					psInsertPeople.setString(1, author);
					psInsertPeople.addBatch();
				}catch (Exception e) {
					e.printStackTrace();
				}
				return;
			}}

			//tempDoc.setPublisher(tempVal);

			if (qName.equalsIgnoreCase("title")) {
				tempDoc.setTitle(tempVal);
			}
			if (qName.equalsIgnoreCase("year")) {
				tempDoc.setYear(tempVal);
			}
			if (qName.equalsIgnoreCase("booktitle")) {
				System.out.println("got to title");
				String title = tempVal;
				if (! hashtable.containsValue(title)) {
					hashtable.put(Integer.toString(hashID), title);
					hashID++;
					try {
						psInsertTitle.setString(1, title);
						psInsertTitle.addBatch();
					}catch (Exception e) {
						e.printStackTrace();
					}
				}
				return; }

				if (qName.equalsIgnoreCase("publisher")) {
					String publisher = tempVal;
					System.out.println("OH HERRO");
					if (! hashtable.containsValue(publisher)) {
						hashtable.put(Integer.toString(hashID), publisher);
						hashID++;
						try {
							psInsertPublisher.setString(1, publisher);
							psInsertPublisher.addBatch();
						}catch (Exception e) {
							e.printStackTrace();
						}
					}return; }
				  if (qName.equalsIgnoreCase("isbn")) {
					tempDoc.setISBN(tempVal);
				} if (qName.equalsIgnoreCase("url")) {
					tempDoc.setURL(tempVal);
				} if (qName.equalsIgnoreCase("pages")) {
					tempDoc.setPages(tempVal);
				} if (qName.equalsIgnoreCase("address")) {
					tempDoc.setAddress(tempVal);
				} if (qName.equalsIgnoreCase("journal")) {
					tempDoc.setJournal(tempVal);
				} if (qName.equalsIgnoreCase("volume")) {
					tempDoc.setVolume(tempVal);
				} if (qName.equalsIgnoreCase("number")) {
					tempDoc.setNumber(tempVal);
				} if (qName.equalsIgnoreCase("month")) {
					tempDoc.setMonth(tempVal);
				} if (qName.equalsIgnoreCase("ee")) {
					tempDoc.setEE(tempVal);
				} if (qName.equalsIgnoreCase("cdrom")) {
					tempDoc.setCDRom(tempVal);
				} if (qName.equalsIgnoreCase("cite")) {
					tempDoc.setCite(tempVal);
				} if (qName.equalsIgnoreCase("note")) {
					tempDoc.setNote(tempVal);
				} if (qName.equalsIgnoreCase("crossref")) {
					tempDoc.setCrossref(tempVal);
				} if (qName.equalsIgnoreCase("series")) {
					tempDoc.setSeries(tempVal);
				} if (qName.equalsIgnoreCase("school")) {
					tempDoc.setSchool(tempVal);
				} if (qName.equalsIgnoreCase("chapter")) {
					tempDoc.setChapter(tempVal);
				}
				//		}
				//			else if (qName.equalsIgnoreCase("editor")) {
				////		tempDoc.setEditor(tempVal);
				//		String editor = tempVal;
				//		if (! hashtable.containsValue(editor)) {
				//			hashtable.put(Integer.toString(hashID), editor);
				//			hashID++;
				//			try {
				//				psInsertPeople.setString(1, editor+" Editor");
				//				psInsertPeople.addBatch();
				//			}catch (Exception e) {
				//				e.printStackTrace();
				//			}
				//		}
				//	}


			}


		}