#!/usr/bin/env bash
set -e
WORKDIR="$(cd "$(dirname "$0")" && pwd)"
cd "$WORKDIR"

# 加载本机敏感环境变量（如短信密钥），不存在则跳过（该文件不入库）
[ -f .smssend ] && source .smssend
# 加载本机部署环境（如 BASEURL：生成物/媒体文件的公网回源地址前缀），不入库
[ -f .pipelineenv ] && source .pipelineenv

# 运行模式：all(默认，HTTP+poller) / web(仅HTTP，无poller) / worker(仅poller，分布式worker节点)
# 多 worker 本机/跨机测试：bash start.sh worker <id>（id=1,2,3...，端口 9090+id，pid 独立）
MODE="${1:-all}"
case "$MODE" in
  web)    PORT=9090; PIDFILE="pipeline-web.pid"; PIPELINE_MODE=web ;;
  worker)
      WID="${2:-1}"
      PORT=$((9090 + WID))
      PIDFILE="pipeline-worker-${WID}.pid"
      PIPELINE_MODE=worker ;;
  all)    PORT=9090; PIDFILE="pipeline.pid"; PIPELINE_MODE=all ;;
  *) echo "用法: $0 [web|worker <id>|all]"; exit 1 ;;
esac
export PIPELINE_MODE

if [ -f "$PIDFILE" ]; then
    pid=$(cat "$PIDFILE")
    if kill -0 "$pid" 2>/dev/null; then
        echo "Already running (PID $pid)"
        exit 0
    fi
    rm -f "$PIDFILE"
fi

echo "Starting pipeline-app ($MODE mode) on port $PORT..."
export PATH="$WORKDIR/bin:$PATH"
$WORKDIR/py3/bin/python $WORKDIR/app/pipeline_app.py -p $PORT -w $WORKDIR >> $WORKDIR/logs/pipeline.log 2>&1 &
echo $! > "$PIDFILE"
echo "Started PID $(cat $PIDFILE)"

# 独立异步记账进程（模型用量出账：客户付/商户营收/供应商成本）
# 只随主进程/worker 启动；web 单独模式不启动（同步记账在 web 内）
if [ "$MODE" != "web" ]; then
    ACC_PIDFILE="pipeline-accounting.pid"
    if [ -f "$ACC_PIDFILE" ]; then
        acc_pid=$(cat "$ACC_PIDFILE")
        if kill -0 "$acc_pid" 2>/dev/null; then
            echo "Accounting worker already running (PID $acc_pid)"
        else
            rm -f "$ACC_PIDFILE"
        fi
    fi
    if [ ! -f "$ACC_PIDFILE" ]; then
        $WORKDIR/py3/bin/python $WORKDIR/app/pipeline_accounting_worker.py -w $WORKDIR >> $WORKDIR/logs/pipeline-accounting.log 2>&1 &
        echo $! > "$ACC_PIDFILE"
        echo "Started accounting worker PID $(cat $ACC_PIDFILE)"
    fi
fi