#!/bin/bash
INDEX=$1
spider_code=$2
PUBLIC_DIR="${3:-/shared}"
username="user$INDEX"
export DISPLAY=:$INDEX
echo "当前显示桌面:$DISPLAY"

echo "正在启动VNC服务"
sudo su - "$username" -c "vncserver :$INDEX -geometry 1280x800 -depth 24 -localhost no"
echo "授权桌面访问"
sudo su - "$username" -c "export DISPLAY=:$INDEX && xhost +" 2>/dev/null
echo "等待 VNC 启动"
sleep 3

start_spider(){
  cd $PUBLIC_DIR/learn-spider
  echo "$PUBLIC_DIR/learn-spider"
  echo "正在启动爬虫$spider_code"
  sudo -E ./start_spider.sh $spider_code  # -E 保留当前环境变量
}

echo "启动爬虫"
start_spider
echo "爬虫执行完毕,20秒后关闭VNC"
sleep 20
sudo su - "$username" -c "vncserver -kill :$INDEX" 2>/dev/null