/*
 * Copyright 
 *   2007 axYus - www.axyus.com
 *   2007 N.Le Corre - nicolas.lecorre@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.ui;

import java.awt.Cursor;
import java.util.ArrayList;
import java.util.Vector;

import javax.swing.event.TreeModelListener;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;

import org.apache.log4j.Logger;

import fr.gouv.finances.cp.utils.Pair;
import fr.gouv.finances.cp.xemelios.common.config.DocumentModel;
import fr.gouv.finances.cp.xemelios.common.config.DocumentsModel;
import fr.gouv.finances.cp.xemelios.data.DataAccessException;
import fr.gouv.finances.cp.xemelios.data.DataConfigurationException;
import fr.gouv.finances.cp.xemelios.data.DataLayerManager;

public class DatabaseTreeDataDonnee implements TreeModel {
	private static Logger logger = Logger.getLogger(DatabaseTreeDataDonnee.class);
    private ArrayList<TreeModelListener> listeners;
    private DocumentsModel dm;
    private DataNode root;
    
    public DatabaseTreeDataDonnee(DocumentsModel dm) {
        super();
        this.dm = dm;
        root = new DataNode(dm,DataNode.TYPE_ROOT);
        root.tree=this;
        loadChildren(root);
        listeners = new ArrayList<TreeModelListener>();
    }
    
    public void addTreeModelListener(TreeModelListener l) {
        listeners.add(l);
    }
    public Object getChild(Object parent, int index) {
        DataNode dn = (DataNode)parent;
        if(!dn.areChildrenLoaded()) {
            loadChildren(dn);
        }
        return dn.getChildren().elementAt(index);
    }
    public int getChildCount(Object parent) {
        DataNode dn = (DataNode)parent;
        if(!dn.areChildrenLoaded()) {
            loadChildren(dn); 
        }
        return dn.getChildren().size();
    }
    public int getIndexOfChild(Object parent, Object child) {
        DataNode dn = (DataNode)parent;
        return dn.getChildren().indexOf(child);
    }
    public Object getRoot() {
        return root;
    }
    public boolean isLeaf(Object node) {
        DataNode dn = (DataNode)node;
        //return dn.getType()==DataNode.TYPE_PERSO2 || dn.getChildCount()==0;
        if (dn.getType()==DataNode.TYPE_PERSO2) {
        	return true;
        } else if (dn.getType()==DataNode.TYPE_PERSO1 && dn.getChildCount()==0) {
        	return true;
        } else if (dn.getType()==DataNode.TYPE_BUDGET && dn.getChildCount()==0) {
        	return true;
        } else {
        	return false;
        }
    }
    public void removeTreeModelListener(TreeModelListener l) {
        listeners.remove(l);
    }
    public void valueForPathChanged(TreePath path, Object newValue) {}
    protected Vector<Pair> getCollectivitesFromDocumentModel(DocumentModel dm) {
        Vector<Pair> ret = null;
        try {
            ret = DataLayerManager.getImplementation().getCollectivites(dm);
        } catch(DataAccessException daEx) {
            logger.warn("",daEx);
            ret = new Vector<Pair>();
        } catch(DataConfigurationException dcEx) {
            logger.error("",dcEx);
            ret = new Vector<Pair>();
        }
        return ret;
    }
    protected void loadChildren(DataNode dn) {
        switch(dn.getType()) {
            case DataNode.TYPE_ROOT: {
                for(DocumentModel doc:dm.getDocuments()) {
                    if(doc.getDisplayInMenuIf()==null || "true".equals(System.getProperty(doc.getDisplayInMenuIf()))) dn.addChild(doc);
                }
                dn.childrenLoaded();
                break;
            }
            case DataNode.TYPE_DOCUMENT: {
                Vector<Pair> collectivites = getCollectivitesFromDocumentModel((DocumentModel)dn.getData());
                for(Pair p:collectivites) dn.addChild(p);
                dn.childrenLoaded();
                break;
            }
            case DataNode.TYPE_COLLECTIVITE: {
                Vector<Pair> budgets = null;
                try {
                    Pair collectivite = (Pair)dn.getData();
                    DocumentModel dm = (DocumentModel)dn.getParent().getData();
                    budgets = DataLayerManager.getImplementation().getBudgets(dm,collectivite);
                } catch(DataAccessException daEx) {
                    logger.error("",daEx);
                    budgets = new Vector<Pair>();
                } catch(DataConfigurationException dcEx) {
                    logger.error("",dcEx);
                    budgets = new Vector<Pair>();
                }
                for(Pair budget:budgets) dn.addChild(budget);
                dn.childrenLoaded();
                break;
            }
            case DataNode.TYPE_BUDGET: {
                // il faut aller chercher la liste des special-key
                MainWindow.getInstance().setCursor(new Cursor(Cursor.WAIT_CURSOR));
                DocumentModel doc = (DocumentModel)(dn.getParent().getParent().getData());
                Pair collectivite = (Pair)dn.getParent().getData();
                Pair budget = (Pair)dn.getData();
                Vector<Pair> ret = new Vector<Pair>();
                try {
                    ret = DataLayerManager.getImplementation().getSpecialKeys1(doc,collectivite,budget,true);
                } catch(DataAccessException daEx) {
                    logger.error("",daEx);
                    ret = new Vector<Pair>();
                } catch(DataConfigurationException dcEx) {
                    logger.error("",dcEx);
                    ret = new Vector<Pair>();
                } finally {
                	MainWindow.getInstance().setCursor(Cursor.getDefaultCursor());
                }
                for(Pair p:ret) dn.addChild(p);
                dn.childrenLoaded();
                break;
            }
            case DataNode.TYPE_PERSO1: {
                // il faut aller chercher la liste des special-key
            	MainWindow.getInstance().setCursor(new Cursor(Cursor.WAIT_CURSOR));
                DocumentModel doc = (DocumentModel)(dn.getParent().getParent().getParent().getData());
                Pair collectivite = (Pair)dn.getParent().getParent().getData();
                Pair budget = (Pair)dn.getParent().getData();
                Pair key1 = (Pair)dn.getData();
                Vector<Pair> ret = null;
                try {
                    ret = DataLayerManager.getImplementation().getSpecialKeys2(doc,collectivite,budget,key1,true);
                } catch(DataAccessException daEx) {
                    logger.error("",daEx);
                    ret = new Vector<Pair>();
                } catch(DataConfigurationException dcEx) {
                    logger.error("",dcEx);
                    ret = new Vector<Pair>();
                } finally {
                	MainWindow.getInstance().setCursor(Cursor.getDefaultCursor());
                }
                for(Pair p:ret) dn.addChild(p);
                dn.childrenLoaded();
                break;
            }
        }
    }

	public class DataNode {
	    public static final transient int TYPE_ROOT = 0; 
	    public static final transient int TYPE_DOCUMENT = 1;
	    public static final transient int TYPE_COLLECTIVITE = 2;
	    public static final transient int TYPE_BUDGET = 3;
	    public static final transient int TYPE_PERSO1 = 4;
	    public static final transient int TYPE_PERSO2 = 5;
	    public static final transient int TYPE_PERSO3 = 6;
	    private boolean areChildrenLoaded = false;
	    private Object data;
	    private int type;
	    private Vector<DataNode> children;
	    private DataNode parent;
	    private DatabaseTreeDataDonnee tree = null;
	    
	    public DataNode(Object data, int type) {
	        super();
	        this.data = data;
	        this.type = type;
	        children = new Vector<DataNode>();
	    }
	    @Override
	    public boolean equals(Object o) {
	        if(o==this) return true;
	        if(o instanceof DataNode) return data.equals(((DataNode)o).data);
	        return o.equals(data);
	    }
	    @Override
	    public String toString() {
	        // mise en majuscule de la première lettre de data
//	    	if (this.data.toString()==null || "".equals(this.data.toString())) return null;
//	        String ret = this.data.toString().substring(0, 1).toUpperCase(); 
//	        ret += this.data.toString().substring(1, this.data.toString().length());
//	    	return ret;
	    	return this.data.toString();
	    }
	    public DataNode getParent() { return parent; }
	    public Vector<DataNode> getChildren() { return children; }
	    public void addChild(Object child) {
	    	if(child==null) return;
	        DataNode node = null;
	        if(!(child instanceof DataNode)) {
	            node = new DataNode(child,type+1);	            
	        } else node = (DataNode)child;
	        node.parent = this;
	        children.add(node);
	        node.tree=this.tree;
	    }
	    public int getChildCount() {
	    	if(!areChildrenLoaded) {
	    		tree.loadChildren(this);
	    	}
	    	return children.size();
	    }
	    @Override
	    public int hashCode() { return data.hashCode(); }
	    public void childrenLoaded() { areChildrenLoaded = true; }
	    public boolean areChildrenLoaded() { return areChildrenLoaded; }
	    public int getType() { return type; }
	    public Object getData() { return data; }
	}    
}
