#!/bin/bash

BACKUP_DIR="/data/chroma/backups"
DATE=$(date +%Y%m%d_%H%M%S)
BACKUP_FILE="chroma_backup_$DATE.tar.gz"

echo "======================================"
echo "  Chroma 数据备份脚本"
echo "======================================"
echo ""

mkdir -p $BACKUP_DIR

echo "[1/3] 创建备份..."
docker exec chroma_server tar -czf /tmp/$BACKUP_FILE -C /chroma chroma_data

echo "[2/3] 复制备份文件..."
docker cp chroma_server:/tmp/$BACKUP_FILE $BACKUP_DIR/

echo "[3/3] 清理旧备份..."
find $BACKUP_DIR -name "*.tar.gz" -mtime +7 -delete

BACKUP_SIZE=$(du -h $BACKUP_DIR/$BACKUP_FILE | cut -f1)

echo ""
echo "======================================"
echo "  备份完成！"
echo "======================================"
echo ""
echo "备份文件: $BACKUP_DIR/$BACKUP_FILE"
echo "文件大小: $BACKUP_SIZE"
echo ""