/*
 * Copyright
 *   2007 axYus - www.axyus.com
 *   2007 JP.Tessier - jean-philippe.tessier@axyus.com
 *
 * This file is part of XEMELIOS.
 *
 * XEMELIOS is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * XEMELIOS is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with XEMELIOS; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package fr.gouv.finances.cp.xemelios.controls;

import java.util.ArrayList;
import java.util.Hashtable;
import java.util.Stack;
import java.util.TreeSet;
import java.util.Vector;

import javax.swing.JFrame;
import javax.swing.SwingUtilities;
import javax.xml.namespace.NamespaceContext;

import org.apache.log4j.Logger;
import org.xml.sax.Attributes;
import org.xml.sax.ContentHandler;
import org.xml.sax.Locator;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;

import fr.gouv.finances.dgfip.utils.Pair;
import fr.gouv.finances.cp.utils.ui.DlgGetPair;
import fr.gouv.finances.dgfip.utils.xml.XmlAttributesImpl;
import fr.gouv.finances.dgfip.xemelios.common.ToolException;
import fr.gouv.finances.dgfip.xemelios.common.config.DocumentModel;
import fr.gouv.finances.cp.xemelios.controls.comm.COLLBUDG;
import fr.gouv.finances.cp.xemelios.controls.models.ControlModel;
import fr.gouv.finances.cp.xemelios.controls.models.DocumentControlModel;
import fr.gouv.finances.cp.xemelios.ui.MainWindow;
import fr.gouv.finances.dgfip.xemelios.auth.UnauthorizedException;
import fr.gouv.finances.dgfip.xemelios.data.DataAccessException;
import fr.gouv.finances.dgfip.xemelios.data.DataConfigurationException;
import fr.gouv.finances.dgfip.xemelios.data.DataLayerManager;
import java.util.Iterator;

/**
 * @author Jean-Philippe TESSIER
 *
 * Permet d'appeler les contrôles unitaires afin de
 * traiter les anomalies rencontrées que renvoient ces controles.
 * Ces anomalies sont envoyées à un serializer qui les ajoutent au fichier contôler
 * par l'intermédiares de tags namespacisés en anos afin de pouvoir les traités dans XéMèLios
 */
public class ControlContentHandler extends DefaultHandler implements ContentHandler {
	private static final Logger logger = Logger.getLogger(ControlContentHandler.class);
	private Pair collectivite,budget;
	public int MAX_ANOS = 300;
	private ArrayList<AbstractUnitControl> controlsToCall;
	private DocumentControlModel docControlMod;
	private DocumentModel docModel;
	//private DocumentModel docsModel;
	private Hashtable<String,Object> hParams;
	private Stack<String> stack;
	private MainWindow owner;
	//private String etat;
	private Vector<Anomalie> anos = new Vector<Anomalie>();
	private Vector<Anomalie> vTotalAnomalies;
	private Vector<ControlModel> controlsModel;
	private SaxXmlWriter saxXmlWriter;
	private NamespaceContext nsCtx;
	private boolean startEtatPassed=false;
	public boolean importInterupted=false;
	private String fileName;
//	private DlgControl dlgControl;
        private ControlProgressListener cpl;
	

	public void setControlProgressListener(ControlProgressListener cpl) {
		this.cpl=cpl;
	}
	public ControlContentHandler(DocumentModel dm,DocumentControlModel docCtrlMod,Hashtable<String,Object> parameters,MainWindow frame,String file){
		super();
		this.docModel = dm;
		this.docControlMod=docCtrlMod;
		this.owner=frame;
		this.fileName=file;
		this.nsCtx=dm.getNamespaces();
		vTotalAnomalies=new Vector<Anomalie>();
		stack=new Stack<String>();
		controlsToCall = new ArrayList<AbstractUnitControl>();
		hParams=parameters;
				
			// Le premier contrôle lit le code collectivité ainsi que le code budgétaire.
			try {
				COLLBUDG cbc = new COLLBUDG();
				cbc.setDocumentModel(docControlMod);
				cbc.setCch(this);
				cbc.setColl_path(docModel.getCollectivitePath().getCodePath().getPath());
				cbc.setColl_lib(docModel.getCollectivitePath().getLibellePath().getPath());
				cbc.setBudg_path(docModel.getBudgetPath().getCodePath().getPath());
				cbc.setBudg_lib(docModel.getBudgetPath().getLibellePath().getPath());
				controlsToCall.add(cbc);
			} catch (Exception e) {
				e.printStackTrace();
			}
			for(ControlModel ctrlm:docCtrlMod.getControls()){
				try {
					Object o=new Object();
					if(!ctrlm.id.startsWith("TECH")){
						o = Class.forName(ctrlm.getClasse()).newInstance();
					}
					if(o instanceof AbstractUnitControl) {
						AbstractUnitControl auc = (AbstractUnitControl)o;
						auc.setParameters(getParameters());
						auc.setDocumentModel(docControlMod);
						auc.setFileName(fileName);
						controlsToCall.add(auc);
					}
				} catch (InstantiationException e) {
					e.printStackTrace();
				} catch (IllegalAccessException e) {
					e.printStackTrace();
				} catch (ClassNotFoundException e) {
					e.printStackTrace();
				}
			}
	}

	public void startDocument() throws SAXException {
		for(AbstractUnitControl auc:controlsToCall) {
			try {
				auc.startDocument();
			} catch(SAXException saxEx) {
				logger.debug("in startDocument cch !!!",saxEx);
			}
		}
		saxXmlWriter.startDocument();
	}

	public void startElement(String uri, String localName, String qName,Attributes atts) throws SAXException {
		stack.push(localName);
		String xpath = getXpathExpression();
		XmlAttributesImpl attrs = new XmlAttributesImpl();
		for(int i=0;i<atts.getLength();i++) {
			attrs.addAttribute(atts.getURI(i),atts.getLocalName(i),atts.getQName(i),atts.getType(i),atts.getValue(i));
		}
		if(stack.size()==1){
			// Ajout du namespace ano: au document
			startEtatPassed=true;
			saxXmlWriter.startPrefixMapping(nsCtx.getPrefix(Anomalie.ANOMALY_NS_URI),Anomalie.ANOMALY_NS_URI);
		}
		attrs.addAttribute(Anomalie.ANOMALY_NS_URI,"node-id",
				(nsCtx.getPrefix(Anomalie.ANOMALY_NS_URI)!=null?nsCtx.getPrefix(Anomalie.ANOMALY_NS_URI)+":":"")+"node-id",
				"CDATA",
				IdGenerator.nextId());
		for(AbstractUnitControl auc:controlsToCall) {
			auc.startElement(uri,localName,qName,attrs,getXpathExpression());
		}
		saxXmlWriter.startElement(uri,localName,qName,attrs);
	}

	public void endElement(String uri, String localName, String qName) throws SAXException {
		Vector<Anomalie> vAno = new Vector<Anomalie>();
		for(AbstractUnitControl auc:controlsToCall) {
			try {
				anos = auc.endElement(uri,localName,qName,getXpathExpression());
				if(anos!=null && !anos.isEmpty()) {
					if(!(vTotalAnomalies.size() > MAX_ANOS)){
						vTotalAnomalies.addAll(anos);
						vAno.addAll(anos);
					}
				}
			} catch(SAXException saxEx) {
				logger.debug("in endElement!!!",saxEx);
			}
		}
		if(!vAno.isEmpty()){
			for(Anomalie ano:vAno){
				saxXmlWriter.writeAnomally(ano);
			}
		}
		if(stack.size()==1) saxXmlWriter.endPrefixMapping(nsCtx.getPrefix(Anomalie.ANOMALY_NS_URI));

		saxXmlWriter.endElement(uri,localName,qName);
		stack.pop();

	}

	public void endDocument() throws SAXException {
        try {
            if(collectivite==null || collectivite.key==null || collectivite.libelle==null){
                try {
                    collectivite = getCollectivite(docModel,fileName);
                    if(!owner.getConnectedUser().hasCollectivite(collectivite.key,docModel)) throw new UnauthorizedException("Voux n'êtes pas autorisé à contrôler des fichier de cette collectivite"+collectivite.libelle!=null?" ("+collectivite.libelle+")":"");
                } catch (ToolException e) {
                    e.printStackTrace();
                }
            }
            if(budget==null  || budget.key==null || budget.libelle==null || budget.libelle.equals("")){
                try {
                    budget = getBudget(docModel,collectivite,fileName);
                } catch (ToolException e) {
                    e.printStackTrace();
                }

            }
        } catch(UnauthorizedException ex) {
            throw new SAXException(ex);
        }
		saxXmlWriter.endDocument();
	}

	private Hashtable<String,Object> getParameters(){
		return hParams;
	}

	public void setParameters(Hashtable<String,Object> ho){
		hParams=ho;
	}

	private String getXpathExpression(){
		StringBuffer sb = new StringBuffer();
		for(String s:stack){
			sb.append(s+"/");
		}
		return sb.toString();
	}

	public void setControlsModel(Vector<ControlModel> cm){
		this.controlsModel=cm;
	}

	public void setSaxWriter(SaxXmlWriter sxw){
		saxXmlWriter=sxw;
	}

	public Vector<Anomalie> getVTotalAnomalies() {
		return vTotalAnomalies;
	}

	public Pair getCollectivite(DocumentModel dm, String fileName) throws ToolException {
		logger.debug("in getCollectivite");
		Pair ret = null;
		if(ret==null) {
			QueryCollectivite qc = new QueryCollectivite(fileName);
			qc.setDefaultCollectivite(collectivite);
			if(SwingUtilities.isEventDispatchThread()) qc.run();
			else try { SwingUtilities.invokeAndWait(qc); } catch(Throwable t) {}
			ret = qc.getRet();
			if(ret==null) {
				throw new ToolException("Action interrompue",ToolException.ERROR_INTERRUPTED);

			}
		}
		return ret;
	}

	public Pair getBudget(DocumentModel dm, Pair collectivite, String fileName) throws ToolException, UnauthorizedException {
		logger.debug("in getBudget");
		Pair ret = null;
		if(ret==null) {
			TreeSet<Pair> existing = new TreeSet<Pair>();
			try {
				existing.addAll(DataLayerManager.getImplementation().getBudgets(dm,collectivite,owner.getConnectedUser()));
			} catch (DataConfigurationException e) {
				logger.error("in getBudget",e);
			} catch (DataAccessException e) {
				logger.error("in getBudget",e);
			}
			Vector<Pair> budgetsDm = dm.getDefaultBudgets();
			existing.addAll(budgetsDm);
			if(docModel.getDefaultBudgets().firstElement() != null) existing.add(docModel.getDefaultBudgets().firstElement());
            if(budget!=null && (budget.libelle==null || budget.libelle.length()==0)) {
                if(existing.contains(budget)) {
                    for(Iterator<Pair> it=existing.iterator();it.hasNext();) {
                        Pair p = it.next();
                        if(p.key.equals(budget.key)) return p;
                    }
                }
            }
			QueryBudget qb = new QueryBudget(fileName,existing,collectivite.libelle);
			if(SwingUtilities.isEventDispatchThread()) qb.run();
			else try { SwingUtilities.invokeAndWait(qb); } catch(Throwable t) {}
			ret=qb.getRet();
			if(ret==null) {
				throw new ToolException("Action interrompue",ToolException.ERROR_INTERRUPTED);
			}
		}
		return ret;
	}


	private class QueryCollectivite implements Runnable {
		private Pair ret;
		private String fileName;
		private Pair defaultCollectivite = null;
		public Pair getDefaultCollectivite() {
			return defaultCollectivite;
		}
		public void setDefaultCollectivite(
				Pair defaultCollectivite) {
			this.defaultCollectivite = defaultCollectivite;
		}
		public QueryCollectivite(String fileName) {
			super();
			this.fileName=fileName;
		}
		public void run() {
			DlgGetPair dlg = new DlgGetPair(owner,"Sur quelle collectivité porte ce document : "+fileName+" ?",null,null,collectivite);
			if(dlg.run()) {
				ret = new Pair(dlg.getCode(),dlg.getLibelle());
			} else {
				importInterupted=true;
//				dlgControl.notifyImportInterupted();
//				dlgControl.setVisible(false);
                                cpl.notifyImportInterrupted(true);
			}
		}
		public Pair getRet() { return ret; }
	}

	private class QueryBudget implements Runnable {
		private String fileName;
		private TreeSet<Pair> existing;
		private Pair ret;
		private String collectivite = null;

		public QueryBudget(String fileName, TreeSet<Pair> existing, String collectivite) {
			super();
			this.fileName=fileName;
			this.existing=existing;
			this.collectivite=collectivite;
		}
		public void run() {
			DlgGetPair dlg = null;
			dlg= new DlgGetPair(owner,"Sur quel budget porte ce document : "+fileName+" ?",existing,collectivite,budget);

			if(dlg.run()) {
				ret = new Pair(dlg.getCode(),dlg.getLibelle());
			} else {
				if(!importInterupted){
//					dlgControl.notifyImportInterupted();
					importInterupted=true;
                                        cpl.notifyImportInterrupted(false);
				}
			}
			logger.debug("fin de QueryBudget.run()");
		}
		public Pair getRet() {
			logger.debug("QueryBudget.getRet()");
			return ret;
		}
	}
	public void startPrefixMapping(String prefix, String uri) throws SAXException {
		super.startPrefixMapping(prefix,uri);
		for(AbstractUnitControl auc:controlsToCall) {
			auc.startPrefixMapping(prefix,uri,getXpathExpression());
		}
		saxXmlWriter.startPrefixMapping(prefix,uri);
	}

	public void endPrefixMapping(String prefix) throws SAXException {
		super.endPrefixMapping(prefix);
		for(AbstractUnitControl auc:controlsToCall) {
			auc.endPrefixMapping(prefix,getXpathExpression());
		}
		saxXmlWriter.endPrefixMapping(prefix);
	}

	public void characters(char[] ch, int start, int length) throws SAXException {
		saxXmlWriter.characters(ch,start,length);
	}

	public void setDocumentLocator(Locator locator) {
		saxXmlWriter.setDocumentLocator(locator);
	}

	public void ignorableWhitespace(char[] ch, int start, int length) throws SAXException {
		saxXmlWriter.ignorableWhitespace(ch,start,length);
	}

	public void processingInstruction(String target, String data){
		try {
			saxXmlWriter.processingInstruction(target,data);
		} catch (SAXException e) {
			e.printStackTrace();
		}
	}

	public void skippedEntity(String name) throws SAXException {
		saxXmlWriter.skippedEntity(name);
	}

	public void setNamespaceContext(NamespaceContext nsCtx) {
		this.nsCtx = nsCtx;
	}

	public Pair getBudget() {
		return budget;
	}

	public void setBudget(Pair budget) {
		this.budget = budget;
	}

	public Pair getCollectivite() {
		return collectivite;
	}

	public void setCollectivite(Pair collectivite) {
		this.collectivite = collectivite;
                logger.debug("setting collectivite: "+collectivite);
	}
}