package com.yb.controller;


import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.yb.config.R;
import com.yb.config.XzConfig;
import com.yb.service.AgentService;
import com.yb.service.XzService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;

@Tag(name = "Agent意图")
@RestController
@RequestMapping("/api/agent")
@Api(tags = "Agent意图")
@Component
public class agentController {

    @Autowired
    private XzConfig xzConfig;
    @Autowired
    private XzService xzService;
    @Autowired
    private AgentService agentService;

    @ApiOperation("意图识别")
    @PostMapping("/intention")
    @Operation(summary = "意图识别")
    public R getIntention(@RequestParam Map<String, String> allParams) throws Exception {
        System.out.println("rec getMessage:" + allParams);
        String msg = allParams.get("message");
        String sessionId = allParams.get("session_id");
        String libId = allParams.get("lib_id");
        String url = xzConfig.agentUrl;
        String intent = xzConfig.agentIntent;
        String rag = xzConfig.agentRag;
        String chart = xzConfig.agentChat;
//       判断msg是否为空
        if (StringUtils.isEmpty(msg)) {
            String info = agentService.getAgentIntention(url, intent, "''");
            JSONObject obj = JSONObject.parseObject(info);
            String cont = obj.getString("content");
            System.out.println("rec isEmpty:" + cont);
            return R.error(cont);
//
        }
//       判断是否调用地图
        boolean isMap = msg.matches("(?i).*@map.*");
        HashMap<String, Object> hashMap = new HashMap<String, Object>();
        if (isMap) {
            hashMap.put("type", "Map");
            HashMap<String, Object> mapObj = new HashMap<String, Object>();
            String funcInfo = agentService.getAgentIntention(url, intent, msg);
            JSONObject jsonObject = JSON.parseObject(funcInfo);
            String orderString = jsonObject.getString("order");
            if(orderString == null){
                hashMap.put("type", "text");
                JSONObject errInfo = JSON.parseObject(funcInfo);
                String erroContent = errInfo.getString("content");
                hashMap.put("content", erroContent);
                return R.error(erroContent);

            }
            JSONObject orderObject = JSON.parseObject(orderString);

            String func = orderObject.getString("name");
            JSONObject argsObject = JSON.parseObject(orderObject.getString("args"));
            String place = argsObject.getString("place");
            String type = argsObject.getString("attris");
            String dis = argsObject.getString("redius");
            int radius = 5000;
            if (dis != null) {
                dis = dis.replace("m", "");
                radius = Integer.parseInt(dis);
            }
            String queryFiled = "enti_name";
            String typeFiled = "sjnr";
            String func1 = "poiMap";
            String func2 = "aroundPoi";
            String func3 = "queryMeta";
            if (func.equals(func1)) {
                String poiMap = agentService.getAgentPoiMap(place, queryFiled);
                System.out.println("poiMap: " + poiMap);
                JSONObject poiObj = JSONObject.parseObject(poiMap);
                System.out.println("poiMap: " + poiObj);
                mapObj.put("func", func1);
                mapObj.put("data", poiObj);
            } else if (func.equals(func2)) {
                mapObj.put("func", func2);
                String aroundPoi = agentService.getAgentAroundPoi(place, type, radius, queryFiled, typeFiled);
                JSONArray dataArray = JSONArray.parseArray(aroundPoi);
                System.out.println("aroundPoi: " + dataArray);
                mapObj.put("data", dataArray);
            } else if (func.equals(func3)) {
                mapObj.put("func", func3);
                String filedString= xzService.getQueryMeta(xzConfig.layerId, xzConfig.dbid,typeFiled,xzConfig.queryEntity);
                JSONObject metaObject = JSON.parseObject(filedString);
                JSONArray dataArray = metaObject.getJSONArray("data");
                String markDown = "| 数据类型 | 数据量 |\\n|------|------|\\n";
                for (int i = 0; i < dataArray.size(); i++) {
                    JSONObject dataObj = dataArray.getJSONObject(i);
                    System.out.println("rec queryMeta:" + dataObj);
                    String[] filedValue = new String[2];

                    for (String key : dataObj.keySet()) {
                        if (key == typeFiled) {
                            filedValue[0] = dataObj.getString(key);
                        } else {
                            filedValue[1] = dataObj.getString(key);
                        }
                    }
                    markDown += "| " + filedValue[0] + " | " + filedValue[1] + " |\n";
                }
                mapObj.put("data", markDown);
            }

            hashMap.put("content", mapObj);
            return R.ok(hashMap);

        } else {
            String ragInfo = agentService.getAgentIntention(url, chart, msg);
            JSONObject jsonObject = JSON.parseObject(ragInfo);
            String answer = jsonObject.getString("content");
            System.out.println("Answer: " + answer);
            hashMap.put("content", answer);
            hashMap.put("type", "Rag");
            return R.ok(hashMap);
        }

//        return R.error("未查询到相关信息");

    }


}