package com.smtaiserver.smtaiserver.control;

import java.util.*;
import java.util.regex.Pattern;

import com.smtservlet.util.Json;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;
import org.dom4j.Document;
import org.springframework.web.servlet.ModelAndView;

import com.smtaiserver.smtaiserver.core.SMTAIServerApp;
import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
import com.smtaiserver.smtaiserver.core.SMTCheckChatStreamView;
import com.smtaiserver.smtaiserver.database.SMTDatabase;
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecords;
import com.smtaiserver.smtaiserver.javaai.qwen.agent.SMTQwenAgent;
import com.smtservlet.core.SMTRequest;
import com.smtservlet.util.SMTJsonWriter;
import com.smtservlet.util.SMTStatic;

public class SMTWorkflowManagerControl
{

	private static String[] _flowFuncNames = new String[] {
		"assessment",		"评估"
	};
	
	public ModelAndView publishWorkflowAgent(SMTAIServerRequest tranReq) throws Exception 
	{
		String id = tranReq.convParamToString("agent_id", true);
		String publish = tranReq.convParamToString("publish", true);
		
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{
    		DBRecords recs = db.querySQL("SELECT * FROM ai_agent_workflow WHERE agent_id=?", new Object[] {
        		id
    		});
    		if(recs.getRowCount() == 0)
    			return tranReq.returnJsonState(false, "未发现工作流id", null);
    		
    		String errmsg = null;
    		
    		// 检查工作流是否合法，如果不合法直接设置为"N"
    		try
    		{
				DBRecord rec = recs.getRecord(0);
				SMTQwenAgent agent = (SMTQwenAgent) Class.forName(rec.getString("clz_name")).newInstance();
				agent.initInstance(rec);
    		}
    		catch(Exception ex)
    		{
    			errmsg = "工作流配置错误";
    			publish = "N";
    		}
    		
    		db.executeSQL("UPDATE ai_agent_workflow SET agent_order=? WHERE agent_id=?", new Object[] {
        		"Y".equals(publish) ? 1 : 0,
        		id
    		});
    		
    		String curPublish = (recs.getRecord(0).getInteger("agent_order") > 0) ? "Y" : "N";
    		if(!publish.equals(curPublish))
    			SMTAIServerApp.getApp().clearQwenAgentManager();
    		
    		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, errmsg, null);
    		jsonWr.addKeyValue("publish", publish);
    		return tranReq.returnJson(jsonWr);
    	}
    	finally
    	{
    		db.close();
    	}
	} 
	
	public ModelAndView getFlowJsonFuncNames(SMTAIServerRequest tranReq) throws Exception 
	{
		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);
		jsonWr.beginArray("funcs");
		{
			for(int i = 0; i < _flowFuncNames.length; i += 2)
			{
				jsonWr.beginMap(null);
				{
					jsonWr.addKeyValue("id", _flowFuncNames[i + 0]);
					jsonWr.addKeyValue("title", _flowFuncNames[i + 1]);
				}
				jsonWr.endMap();
			}
		}
		jsonWr.endArray();
		
		return tranReq.returnJson(jsonWr);
	}
	
	public ModelAndView checkWorkflowAgentValidate(SMTAIServerRequest tranReq) throws Exception 
	{
		String agentId = tranReq.convParamToString("agent_id", true);
		String question = tranReq.convParamToString("question", true);
		
		DBRecords recs;
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{
			recs = db.querySQL("SELECT * FROM ai_agent_workflow WHERE agent_id=?", new Object[] {
				agentId
			});
			if(recs.getRowCount() == 0)
				return tranReq.returnJsonState(false, "未发现记录", null);	
    	}
    	finally
    	{
    		db.close();
    	}
    	
		DBRecord rec = recs.getRecord(0);
		SMTQwenAgent agent = (SMTQwenAgent) Class.forName(rec.getString("clz_name")).newInstance();
		agent.initInstance(rec);
		Map<String, SMTQwenAgent> mapId2Agent = new HashMap<>();
		mapId2Agent.put(agent.getAgentId(), agent);
		
		return new ModelAndView(new SMTCheckChatStreamView(tranReq, question, mapId2Agent));

	}
	
	public ModelAndView getFlowJsonAgentNames(SMTRequest tranReq) throws Exception 
	{
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{	
    		DBRecords recs = db.querySQL(
    			  " SELECT agent_id, agent_title FROM ("
    			+ " SELECT agent_id, agent_title FROM ai_agent_amis"
    			+ " UNION ALL"
    			+ " SELECT knowlg_id, knowlg_title FROM ai_agent_knowlg"
    			+ " UNION ALL"
    			+ " SELECT agent_id, agent_title FROM ai_agent_metrics"
    			+ " UNION ALL"
    			+ " SELECT agent_id, agent_title FROM ai_agent_workflow"
    			+ ") T WHERE agent_title IS NOT NULL ORDER BY agent_title"
    			, null);
    		
    		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);
    		jsonWr.beginArray("agents");
    		for(DBRecord rec : recs.getRecords())
    		{
    			jsonWr.beginMap(null);
    			{
    				jsonWr.addKeyValue("id", rec.getString("agent_id"));
    				jsonWr.addKeyValue("title", rec.getString("agent_title"));
    			}
    			jsonWr.endMap();
    		}
    		jsonWr.endArray();
    		
    		
   			return tranReq.returnJson(jsonWr);
    	}
    	finally
    	{
    		db.close();
    	}		

	}
	
	public ModelAndView updateWorkflowJsonFlow(SMTRequest tranReq) throws Exception 
	{
		String agentId = tranReq.convParamToString("agent_id", true);
		String sJson = tranReq.convParamToString("json_flow", true);

    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{
    		DBRecords recs = db.querySQL("SELECT * FROM ai_agent_workflow WHERE agent_id=?", new Object[] {
    			agentId
        	});
        	if(recs.getRowCount() == 0)
        		return tranReq.returnJsonState(false, "未发现工作流id", null);

    		db.executeSQL("UPDATE ai_agent_workflow SET clz_arguments = ? WHERE agent_id=?", new Object[] {
    			sJson,
    			agentId
    			});
    		
    		if(recs.getRecord(0).getInteger("agent_order") > 0)
    			SMTAIServerApp.getApp().clearQwenAgentManager();
    		
   			return tranReq.returnJsonState(true, null, null);
    	}
    	finally
    	{
    		db.close();
    	}		
	}
	
	public ModelAndView getWorkflowJsonFlow(SMTRequest tranReq) throws Exception 
	{
		String agentId = tranReq.convParamToString("agent_id", true);

    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{	
    		DBRecords recs = db.querySQL("SELECT clz_arguments FROM ai_agent_workflow WHERE agent_id=?", new Object[] {
    			agentId
    			});
    		
    		if(recs.getRowCount() == 0)
    			throw new Exception("can't find agent");
    		
    		String sJson = recs.getRecord(0).getString("clz_arguments");
    		if(SMTStatic.isNullOrEmpty(sJson))
    			return tranReq.returnJsonState(true, null, null);
    		
    		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);
    		jsonWr.addKeyRaw("json_flow", sJson);
    		
    		return tranReq.returnJson(jsonWr);
    	}
    	finally
    	{
    		db.close();
    	}		
	}

	
	public ModelAndView updateWorkflowAgent(SMTRequest tranReq) throws Exception 
	{
		String agentId = tranReq.convParamToString("agent_id", true);
		String title = tranReq.convParamToString("title", true);
		String prompt = tranReq.convParamToString("prompt", true);

    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{	
    		db.executeSQL("UPDATE ai_agent_workflow SET agent_title=?, agent_xml=?, update_time=? WHERE agent_id=?", new Object[] {
    			title,
    			"<TITLE>" + SMTStatic.convHtmlBodyToString(prompt) + "</TITLE>",
    			new Date(),
    			agentId
    			});
   		
    		return tranReq.returnJsonState(true, null, null);
    	}
    	finally
    	{
    		db.close();
    	}		
	}
	
	public ModelAndView deleteWorkflowAgent(SMTRequest tranReq) throws Exception 
    {
		String agentId = tranReq.convParamToString("agent_id", true);
		
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{	
    		db.executeSQL("DELETE FROM ai_agent_workflow WHERE agent_id=?", new Object[] {agentId});
   		
    		return tranReq.returnJsonState(true, null, null);
    	}
    	finally
    	{
    		db.close();
    	}
    }
	
	public ModelAndView addWorkflowAgent(SMTRequest tranReq) throws Exception 
    {
		String groupId = tranReq.convParamToString("group_id", true);
		String title = tranReq.convParamToString("title", true);
		String prompt = tranReq.convParamToString("prompt", true);
		String innerCall = tranReq.convParamToString("inner_call", true);
		String agentId = "a_" + SMTStatic.newUUID();
		Date time = new Date();
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{
    		db.executeSQL(
    			  " INSERT INTO ai_agent_workflow"
    			+ " (agent_id,agent_group,agent_type,agent_title,agent_xml,clz_name,clz_arguments,agent_order,is_debug,create_user,create_time,update_time,agent_note,inner_call)"
    			+ " VALUES"
    			+ " (?,?,?,?,?,?,?,?,?,?,?,?,?,?)"
    			, new Object[] {
    				agentId,
    				groupId,
    				"query",
    				title,
    				"<TITLE>" + SMTStatic.convHtmlBodyToString(prompt) + "</TITLE>",
    				"com.smtaiserver.smtaiserver.javaai.qwen.agent.SMTQwenAgentJsonWorkflow",
    				null,
    				0,
    				null,
    				tranReq.getLoginUserId(),
    				time,
    				time,
    				null,
    				innerCall
    			});	
    		
    		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);
    		jsonWr.addKeyValue("agent_id", agentId);
    		
    		return tranReq.returnJson(jsonWr);
    	}
    	finally
    	{
    		db.close();
    	}

    }
	
	
    public ModelAndView getWorkflowAgentList(SMTRequest tranReq) throws Exception 
    {
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{
    		DBRecords recs = db.querySQL(
    			  " SELECT A.agent_id, A.agent_title, A.agent_xml, A.clz_arguments, A.create_time, A.update_time, A.agent_note, A.agent_order, A.agent_group, A.inner_call, B.user_name"
    			+ " FROM (SELECT * FROM ai_agent_workflow WHERE clz_name='com.smtaiserver.smtaiserver.javaai.qwen.agent.SMTQwenAgentJsonWorkflow') "
    			+ " A LEFT JOIN sys_user_info B ON A.create_user=B.user_id", null);
			
    		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);
    		
    		jsonWr.beginArray("values");
    		for(DBRecord rec : recs.getRecords())
    		{
    			jsonWr.beginMap(null);
    			{
    				jsonWr.addKeyValue("id", rec.getString("agent_id"));
    				jsonWr.addKeyValue("title", rec.getString("agent_title"));
    				jsonWr.addKeyValue("create_user", rec.getString("user_name"));
    				jsonWr.addKeyValue("create_time", rec.getString("create_time"));
    				jsonWr.addKeyValue("note", rec.getString("agent_note"));
    				jsonWr.addKeyValue("agent_group", rec.getString("agent_group"));
    				jsonWr.addKeyValue("inner_call", rec.getString("inner_call"));
    				jsonWr.addKeyValue("published", (rec.getInteger("agent_order") > 0) ? "Y" : "N");
    				
					jsonWr.beginMap("supervisor");
		    		{
						Document doc = SMTStatic.convStrToXmlDoc("<ROOT>" + rec.getString("agent_xml") + "</ROOT>");
						String prompt = SMTStatic.trimStrLines(doc.selectSingleNode("ROOT/TITLE").getText());
						jsonWr.addKeyValue("prompt", prompt);
					}
					jsonWr.endMap();
    			}
	    		jsonWr.endMap();
    		}
    		jsonWr.endArray();
    		
            return tranReq.returnJson(jsonWr);
    	}
    	finally
    	{
    		db.close();
    	}
    }

    public ModelAndView getLLMInfoList(SMTRequest tranReq) throws Exception 
    {
    	SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();
    	try
    	{
    		DBRecords recsFactory = db.querySQL("SELECT * FROM ai_llm_factory", null);
    		DBRecords recsConnection = db.querySQL("SELECT * FROM ai_llm_connect", null);
			
    		SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);
    		
    		jsonWr.beginArray("factorys");
    		for(DBRecord rec : recsFactory.getRecords())
    		{
    			jsonWr.beginMap(null);
    			{
    				jsonWr.addKeyValue("factory_id", rec.getString("factory_id"));
    				jsonWr.addKeyValue("factory_title", rec.getString("factory_title"));
    			}
    			jsonWr.endMap();
    		}
    		jsonWr.endArray();
    		
    		jsonWr.beginArray("connections");
    		for(DBRecord rec : recsConnection.getRecords())
    		{
    			jsonWr.beginMap(null);
    			{
    				jsonWr.addKeyValue("connect_id", rec.getString("connect_id"));
    				jsonWr.addKeyValue("factory_id", rec.getString("factory_id"));
    				jsonWr.addKeyValue("connect_title", rec.getString("connect_title"));
    			}
    			jsonWr.endMap();
    		}
    		jsonWr.endArray();

    		
            return tranReq.returnJson(jsonWr);
    	}
    	finally
    	{
    		db.close();
    	}
    }

	/**
	 * 查询n8n工作流列表
	 * @param tranReq
	 * @return
	 * @throws Exception
	 */
	public ModelAndView getN8NNameList(SMTRequest tranReq) throws Exception {
		String active = tranReq.convParamToString("active", true);
		String tags = tranReq.convParamToString("tags", false);
		String name = tranReq.convParamToString("name", false);
		String projectId = tranReq.convParamToString("projectId", false);
		String excludePinnedData = tranReq.convParamToString("excludePinnedData", false);
		int limit = 250; // 上限250条

		// 构建请求URL
		String requestUrl = SMTAIServerApp.buildUrl(active, tags, name, projectId, excludePinnedData, limit);

		// 创建OkHttpClient对象
		OkHttpClient client = new OkHttpClient();

		String n8nApiKey = (String) SMTAIServerApp.getApp().getGlobalConfig("n8n_api_key", null);
		// 创建请求头
		Request request = new Request.Builder()
				.url(requestUrl)
				.addHeader("X-N8N-API-KEY", n8nApiKey)
				.addHeader("accept", "application/json")
				.build();
		Response response = client.newCall(request).execute();
		String res = response.body().string();
		Json.Reader reader = new Json.Reader();
		Json read = (Json) reader.read(res, null);
		SMTJsonWriter jsonWriter = new SMTJsonWriter(false);

		// 构建最终的返回数据结构
		List<Json> resultList = new ArrayList<>();

		try {
			if (response.body() != null) {
				List<Json> data = read.safeGetJsonList("data");
				for (Json date : data) {
					String flOWName = date.safeGetStr("name", null);
					String id = date.safeGetStr("id", null);

					// 创建一个新的 JSON 对象
					SMTJsonWriter flowObj = new SMTJsonWriter(false);
					flowObj.addKeyValue("id", id);
					flowObj.addKeyValue("flow_name", flOWName);

					List<Json> nodeList = date.safeGetJsonList("nodes");
					for (Json node : nodeList) {
						String type = node.safeGetStr("type", null);
						if (type.equals("n8n-nodes-base.webhook")) {
							Json parameters = node.safeGetJson("parameters");
							String path = parameters.safeGetStr("path", null);
							Pattern pattern = Pattern.compile("/:?([^/]+)");
							List<String> pathRes = SMTAIServerApp.extractParts(path, pattern);
							flowObj.addKeyValue("parameters", pathRes);
						}
					}
					// 将创建的 flowObj 添加到 resultList 中
					resultList.add(flowObj.getRootJson());
				}

				// 将 resultList 设置为 "data" 键的值
				jsonWriter.addKeyValue("data", resultList);
			}
			jsonWriter.addKeyValue("json_ok", true);
			return tranReq.returnJson(jsonWriter);
		} catch (Exception e) {
			throw new Exception(e);
		}
	}
}