#!/bin/bash
set -e

echo "============================================="
echo "  requests + urllib3 离线安装"
echo "============================================="

# 检查文件是否存在
if [ ! -f "urllib3-1.26.9-py2.py3-none-any.whl" ]; then
    echo "ERROR: 缺少 urllib3-1.26.9-py2.py3-none-any.whl"
    exit 1
fi

if [ ! -f "requests-2.27.1-py2.py3-none-any.whl" ]; then
    echo "ERROR: 缺少 requests-2.27.1-py2.py3-none-any.whl"
    exit 1
fi

# 自动获取 Python 库路径
PY_SITE=$(python2 -c "import site; print(site.getsitepackages()[0])" 2>/dev/null || \
python2 -c "import distutils.sysconfig; print(distutils.sysconfig.get_python_lib())")

echo "Python 库路径: $PY_SITE"

# 临时目录
TMP=$(mktemp -d)

# ==================== 安装 urllib3 ====================
echo ">>> 安装 urllib3..."
unzip -q -o urllib3-1.26.9-py2.py3-none-any.whl -d "$TMP"
cp -r "$TMP/urllib3" "$PY_SITE/"

# ==================== 安装 requests ====================
echo ">>> 安装 requests..."
unzip -q -o requests-2.27.1-py2.py3-none-any.whl -d "$TMP"
cp -r "$TMP/requests" "$PY_SITE/"

# 权限（产线必须）
chmod -R 755 "$PY_SITE/urllib3"
chmod -R 755 "$PY_SITE/requests"

rm -rf "$TMP"

echo "============================================="
echo " ✅ 安装完成！"
echo "============================================="