CMVR-IOT/cmvr-iot-test/src/main/java/com/cmvr/test/service/FlowiseActionService.java

package com.cmvr.test.service;

import cn.hutool.http.HttpRequest;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.cmvr.common.exception.GlobalException;
import com.cmvr.common.utils.http.CallAPIUtil;
import com.cmvr.common.utils.uuid.IdUtils;
import com.cmvr.edge.client.service.EdgeBioHeadService;
import com.cmvr.test.enums.FlowiseActionEnum;
import com.cmvr.test.flow.context.TaskContext;
import com.cmvr.test.flow.context.TaskContextManager;
import com.cmvr.test.flow.control.FlowControlService;
import com.cmvr.test.model.vo.FlowiseActionRequestVO;
import com.cmvr.test.model.vo.FlowiseChatRequestVO;
import com.cmvr.test.model.vo.FlowiseStartRequestVO;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.Map;

@Slf4j
@Service
@RequiredArgsConstructor
public class FlowiseActionService {

    private final EdgeBioHeadService edgeBioHeadService;
    private final TaskContextManager taskContextManager;
    private final FlowControlService flowControlService;

    @Value("${flowise.tts}")
    private String flowiseTts;

    @Value("${flowise.query}")
    private String flowiseQuery;

    @Value("${flowise.start}")
    private String flowiseStart;

    @Value("${flowise.abort}")
    private String flowiseAbort;

    @Value("${flowise.api-key}")
    private String flowiseApiKey;

    @Resource(name = "threadPoolTaskExecutor")
    private final ThreadPoolTaskExecutor executor;


    private final Object lock = new Object();
    private String chatId = null;

    public String start(FlowiseStartRequestVO request) {
        log.info("flowise start: {}", System.currentTimeMillis());
        abort();
        synchronized (lock) {
            if (chatId != null) {
                throw new GlobalException("Flow is already running: " + chatId);
            }

            chatId = IdUtils.fastUUID();

            JSONObject body = new JSONObject();
            body.put("question", request.getQuestion());
            body.put("chatId", chatId);
            body.put("streaming", true);

            executor.submit(() -> {
                try {
                    HttpRequest.post(flowiseStart)
                            .header("Content-Type", "application/json")
                            .header("Authorization", "Bearer " + flowiseApiKey)
                            .body(body.toJSONString())
                            .timeout(0)
                            .execute()
                            .body();
                } catch (Exception e) {
                    throw new GlobalException("flowise 流程执行失败: ", e.getMessage());
                }
            });
            return chatId;
        }
    }

    public String abort() {
        CallAPIUtil.doPostJson(flowiseTts + "stop", null, null);
        Map<String, TaskContext> instContextMap = taskContextManager.getInstContextMap();
        instContextMap.forEach((key, value) -> flowControlService.stop(key));
        synchronized (lock) {

            if (chatId == null) {
                return "No flow running.";
            }

            String id = chatId;

            String result = HttpRequest.put(flowiseAbort + id)
                    .header("Authorization", "Bearer " + flowiseApiKey)
                    .execute()
                    .body();

            chatId = null;

            return result;
        }
    }

    public String command(FlowiseActionRequestVO request) {
        log.info("command start: {}", System.currentTimeMillis());
        String action = request.getAction();
        FlowiseActionEnum actionEnum = FlowiseActionEnum.fromAction(action);
        String terminalId = "4ed1246c465b97975f96c9ef8371a3bd";
        String bioId = "bio_head";
        switch (actionEnum) {
            // ==== 表情 ====
            case SMILE:
                return edgeBioHeadService.specialExpression(terminalId, bioId, 1);
            case SURPRISED:
                return edgeBioHeadService.specialExpression(terminalId, bioId, 2);
            case ANGRY:
                return edgeBioHeadService.specialExpression(terminalId, bioId, 3);
            case SAD:
                return edgeBioHeadService.specialExpression(terminalId, bioId, 4);
            case SLEEP:
                return edgeBioHeadService.specialExpression(terminalId, bioId, 5);
            case YAWN:
                return edgeBioHeadService.specialExpression(terminalId, bioId, 6);

            // ==== 动作 ====

            default:
                throw new UnsupportedOperationException("未实现的 Flowise Action: " + action);
        }
    }

    public String chat(FlowiseChatRequestVO requestVO) {
        log.info("chat start: {}", System.currentTimeMillis());
        Map<String, String> body = new HashMap<>();
        body.put("text", requestVO.getText());
        body.put("voice", "x4_yezi");

        CallAPIUtil.doPostJson(flowiseTts + "play", null, body);
        return "ok";
    }

    public JSONObject query(String chatId) {
        String body = HttpRequest.get(flowiseQuery + chatId)
                .header("Authorization", "Bearer " + flowiseApiKey)
                .execute()
                .body();
        JSONObject root = JSON.parseObject(body);
        String execStr = root.getString("executionData");
        JSONArray execJson = JSON.parseArray(execStr);
        root.put("executionData", execJson);

        return root;
    }
}