#!/bin/bash
# 导出行课数据 wrapper 脚本
# 用法: ./scripts/export_user_data.sh --user-id 33123
#       ./scripts/export_user_data.sh --user-ids 33123,33124
#       ./scripts/export_user_data.sh --account-ids 2148
#       ./scripts/export_user_data.sh --user-id 33123 --output-dir ./my_output/

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
WORKSPACE="$(dirname "$SCRIPT_DIR")"

# 数据库凭证（从 secrets.md）
export MYSQL_HOST="${MYSQL_HOST:-bj-cdb-8frbdwju.sql.tencentcdb.com}"
export MYSQL_USERNAME="${MYSQL_USERNAME:-read_only}"
export MYSQL_PASSWORD='fdsfiidier^$*hjfdijjd232'
export MYSQL_PORT="${MYSQL_PORT:-25413}"

export MYSQL_HOST_online="${MYSQL_HOST_online:-bj-cdb-dh2fkqa0.sql.tencentcdb.com}"
export MYSQL_USERNAME_online="${MYSQL_USERNAME_online:-read_only}"
export MYSQL_PASSWORD_online='fsdo45ijfmfmuu77$%^&'
export MYSQL_PORT_online="${MYSQL_PORT_online:-27751}"

export PG_DB_HOST="${PG_DB_HOST:-bj-postgres-16pob4sg.sql.tencentcdb.com}"
export PG_DB_PORT="${PG_DB_PORT:-28591}"
export PG_DB_USER="${PG_DB_USER:-ai_member}"
export PG_DB_PASSWORD='LdfjdjL83h3h3^$&**YGG*'
export PG_DB_DATABASE="${PG_DB_DATABASE:-vala}"

export ES_HOST="${ES_HOST:-es-7vd7jcu9.public.tencentelasticsearch.com}"
export ES_PORT="${ES_PORT:-9200}"
export ES_SCHEME="${ES_SCHEME:-https}"
export ES_USER="${ES_USER:-elastic}"
export ES_PASSWORD='F%?QDcWes7N2WTuiYD11'

# 默认输出目录
DEFAULT_OUTPUT="${WORKSPACE}/output"
mkdir -p "$DEFAULT_OUTPUT"

# 如果未指定 --output-dir，自动加默认值
HAS_OUTPUT=false
for arg in "$@"; do
    if [[ "$arg" == "--output-dir" ]]; then
        HAS_OUTPUT=true
        break
    fi
done

PYTHON_SCRIPT="${WORKSPACE}/git_repos/llm_offline_production/config_user_data_extract_and_analyze/export_user_id_data.py"

if [ "$HAS_OUTPUT" = false ]; then
    exec python3 "$PYTHON_SCRIPT" --output-dir "$DEFAULT_OUTPUT" "$@"
else
    exec python3 "$PYTHON_SCRIPT" "$@"
fi