#!/bin/bash
# 记忆系统快速查询脚本

MEMORY_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
CLI_SCRIPT="$MEMORY_ROOT/memory_cli.py"

# 帮助信息
show_help() {
    cat << EOF
记忆系统快速查询工具

用法: $0 [命令] [选项]

命令:
    today           查看今天的事件
    week            查看本周的事件
    recent <天数>   查看最近N天的事件
    search <关键词> 搜索标题包含关键词的事件
    types           统计事件类型分布
    tags            统计标签使用情况
    stats           显示系统统计信息
    help            显示此帮助信息

示例:
    $0 today
    $0 week
    $0 recent 3
    $0 search "认证"
    $0 types
    $0 tags
    $0 stats

EOF
}

# 检查Python环境
check_python() {
    if ! command -v python3 &> /dev/null; then
        echo "错误: 未找到python3，请确保Python 3已安装"
        exit 1
    fi

    # 检查必要的Python模块
    python3 -c "import sys; sys.exit(0)" 2>/dev/null || {
        echo "错误: Python环境异常"
        exit 1
    }
}

# 检查CLI脚本是否存在并可执行
check_cli() {
    if [ ! -f "$CLI_SCRIPT" ]; then
        echo "错误: 找不到 memory_cli.py 脚本 ($CLI_SCRIPT)"
        exit 1
    fi

    if [ ! -x "$CLI_SCRIPT" ] && [ ! -x "$(which python3)" ]; then
        echo "错误: 无法执行Python脚本，请检查权限"
        exit 1
    fi
}

# 执行查询并处理错误
run_query() {
    if ! python3 "$CLI_SCRIPT" query "$@" 2>&1; then
        echo "查询执行失败，可能的原因:"
        echo "1. 记忆系统尚未初始化"
        echo "2. 查询语法错误"
        echo "3. 文件权限问题"
        echo "请检查 memory_cli.py 是否正常工作"
        return 1
    fi
}

# 主逻辑
main() {
    check_python
    check_cli

    case "${1:-help}" in
        "today")
            echo "=== 今天的事件 ==="
            run_query --where "date='$(date +%Y-%m-%d)'" --order-by "timestamp desc"
            ;;
        "week")
            echo "=== 本周的事件 ==="
            run_query --where "date>='$(date -d 'last monday' +%Y-%m-%d)'" --order-by "timestamp desc"
            ;;
        "recent")
            days="${2:-7}"
            echo "=== 最近 $days 天的事件 ==="
            run_query --where "timestamp >= '$(date -d "$days days ago" +%Y-%m-%dT%H:%M:%S)'" --order-by "timestamp desc"
            ;;
        "search")
            keyword="${2:-}"
            if [ -z "$keyword" ]; then
                echo "错误: 请提供搜索关键词"
                exit 1
            fi
            echo "=== 搜索包含 '$keyword' 的事件 ==="
            # 注意: 当前CLI不支持LIKE操作，这里用简单的方式
            run_query --where "title LIKE '%$keyword%'" 2>/dev/null || run_query | grep -i "$keyword"
            ;;
        "types")
            echo "=== 事件类型统计 ==="
            if run_query --select "type" --format json 2>/dev/null | python3 -c "
import sys, json
try:
    data = json.load(sys.stdin)
    types = {}
    for item in data:
        t = item.get('type', 'unknown')
        types[t] = types.get(t, 0) + 1
    if not types:
        print('暂无事件数据')
    else:
        for t, count in sorted(types.items(), key=lambda x: x[1], reverse=True):
            print(f'{t}: {count}')
except Exception as e:
    print(f'统计失败: {e}')
    print('请检查Python环境和JSON数据格式')
" 2>/dev/null; then
                echo "统计功能需要Python 3和JSON支持"
            fi
            ;;
        "tags")
            echo "=== 标签使用统计 ==="
            if run_query --select "tags" --format json 2>/dev/null | python3 -c "
import sys, json
try:
    data = json.load(sys.stdin)
    tags = {}
    for item in data:
        for tag in item.get('tags', []):
            tags[tag] = tags.get(tag, 0) + 1
    if not tags:
        print('暂无标签数据')
    else:
        for tag, count in sorted(tags.items(), key=lambda x: x[1], reverse=True)[:20]:
            print(f'{tag}: {count}')
except Exception as e:
    print(f'统计失败: {e}')
    print('请检查Python环境和JSON数据格式')
" 2>/dev/null; then
                echo "统计功能需要Python 3和JSON支持"
            fi
            ;;
        "stats")
            echo "=== 记忆系统统计 ==="
            echo "Episodic事件数量: $(find "$MEMORY_ROOT/episodic" -name "*.md" 2>/dev/null | wc -l)"
            echo "Long-term文档数量: $(find "$MEMORY_ROOT/long-term" -name "*.md" 2>/dev/null | wc -l)"
            echo "Short-term文件数量: $(find "$MEMORY_ROOT/short-term" -name "*.md" 2>/dev/null | wc -l)"
            echo "总计: $(find "$MEMORY_ROOT" -name "*.md" 2>/dev/null | wc -l)"
            ;;
        "help"|*)
            show_help
            ;;
    esac
}

main "$@"