#!/bin/bash
source ~/.bash_profile

# ============================================================================
# 配置区域 - 可根据实际情况修改
# ============================================================================

# 数据路径配置
# 修改这个路径指向实际的session文件位置
SESSION_DATA_DIR="../output"

# Swing算法参数
ALPHA=0.7          # Swing算法的alpha参数
THRESHOLD1=1       # 交互强度阈值1
THRESHOLD2=3       # 交互强度阈值2
THREAD_NUM=4       # 线程数
SHOW_PROGRESS=1    # 是否显示进度 (0/1)

# Python环境（如果需要特定的Python环境，在这里配置）
PYTHON_CMD="python3"

# ============================================================================
# 脚本执行区域
# ============================================================================

# 编译C++程序
echo "编译Swing程序..."
make
if [[ $? -ne 0 ]]; then
    echo "编译失败，退出"
    exit 1
fi

# 获取日期
DAY=`date +"%Y%m%d"`
# 如果需要使用特定日期，取消下面的注释
# DAY=20241017

echo "处理日期: ${DAY}"

# 清理旧的输出目录（365天前）和日志（180天前）
find . -type d -name 'output_*' -ctime +365 -exec rm -rf {} \; 2>/dev/null
mkdir -p logs
find logs/ -type f -mtime +180 -exec rm -f {} \; 2>/dev/null

# 创建输出目录
output_dir=output_${DAY}
mkdir -p ${output_dir}

# 确定session文件路径
# 优先使用带日期的文件，如果不存在则使用.cpp格式的文件
SESSION_FILE="${SESSION_DATA_DIR}/session.txt.${DAY}.cpp"
if [[ ! -f ${SESSION_FILE} ]]; then
    SESSION_FILE="${SESSION_DATA_DIR}/session.txt.${DAY}"
fi

if [[ ! -f ${SESSION_FILE} ]]; then
    echo "错误: Session文件不存在: ${SESSION_FILE}"
    echo "请先运行 generate_session.py 生成session文件"
    exit 1
fi

echo "使用session文件: ${SESSION_FILE}"
echo "Swing参数: alpha=${ALPHA}, threshold1=${THRESHOLD1}, threshold2=${THRESHOLD2}, threads=${THREAD_NUM}"

# 运行Swing算法
# 如果session文件格式是 "uid \t json"，需要用cut -f 2提取json部分
# 如果session文件格式是纯json（每行一个），直接cat即可
echo "开始运行Swing算法..."
if grep -q $'\t' ${SESSION_FILE}; then
    # 包含tab，需要提取第二列
    echo "检测到session文件包含uid，提取json部分..."
    cat ${SESSION_FILE} | cut -f 2 | bin/swing ${ALPHA} ${THRESHOLD1} ${THRESHOLD2} ${THREAD_NUM} ${output_dir} ${SHOW_PROGRESS}
else
    # 纯json格式
    echo "检测到session文件为纯json格式..."
    cat ${SESSION_FILE} | bin/swing ${ALPHA} ${THRESHOLD1} ${THRESHOLD2} ${THREAD_NUM} ${output_dir} ${SHOW_PROGRESS}
fi

# 检查Swing算法是否成功执行
if [[ $? -eq 0 ]]; then
    echo "Swing算法执行成功"
    
    # 更新软链接指向最新输出
    if [[ -e output ]]; then
        rm -rf output
    fi
    ln -s "${output_dir}" output
    echo "软链接已更新为指向 ${output_dir}"
    
    # 合并结果文件
    echo "合并结果文件..."
    cat output/sim_matrx.* > output/swing_similar.txt
    echo "结果已合并到 output/swing_similar.txt"
    
    # 生成可读的debug文件（添加商品名称）
    echo "生成可读的debug文件..."
    DEBUG_SCRIPT="../scripts/add_names_to_swing.py"
    
    if [[ -f ${DEBUG_SCRIPT} ]]; then
        ${PYTHON_CMD} ${DEBUG_SCRIPT} output/swing_similar.txt output/swing_similar_readable.txt --debug
        
        if [[ $? -eq 0 ]]; then
            echo "Debug文件已生成: output/swing_similar_readable.txt"
        else
            echo "警告: 生成debug文件失败，但Swing结果已保存"
        fi
    else
        echo "警告: Debug脚本不存在: ${DEBUG_SCRIPT}"
        echo "跳过生成可读文件"
    fi
    
else
    echo "Swing算法执行失败，未更新软链接"
    exit 1
fi

# ============================================================================
# 用户协同过滤（UCF）- 可选
# ============================================================================

# 如果需要运行UCF，取消下面的注释
# echo "运行用户协同过滤..."
# # 仅使用最新的5万条数据
# tail -n 50000 ${SESSION_FILE} > output/ucf.input
# python3 src/ucf.py output/ucf.input output/ucf.txt

echo "全部完成！"
echo "结果文件:"
echo "  - Swing相似度: ${output_dir}/swing_similar.txt"
echo "  - Swing可读版: ${output_dir}/swing_similar_readable.txt"