package com.agibot.aiem.teleop;

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.teleop.TeleopEnterCallback;
import com.agibot.aiem.sdk.teleop.TeleopExitCallback;
import com.agibot.aiem.sdk.teleop.TeleopKeepaliveCallback;
import com.agibot.aiem.sdk.teleop.TeleopRequest;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class TeleopExample {

    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 teleopId = IdGenerator.generateTeleopId();
        TeleopRequest teleopRequest = new TeleopRequest(agentSdk, agentId, teleopId);
        agentSdk.registerTeleop(teleopRequest);

        // 进入遥操
        AgentParam enterParam = AgentParam.create();
        teleopRequest.enter(enterParam, new TeleopEnterCallback() {
            @Override
            public void onEnterAck(String teleopId, String eventId, int code, String msg) {
                log.info("onEnterAck => teleopId {}, eventId {}, code {}, msg {}", teleopId, eventId, code, msg);
            }
        }, 2000);

        // 开始说话
        AgentParam startParam = AgentParam.create();
        startParam.setString("role", "male");       // male/female
        startParam.setInteger("threshold", 0);     // -14 - 14
        String eventId = teleopRequest.startAudio(startParam);
        // 持续说话（这里以空字符串占位，实际传 base64 编码后的 PCM/Opus 分片）
        teleopRequest.appendAudioDelta(eventId, "");
        // 结束说话
        teleopRequest.doneAudio(eventId);

        // 检查机器人是否在线（详见「定时探活」小节，遥操期间应每 10s 探活一次）
        teleopRequest.keepalive(new TeleopKeepaliveCallback() {
            @Override
            public void onKeepaliveAck(String teleopId, String eventId, int code, String msg) {
                log.info("onKeepaliveAck => teleopId {}, eventId {}, code {}, msg {}", teleopId, eventId, code, msg);
            }
        }, 2000);

        // 退出遥操
        teleopRequest.exit(new TeleopExitCallback() {
            @Override
            public void onExitAck(String teleopId, String eventId, int code, String msg) {
                log.info("onExitAck => teleopId {}, eventId {}, code {}, msg {}", teleopId, eventId, code, msg);
            }
        }, 2000);

        // 注销并释放（示例省略）
        // agentSdk.unregisterTeleop(teleopId);
        // agentSdk.release();
    }
}

import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;

ScheduledExecutorService keepaliveScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
    Thread t = new Thread(r, "teleop-keepalive");
    t.setDaemon(true);
    return t;
});
AtomicBoolean robotOnline = new AtomicBoolean(true);

keepaliveScheduler.scheduleAtFixedRate(() -> {
    if (!robotOnline.get()) return;
    teleopRequest.keepalive(new TeleopKeepaliveCallback() {
        @Override
        public void onKeepaliveAck(String teleopId, String eventId, int code, String msg) {
            if (code == 0) return;
            if (!robotOnline.compareAndSet(true, false)) return;
            log.warn("robot offline => code={}, msg={}", code, msg);
            keepaliveScheduler.shutdown();
            teleopRequest.exit(new TeleopExitCallback() {
                @Override
                public void onExitAck(String teleopId, String eventId, int code, String msg) {
                    agentSdk.unregisterTeleop(teleopId);
                }
            }, 2000);
        }
    }, 2000);
}, 10, 10, TimeUnit.SECONDS);