package cn.com.smartbi;

import java.io.*;
import java.util.*;
import java.util.Map.Entry;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import org.w3c.dom.*;
import smartbi.SmartbiException;
import smartbi.freequery.FreeQueryErrorCode;
import smartbi.freequery.metadata.IJavaQueryData;
import smartbi.freequery.metadata.JavaQueryConfig;
import smartbi.freequery.metadata.JavaQueryOutputField;
import smartbi.freequery.metadata.JavaQueryParameter;
import smartbi.freequery.querydata.CellData;
import smartbi.freequery.querydata.GridData;
import smartbi.net.sf.json.JSONObject;
import smartbi.util.StringUtil;
import smartbi.util.ValueType;
public class JavaQueryReadXml implements IJavaQueryData {
	private Map<String, JavaQueryConfig> configs = new LinkedHashMap<String, JavaQueryConfig>();
	private List<JavaQueryOutputField> outputFields;
	Element element = null;
	DocumentBuilder db = null;
	DocumentBuilderFactory dbf = null;
	public JavaQueryReadXml() {
		addConfig("FileName", "XML地址", "给定一个文件名","", true);
		addConfig("parentNode", "节点路径", "格式为:根节点/xx/xx...", "", true);
	}
	private void addConfig(String name, String alias, String desc,
			String defaultValue, boolean notNull) {
		JavaQueryConfig p = new JavaQueryConfig();
		p.setName(name);
		p.setAlias(alias);
		p.setDesc(desc);
		p.setValue(defaultValue);
		p.setNotNull(notNull);
		configs.put(name, p);
	}
	@Override
	public List<JavaQueryConfig> getConfigs() {
		return new ArrayList<JavaQueryConfig>(configs.values());
	}
	@Override
	public GridData getGridData(int arg0, int arg1) {
		Element parentNode = getParentNode(configs.get("parentNode").getValue());
		NodeList nodelist;
		List<List<CellData>> datas = new ArrayList<List<CellData>>();
		if(parentNode.getParentNode()==null||parentNode.getParentNode().getNodeType()==9){
			nodelist = parentNode.getChildNodes();
			 List<CellData> row = new ArrayList<CellData>();
			 for(JavaQueryOutputField j:outputFields){
					for(int k = 0;k<nodelist.getLength();k++){
						if(nodelist.item(k).getNodeName() == j.getId()){
							CellData c = new CellData();
							c.setStringValue(nodelist.item(k).getTextContent());
							row.add(c);
						}
					}
				}
			 datas.add(row);
		}else{
			Element parent_parentNode = (Element) parentNode.getParentNode();
			NodeList pChildren = parent_parentNode.getChildNodes();
			nodelist =  parent_parentNode.getElementsByTagName(parentNode.getNodeName());
			for(int i = 0;i<nodelist.getLength();i++){
				List<CellData> row = new ArrayList<CellData>();
				NodeList childlist = nodelist.item(i).getChildNodes();
				for(JavaQueryOutputField j:outputFields){
					for(int k = 0;k<pChildren.getLength();k++){
						if(pChildren.item(k).getNodeName() == j.getId()){
							CellData c = new CellData();
							c.setStringValue(pChildren.item(k).getTextContent());
							row.add(c);
						}
					}
					for(int k = 0;k<childlist.getLength();k++){
						if(childlist.item(k).getNodeName() == j.getId()){
							CellData c = new CellData();
							c.setStringValue(childlist.item(k).getTextContent());
							row.add(c);
						}
					}
				}
				datas.add(row);
			}
		}
		
		GridData d = new GridData();
		List<String> headers = new ArrayList<String>();
		for (JavaQueryOutputField of : outputFields)
			headers.add(of.getName());
		d.setStringHeaders(headers);
		d.setData(datas);
		return d;
	}
	@Override
	public List<JavaQueryOutputField> getOutputFields() {
		return outputFields;
	}
	@Override
	public List<JavaQueryParameter> getParameters() {
		return new ArrayList<JavaQueryParameter>();
	}
	@Override
	public int getRowCount() {
		return Integer.MAX_VALUE;
	}
	@Override
	public void init() {
		Element parentNode = getParentNode(configs.get("parentNode").getValue());
		if(parentNode==null){
			throw new SmartbiException(FreeQueryErrorCode.JAVA_QUERY_DEFINE_CLASS_ERROR).setDetail("未查询到记录！");
		}
		NodeList childNodes = parentNode.getChildNodes();
		outputFields = new ArrayList<JavaQueryOutputField>();
		if(parentNode.getParentNode()==null||parentNode.getParentNode().getNodeType()==9){
			addOutput(childNodes);
		}else{
			Element pNode = (Element) parentNode.getParentNode();
			NodeList pChildren = pNode.getChildNodes();
			addOutput(pChildren);
			addOutput(childNodes);
		}
	}
	@Override
	public void loadConfigs(String configStr) {
		if (StringUtil.isNullOrEmpty(configStr))
			return;
		JSONObject obj = JSONObject.fromString(configStr);
		configs.get("FileName").setValue(
				obj.has("FileName") ? obj.getString("FileName") : null);
		configs.get("parentNode").setValue(
				obj.has("parentNode") ? obj.getString("parentNode") : null);
	}
	@Override
	public String saveConfigs() {
		JSONObject json = new JSONObject();
		for (JavaQueryConfig config : configs.values())
			json.put(config.getName(), config.getValue());
		return json.toString();
	}
	@Override
	public void setConfigValue(String key, String value) {
		configs.get(key).setValue(value);
	}
	@Override
	public void setConfigValues(Map<String, String> configValues) {
		for (Entry<String, String> config : configValues.entrySet())
			configs.get(config.getKey()).setValue(config.getValue());
	}
	@Override
	public void setParameterValue(String arg0, String arg1, String arg2) {
	}
	
	@Override
	public void close() {
	}
	
	public void addOutput(NodeList childNodes){
		for(int j = 0; j < childNodes.getLength(); j++){
			Node e = childNodes.item(j);
			if (e.getNodeType() == 1 && isValidateNode(e)) {
				JavaQueryOutputField jf = new JavaQueryOutputField();
				jf.setId(e.getNodeName());
				jf.setName(e.getNodeName());
				jf.setAlias(e.getNodeName());
				jf.setDataType(ValueType.STRING);
				outputFields.add(jf);
			}
		}
	}
	// 判断是否合法节点
	private boolean isValidateNode(Node e) {
		if (e.getChildNodes().getLength() == 1) {
			Node child_node = e.getChildNodes().item(0);
			if (child_node.getNodeType() == 3||child_node.getNodeType() == 4) { //4是CDATA类型的数据，3是文本text
				return true;
			} else {
				return false;
			}
		} else if(e.getChildNodes().getLength()==0){
			return true;
		}else{
			return false;
		}
	}
	// 获取指定根节点
	private Element getParentNode(String str) {
		File f = new File(configs.get("FileName").getValue());
		try {
			dbf = DocumentBuilderFactory.newInstance();
			// 返回db对象用documentBuilderFatory对象获得返回documentBuildr对象
			db = dbf.newDocumentBuilder();
			// 得到一个DOM并返回给document对象
			Document dt = db.parse(f);
			// 得到一个element根元素
			element = dt.getDocumentElement();
			if(str.indexOf("/") != -1){
				String[] node_val = str.split("/");
				Element parentNode = element;
				for(int i = 0; i < node_val.length; i++){
					NodeList childNodes = parentNode.getChildNodes();
					for(int j = 0; j < childNodes.getLength(); j++){
						if(node_val[i].equals(childNodes.item(j).getNodeName())){
							parentNode = (Element) childNodes.item(j);
							break;
						}
					}
				}
				return parentNode;
			}else{
				return element;
			}
		} catch (Exception e) {
			e.printStackTrace();
			return element;
		}
	}
}