package org.madore.damlengine;

import java.util.regex.Pattern;
import java.io.InputStream;
import java.security.MessageDigest;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import javax.xml.xpath.*;
import org.w3c.dom.*;
import org.w3c.dom.ls.DOMImplementationLS;
import org.w3c.dom.ls.LSParser;
import org.w3c.dom.ls.LSSerializer;
import org.w3c.dom.ls.LSInput;

public final class WeblogPopulate {

    private WeblogPopulate() { // Forbid instantiation
	throw new AssertionError("WeblogPopulate cannot be instantiated");
    }

    public static final String toHex(byte[] digest) {
	StringBuilder sb = new StringBuilder();
	for ( byte b : digest )
	    sb.append(String.format("%02x", b));
	return sb.toString();
    }

    public static void populate(InputStream in)
	throws Exception {

	final DOMImplementationLS domils
	    = (DOMImplementationLS)(DamlEngine.IncantDOM.getDOMI());
	LSSerializer ser = domils.createLSSerializer();
	ser.getDomConfig().setParameter("xml-declaration", false);
	LSParser par = domils.createLSParser(DOMImplementationLS.MODE_SYNCHRONOUS, null);
	par.getDomConfig().setParameter("resource-resolver", new Resolver());
	par.getDomConfig().setParameter("http://xml.org/sax/features/validation", true);
	par.getDomConfig().setParameter("http://xml.org/sax/features/namespaces", true);
	par.getDomConfig().setParameter("http://apache.org/xml/properties/internal/error-handler", new DamlEngine.SelectiveErrorHandler());

	MessageDigest sha1 = MessageDigest.getInstance("SHA-1");

	final Connection conn = WeblogDatabaseConnection.getConnection();

	final PreparedStatement checkSt
	    = conn.prepareStatement("SELECT sha1 FROM entries WHERE id=?");
	final PreparedStatement insertNSt
	    = conn.prepareStatement("INSERT INTO entries(id,edate,lang,special_name,title,title_xml,content,sha1,has_cut) VALUES (?,?,?,?,?,?::xml,?::xml,?,?)");
	final PreparedStatement insertCSt
	    = conn.prepareStatement("INSERT INTO entries(id,edate,lang,special_name,title,title_xml,content,sha1,has_cut,cdate) VALUES (?,?,?,?,?,?::xml,?::xml,?,?,?::timestamptz)");
	final PreparedStatement updateNSt
	    = conn.prepareStatement("UPDATE entries SET (edate,mdate,lang,special_name,title,title_xml,content,sha1,has_cut)=(?,DEFAULT,?,?,?,?::xml,?::xml,?,?) WHERE id=?");
	final PreparedStatement updateCSt
	    = conn.prepareStatement("UPDATE entries SET (edate,mdate,lang,special_name,title,title_xml,content,sha1,has_cut,cdate)=(?,DEFAULT,?,?,?,?::xml,?::xml,?,?,?::timestamptz) WHERE id=?");
	final PreparedStatement checkDateSt
	    = conn.prepareStatement("SELECT to_char(cdate,'YYYY-MM-DD\"T\"HH24:MI:SS\"Z\"') FROM entries WHERE id=?");
	final PreparedStatement clearCatSt
	    = conn.prepareStatement("DELETE FROM incat WHERE id=?");
	final PreparedStatement setCatSt
	    = conn.prepareStatement("INSERT INTO incat(id,code) VALUES (?,?)");

	LSInput input = domils.createLSInput();
	input.setByteStream(in);
	Document doc = par.parse(input);
	XPathFactory xpf = XPathFactory.newInstance();
	XPath xp = xpf.newXPath();
	xp.setNamespaceContext(new DamlEngine.DamlNSMapping());
	XPathExpression expr = xp.compile("//d:weblog/d:entry");
	XPathExpression texpr = xp.compile("d:title");
	XPathExpression cutexpr = xp.compile(".//d:cut-here");
	NodeList entries = (NodeList)(expr.evaluate(doc, XPathConstants.NODESET));
	for ( int i=0 ; i<entries.getLength() ; i++ ) {
	    Element ent = (Element)(entries.item(i));
	    String idStr = ent.getAttributeNS(null, "number");
	    if ( ! Pattern.matches("^\\d{4}$", idStr) )
		throw new IllegalArgumentException("entry number attribute must be of the form NNNN");
	    int id = Integer.parseInt(idStr);
	    String date = ent.getAttributeNS(null, "date");
	    if ( ! Pattern.matches("^\\d{4}-\\d{2}-\\d{2}$", date) )
		throw new IllegalArgumentException("entry date attribute must be of the form YYYY-MM-DD");
	    String cdate;
	    if ( ent.hasAttributeNS(null, "cdate") )
		cdate = ent.getAttributeNS(null, "cdate");
	    else
		cdate = null;
	    String catStr = ent.getAttributeNS(null, "cat");
	    String[] catList = catStr.split("\\s+");
	    String specialName = ent.getAttributeNS(null, "special-name");
	    if ( specialName.equals("") )
		specialName = null;
	    String lang = LangHelper.getLangRec(ent);
	    String content = ser.writeToString(ent);
	    sha1.reset();
	    String digest = toHex(sha1.digest(content.getBytes("UTF-8")));

	    checkSt.setInt(1, id);
	    ResultSet checkRes = checkSt.executeQuery();
	    boolean exists = checkRes.next();
	    if ( exists && checkRes.getString(1).equals(digest) )
		continue;
	    Node titleNode = (Node)(texpr.evaluate(ent, XPathConstants.NODE));
	    String titleTxt = (titleNode != null) ? titleNode.getTextContent() : null;
	    String titleXml = (titleNode != null) ? ser.writeToString(titleNode) : null;
	    boolean hasCut = cutexpr.evaluate(ent, XPathConstants.NODE) != null;
	    conn.setAutoCommit(false);
	    if ( exists ) {
		System.err.println("Updating entry "+id);
		PreparedStatement updateSt;
		if ( cdate == null )
		    updateSt = updateNSt;
		else
		    updateSt = updateCSt;
		updateSt.setString(1, date);
		updateSt.setString(2, lang);
		updateSt.setString(3, specialName);
		updateSt.setString(4, titleTxt);
		updateSt.setString(5, titleXml);
		updateSt.setString(6, content);
		updateSt.setString(7, digest);
		updateSt.setBoolean(8, hasCut);
		if ( cdate != null ) {
		    updateSt.setString(9, cdate);
		    updateSt.setInt(10, id);
		} else {
		    updateSt.setInt(9, id);
		}
		updateSt.executeUpdate();
	    } else {
		System.err.println("Registering entry "+id);
		PreparedStatement insertSt;
		if ( cdate == null )
		    insertSt = insertNSt;
		else
		    insertSt = insertCSt;
		insertSt.setInt(1, id);
		insertSt.setString(2, date);
		insertSt.setString(3, lang);
		insertSt.setString(4, specialName);
		insertSt.setString(5, titleTxt);
		insertSt.setString(6, titleXml);
		insertSt.setString(7, content);
		insertSt.setString(8, digest);
		insertSt.setBoolean(9, hasCut);
		if ( cdate != null )
		    insertSt.setString(10, cdate);
		insertSt.executeUpdate();
	    }
	    if ( ! Pattern.matches("^\\d{4}-\\d{2}-\\d{2}T\\d{2}\\:\\d{2}(?:\\:\\d{2})?(?:Z|[\\+\\-]\\d{2}\\:\\d{2})$", cdate) ) {
		checkDateSt.setInt(1, id);
		ResultSet checkDate = checkDateSt.executeQuery();
		checkDate.next();
		String checkDateStr = checkDate.getString(1);
		System.err.println("cdate for entry "+id+": "
				   +checkDateStr);
	    }
	    clearCatSt.setInt(1, id);
	    clearCatSt.executeUpdate();
	    for ( String cat : catList ) {
		if ( ! cat.equals("") ) {
		    setCatSt.setInt(1, id);
		    setCatSt.setString(2, cat);
		    setCatSt.executeUpdate();
		}
	    }
	    conn.commit();
	}

	checkSt.close();
	insertNSt.close();
	insertCSt.close();
	updateNSt.close();
	updateCSt.close();
	checkDateSt.close();
	clearCatSt.close();
	setCatSt.close();

    }

}