#!/bin/bash
# 清理测试用户脚本

echo "检查测试用户数量..."

# 统计测试用户
docker compose -f docker-compose.dev.yaml exec -T postgres psql -U postgres -d claude_code_hub -c "
SELECT
  COUNT(*) as 测试用户数量
FROM users
WHERE (name LIKE '测试用户%' OR name LIKE '%test%' OR name LIKE 'Test%')
  AND deleted_at IS NULL;
"

echo ""
echo "预览将要删除的用户（前 10 个）..."
docker compose -f docker-compose.dev.yaml exec -T postgres psql -U postgres -d claude_code_hub -c "
SELECT id, name, created_at
FROM users
WHERE (name LIKE '测试用户%' OR name LIKE '%test%' OR name LIKE 'Test%')
  AND deleted_at IS NULL
ORDER BY created_at DESC
LIMIT 10;
"

echo ""
read -p "确认删除这些测试用户吗？(y/N): " confirm

if [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]; then
  echo "开始清理..."

  # 软删除关联的 keys
  docker compose -f docker-compose.dev.yaml exec -T postgres psql -U postgres -d claude_code_hub -c "
  UPDATE keys
  SET deleted_at = NOW(), updated_at = NOW()
  WHERE user_id IN (
    SELECT id FROM users
    WHERE (name LIKE '测试用户%' OR name LIKE '%test%' OR name LIKE 'Test%')
      AND deleted_at IS NULL
  )
  AND deleted_at IS NULL;
  "

  # 软删除测试用户
  docker compose -f docker-compose.dev.yaml exec -T postgres psql -U postgres -d claude_code_hub -c "
  UPDATE users
  SET deleted_at = NOW(), updated_at = NOW()
  WHERE (name LIKE '测试用户%' OR name LIKE '%test%' OR name LIKE 'Test%')
    AND deleted_at IS NULL;
  "

  echo "清理完成！"
  echo ""
  echo "剩余用户统计："
  docker compose -f docker-compose.dev.yaml exec -T postgres psql -U postgres -d claude_code_hub -c "
  SELECT COUNT(*) as 总用户数 FROM users WHERE deleted_at IS NULL;
  "
else
  echo "取消清理"
fi