#!/bin/bash

single_server_name=$1

go mod tidy
go mod vendor

# 定义一个函数来在每个服务器目录下运行 go run <server_name>.go
run_server() {
  server_name=$1
  
  # 导航到相应的目录
  cd server/$server_name
  go build 
  echo "build $server_name"

  # 如果之前存在相同名字的 screen 会话，先将其终止
  # 首先尝试关闭已存在的screen会话
  existing_session=$(screen -ls | grep -w "$server_name")
  if [ -n "$existing_session" ]; then
    echo "Terminating existing screen session for $server_name"
    screen -S "$server_name" -X quit
    while [[ $(screen -ls | grep "\.$server_name\s") ]]; do
      sleep 0.1s  # 等待0.1秒后再次检查
      echo "wait for $server_name"
    done
  fi

  # 循环检查screen进程是否存在

  [ -f .gitignore ] || echo $server_name > .gitignore
  # 使用 screen 运行 go run <server_name>.go

  echo "Running $server_name"
  screen -dmS $server_name -L  ./$server_name

  # 返回到上一级目录
  cd - > /dev/null
}

find /tmp/go-build* -mmin +5 -exec rm -rf {} +  
find /tmp/go-link* -mmin +5 -exec rm -rf {} +  

server_dirs=() # 初始化一个空数组

if [ -n "$single_server_name" ]; then
    server_dirs=("$single_server_name")
else
    for dir in server/*/ ; do # 遍历 "server/" 下的所有子目录
        dir=${dir%*/}      # 删除末尾的 "/"
        dir=${dir##*/}     # 删除开头的 "server/"
        server_dirs+=("$dir") # 添加到数组
    done
fi

# 在每个服务器目录下运行相应的 go 程序
for server_dir in "${server_dirs[@]}"; do
  run_server $server_dir 
done


if [ -n "$single_server_name" ]; then
    echo "no proxyserver restart"
else
    # 定义目录和screen名称
    dir_path="./proxyserver"
    screen_name="proxyserver"

    # 进入目录
    cd $dir_path
    # 检查是否存在screen session
    if screen -list | grep -q "$screen_name"; then
      # 结束存在的screen session
      screen -S $screen_name -X quit
    fi
    go build
    # 启动新的screen session并运行go程序
    screen -dmS $screen_name -L ./$screen_name
fi