import time

from linksoul_agentsdk import (
    AgentAuthCallback,
    AgentParam,
    AgentSdk,
    IdGenerator,
    TeleopEnterCallback,
    TeleopExitCallback,
    TeleopKeepaliveCallback,
    TeleopRequest,
)

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

# 第一步：创建 SDK 实例
agent_sdk = AgentSdk.create(url, app_id, app_key, app_secret)


# 第二步：注册验签回调
class MyAuthCallback(AgentAuthCallback):
    def on_auth_state(self, app_id: str, code: int, msg: str) -> None:
        print(f"on_auth_state => app_id: {app_id}, code: {code}, msg: {msg}")


agent_sdk.register_auth(MyAuthCallback())

# 第三步：初始化
agent_sdk.initialize()
time.sleep(2)

# 注册遥操请求实例
teleop_id = IdGenerator.generate_teleop_id()
teleop_request = TeleopRequest(agent_sdk, agent_id, teleop_id)
agent_sdk.register_teleop(teleop_request)


# 进入遥操
class MyEnterCallback(TeleopEnterCallback):
    def on_enter_ack(self, teleop_id: str, event_id: str, code: int, msg: str) -> None:
        print(f"on_enter_ack => teleop_id={teleop_id}, event_id={event_id}, code={code}, msg={msg}")


enter_param = AgentParam.create()
teleop_request.enter(enter_param, MyEnterCallback(), 2000)

# 开始说话
start_param = (
    AgentParam.create()
    .set_string("role", "male")       # male / female
    .set_integer("threshold", 0)      # -14 ~ 14
)
event_id = teleop_request.start_audio(start_param)
# 持续说话（这里以空字符串占位，实际传 base64 编码后的 PCM/Opus 分片）
teleop_request.append_audio_delta(event_id, "")
# 结束说话
teleop_request.done_audio(event_id)


# 检查机器人是否在线（详见「定时探活」小节，遥操期间应每 10s 探活一次）
class MyKeepaliveCallback(TeleopKeepaliveCallback):
    def on_keepalive_ack(self, teleop_id: str, event_id: str, code: int, msg: str) -> None:
        print(f"on_keepalive_ack => teleop_id={teleop_id}, event_id={event_id}, code={code}, msg={msg}")


teleop_request.keepalive(MyKeepaliveCallback(), 2000)


# 退出遥操
class MyExitCallback(TeleopExitCallback):
    def on_exit_ack(self, teleop_id: str, event_id: str, code: int, msg: str) -> None:
        print(f"on_exit_ack => teleop_id={teleop_id}, event_id={event_id}, code={code}, msg={msg}")


teleop_request.exit(MyExitCallback(), 2000)

# 注销并释放（示例省略）
# agent_sdk.unregister_teleop(teleop_id)
# agent_sdk.release()

import threading
from linksoul_agentsdk import TeleopKeepaliveCallback, TeleopExitCallback


def start_keepalive(teleop_request, agent_sdk, interval_sec: int = 10):
    """enter 成功后调用；每 interval_sec 秒探活一次"""
    state = {"online": True}

    def tick():
        if not state["online"]:
            return

        class _Cb(TeleopKeepaliveCallback):
            def on_keepalive_ack(self, teleop_id, event_id, code, msg):
                if code != 0 and state["online"]:
                    state["online"] = False
                    print(f"robot offline => code={code}, msg={msg}")

                    class _ExitCb(TeleopExitCallback):
                        def on_exit_ack(self, teleop_id, event_id, code, msg):
                            agent_sdk.unregister_teleop(teleop_id)
                    teleop_request.exit(_ExitCb(), 2000)

        teleop_request.keepalive(_Cb(), 2000)
        if state["online"]:
            threading.Timer(interval_sec, tick).start()

    threading.Timer(interval_sec, tick).start()
    return state  # 业务侧可通过 state["online"] = False 主动停止