#!/bin/bash

echo "🧪 测试data目录功能"
echo "=================================="
echo ""

API_URL="http://10.10.10.77:8001/api"

echo "1. 测试获取文件列表..."
curl -s "${API_URL}/datasets/files" | python3 -c "
import json, sys
data = json.load(sys.stdin)
print(f'✅ 共 {data[\"data\"][\"total\"]} 个文件')
print('')
for f in data['data']['files']:
    print(f'  📄 {f[\"filename\"]} - {f[\"size_human\"]}')
"
echo ""

echo "2. 测试上传新文件..."
cat > /tmp/test_upload.json << 'INNER_EOF'
[{"text": "测试上传", "label": "test"}]
INNER_EOF

curl -s -X POST "${API_URL}/datasets/upload" \
  -F "file=@/tmp/test_upload.json" \
  -F "description=测试data目录上传" | python3 -c "
import json, sys
data = json.load(sys.stdin)
if data['code'] == 200:
    print('✅ 文件上传成功')
    print(f'  保存路径: {data[\"data\"][\"dataset\"][\"saved_path\"]}')
else:
    print('❌ 上传失败')
"
echo ""

echo "3. 再次获取文件列表..."
curl -s "${API_URL}/datasets/files" | python3 -c "
import json, sys
data = json.load(sys.stdin)
print(f'✅ 当前共 {data[\"data\"][\"total\"]} 个文件')
"

echo ""
echo "=================================="
echo "✅ 测试完成！"
echo ""
echo "📁 data目录位置："
echo "   /data/code/FT_Platform/YG_FT_Platform/data/"
echo ""
echo "🔍 查看文件："
echo "   ls -lh /data/code/FT_Platform/YG_FT_Platform/data/"