package com.smtaiserver.smtaiserver.javaai.metrics.base;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.tree.DefaultText;

import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
import com.smtaiserver.smtaiserver.javaai.ast.ASTDBMap;

import com.smtservlet.util.Json;
import com.smtservlet.util.SMTStatic;



@SuppressWarnings("unused")
public class SMTMetricsSampleXml
{

	///////////////////////////////////////////////////////////////////////////////////////
	private static class SQLXMLExecArg
	{
		public SMTMetricsDefXml 	_metricsDef;
		public ASTDBMap 			_dbMap;
		public Json 				_jsonAST;
		public SMTAIServerRequest 	_tranReq;
		public StringBuilder		_sbQuestion = new StringBuilder();
		public StringBuilder		_sbSqlText;
		public List<Object>			_listSqlArgs;
		public List<String>			_listQuestion;
		
		public SQLXMLExecArg(SMTMetricsDefXml metricsDef, ASTDBMap dbMap, Json jsonAST, SMTAIServerRequest tranReq, List<String> r_list)
		{
			_metricsDef = metricsDef;
			_dbMap = dbMap;
			_jsonAST = jsonAST;
			_tranReq = tranReq;
			_listQuestion = r_list;
		}
	}
	
	///////////////////////////////////////////////////////////////////////////////////////
	private static abstract class SQLXMLNode
	{
		public abstract SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception;
	}
	
	///////////////////////////////////////////////////////////////////////////////////////
	private static abstract class SQLXMLNodeSQLXml extends SQLXMLNode
	{
		private List<Object>	_listChildren = new ArrayList<>();
		

		public SQLXMLNodeSQLXml(Element xmlRoot) throws Exception
		{
			for (Iterator<Node> iterInner = xmlRoot.nodeIterator(); iterInner.hasNext();)
			{			
				Node nodeInner = iterInner.next();
				if(nodeInner.getNodeType() == Node.TEXT_NODE)
				{
					String text = ((DefaultText)nodeInner).getText();
					int lastPos = _listChildren.size() - 1;
					if(lastPos >= 0 && _listChildren.get(lastPos) instanceof String)
					{
						_listChildren.set(lastPos, (String)_listChildren.get(lastPos) + text);
					}
					else
					{
						_listChildren.add(text);
					}
				}	
				else
				{
					_listChildren.add(createSQLXMLNode((Element)nodeInner));
				}
			}		
		}
		
		protected SMTJavaAIError parseSQLText(SQLXMLExecArg execArg, StringBuilder r_sbSQL, List<Object> r_sqlArgs) throws Exception
		{
			// 生成查询SQL
			execArg._sbSqlText = r_sbSQL;
			execArg._listSqlArgs = r_sqlArgs;
			
			for(Object oxmlNode : _listChildren)
			{
				if(oxmlNode instanceof String)
				{
					execArg._sbSqlText.append(oxmlNode);
				}
				else if(oxmlNode instanceof SQLXMLNode)
				{
					SMTJavaAIError error = ((SQLXMLNode)oxmlNode).execute(execArg);
					if(error != null)
						return error;
				}
			}			
			
			return null;
		}
	}
	
	///////////////////////////////////////////////////////////////////////////////////////
	private static class SQLXMLNodePARAM extends SQLXMLNode
	{
		private String	_paramKey;
		private String	_defValue;
		
		public SQLXMLNodePARAM(Element xmlRoot) throws Exception
		{
			_paramKey = SMTStatic.getXmlAttr(xmlRoot, "key");
			_defValue = SMTStatic.getXmlAttr(xmlRoot, "default");
		}
		
		@Override
		public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
		{
			String value = execArg._jsonAST.safeGetStr(_paramKey, null);
			if(value == null)
				value = _defValue;
			
			execArg._sbQuestion.append(value);
			
			return null;
		}
	}
	
	///////////////////////////////////////////////////////////////////////////////////////
	private static class SQLXMLNodeMETRIC_NAME extends SQLXMLNode
	{
		public SQLXMLNodeMETRIC_NAME(Element xmlRoot)
		{
		}
		
		@Override
		public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
		{
			execArg._sbQuestion.append(execArg._metricsDef.getTitle());
			
			return null;
		}
	}	
	
	///////////////////////////////////////////////////////////////////////////////////////
	private static class SQLXMLNodeQuestion extends SQLXMLNode
	{
		private List<Object>	_listChildren = new ArrayList<>();
		
		public SQLXMLNodeQuestion(Element xmlRoot) throws Exception
		{
			for (Iterator<Node> iterInner = xmlRoot.nodeIterator(); iterInner.hasNext();)
			{			
				Node nodeInner = iterInner.next();
				if(nodeInner.getNodeType() == Node.TEXT_NODE)
				{
					String text = ((DefaultText)nodeInner).getText();
					int lastPos = _listChildren.size() - 1;
					if(lastPos >= 0 && _listChildren.get(lastPos) instanceof String)
					{
						_listChildren.set(lastPos, (String)_listChildren.get(lastPos) + text);
					}
					else
					{
						_listChildren.add(text);
					}
				}	
				else
				{
					_listChildren.add(createSQLXMLNode((Element)nodeInner));
				}
			}
		}

		@Override
		public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
		{
			execArg._sbQuestion.setLength(0);
			for(Object oxmlNode : _listChildren)
			{
				if(oxmlNode instanceof String)
				{
					execArg._sbQuestion.append(oxmlNode);
				}
				else if(oxmlNode instanceof SQLXMLNode)
				{
					SMTJavaAIError error = ((SQLXMLNode)oxmlNode).execute(execArg);
					if(error != null)
						return error;
				}
			}
			if(execArg._sbQuestion.length() > 0)
				execArg._listQuestion.add(
					execArg._sbQuestion.toString()
					.replace("\r", "")
					.replace("\n", "")
					.replace("\t", "")
					.replace(" ", "")
					);
			
			return null;
		}
	}

	
	
	///////////////////////////////////////////////////////////////////////////////////////
	private static class SQLXMLNodeSAMPLE_QUESTIONS extends SQLXMLNode
	{
		private List<Object>	_listChildren = new ArrayList<>();
		
		public SQLXMLNodeSAMPLE_QUESTIONS(Element xmlRoot) throws Exception
		{
			for (Iterator<Node> iterInner = xmlRoot.nodeIterator(); iterInner.hasNext();)
			{			
				Node nodeInner = iterInner.next();
				if(nodeInner.getNodeType() != Node.TEXT_NODE)
				{
					_listChildren.add(createSQLXMLNode((Element)nodeInner));
				}
			}
		}

		@Override
		public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
		{
			for(Object oxmlNode : _listChildren)
			{
				if(oxmlNode instanceof String)
				{
					execArg._sbQuestion.append(oxmlNode);
				}
				else if(oxmlNode instanceof SQLXMLNode)
				{
					SMTJavaAIError error = ((SQLXMLNode)oxmlNode).execute(execArg);
					if(error != null)
						return error;
				}
			}

			
			return null;
		}
	}

	
	//////////////////////////////////////////////////////////////////////////////
	private SQLXMLNodeSAMPLE_QUESTIONS		_SQLXMLNode;
	
	private static SQLXMLNode createSQLXMLNode(Element xmlRoot) throws Exception
	{
		String name = xmlRoot.getName().toUpperCase();
		if("QUESTION".equals(name))
			return new SQLXMLNodeQuestion(xmlRoot);
		else if("PARAM".equals(name))
			return new SQLXMLNodePARAM(xmlRoot);
		else if("METRIC_NAME".equals(name))
			return new SQLXMLNodeMETRIC_NAME(xmlRoot);		
		else
			throw new Exception("unknow SQLXML : " + name);
	}

	public SMTMetricsSampleXml(Element rootElement) throws Exception
	{
		_SQLXMLNode = new SQLXMLNodeSAMPLE_QUESTIONS(rootElement);
	}
	
	public void createSampleQuestion(SMTMetricsDefXml metricsDef, ASTDBMap dbMap, Json jsonAST, SMTAIServerRequest tranReq, List<String> r_list) throws Exception
	{
		SQLXMLExecArg execArg = new SQLXMLExecArg(metricsDef, dbMap, jsonAST, tranReq, r_list);
		
		_SQLXMLNode.execute(execArg);
	}
}