/*
    * Created on Oct 3, 2004
    *
    */
   package org.extremecomponents.tree.tag;
   
   import java.util.Iterator;
   import java.util.List;
   
  import javax.servlet.jsp.JspException;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.apache.taglibs.standard.lang.support.ExpressionEvaluatorManager;
  import org.extremecomponents.table.bean.Column;
  import org.extremecomponents.table.bean.Table;
  import org.extremecomponents.table.model.ExportHandler;
  import org.extremecomponents.table.model.TableModel;
  import org.extremecomponents.table.tag.TableTag;
  import org.extremecomponents.table.tag.TagUtils;
  import org.extremecomponents.tree.model.TreeModel;
  import org.extremecomponents.util.ExceptionUtils;
  
  /***
   * org.extremecomponents.tree.tag.TreeTag.java - 
   * 
   * @author Paul Horn
   */
  public class TreeTag extends TableTag
  {
  	private static final long serialVersionUID = -6626881785728133686L;
  	private static Log logger = LogFactory.getLog(TreeTag.class);
  	
  	/***
  	 * Will iterate over the columns the number of times specified by the maxRows attribute.
  	 */
  	private Iterator iterator;
  	/***
  	 * Attribute of the bean holding a reference to it's parent.
  	 */
  	public String parentAttribute;
  	/***
  	 * Attribute of the bean holding a reference to itself.
  	 */
  	private String identifier;
  	/***
  	 * The Model represents the internal plumbing of the table
  	 */
  	private TreeModel model;
  	
  	/***
  	 * @return Returns the parentAttribute.
  	 */
  	public String getParentAttribute()
  	{
  		return parentAttribute;
  	}
  	/***
  	 * @param parentAttribute The parentAttribute to set.
  	 */
  	public void setParentAttribute(String parentAttribute)
  	{
  		this.parentAttribute = parentAttribute;
  	}
  	/***
  	 * @return Returns the identifier.
  	 */
  	public String getIdentifier()
  	{
  		return identifier;
  	}
  	/***
  	 * @param identifier The identifier to set.
  	 */
  	public void setIdentifier(String identifier)
  	{
  		this.identifier = identifier;
  	}		
  	/***
  	 * @return Returns the model.
  	 */
  	public TableModel getModel()
  	{
  		return (TableModel) model;
  	}
  	/***
  	 * @param model The model to set.
  	 */
  	public void setModel(TreeModel model)
  	{
  		this.model = model;
  	}
  	
  	public boolean hasMetaData()
  	{
  		return model.getViewHandler().rowcount() > 0;
  	}
  	
  	/***
 	 * Add the Column meta data to the Model
 	 */
 	public void addColumnMetaData(Column column)
 	{
 		model.getColumnMetaData().addColumn(column);
 	}
 	
 	/***
 	 * Add a Column value to the Model
 	 */
 	public void addColumnValue(String property, Object value)
 	{
 		model.getViewHandler().addColumnValue(property, value);
 	}	
 	
 	/***
 	 * Find out how many rows have been added to the Model.
 	 */
 	public int getModelRowsSize()
 	{
 		return model.getViewHandler().rowcount();
 	}	
 	
 	public void addProperty(String name, String value)
 	{
 		model.getTreeProperties().setProperty(name, value);
 	}
 	
 	/***
 	 * The start of the tag. First fire up the Model and
 	 * then make a copy of the table tag attributes to
 	 * send to the model.
 	 */
 	public int doStartTag() throws JspException
 	{
 		logger.debug("TreeTag.doStartTag()");
 		
 		try
 		{
 			model = new TreeModel(pageContext); //just fire up the tablemodel
 			model.setParentAttribute(parentAttribute);
 			model.setIdentifier(identifier);
 			
 			if (getAction() != null)
 			{
 				setAction((String) ExpressionEvaluatorManager.evaluate("action", getAction(), String.class, this, pageContext));
 			}
 			
 			if (getImagePath() != null)
 			{
 				setImagePath((String) ExpressionEvaluatorManager.evaluate("imagePath", getImagePath(), String.class, this, pageContext));
 			}
 			
 			Table table = new Table((TableModel) model); //make a copy
 			table.setId(getId());
 			table.setCollection(getCollection());
 			table.setScope(getScope());
 			table.setAction(getAction());
 			table.setStyleClass(getStyleClass());
 			table.setHeaderClass(getHeaderClass());
 			table.setBorder(getBorder());
 			table.setCellpadding(getCellpadding());
 			table.setCellspacing(getCellspacing());
 			table.setMaxRows("0");
 			table.setFilter(getFilter());
 			table.setPagination(getPagination());
 			table.setImagePath(getImagePath());
 			table.setSort(getSort());
 			table.setTitle(getTitle());
 			table.setStyle(getStyle());
 			table.setWidth(getWidth());
 			model.setTable(table);
 			
 			pageContext.setAttribute("ROWCOUNT", "0");
 			
 			model.getRegistry().init(pageContext);
 		}
 		catch (Exception e)
 		{
 			throw new JspException("TreeTag.doStartTag() Problem: " + ExceptionUtils.formatStackTrace(e));
 		}
 
 		return EVAL_BODY_INCLUDE;
 	}
 	
 	/***
 	 * Two things need to be accomplished here.
 	 * First, need to iterate once over the columns to load up all the attributes.
 	 * Second, need to iterate over the columns as many times as specified by the maxRows
 	 * attribute so the columns rows can be resolved. On each iteration over the columns
 	 * the current bean in the Collection is passed via the PageContext.
 	 */
 	public int doAfterBody() throws JspException
 	{
 		logger.debug("TreeTag.doAfterBody()");
 
 		TreeModel model = (TreeModel) getModel();
 		try
 		{
 			if (model.getViewHandler().rowcount() == 0)
 			{
 				List rows = model.init();
 				iterator = rows.iterator();
 			}
 		}
 		catch (Exception e)
 		{
 			e.printStackTrace();
 			throw new JspException("TreeTag.doAfterBody() Problem: " + ExceptionUtils.formatStackTrace(e));
 		}
 
 		if ((iterator != null) && iterator.hasNext())
 		{
 			model.getViewHandler().addRow();
 			Object rowItem = iterator.next();
 			logger.debug("Row: " + rowItem.toString());
 			pageContext.setAttribute(getCollection(), rowItem);
 			pageContext.setAttribute("ROWCOUNT", "" + getModelRowsSize());
 
 			return EVAL_BODY_AGAIN;
 		}
 		else
 		{
 			return SKIP_BODY;
 		}
 	}
 	
 	/***
 	 * As a convenience use the EL expression langauge for the columns that need it.
 	 * Also execute the Model so that it can do all its processing and build the table.
 	 */
 	public int doEndTag() throws JspException
 	{
 		logger.debug("TreeTag.doEndTag()");
 
 		try
 		{
 			if (!model.getColumnMetaData().hasMetatData())
 			{
 				List rows = model.init();
 				TagUtils.autoSetColumns(this, rows);
 			}
 
 			Object output = model.getViewHandler().getView().afterBody(model);
 
 			if (output instanceof String)
 			{
 				pageContext.getOut().println(output);
 			}
 
 			pageContext.getRequest().setAttribute(ExportHandler.EXPORT_DATA, output);
 		}
 		catch (Exception e)
 		{
 			throw new JspException("TreeTag.doEndTag() Problem: " + ExceptionUtils.formatStackTrace(e));
 		}
 		
 		model.destroy();
 		cleanup();
 
 		return EVAL_PAGE;
 	}	
 	
 	public void cleanup()
 	{
 		iterator = null;
 		parentAttribute = null;
 		super.cleanup();
 	}
 
 }