/*
    * Copyright 2004-2006 the Seasar Foundation and the Others.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *     http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
   * either express or implied. See the License for the specific language
   * governing permissions and limitations under the License.
   */
  /*
   * Created on 2005/08/23
   */
  package org.seasar.tuigwaa.controller;
  
  import java.util.Collection;
  import java.util.Iterator;
  import java.util.Map;
  
  import org.apache.commons.lang.ArrayUtils;
  import org.seasar.tuigwaa.database.CriteriaFunctionFactory;
  import org.seasar.tuigwaa.database.function.CriteriaExeFunction;
  import org.seasar.tuigwaa.database.function.aggregation.AggregationFunction;
  import org.seasar.tuigwaa.database.function.aggregation.GroupProjectionFunction;
  import org.seasar.tuigwaa.database.function.aggregation.ProjectionFunction;
  import org.seasar.tuigwaa.database.function.criteria.CriteriaFunction;
  import org.seasar.tuigwaa.database.function.criteria.CriteriaListFunction;
  import org.seasar.tuigwaa.model.core.TgwAttribute;
  import org.seasar.tuigwaa.model.core.TgwEntity;
  import org.seasar.tuigwaa.model.core.impl.FloatAttribute;
  import org.seasar.tuigwaa.model.core.impl.IntegerAttribute;
  import org.seasar.tuigwaa.model.core.impl.NestedAttribute;
  import org.seasar.tuigwaa.util.ajax.AbstractRowSetForm;
  
  import com.isenshi.util.HtmlBuffer;
  
  /***
   * This class is registerd into JSON Ajax.
   * 集計値を作成するフォームに対応。
   * 
   * @author nishioka
   */
  public class AggregationForm extends AbstractRowSetForm {
  
  	private static final long serialVersionUID = -6793435086829923995L;
  
  	private static final String[] AGGREGATION_TYPES = { "rowcount", "avg",
  			"min", "max", "sum", "countDistinct" };
  
  	private static final String[] NUMFIELD_TYPE = { AGGREGATION_TYPES[1],
  			AGGREGATION_TYPES[2], AGGREGATION_TYPES[3], AGGREGATION_TYPES[4] };
  
  	private String name_;
  
  	private String filterName_;
  
  	private TgwEntity entity_;
  
  	// private String[] groupFields_;
  
  	private String groupField1;
  
  	private String groupField2;
  
  	private String[] dataFilters_;
  
  	private Map dataFilterMap_;
  
  	private boolean valueFlag;
  
  	private boolean extraRowFlag;
  
  	public AggregationForm(TgwEntity entity, Map dataFilterMap) {
  		this.entity_ = entity;
  		this.dataFilterMap_ = dataFilterMap;
  		addRow(createNewRow());
  	}
  
  	// [Start] ----- Getter Setter Methods -----
  
  	public String[] getDataFilters() {
  		return dataFilters_;
  	}
  
  	public void setDataFilters(String[] dataFilters) {
  		this.dataFilters_ = dataFilters;
  	}
  
  	public Collection getDataFilterList() {
  		return dataFilterMap_.values();
  	}
  
  	public void setGroupField1(String groupField1) {
  		this.groupField1 = groupField1;
 	}
 
 	public void setGroupField2(String groupField2) {
 		this.groupField2 = groupField2;
 	}
 
 	public String getGroupField1() {
 		return groupField1;
 	}
 
 	public String getGroupField2() {
 		return groupField2;
 	}
 
 	/*
 	 * public void setGroupFields(String[] groupFields) { this.groupFields_ =
 	 * groupFields; }
 	 * 
 	 * public String[] getGroupFields() { return groupFields_; }
 	 */
 	public TgwEntity getEntity() {
 		return entity_;
 	}
 
 	public String getName() {
 		return name_;
 	}
 
 	public void setName(String name) {
 		this.name_ = name;
 	}
 
 	public void setFilterName(String filterName) {
 		this.filterName_ = filterName;
 	}
 
 	public String getFilterName() {
 		return filterName_;
 	}
 
 	public void setValueFlag(boolean valueFlag) {
 		this.valueFlag = valueFlag;
 	}
 
 	public boolean isValueFlag() {
 		return valueFlag;
 	}
 
 	public void setExtraRowFlag(boolean extraRowFlag) {
 		this.extraRowFlag = extraRowFlag;
 	}
 
 	public boolean isExtraRowFlag() {
 		return extraRowFlag;
 	}
 
 	// [End] ----- Getter Setter Methods -----
 
 	// [Start] ----- Ajax Methods -----
 
 	protected void removeRow(int index, Object row) {
 	}
 
 	protected String[] getHtmlData(int index, Object row) {
 		String[] data = new String[4];
 		FunctionForm form = (FunctionForm) row;
 
 		data[0] = createTypeSelector(index, form);
 		data[1] = createTextField(getPrefix(index) + "displayName");
 		data[2] = createFieldSelector(index, form);
 
 		if (valueFlag) {
 			data[3] = "";
 		} else {
 			data[3] = createRemoveButtonHtml(index, getMessage("button.remove"));
 		}
 		return data;
 	}
 
 	protected Object createRow(int index, String value) {
 		return new FunctionForm(value);
 	}
 
 	private String createTypeSelector(int index, FunctionForm form) {
 		HtmlBuffer buf = new HtmlBuffer();
 		buf.appendStartTag("select");
 		buf.appendAttribute("name", getPrefix(index) + "type");
 		bindTypeEvent(buf);
 		String[] pickTypes = form.getPickTypes();
 
 		for (int i = 0; i < pickTypes.length; i++) {
 			String value = pickTypes[i];
 			String label = getMessage("aggregation.label." + value);
 			buf.appendOption(label, value, form.getType());
 		}
 		buf.endTag();
 
 		return buf.toString();
 	}
 
 	public String createFieldSelector(int index, FunctionForm form) {
 		if (form.getType().equals(AGGREGATION_TYPES[0])) {
 			return "";
 		}
 		boolean onlyNumAttr = false;
 		if (ArrayUtils.contains(NUMFIELD_TYPE, form.getType())) {
 			onlyNumAttr = true;
 		}
 
 		HtmlBuffer buf = new HtmlBuffer();
 		buf.appendStartTag("select");
 		buf.appendAttribute("name", getPrefix(index) + "field");
 
 		// Iterator itr = entity_.getFieldIterator();
 		Iterator itr = entity_.getFirstDepthAllFieldList().iterator();
 		while (itr.hasNext()) {
 			TgwAttribute field = (TgwAttribute) itr.next();
 			if (!onlyNumAttr || isNumField(field)) {
 				buf.appendOption(field.getDisplayName(), field.getName());
 			}
 		}
 		buf.endTag();
 		return buf.toString();
 	} /*
 		 * public void bindNumFieldOption(HtmlBuffer buf, TgwAttribute field,
 		 * String displayPrefix, String prefix) { } }
 		 */
 
 	private boolean isNumField(TgwAttribute field) {
 		if (field instanceof IntegerAttribute
 				|| field instanceof FloatAttribute) {
 			return true;
 		}
 		if (field instanceof NestedAttribute) {
 			return isNumField(((NestedAttribute) field).getRealField());
 		}
 		return false;
 	}
 
 	protected Object createNewRow() {
 		return new FunctionForm();
 	}
 
 	// [End] ----- Ajax Methods -----
 
 	public CriteriaExeFunction getAggregatoinExeFunction() {
 		CriteriaListFunction criteriaList = new CriteriaListFunction();
 		CriteriaExeFunction exe = CriteriaFunctionFactory
 				.createCriteriaExeFunction(getEntity(), getName(), criteriaList);
 
 		if (dataFilters_ != null) {
 			for (int i = 0; i < dataFilters_.length; i++) {
 				CriteriaExeFunction dataFilter = (CriteriaExeFunction) dataFilterMap_
 						.get(dataFilters_[i]);
 				criteriaList
 						.injectCriteriaListFunction((CriteriaFunction) dataFilter
 								.getUnaryCriteriaFunction());
 			}
 		}
 
 		AggregationFunction aggFunction = getAggregaionFunction(exe);
 		criteriaList.addFunction(aggFunction);
 
 		if (isValueFlag()) {
 			exe.setValueFlag(true);
 		}
 
 		if (extraRowFlag) {
 			exe.setExtraRowFlag(true);
 		}
 		if (groupField2 != null && groupField2.length() > 0) {
 			exe.setSpreadSheetFlag(true);
 		}
 		return exe;
 	}
 
 	public AggregationFunction getAggregaionFunction(CriteriaExeFunction exe) {
 		AggregationFunction aggFunction = new AggregationFunction();
 
 		addGroupField(exe, aggFunction, groupField1);
 		addGroupField(exe, aggFunction, groupField2);
 
 		for (Iterator i = getRowSet().iterator(); i.hasNext();) {
 			FunctionForm form = (FunctionForm) i.next();
 			String displayName = form.getDisplayName();
 			String type = form.getType();
 
 			ProjectionFunction function = CriteriaFunctionFactory
 					.getProjectionFunction(type);
 			function.setField(form.getField());
 			function.setAlias(displayName);
 
 			aggFunction.addFunction(function);
 			exe.addProjection(displayName);
 
 		}
 		return aggFunction;
 	}
 
 	private void addGroupField(CriteriaExeFunction exe,
 			AggregationFunction aggFunction, String fieldName) {
 		if (fieldName == null) {
 			return;
 		}
 
 		String[] options = fieldName.split("//?");
 		String option = null;
 		if (options.length == 2) {
 			fieldName = options[0];
 			option = options[1];
 		}
 		TgwAttribute attr = entity_.getField(fieldName);
 		String displayName = attr.getDisplayName();
 		GroupProjectionFunction groupFunction = new GroupProjectionFunction();
 		groupFunction.setField(fieldName);
 		groupFunction.setAlias(fieldName);
 		groupFunction.setOption(option);
 		
 		aggFunction.addFunction(groupFunction);
 		exe.addProjection(displayName);
 	}
 
 	public class FunctionForm {
 		private String type_;
 
 		private String displayName_;
 
 		private String field_;
 
 		public FunctionForm() {
 			this.type_ = AGGREGATION_TYPES[0];
 			// this(type);
 			// this(AGGREGATION_TYPES[0]);
 		}
 
 		public FunctionForm(String type) {
 			this.type_ = type;
 		}
 
 		public String[] getPickTypes() {
 			// Iterator itr = entity_.getFieldIterator();
 			Iterator itr = entity_.getFirstDepthAllFieldList().iterator();
 			boolean existNumField = false;
 			while (itr.hasNext()) {
 				TgwAttribute field = (TgwAttribute) itr.next();
 				if (field instanceof NestedAttribute) {
 					field = ((NestedAttribute) field).getRealField();
 				}
 
 				if (field instanceof IntegerAttribute
 						|| field instanceof FloatAttribute) {
 					existNumField = true;
 					break;
 				}
 			}
 			if (existNumField) {
 				return AGGREGATION_TYPES;
 			} else {
 				return new String[] { "rowcount" };
 			}
 		}
 
 		public void setType(String type) {
 			// this.type_ = type;
 			// Do noting
 		}
 
 		public String getType() {
 			return type_;
 		}
 
 		public String getDisplayName() {
 			return displayName_;
 		}
 
 		public void setDisplayName(String displayName) {
 			this.displayName_ = displayName;
 		}
 
 		public String getField() {
 			return field_;
 		}
 
 		public void setField(String field) {
 			this.field_ = field;
 		}
 	}
 }