Bubbles/function_calls/handlers.py

"""Function Call handlers built on top of structured services."""
from __future__ import annotations

import logging
from commands.context import MessageContext

from .models import (
    WeatherArgs,
    NewsArgs,
    ReminderArgs,
    ReminderListArgs,
    ReminderDeleteArgs,
    PerplexityArgs,
    HelpArgs,
    SummaryArgs,
    ClearMessagesArgs,
    InsultArgs,
)
from .registry import tool_function
from .spec import FunctionResult
from .services import (
    build_help_text,
    build_insult,
    clear_group_messages,
    create_reminder,
    delete_reminder,
    get_news_digest,
    get_weather_report,
    list_reminders,
    run_perplexity,
    summarize_messages,
)

logger = logging.getLogger(__name__)


@tool_function(
    name="weather_query",
    description="查询城市天气预报",
    examples=["北京天气怎么样", "上海天气", "深圳明天会下雨吗"],
    scope="both",
    require_at=True,
)
def handle_weather(ctx: MessageContext, args: WeatherArgs) -> FunctionResult:
    result = get_weather_report(args.city)
    at = ctx.msg.sender if ctx.is_group else ""
    return FunctionResult(handled=True, messages=[result.message], at=at if at else "")


@tool_function(
    name="news_query",
    description="获取今日新闻",
    examples=["看看今天的新闻", "今日要闻", "新闻"],
    scope="both",
    require_at=True,
)
def handle_news(ctx: MessageContext, args: NewsArgs) -> FunctionResult:
    result = get_news_digest()
    at = ctx.msg.sender if ctx.is_group else ""
    return FunctionResult(handled=True, messages=[result.message], at=at if at else "")


@tool_function(
    name="reminder_set",
    description="设置提醒",
    examples=["提醒我明天下午3点开会", "每天早上8点提醒我吃早餐"],
    scope="both",
    require_at=True,
)
def handle_reminder_set(ctx: MessageContext, args: ReminderArgs) -> FunctionResult:
    manager = getattr(ctx.robot, 'reminder_manager', None)
    at = ctx.msg.sender if ctx.is_group else ""
    if not manager:
        return FunctionResult(handled=True, messages=["❌ 内部错误：提醒管理器未初始化。"], at=at)

    service_result = create_reminder(
        manager=manager,
        sender_wxid=ctx.msg.sender,
        data=args.model_dump(),
        roomid=ctx.msg.roomid if ctx.is_group else None,
    )
    return FunctionResult(handled=True, messages=service_result.messages, at=at if at else "")


@tool_function(
    name="reminder_list",
    description="查看所有提醒",
    examples=["查看我的提醒", "我有哪些提醒", "提醒列表"],
    scope="both",
    require_at=True,
)
def handle_reminder_list(ctx: MessageContext, args: ReminderListArgs) -> FunctionResult:
    manager = getattr(ctx.robot, 'reminder_manager', None)
    at = ctx.msg.sender if ctx.is_group else ""
    if not manager:
        return FunctionResult(handled=True, messages=["❌ 内部错误：提醒管理器未初始化。"], at=at)

    service_result = list_reminders(manager, ctx.msg.sender, ctx.all_contacts)
    return FunctionResult(handled=True, messages=service_result.messages, at=at if at else "")


@tool_function(
    name="reminder_delete",
    description="删除提醒",
    examples=["删除开会的提醒", "取消明天的提醒"],
    scope="both",
    require_at=True,
)
def handle_reminder_delete(ctx: MessageContext, args: ReminderDeleteArgs) -> FunctionResult:
    manager = getattr(ctx.robot, 'reminder_manager', None)
    at = ctx.msg.sender if ctx.is_group else ""
    if not manager:
        return FunctionResult(handled=True, messages=["❌ 内部错误：提醒管理器未初始化。"], at=at)

    service_result = delete_reminder(manager, ctx.msg.sender, args.reminder_id)
    return FunctionResult(handled=True, messages=service_result.messages, at=at if at else "")


@tool_function(
    name="perplexity_search",
    description="使用Perplexity进行深度搜索查询",
    examples=["搜索Python最新特性", "查查机器学习教程", "ask什么是量子计算"],
    scope="both",
    require_at=True,
)
def handle_perplexity_search(ctx: MessageContext, args: PerplexityArgs) -> FunctionResult:
    service_result = run_perplexity(ctx, args.query)
    if service_result.handled_externally:
        return FunctionResult(handled=True, messages=[])

    at = ctx.msg.sender if ctx.is_group else ""
    return FunctionResult(handled=True, messages=service_result.messages, at=at if at else "")


@tool_function(
    name="help",
    description="显示机器人帮助信息",
    examples=["help", "帮助", "指令"],
    scope="both",
    require_at=False,
)
def handle_help(ctx: MessageContext, args: HelpArgs) -> FunctionResult:
    help_text = build_help_text()
    return FunctionResult(handled=True, messages=[help_text])


@tool_function(
    name="summary",
    description="总结群聊最近的消息",
    examples=["summary", "总结"],
    scope="group",
    require_at=True,
)
def handle_summary(ctx: MessageContext, args: SummaryArgs) -> FunctionResult:
    result = summarize_messages(ctx)
    return FunctionResult(handled=True, messages=[result.message])


@tool_function(
    name="clear_messages",
    description="清除群聊历史消息记录",
    examples=["clearmessages", "清除历史"],
    scope="group",
    require_at=True,
)
def handle_clear_messages(ctx: MessageContext, args: ClearMessagesArgs) -> FunctionResult:
    result = clear_group_messages(ctx)
    return FunctionResult(handled=True, messages=[result.message])


@tool_function(
    name="insult",
    description="骂指定用户（仅限群聊）",
    examples=["骂一下@某人"],
    scope="group",
    require_at=True,
)
def handle_insult(ctx: MessageContext, args: InsultArgs) -> FunctionResult:
    result = build_insult(ctx, args.target_user)
    return FunctionResult(handled=True, messages=[result.message])