MemOS-MCP

#!/bin/bash

# MemOS 安全启动脚本
# 自动处理进程冲突和锁定问题

set -e

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 日志函数
log_info() {
    echo -e "${BLUE}ℹ️  $1${NC}"
}

log_success() {
    echo -e "${GREEN}✅ $1${NC}"
}

log_warning() {
    echo -e "${YELLOW}⚠️  $1${NC}"
}

log_error() {
    echo -e "${RED}❌ $1${NC}"
}

# 检查依赖
check_dependencies() {
    log_info "检查依赖..."
    
    # 检查Python
    if ! command -v python3 &> /dev/null; then
        log_error "Python3 未安装"
        exit 1
    fi
    
    # 检查虚拟环境
    if [ ! -d "../.venv" ]; then
        log_error "虚拟环境不存在: ../.venv"
        exit 1
    fi
    
    log_success "依赖检查通过"
}

# 激活虚拟环境
activate_venv() {
    log_info "激活虚拟环境..."
    source ../.venv/bin/activate
    log_success "虚拟环境已激活"
}

# 检查并清理现有进程
cleanup_processes() {
    log_info "检查现有MemOS进程..."
    
    # 查找MemOS相关进程
    MEMOS_PIDS=$(ps aux | grep -E "(memos_cli|memos_mcp_server|advanced_examples|usage_examples)" | grep -v grep | awk '{print $2}' || true)
    
    if [ ! -z "$MEMOS_PIDS" ]; then
        log_warning "发现现有MemOS进程: $MEMOS_PIDS"
        echo "是否终止这些进程? (y/N)"
        read -r response
        
        if [[ "$response" =~ ^[Yy]$ ]]; then
            for pid in $MEMOS_PIDS; do
                if kill -0 "$pid" 2>/dev/null; then
                    log_info "终止进程: $pid"
                    kill "$pid" 2>/dev/null || true
                    sleep 1
                    
                    # 如果进程仍在运行，强制杀死
                    if kill -0 "$pid" 2>/dev/null; then
                        log_warning "强制终止进程: $pid"
                        kill -9 "$pid" 2>/dev/null || true
                    fi
                fi
            done
            log_success "已清理现有进程"
        else
            log_error "用户取消，退出"
            exit 1
        fi
    else
        log_success "未发现冲突进程"
    fi
}

# 清理锁文件
cleanup_locks() {
    log_info "清理锁文件..."
    
    # 清理可能的锁文件
    find . -name "*.lock" -delete 2>/dev/null || true
    find . -name "memos.pid" -delete 2>/dev/null || true
    find . -name ".lock*" -delete 2>/dev/null || true
    
    # 清理Qdrant锁
    find ./memos_data/vectors -name "*.lock" -delete 2>/dev/null || true
    find ./mcp_memos_data/vectors -name "*.lock" -delete 2>/dev/null || true
    
    log_success "锁文件清理完成"
}

# 启动MemOS CLI
start_cli() {
    log_info "启动MemOS CLI..."
    echo ""
    echo "🎉 MemOS 安全启动完成！"
    echo "📝 现在可以安全使用MemOS功能"
    echo ""
    
    python3 memos_cli.py
}

# 启动MCP服务器
start_mcp() {
    log_info "启动MemOS MCP服务器..."
    echo ""
    echo "🎉 MemOS MCP服务器安全启动完成！"
    echo "📝 MCP服务器正在运行，等待连接..."
    echo ""
    
    python3 memos_mcp_server.py
}

# 显示帮助
show_help() {
    echo "MemOS 安全启动脚本"
    echo ""
    echo "用法: $0 [选项]"
    echo ""
    echo "选项:"
    echo "  cli     启动MemOS CLI (默认)"
    echo "  mcp     启动MemOS MCP服务器"
    echo "  help    显示此帮助信息"
    echo ""
    echo "功能:"
    echo "  ✅ 自动检测和清理冲突进程"
    echo "  ✅ 清理Qdrant锁文件"
    echo "  ✅ 进程互斥保护"
    echo "  ✅ 优雅的错误处理"
}

# 主函数
main() {
    echo "🚀 MemOS 安全启动脚本"
    echo "========================"
    
    # 进入MemOS目录
    cd "$(dirname "$0")"
    
    # 解析参数
    MODE="${1:-cli}"
    
    case "$MODE" in
        "cli")
            check_dependencies
            activate_venv
            cleanup_processes
            cleanup_locks
            start_cli
            ;;
        "mcp")
            check_dependencies
            activate_venv
            cleanup_processes
            cleanup_locks
            start_mcp
            ;;
        "help"|"-h"|"--help")
            show_help
            ;;
        *)
            log_error "未知选项: $MODE"
            show_help
            exit 1
            ;;
    esac
}

# 信号处理
trap 'log_warning "收到中断信号，正在清理..."; exit 130' INT TERM

# 运行主函数
main "$@"