#!/usr/bin/env bash
# 将 datasets/ 与 algorithms/*/code 软链到外部 workspace（可选，覆盖仓库内嵌副本）
#
# 适用：本地 monorepo 布局 DATA/{HSAP,workspace}
# 克隆单仓库且无外置数据时无需运行本脚本。
#
# 用法:
#   export AS_WORKSPACE_ROOT=/path/to/workspace
#   bash scripts/setup_links.sh
set -euo pipefail

ROOT="$(cd "$(dirname "$0")/.." && pwd)"
WS="${AS_WORKSPACE_ROOT:-$(cd "$ROOT/.." && pwd)/workspace}"

if [[ ! -d "$WS/DMS" ]] && [[ ! -d "$WS/LaneDection" ]]; then
  echo "跳过: workspace 无效 ($WS)，继续使用仓库内嵌 algorithms/datasets"
  exit 0
fi

echo "HSAP: $ROOT"
echo "workspace:     $WS"

link_dir() {
  local target="$1" link="$2"
  if [[ -d "$link" && ! -L "$link" ]]; then
    echo "  备份内嵌目录 → ${link}.embedded.bak"
    mv "$link" "${link}.embedded.bak"
  elif [[ -L "$link" ]]; then
    rm -f "$link"
  fi
  mkdir -p "$(dirname "$link")"
  local rel link_parent
  link_parent="$(dirname "$link")"
  rel="$(python3 -c "import os.path; print(os.path.relpath('''$target''', '''$link_parent'''))")"
  ln -sfn "$rel" "$link"
  echo "  $link -> $rel"
}

mkdir -p "$WS/Lane" "$ROOT/datasets" "$ROOT/algorithms/dms_yolo" "$ROOT/algorithms/lane_ufld"

# workspace 内 Lane 逻辑入口（可选）
[[ -d "$WS/lane" ]] && ln -sfn "$WS/lane" "$WS/Lane/dataset" 2>/dev/null || true
[[ -d "$WS/LaneDection/Code" ]] && ln -sfn "$WS/LaneDection/Code" "$WS/Lane/code" 2>/dev/null || true

echo ">>> datasets/dms（保留仓库内 registry/scripts，仅软链 packs）"
DMS_ROOT="$ROOT/datasets/dms"
if [[ -L "$DMS_ROOT" ]]; then
  rm -f "$DMS_ROOT"
fi
if [[ ! -d "$DMS_ROOT" ]] && [[ -d "${DMS_ROOT}.embedded.bak" ]]; then
  mv "${DMS_ROOT}.embedded.bak" "$DMS_ROOT"
fi
mkdir -p "$DMS_ROOT/packs"
if [[ -d "$WS/DMS/DATASET/packs" ]]; then
  for pack in "$WS/DMS/DATASET/packs"/*; do
    [[ -e "$pack" ]] || continue
    name="$(basename "$pack")"
    link_dir "$pack" "$DMS_ROOT/packs/$name"
  done
fi
link_dir "$WS/lane" "$ROOT/datasets/lane"

echo ">>> algorithms"
link_dir "$WS/DMS/Code/yolo26_rknn_ultralytics-main" "$ROOT/algorithms/dms_yolo/code"
link_dir "$WS/LaneDection/Code" "$ROOT/algorithms/lane_ufld/code"

cat > "$ROOT/manifests/repo_layout.json" <<EOF
{
  "layout": "workspace_symlinks",
  "workspace": "$WS",
  "linked_at": "$(date -Iseconds)"
}
EOF

echo "完成（workspace 软链模式）"