package com.agibot.aiem.extskill;

import com.agibot.aiem.sdk.AgentAuthCallback;
import com.agibot.aiem.sdk.AgentParam;
import com.agibot.aiem.sdk.AgentSdk;
import com.agibot.aiem.sdk.IdGenerator;
import com.agibot.aiem.sdk.extskill.ExtSkillInvokeCallback;
import com.agibot.aiem.sdk.extskill.ExtSkillQueryCallback;
import com.agibot.aiem.sdk.extskill.ExtSkillRequest;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class ExtSkillExample {

    public static void main(String[] args) throws InterruptedException {

        String url       = "wss://open.agibot.com/api/V1/open-portal/app/wss/agent-sdk";
        String appId     = "您通过灵心开放平台创建的应用获取的appId";
        String appKey    = "您通过灵心开放平台创建的应用获取的appKey";
        String appSecret = "您通过灵心开放平台创建的应用获取的appSecret";
        String agentId   = "您通过灵心平台创建的二开智能体获取的agentId";

        // 第一步：创建 SDK 实例
        AgentSdk agentSdk = AgentSdk.create(url, appId, appKey, appSecret);

        // 第二步：注册验签回调接口
        agentSdk.registerAuth(new AgentAuthCallback() {
            @Override
            public void onAuthState(String appId, int code, String msg) {
                System.out.println("onAuthState => appId: " + appId
                        + ", code: " + code + ", msg: " + msg);
            }
        });

        // 第三步：初始化
        agentSdk.initialize();

        Thread.sleep(2000);

        // 第四步：注册扩展技能请求实例
        String requestId = IdGenerator.generateExtSkillRequestId();
        ExtSkillRequest extSkillRequest = new ExtSkillRequest(agentSdk, agentId, requestId);
        agentSdk.registerExtSkill(extSkillRequest);

        // 4.1 查询可用扩展技能清单
        String queryEventId = extSkillRequest.query(new ExtSkillQueryCallback() {
            @Override
            public void onQueryResult(String agentId, String requestId, String eventId,
                                      int code, String msg, AgentParam result) {
                log.info("onQueryResult => agentId {}, requestId {}, eventId {}, code {}, msg {}, result {}",
                        agentId, requestId, eventId, code, msg, result);
            }
        }, 2000);
        log.info("query queryEventId {}", queryEventId);

        // 4.2 触发某个扩展技能执行
        String extSkillId = "test_ext_skill_id_001";           // 扩展技能 ID
        AgentParam input = AgentParam.create();                // 扩展技能入参
        input.setString("role", "male");
        input.setInteger("threshold", 0);

        String invokeEventId = extSkillRequest.invoke(extSkillId, input, new ExtSkillInvokeCallback() {
            @Override
            public void onInvokeAck(String agentId, String requestId, String eventId, int code, String msg) {
                // 第一段：网关已受理
                log.info("onInvokeAck => agentId {}, requestId {}, eventId {}, code {}, msg {}",
                        agentId, requestId, eventId, code, msg);
            }

            @Override
            public void onState(String agentId, String requestId, String eventId, String stateName, String stateValue) {
                // 第二段：执行期间状态上报（0 到多次）
                log.info("onState => agentId {}, requestId {}, eventId {}, stateName {}, stateValue {}",
                        agentId, requestId, eventId, stateName, stateValue);
            }

            @Override
            public void onInvokeResult(String agentId, String requestId, String eventId,
                                       int code, String msg, AgentParam output) {
                // 第三段：技能执行输出
                log.info("onInvokeResult => agentId {}, requestId {}, eventId {}, code {}, msg {}, output {}",
                        agentId, requestId, eventId, code, msg, output);
            }
        }, 2000);
        log.info("invoke invokeEventId {}", invokeEventId);

        Thread.sleep(5000);

        // 第五步：反注册释放
        agentSdk.unregisterExtSkill(requestId);
        // agentSdk.release();
    }
}