#!/usr/bin/env python3
"""
mem0 Commands for OpenClaw
"""

from .mem0_client import Mem0Client

mem0 = Mem0Client()

async def handle_memory_command(args: str, context: dict) -> str:
    """处理 /memory 命令"""
    parts = args.split(' ', 2)
    action = parts[0] if parts else 'help'
    
    user_id = context.get('user_id', 'default')
    agent_id = context.get('agent_id', 'main')
    
    if action == 'add':
        content = parts[1] if len(parts) > 1 else ''
        return await handle_add_memory(content, user_id, agent_id)
    elif action == 'search':
        query = parts[1] if len(parts) > 1 else ''
        return await handle_search_memory(query, user_id, agent_id)
    elif action == 'list':
        return await handle_list_memories(user_id, agent_id)
    elif action == 'delete':
        memory_id = parts[1] if len(parts) > 1 else ''
        return await handle_delete_memory(memory_id, user_id, agent_id)
    elif action == 'sync':
        return await handle_sync_memory(user_id, agent_id)
    elif action == 'status':
        return handle_status(user_id, agent_id)
    else:
        return get_help_text()

async def handle_add_memory(content: str, user_id: str, agent_id: str) -> str:
    if not content:
        return "❌ 请提供记忆内容"
    result = mem0.add([{"role": "user", "content": content}], user_id=user_id, agent_id=agent_id)
    if 'error' in result:
        return f"❌ 添加失败：{result['error']}"
    return f"✅ 记忆已添加：{content[:100]}..."

async def handle_search_memory(query: str, user_id: str, agent_id: str) -> str:
    if not query:
        return "❌ 请提供关键词"
    results = mem0.search(query, user_id=user_id, agent_id=agent_id, limit=5)
    if not results:
        return f"🔍 未找到相关记忆"
    response = f"🔍 找到 {len(results)} 条记忆:\n\n"
    for i, mem in enumerate(results, 1):
        response += f"{i}. {mem.get('memory', 'N/A')}\n"
    return response

async def handle_list_memories(user_id: str, agent_id: str) -> str:
    results = mem0.get_all(user_id=user_id, agent_id=agent_id)
    if not results:
        return "📭 暂无记忆"
    return f"📋 共 {len(results)} 条记忆"

async def handle_delete_memory(memory_id: str, user_id: str, agent_id: str) -> str:
    if not memory_id:
        return "❌ 请提供 ID"
    success = mem0.delete(memory_id, user_id=user_id, agent_id=agent_id)
    return "✅ 已删除" if success else "❌ 删除失败"

async def handle_sync_memory(user_id: str, agent_id: str) -> str:
    return "🔄 同步功能开发中"

def handle_status(user_id: str, agent_id: str) -> str:
    status = mem0.get_status()
    response = "📊 mem0 状态:\n"
    response += f"本地：{'✅' if status['local_initialized'] else '❌'}\n"
    response += f"共享：{'✅' if status['master_initialized'] else '❌'}\n"
    return response

def get_help_text() -> str:
    return """📖 用法：/memory <命令>
命令：add, search, list, delete, sync, status"""