#!/bin/bash
set -e

OFFLINE=/opt/k8s-offline

mkdir -p $OFFLINE
tar xf k8s-offline.tgz -C $OFFLINE

echo "[1] 安装 nvidia driver（离线）"
bash ./install_nvidia_driver.sh

echo "[2] 安装 kubeadm/kubelet/kubectl"
install -m755 $OFFLINE/offline-cache/bin/* /usr/local/bin/

echo "[3] 导入镜像"
for img in $OFFLINE/offline-cache/images/*.tar; do
    ctr -n=k8s.io images import "$img"
done

echo "[4] 加入集群"
kubeadm join {{ nodes.control_plane.ip }}:6443 \
  --token {{ join.token }} \
  --discovery-token-ca-cert-hash sha256:{{ join.hash }}

echo "[5] 自动部署 GPU Operator"
kubectl apply -f $OFFLINE/offline-cache/manifests/gpu-operator.yaml

echo "GPU 工作节点初始化完成"