package com.smtaiserver.smtaiserver.javaai.jsonflow.node;

import com.smtaiserver.smtaiserver.core.SMTAIServerApp;
import com.smtaiserver.smtaiserver.database.SMTDatabase;
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBQueryNotify;
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
import com.smtaiserver.smtaiserver.gismap.SMTGisMapLayerDef;
import com.smtaiserver.smtaiserver.gismap.theme.SMTMapThemeDef;
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowExecArg;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowManager;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowNodeOnlyOutput;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowScriptJet;
import com.smtservlet.util.Json;
import com.smtservlet.util.SMTHttpClient;
import com.smtservlet.util.SMTJsonWriter;
import com.smtservlet.util.SMTStatic;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map.Entry;
import org.mozilla.javascript.Context;
import org.mozilla.javascript.Function;
import org.mozilla.javascript.NativeArray;
import org.mozilla.javascript.NativeObject;

public class SMTJsonFlowNodeScript extends SMTJsonFlowNodeOnlyOutput 
{
	public static class SMTJsonFlowNodeScriptArg
	{
		public SMTJsonFlowExecArg		_execArg;
		
		public SMTJsonFlowNodeScriptArg(SMTJsonFlowExecArg execArg)
		{
			_execArg = execArg;
		}

		public String getGlobalConfig(String key, String defaultValue) throws Exception {
		  String globalConfig = (String) SMTAIServerApp.getApp().getGlobalConfig(key, defaultValue);
		  return globalConfig;
		}
		public String getNowDate() throws Exception {
			return SMTStatic.toString(new Date());
		}
		public String getUUID() throws Exception {
			return  SMTStatic.newUUID();
		}
		public  void webSocketBroadcast(NativeObject nativeObject) throws Exception {
			SMTJsonWriter jsonWr = new SMTJsonWriter(false);
			SMTAIServerApp.convJSToJsonWriter(nativeObject, jsonWr);
			SMTAIServerApp.getApp().webSocketApp(jsonWr.getFullJson());
		}
		
		public Object llmAnswerToJson(String sJson)
		{
			Object object = SMTAIServerApp.convJsonToJS(SMTStatic.convLLMAnswerToJson(sJson, false));
			return object;
		}

		public Object getArg(String key)
		{
			Json jsonValue = _execArg._jsonArgs.safeGetJson(key);
			if(jsonValue == null)
				return null;
			
			if(jsonValue.isPrimitive())
				return jsonValue.getValue();
			
			return jsonValue;
		}
		
		public void setArg(String key, String value)
		{
			_execArg._jsonArgs.set(key, value);
		}
		
		public String getGlobalConfig(String key) throws Exception
		{
			Object value = SMTAIServerApp.getApp().getGlobalConfig(key, null);
			if(value == null)
				return null;
			
			return SMTStatic.toString(value);
		}
		
		public Object queryWeb(NativeObject nvConfig) throws Exception
		{
			String url = (String) SMTAIServerApp.getJSValue(nvConfig, "url");
			String method = (String) SMTAIServerApp.getJSValue(nvConfig, "method", "GET");
			String resultType = (String) SMTAIServerApp.getJSValue(nvConfig, "result_type", "String");
			
			String result;
			if("GET".equalsIgnoreCase(method))
			{
				List<String> listArgs = new ArrayList<>();
				NativeObject nvArgs = (NativeObject) SMTAIServerApp.getJSValue(nvConfig, "args", null);
				
				if(nvArgs != null)
				{
					for(Entry<Object, Object> entry : nvArgs.entrySet())
					{
						listArgs.add(SMTStatic.toString(entry.getKey()));
						listArgs.add(SMTStatic.toString(SMTAIServerApp.unwrapObject(entry.getValue())));
					}
				}
				
				SMTHttpClient web = new SMTHttpClient();
				result = web.getHttpString(url, listArgs.toArray(new String[listArgs.size()]), null);
			}
			else
			{
				throw new Exception("unknow web method : " + method);
			}
			
			if("String".equalsIgnoreCase(resultType))
			{
				return result;
			}
			else if("Json".equalsIgnoreCase(resultType))
			{
				return SMTAIServerApp.convJsonToJS(Json.read(result));
			}
			else
			{
				throw new Exception("unknow result type : " + resultType);
			}
		}
		
		public NativeArray getMapLayerList() throws Exception
		{
			List<SMTGisMapLayerDef> list = new ArrayList<>();
			for(SMTGisMapLayerDef layerDef : SMTAIServerApp.getApp().getMapLayerDefMap().values())
			{
				list.add(layerDef);
			}
			return new NativeArray(list.toArray(new SMTGisMapLayerDef[list.size()]));
		}
		
		public NativeArray getMapThemeList() throws Exception
		{
			List<SMTMapThemeDef> list = new ArrayList<>();
			for(SMTMapThemeDef themeDef : SMTAIServerApp.getApp().getMapThemeDefMap().values())
			{
				list.add(themeDef);
			}
			return new NativeArray(list.toArray(new SMTMapThemeDef[list.size()]));
		}
		
		public void sendTableToClent(NativeObject nvArgs) throws Exception
		{
			String title = (String) SMTAIServerApp.getJSValue(nvArgs, "title");
			NativeArray arrCols = (NativeArray)SMTAIServerApp.getJSValue(nvArgs, "columns");
			NativeArray arrRecs = (NativeArray)SMTAIServerApp.getJSValue(nvArgs, "recs");
			
			SMTJsonWriter jsonWrResult = _execArg._tranReq.getResultJsonWr();
			jsonWrResult.addKeyValue("answer_type", "summary");
			jsonWrResult.beginArray("summary");
			{
				jsonWrResult.beginMap(null);
				{
					// 加入基本信息
					jsonWrResult.addKeyValue("type", "recordset");
					jsonWrResult.addKeyValue("agent_key", "#0/" + SMTStatic.newUUID());
					jsonWrResult.addKeyValue("chart", "table");
					jsonWrResult.addKeyValue("title", title);
					
					// 加入字段列
					List<String> listColName = new ArrayList<>();
					jsonWrResult.beginArray("cols");
					for(int i = 0; i < arrCols.size(); i ++)
					{
						NativeObject nvCol = (NativeObject)SMTAIServerApp.unwrapObject(arrCols.get(i));
						jsonWrResult.beginMap(null);
						{
							jsonWrResult.addKeyValue("title", (String)SMTAIServerApp.getJSValue(nvCol, "title"));
							jsonWrResult.addKeyValue("name", (String)SMTAIServerApp.getJSValue(nvCol, "name"));
							jsonWrResult.addKeyValue("type", "text");
							jsonWrResult.addKeyValue("group", false);
							listColName.add((String)SMTAIServerApp.getJSValue(nvCol, "name"));
						}
						jsonWrResult.endMap();
					}
					jsonWrResult.endArray();
					
					// 加入值列
					jsonWrResult.beginArray("values");
					for(int i = 0; i < arrRecs.size(); i ++)
					{
						NativeObject nvRec = (NativeObject)SMTAIServerApp.unwrapObject(arrRecs.get(i));
						jsonWrResult.beginArray(null);
						{
							for(int j = 0; j < listColName.size(); j ++)
							{
								String colName = listColName.get(j);
								jsonWrResult.addKeyValue(null, SMTAIServerApp.getJSValue(nvRec, colName, null));
							}
						}
						jsonWrResult.endArray();
					}
					jsonWrResult.endArray();
				}
				jsonWrResult.endMap();
			}
			jsonWrResult.endArray();
			
			_execArg._tranReq.sendChunkedResultBlock();			
		}
		
		public void sendChartToClient(NativeObject nvArgs) throws Exception
		{
			String title = (String) SMTAIServerApp.getJSValue(nvArgs, "title");
			String colTime = (String) SMTAIServerApp.getJSValue(nvArgs, "col_time");
			String colName = (String) SMTAIServerApp.getJSValue(nvArgs, "col_name");
			String colValue = (String) SMTAIServerApp.getJSValue(nvArgs, "col_value");
			NativeArray arrRecs = (NativeArray) SMTAIServerApp.getJSValue(nvArgs, "recs");
			
			
			SMTJsonWriter jsonWrResult = _execArg._tranReq.getResultJsonWr();
			jsonWrResult.addKeyValue("answer_type", "summary");
			jsonWrResult.beginArray("summary");
			{
				jsonWrResult.beginMap(null);
				{
					// 加入基本信息
					jsonWrResult.addKeyValue("type", "recordset");
					jsonWrResult.addKeyValue("agent_key", "#0/" + SMTStatic.newUUID());
					jsonWrResult.addKeyValue("chart", "muli_line");
					jsonWrResult.addKeyValue("title", title);
					
					// 加入字段名
					jsonWrResult.beginArray("cols");
					{
						jsonWrResult.beginMap(null);
						{
							jsonWrResult.addKeyValue("title", "时间");
							jsonWrResult.addKeyValue("type", "time");
						}
						jsonWrResult.endMap();
						
						jsonWrResult.beginMap(null);
						{
							jsonWrResult.addKeyValue("title", "名称");
							jsonWrResult.addKeyValue("type", "name");
						}
						jsonWrResult.endMap();
						
						jsonWrResult.beginMap(null);
						{
							jsonWrResult.addKeyValue("title", "值");
							jsonWrResult.addKeyValue("type", "value");
						}
						jsonWrResult.endMap();
					}
					jsonWrResult.endArray();
					
					// 加入值
					jsonWrResult.beginArray("values");
					for(int i = 0; i < arrRecs.size(); i ++)
					{
						NativeObject nvRec = (NativeObject)SMTAIServerApp.unwrapObject(arrRecs.get(i));
						jsonWrResult.beginArray(null);
						{
							jsonWrResult.addKeyValue(null, SMTAIServerApp.getJSValue(nvRec, colTime, null));
							jsonWrResult.addKeyValue(null, SMTAIServerApp.getJSValue(nvRec, colName, null));
							jsonWrResult.addKeyValue(null, SMTAIServerApp.getJSValue(nvRec, colValue, null));
						}
						jsonWrResult.endArray();
					}
					jsonWrResult.endArray();
				}
				jsonWrResult.endMap();
			}
			jsonWrResult.endArray();
			
			_execArg._tranReq.sendChunkedResultBlock();
		}
		
		public void sendKnowledgeToClient(NativeObject nvArgs) throws Exception
		{
			String result = (String) SMTAIServerApp.getJSValue(nvArgs, "result");
			SMTJsonWriter jsonWrResult = _execArg._tranReq.getResultJsonWr();
			jsonWrResult.addKeyValue("answer_type", "knowledge");
			jsonWrResult.beginArray("knowledge");
			{
				jsonWrResult.beginMap(null);
				{
					jsonWrResult.addKeyValue("answer", result);
				}
				jsonWrResult.endMap();
			}
			jsonWrResult.endArray();
			
			_execArg._tranReq.sendChunkedResultBlock();
		}
		
		public void sendResultToClient(NativeObject nvArgs) throws Exception
		{
			SMTAIServerApp.convJSToJsonWriter(nvArgs, _execArg._tranReq.getResultJsonWr());
			_execArg._tranReq.sendChunkedResultBlock();
		}
		
		public void sendErrorToClient(String errMsg)throws Exception
		{
			_execArg._tranReq.sendChunkedBlock("summary", Json.object(
				"json_ok", false, 
				"err_code", "MESSAGE",
				"json_msg", errMsg
				));
			_execArg._tranReq.sendChunkedBlock("error", errMsg);
		}
		
		public void sendJsonToClient(String type, NativeObject nvArgs)throws Exception
		{
			SMTJsonWriter jsonWr = new SMTJsonWriter(false);
			SMTAIServerApp.convJSToJsonWriter(nvArgs, jsonWr);
			_execArg._tranReq.sendChunkedBlock(type, jsonWr.getFullJson());
		}
		
		public NativeArray querySQL(String sql, NativeObject nvParams) throws Exception
		{
			SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
			return querySQL(db, sql, nvParams);
		}

		public NativeArray querySQL(String dsId, String sql, NativeObject nvParams) throws Exception
		{
			SMTDatabase db = SMTAIServerApp.getApp().getDataSource(dsId).allocDatabase();
			return querySQL(db, sql, nvParams);
		}
		
		public int executeSQL(String sql, NativeObject nvParams) throws Exception
		{
			SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
			return executeSQL(db, sql, nvParams);
		}
		
		public int executeSQL(String dsId, String sql, NativeObject nvParams) throws Exception
		{
			SMTDatabase db = SMTAIServerApp.getApp().getDataSource(dsId).allocDatabase();
			return executeSQL(db, sql, nvParams);
		}

		
		private int executeSQL(SMTDatabase db, String sql, NativeObject nvParams) throws Exception
		{
			List<Object> sqlParams = new ArrayList<>();
			try
			{
				// 解析出原始sql
				String rawSQL = SMTStatic.stringFormat(sql, new SMTStatic.StringNamedNotify()
				{
					@Override
					public Object getNamedValue(String name, Object[] args) throws Exception 
					{
						char type;
						String value = SMTStatic.toString(SMTAIServerApp.getJSValue(nvParams, name));
						if(value.length() == 1)
						{
							type = value.charAt(0);
							value = "";
						}
						else
						{
							type = value.charAt(0);
							value = value.substring(1);
							
						}
						
						Object ovalue;
						
						switch(type)
						{
						case 'N':
							ovalue = null;
							break;
							
						case 'I':
							ovalue = SMTStatic.toInt(value);
							break;
							
						case 'D':
							ovalue = SMTStatic.toDouble(value);
							break;
							
						case 'L':
							ovalue = SMTStatic.toLong(value);
							break;
							
						case 'T':
							ovalue = SMTStatic.toDate(value);
							break;
							
						default:
							ovalue = value;
							break;
						}
						
						sqlParams.add(ovalue);
						return "?";
					}
				});
				

				// 执行查询
				int ret = db.executeSQL(rawSQL, sqlParams.toArray(new Object[sqlParams.size()]));
				
				return ret;
			}
			finally
			{
				db.close();
			}
		}

		
		private NativeArray querySQL(SMTDatabase db, String sql, NativeObject nvParams) throws Exception
		{
			List<Object> sqlParams = new ArrayList<>();
			List<NativeObject> listRecs = new ArrayList<>();
			try
			{
				// 解析出原始sql
				String rawSQL = SMTStatic.stringFormat(sql, new SMTStatic.StringNamedNotify()
				{
					@Override
					public Object getNamedValue(String name, Object[] args) throws Exception 
					{
						char type;
						String value = SMTStatic.toString(SMTAIServerApp.getJSValue(nvParams, name));
						if(value.length() == 1)
						{
							type = value.charAt(0);
							value = "";
						}
						else
						{
							type = value.charAt(0);
							value = value.substring(1);
							
						}
						
						Object ovalue;
						
						switch(type)
						{
						case 'N':
							ovalue = null;
							break;
							
						case 'I':
							ovalue = SMTStatic.toInt(value);
							break;
							
						case 'D':
							ovalue = SMTStatic.toDouble(value);
							break;
							
						case 'L':
							ovalue = SMTStatic.toLong(value);
							break;
							
						case 'T':
							ovalue = SMTStatic.toDate(value);
							break;
							
						default:
							ovalue = value;
							break;
						}
						
						sqlParams.add(ovalue);
						return "?";
					}
				});
				

				// 执行查询
				db.querySQLNotify(rawSQL, sqlParams.toArray(new Object[sqlParams.size()]), new DBQueryNotify() {

					@Override
					public boolean onNextRecord(DBRecord rec) throws Exception 
					{
						NativeObject nvRec = new NativeObject();
						for(Entry<String, Integer> entry : rec.getFieldMap().entrySet())
						{
							String colName = entry.getKey();
							String value = rec.getString(entry.getValue());
							if(value == null)
								continue;
							SMTAIServerApp.putJSNotNullValue(nvRec, colName, value);
						}
						listRecs.add(nvRec);
						
						return true;
					}
					
				});
				
				return new NativeArray(listRecs.toArray(new NativeObject[listRecs.size()]));
			}
			finally
			{
				db.close();
			}
		}
	}
	
	//////////////////////////////////////////////////////////////////////////////////////////
	
	protected Function		_func;
	
	@Override
	public void initInstane(SMTJsonFlowManager manager, Json jsonNode) throws Exception
	{
		super.initInstane(manager, jsonNode);
		
		List<Json> jsonGroupParams = jsonNode.getJsonPath("data|group_params|", false).asJsonList();
		for(Json jsonParams : jsonGroupParams)
		{
			String type = jsonParams.getJsonPath("params|0|type", false).asString();
			if("code".equals(type))
			{
				SMTJsonFlowScriptJet scriptJet = this._manager.allocScriptJet();
				try
				{
					String code = "function(args){" + jsonParams.getJsonPath("params|0|value", false).asString() + "}";
					Context cx = scriptJet.entryContext();
					try
					{
						_func = scriptJet.compileFunction(cx, this.getTitle(), code);
					}
					catch(Exception ex)
					{
						throw new Exception("compile js code error : \n" + code, ex);
					}
					finally
					{
						Context.exit();
					}
				}
				finally
				{
					scriptJet.close();
				}
				
			}
		}
	}
	
	@Override
	public void afterInstance() throws Exception
	{
		super.afterInstance();
		
		if(_func == null)
			throw new Exception("script function code is not define");
	}
	
	@Override
	public SMTJavaAIError executeFlowNode(SMTJsonFlowExecArg execArg) throws Exception 
	{
		SMTJsonFlowScriptJet scriptJet = this._manager.allocScriptJet();
		try
		{
			Context cx = scriptJet.entryContext();
			try
			{
				SMTJsonFlowNodeScriptArg arg = new SMTJsonFlowNodeScriptArg(execArg);
				
				scriptJet.callFunction(cx, _func, new Object[] {arg});
			}
			finally
			{
				Context.exit();
			}		
		}
		finally
		{
			scriptJet.close();
		}
		return super.executeFlowNode(execArg);
	}
}