MemOS-MCP

#!/bin/bash

# MemOS智能备份脚本
# 专为个人电脑设计，支持开机自动备份、使用时提醒备份等策略

set -e

MEMOS_DIR="/home/qqinshu/视频/MemOS"
BACKUP_SCRIPT="$MEMOS_DIR/backup_simple.sh"
BACKUP_STATE_FILE="$MEMOS_DIR/memos_data/.backup_state"
LAST_BACKUP_FILE="$MEMOS_DIR/memos_data/.last_backup"

# 颜色定义
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}"
}

# 获取上次备份时间
get_last_backup_time() {
    if [ -f "$LAST_BACKUP_FILE" ]; then
        cat "$LAST_BACKUP_FILE"
    else
        echo "0"
    fi
}

# 更新上次备份时间
update_last_backup_time() {
    echo "$(date +%s)" > "$LAST_BACKUP_FILE"
}

# 检查是否需要备份
need_backup() {
    local last_backup=$(get_last_backup_time)
    local current_time=$(date +%s)
    local hours_since_backup=$(( (current_time - last_backup) / 3600 ))
    
    # 如果超过24小时没有备份，则需要备份
    if [ "$hours_since_backup" -gt 24 ]; then
        return 0  # 需要备份
    else
        return 1  # 不需要备份
    fi
}

# 静默备份（后台执行）
silent_backup() {
    log_info "执行静默备份..."
    
    if "$BACKUP_SCRIPT" > /dev/null 2>&1; then
        update_last_backup_time
        log_success "静默备份完成"
        return 0
    else
        log_error "静默备份失败"
        return 1
    fi
}

# 交互式备份提醒
interactive_backup_reminder() {
    local last_backup=$(get_last_backup_time)
    local current_time=$(date +%s)
    local hours_since_backup=$(( (current_time - last_backup) / 3600 ))
    
    echo "🔔 MemOS备份提醒"
    echo "================"
    
    if [ "$last_backup" = "0" ]; then
        echo "📋 状态: 尚未进行过备份"
    else
        echo "📋 上次备份: $(date -d @$last_backup '+%Y-%m-%d %H:%M:%S') (${hours_since_backup}小时前)"
    fi
    
    if need_backup; then
        echo "⚠️  建议立即执行备份！"
        echo ""
        echo "选择操作:"
        echo "  1) 立即备份"
        echo "  2) 稍后提醒 (1小时后)"
        echo "  3) 跳过此次提醒"
        echo ""
        read -p "请选择 (1-3): " choice
        
        case "$choice" in
            1)
                log_info "开始执行备份..."
                if "$BACKUP_SCRIPT"; then
                    update_last_backup_time
                    log_success "备份完成！"
                else
                    log_error "备份失败，请检查系统状态"
                fi
                ;;
            2)
                # 设置1小时后提醒
                echo "$(date -d '+1 hour' +%s)" > "$BACKUP_STATE_FILE"
                log_info "已设置1小时后提醒"
                ;;
            3)
                log_info "已跳过此次备份提醒"
                ;;
            *)
                log_warning "无效选择，跳过备份"
                ;;
        esac
    else
        echo "✅ 备份状态正常，无需立即备份"
    fi
}

# 开机自动检查
boot_check() {
    log_info "开机备份检查..."
    
    if need_backup; then
        log_warning "检测到需要备份，将在后台执行..."
        
        # 等待系统稳定（开机后30秒）
        sleep 30
        
        if silent_backup; then
            # 发送桌面通知（如果支持）
            if command -v notify-send > /dev/null 2>&1; then
                notify-send "MemOS备份" "开机自动备份已完成" --icon=dialog-information
            fi
        else
            # 备份失败时发送通知
            if command -v notify-send > /dev/null 2>&1; then
                notify-send "MemOS备份" "开机自动备份失败，请手动检查" --icon=dialog-error
            fi
        fi
    else
        log_info "备份状态正常，无需执行开机备份"
    fi
}

# 关机前备份检查
shutdown_check() {
    log_info "关机前备份检查..."
    
    if need_backup; then
        log_warning "检测到需要备份！"
        echo ""
        echo "系统即将关机，建议先执行备份。"
        echo ""
        read -p "是否现在备份？(Y/n): " choice
        
        if [[ "$choice" =~ ^[Yy]$ ]] || [[ -z "$choice" ]]; then
            log_info "执行关机前备份..."
            if "$BACKUP_SCRIPT"; then
                update_last_backup_time
                log_success "备份完成，可以安全关机"
            else
                log_error "备份失败！"
                read -p "是否仍要继续关机？(y/N): " force_choice
                if [[ ! "$force_choice" =~ ^[Yy]$ ]]; then
                    log_info "已取消关机"
                    exit 1
                fi
            fi
        else
            log_warning "已跳过备份，直接关机"
        fi
    else
        log_info "备份状态正常，可以安全关机"
    fi
}

# 设置开机自启动
setup_autostart() {
    log_info "设置开机自动备份检查..."
    
    # 创建systemd用户服务
    local service_dir="$HOME/.config/systemd/user"
    mkdir -p "$service_dir"
    
    cat > "$service_dir/memos-backup-check.service" << EOF
[Unit]
Description=MemOS开机备份检查
After=graphical-session.target

[Service]
Type=oneshot
ExecStart=$MEMOS_DIR/smart_backup.sh boot
Environment=DISPLAY=:0

[Install]
WantedBy=default.target
EOF
    
    # 启用服务
    systemctl --user daemon-reload
    systemctl --user enable memos-backup-check.service
    
    log_success "开机自动备份检查已设置"
}

# 移除开机自启动
remove_autostart() {
    log_info "移除开机自动备份检查..."
    
    systemctl --user disable memos-backup-check.service 2>/dev/null || true
    rm -f "$HOME/.config/systemd/user/memos-backup-check.service"
    systemctl --user daemon-reload
    
    log_success "开机自动备份检查已移除"
}

# 显示帮助信息
show_help() {
    echo "MemOS智能备份脚本 - 个人电脑优化版"
    echo ""
    echo "用法: $0 [命令]"
    echo ""
    echo "命令:"
    echo "  check          检查备份状态并提供交互式选项"
    echo "  boot           开机自动检查（静默模式）"
    echo "  shutdown       关机前检查"
    echo "  force          强制执行备份"
    echo "  status         显示备份状态"
    echo "  setup-auto     设置开机自动检查"
    echo "  remove-auto    移除开机自动检查"
    echo "  help           显示此帮助信息"
    echo ""
    echo "个人电脑使用建议:"
    echo "  1. 运行 '$0 setup-auto' 设置开机自动检查"
    echo "  2. 定期运行 '$0 check' 进行交互式备份"
    echo "  3. 关机前运行 '$0 shutdown' 确保数据安全"
}

# 显示备份状态
show_status() {
    local last_backup=$(get_last_backup_time)
    local current_time=$(date +%s)
    
    echo "📊 MemOS备份状态"
    echo "================"
    
    if [ "$last_backup" = "0" ]; then
        echo "📋 备份状态: 尚未备份"
        echo "⚠️  建议: 立即执行首次备份"
    else
        local hours_since_backup=$(( (current_time - last_backup) / 3600 ))
        echo "📋 上次备份: $(date -d @$last_backup '+%Y-%m-%d %H:%M:%S')"
        echo "⏰ 距今时间: ${hours_since_backup}小时"
        
        if need_backup; then
            echo "⚠️  状态: 需要备份"
        else
            echo "✅ 状态: 正常"
        fi
    fi
    
    # 显示最近的备份文件
    echo ""
    echo "📁 最近的备份文件:"
    if ls "$MEMOS_DIR/memos_data/backups/"memos_backup_*.tar.gz 1> /dev/null 2>&1; then
        ls -lht "$MEMOS_DIR/memos_data/backups/"memos_backup_*.tar.gz | head -3
    else
        echo "   未找到备份文件"
    fi
}

# 主函数
main() {
    case "${1:-check}" in
        check)
            interactive_backup_reminder
            ;;
        boot)
            boot_check
            ;;
        shutdown)
            shutdown_check
            ;;
        force)
            log_info "强制执行备份..."
            if "$BACKUP_SCRIPT"; then
                update_last_backup_time
                log_success "强制备份完成"
            else
                log_error "强制备份失败"
                exit 1
            fi
            ;;
        status)
            show_status
            ;;
        setup-auto)
            setup_autostart
            ;;
        remove-auto)
            remove_autostart
            ;;
        help|--help|-h)
            show_help
            ;;
        *)
            log_error "未知命令: $1"
            show_help
            exit 1
            ;;
    esac
}

# 运行主函数
main "$@"