MemOS-MCP

#!/bin/bash

# MemOS systemd服务安装脚本
# 用于安装、配置和管理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_permissions() {
    if [[ $EUID -eq 0 ]]; then
        log_error "请不要使用root权限运行此脚本"
        log_info "使用普通用户权限运行，需要sudo时会自动提示"
        exit 1
    fi
}

# 检查依赖
check_dependencies() {
    log_info "检查系统依赖..."
    
    # 检查systemctl
    if ! command -v systemctl &> /dev/null; then
        log_error "systemctl 未找到，请确保运行在systemd系统上"
        exit 1
    fi
    
    # 检查MemOS文件
    if [ ! -f "memos.service" ]; then
        log_error "memos.service 文件不存在"
        exit 1
    fi
    
    if [ ! -f "memos_mcp_server.py" ]; then
        log_error "memos_mcp_server.py 文件不存在"
        exit 1
    fi
    
    if [ ! -d "memos_venv" ]; then
        log_error "虚拟环境 memos_venv 不存在"
        exit 1
    fi
    
    log_success "依赖检查通过"
}

# 安装服务
install_service() {
    log_info "安装MemOS systemd服务..."
    
    # 复制服务文件到系统目录
    sudo cp memos.service /etc/systemd/system/
    log_success "服务文件已复制到 /etc/systemd/system/"
    
    # 重新加载systemd配置
    sudo systemctl daemon-reload
    log_success "systemd配置已重新加载"
    
    # 启用服务（开机自启动）
    sudo systemctl enable memos.service
    log_success "MemOS服务已设置为开机自启动"
}

# 启动服务
start_service() {
    log_info "启动MemOS服务..."
    
    # 清理可能的锁文件
    find . -name "*.lock" -delete 2>/dev/null || true
    find . -name "memos.pid" -delete 2>/dev/null || true
    
    # 启动服务
    sudo systemctl start memos.service
    
    # 等待服务启动
    sleep 3
    
    # 检查服务状态
    if sudo systemctl is-active --quiet memos.service; then
        log_success "MemOS服务启动成功"
    else
        log_error "MemOS服务启动失败"
        log_info "查看详细日志: sudo journalctl -u memos.service -f"
        exit 1
    fi
}

# 停止服务
stop_service() {
    log_info "停止MemOS服务..."
    sudo systemctl stop memos.service
    log_success "MemOS服务已停止"
}

# 重启服务
restart_service() {
    log_info "重启MemOS服务..."
    sudo systemctl restart memos.service
    
    # 等待服务启动
    sleep 3
    
    # 检查服务状态
    if sudo systemctl is-active --quiet memos.service; then
        log_success "MemOS服务重启成功"
    else
        log_error "MemOS服务重启失败"
        log_info "查看详细日志: sudo journalctl -u memos.service -f"
        exit 1
    fi
}

# 查看服务状态
status_service() {
    log_info "MemOS服务状态:"
    sudo systemctl status memos.service --no-pager
}

# 查看服务日志
logs_service() {
    log_info "MemOS服务日志 (按Ctrl+C退出):"
    sudo journalctl -u memos.service -f
}

# 卸载服务
uninstall_service() {
    log_warning "卸载MemOS服务..."
    
    # 停止服务
    sudo systemctl stop memos.service 2>/dev/null || true
    
    # 禁用服务
    sudo systemctl disable memos.service 2>/dev/null || true
    
    # 删除服务文件
    sudo rm -f /etc/systemd/system/memos.service
    
    # 重新加载systemd配置
    sudo systemctl daemon-reload
    
    log_success "MemOS服务已卸载"
}

# 测试服务重启
test_restart() {
    log_info "测试服务自动重启功能..."
    
    # 获取服务主进程PID
    MAIN_PID=$(sudo systemctl show memos.service --property=MainPID --value)
    
    if [ "$MAIN_PID" = "0" ] || [ -z "$MAIN_PID" ]; then
        log_error "服务未运行，无法测试重启"
        exit 1
    fi
    
    log_info "当前主进程PID: $MAIN_PID"
    
    # 杀死主进程模拟异常
    log_info "模拟进程异常..."
    sudo kill -9 "$MAIN_PID"
    
    # 等待systemd重启服务
    log_info "等待systemd自动重启服务..."
    sleep 15
    
    # 检查服务是否重新启动
    if sudo systemctl is-active --quiet memos.service; then
        NEW_PID=$(sudo systemctl show memos.service --property=MainPID --value)
        log_success "服务自动重启成功，新PID: $NEW_PID"
    else
        log_error "服务自动重启失败"
        exit 1
    fi
}

# 显示帮助
show_help() {
    echo "MemOS systemd服务管理脚本"
    echo ""
    echo "用法: $0 [命令]"
    echo ""
    echo "命令:"
    echo "  install     安装并启用MemOS服务"
    echo "  start       启动MemOS服务"
    echo "  stop        停止MemOS服务"
    echo "  restart     重启MemOS服务"
    echo "  status      查看服务状态"
    echo "  logs        查看服务日志"
    echo "  test        测试自动重启功能"
    echo "  uninstall   卸载MemOS服务"
    echo "  help        显示此帮助信息"
    echo ""
    echo "示例:"
    echo "  $0 install    # 安装并启动服务"
    echo "  $0 logs       # 查看实时日志"
    echo "  $0 test       # 测试自动重启"
}

# 主函数
main() {
    echo "🔧 MemOS systemd服务管理"
    echo "========================="
    
    # 进入MemOS目录
    cd "$(dirname "$0")"
    
    # 检查权限
    check_permissions
    
    # 解析命令
    COMMAND="${1:-help}"
    
    case "$COMMAND" in
        "install")
            check_dependencies
            install_service
            start_service
            status_service
            ;;
        "start")
            start_service
            ;;
        "stop")
            stop_service
            ;;
        "restart")
            restart_service
            ;;
        "status")
            status_service
            ;;
        "logs")
            logs_service
            ;;
        "test")
            test_restart
            ;;
        "uninstall")
            uninstall_service
            ;;
        "help"|"-h"|"--help")
            show_help
            ;;
        *)
            log_error "未知命令: $COMMAND"
            show_help
            exit 1
            ;;
    esac
}

# 信号处理
trap 'log_warning "收到中断信号"; exit 130' INT TERM

# 运行主函数
main "$@"