#!/bin/bash

echo "======================================================================"
echo "测试修复是否成功"
echo "======================================================================"
echo ""

cd /home/tw/recommendation/offline_tasks

# 测试 1: 检查文件结构
echo "[测试 1] 检查文件结构..."
if [ -f "db_service.py" ]; then
    echo "  ✓ db_service.py 在 offline_tasks/ 根目录"
else
    echo "  ✗ db_service.py 未找到"
    exit 1
fi

if [ -f "config/offline_config.py" ]; then
    echo "  ✓ config/offline_config.py 存在"
else
    echo "  ✗ config/offline_config.py 未找到"
    exit 1
fi

if [ -f "deepwalk/deepwalk.py" ] && [ -f "deepwalk/alias.py" ]; then
    echo "  ✓ DeepWalk 文件已移动到 offline_tasks/deepwalk/"
else
    echo "  ✗ DeepWalk 文件未找到"
    exit 1
fi

# 测试 2: 检查 Python 语法
echo ""
echo "[测试 2] 检查 Python 脚本语法..."
python3 -m py_compile scripts/i2i_item_behavior.py 2>/dev/null
if [ $? -eq 0 ]; then
    echo "  ✓ i2i_item_behavior.py 语法正确"
else
    echo "  ✗ i2i_item_behavior.py 语法错误"
    exit 1
fi

python3 -m py_compile scripts/tag_category_similar.py 2>/dev/null
if [ $? -eq 0 ]; then
    echo "  ✓ tag_category_similar.py 语法正确"
else
    echo "  ✗ tag_category_similar.py 语法错误"
    exit 1
fi

python3 -m py_compile scripts/i2i_deepwalk.py 2>/dev/null
if [ $? -eq 0 ]; then
    echo "  ✓ i2i_deepwalk.py 语法正确"
else
    echo "  ✗ i2i_deepwalk.py 语法错误"
    exit 1
fi

python3 -m py_compile scripts/fetch_item_attributes.py 2>/dev/null
if [ $? -eq 0 ]; then
    echo "  ✓ fetch_item_attributes.py 语法正确"
else
    echo "  ✗ fetch_item_attributes.py 语法错误"
    exit 1
fi

python3 -m py_compile scripts/generate_session.py 2>/dev/null
if [ $? -eq 0 ]; then
    echo "  ✓ generate_session.py 语法正确"
else
    echo "  ✗ generate_session.py 语法错误"
    exit 1
fi

# 测试 3: 检查是否还有 sys.path hack
echo ""
echo "[测试 3] 检查是否清理了 sys.path hack..."
sys_path_count=$(grep -r "sys.path.append" scripts/*.py 2>/dev/null | wc -l)
if [ $sys_path_count -eq 0 ]; then
    echo "  ✓ 所有 sys.path hack 已清理"
else
    echo "  ⚠️  仍有 $sys_path_count 个文件包含 sys.path.append"
    grep -r "sys.path.append" scripts/*.py
fi

# 测试 4: 检查导入语句
echo ""
echo "[测试 4] 检查导入语句..."
if grep -q "^from db_service import" scripts/i2i_item_behavior.py; then
    echo "  ✓ i2i_item_behavior.py 正确导入 db_service"
else
    echo "  ✗ i2i_item_behavior.py 未导入 db_service"
    exit 1
fi

if grep -q "^from config.offline_config import" scripts/fetch_item_attributes.py; then
    echo "  ✓ fetch_item_attributes.py 正确导入 config.offline_config"
else
    echo "  ✗ fetch_item_attributes.py 未导入 config.offline_config"
    exit 1
fi

if grep -q "^from scripts.debug_utils import" scripts/fetch_item_attributes.py; then
    echo "  ✓ fetch_item_attributes.py 正确导入 scripts.debug_utils"
else
    echo "  ✗ fetch_item_attributes.py 未导入 scripts.debug_utils"
    exit 1
fi

if grep -q "^from deepwalk.deepwalk import DeepWalk" scripts/i2i_deepwalk.py; then
    echo "  ✓ i2i_deepwalk.py 正确导入 deepwalk.deepwalk.DeepWalk"
else
    echo "  ✗ i2i_deepwalk.py 未导入 deepwalk.deepwalk.DeepWalk"
    exit 1
fi

# 测试 5: 检查 run.sh 是否设置了 PYTHONPATH
echo ""
echo "[测试 5] 检查 run.sh 配置..."
if grep -q "export PYTHONPATH" run.sh; then
    echo "  ✓ run.sh 已设置 PYTHONPATH"
else
    echo "  ⚠️  run.sh 未设置 PYTHONPATH"
fi

echo ""
echo "======================================================================"
echo "✓ 所有测试通过！"
echo "======================================================================"
echo ""
echo "文件结构:"
echo "  offline_tasks/  (Python 根目录)"
echo "    ├── db_service.py"
echo "    ├── config/offline_config.py"
echo "    ├── deepwalk/deepwalk.py, alias.py"
echo "    ├── scripts/*.py"
echo "    └── run.sh (设置了 PYTHONPATH)"
echo ""
echo "现在可以运行："
echo "  cd /home/tw/recommendation/offline_tasks"
echo "  bash run.sh"
echo ""