#!/bin/bash
ln -sf ~/palworld.sh /usr/local/bin/p

ip_address() {
ipv4_address=$(curl -s ipv4.ip.sb)
ipv6_address=$(curl -s --max-time 1 ipv6.ip.sb)
}


install() {
    if [ $# -eq 0 ]; then
        echo "未提供软件包参数!"
        return 1
    fi

    for package in "$@"; do
        if ! command -v "$package" &>/dev/null; then
            if command -v apt &>/dev/null; then
                apt update -y && apt install -y "$package"
            elif command -v yum &>/dev/null; then
                yum -y update && yum -y install "$package"
            elif command -v apk &>/dev/null; then
                apk update && apk add "$package"
            else
                echo "未知的包管理器!"
                return 1
            fi
        fi
    done

    return 0
}


remove() {
    if [ $# -eq 0 ]; then
        echo "未提供软件包参数!"
        return 1
    fi

    for package in "$@"; do
        if command -v apt &>/dev/null; then
            apt purge -y "$package"
        elif command -v yum &>/dev/null; then
            yum remove -y "$package"
        elif command -v apk &>/dev/null; then
            apk del "$package"
        else
            echo "未知的包管理器!"
            return 1
        fi
    done

    return 0
}


break_end() {
      echo -e "\033[0;32m操作完成\033[0m"
      echo "按任意键继续..."
      read -n 1 -s -r -p ""
      echo ""
      clear
}

palworld() {
            p
            exit
}


install_add_docker() {
    if [ -f "/etc/alpine-release" ]; then
        apk update
        apk add docker docker-compose
        rc-update add docker default
        service docker start
    else
        curl -fsSL https://get.docker.com | sh && ln -s /usr/libexec/docker/cli-plugins/docker-compose /usr/local/bin
        systemctl start docker
        systemctl enable docker
    fi
}

install_docker() {
    if ! command -v docker &>/dev/null; then
        install_add_docker
    else
        echo "Docker 已经安装"
    fi
}

pal_start() {
    ip_address
    tmux new -d -s my1 "docker exec -it steamcmd bash -c '/home/steam/Steam/steamapps/common/PalServer/PalServer.sh'"
    echo -e "\033[0;32m幻兽帕鲁服务启动啦！\033[0m"
    echo -e "\033[0;32m游戏下载地址: https://store.steampowered.com/app/1623730\033[0m"
    echo -e "\033[0;32m进入游戏连接:\033[93m $ipv4_address:8255 \033[0;32m开始冒险吧！\033[0m"

}

pal_backup() {
  cd ~
  curl -sS -O https://kejilion.pro/pal_backup.sh && chmod +x pal_backup.sh
}

pal_install_status() {
  CONTAINER_NAME="steamcmd"

  # 检查容器是否已安装
  if [ "$(docker ps -a -q -f name=$CONTAINER_NAME 2>/dev/null)" ]; then
      container_status="\e[32m幻兽帕鲁服务已安装\e[0m"  # 绿色
  else
      container_status="\e[90m幻兽帕鲁服务未安装\e[0m"  # 灰色
  fi

  SESSION_NAME="my1"

  ip_address
  # 检查 tmux 中是否存在指定的工作区
  if tmux has-session -t $SESSION_NAME 2>/dev/null; then
      tmux_status="\e[32m已开服:\033[93m $ipv4_address:8255\e[0m"  # 绿色
  else
      tmux_status="\e[90m未开服\e[0m"  # 灰色
  fi

}

while true; do
clear
pal_install_status
echo -e "\033[93m      .            .  ."
echo "._  _.|.    , _ ._.| _|"
echo "[_)(_]| \/\/ (_)[  |(_]"
echo "|                      "
echo -e "\033[96m幻兽帕鲁开服一键脚本工具v1.0.2  by KEJILION\033[0m"
echo -e "\033[96m-输入\033[93mp\033[96m可快速启动此脚本-\033[0m"
echo -e "$container_status $tmux_status"
echo "------------------------"
echo "1. 安装幻兽帕鲁服务"
echo "2. 开启幻兽帕鲁服务"
echo "3. 关闭幻兽帕鲁服务"
echo "4. 重启幻兽帕鲁服务"
echo "------------------------"
echo "5. 查看服务器状态"
echo "6. 设置虚拟内存"
echo "------------------------"
echo "7. 导出游戏存档"
echo "8. 导入游戏存档"
echo "9. 定时备份游戏存档"
echo "------------------------"
echo "10. 修改游戏配置"
echo "------------------------"
echo "11. 更新幻兽帕鲁服务"
echo "12. 卸载幻兽帕鲁服务"
echo "------------------------"
echo "k. 科技lion脚本工具箱"
echo "------------------------"
echo "00. 脚本更新"
echo "------------------------"
echo "0. 退出脚本"
echo "------------------------"
read -p "请输入你的选择: " choice

case $choice in
  1)
    clear
    install_docker
    install tmux
    docker run -dit --name steamcmd -p 8255:8211/udp --restart=always cm2network/steamcmd
    docker exec -it steamcmd bash -c "/home/steam/steamcmd/steamcmd.sh +login anonymous +app_update 2394010 validate +quit"
    clear
    pal_start
    ;;

  2)
    clear
    docker start steamcmd > /dev/null 2>&1
    pal_start
    ;;

  3)
    clear
    tmux kill-session -t my1
    docker stop steamcmd > /dev/null 2>&1
    echo -e "\033[0;32m幻兽帕鲁服务已关闭\033[0m"
    ;;

  4)
    clear
    tmux kill-session -t my1
    docker restart steamcmd > /dev/null 2>&1
    pal_start
    ;;

  5)
    clear
    install btop
    clear
    btop
    ;;

  6)
            clear
            swap_used=$(free -m | awk 'NR==3{print $3}')
            swap_total=$(free -m | awk 'NR==3{print $2}')

            if [ "$swap_total" -eq 0 ]; then
              swap_percentage=0
            else
              swap_percentage=$((swap_used * 100 / swap_total))
            fi

            swap_info="${swap_used}MB/${swap_total}MB (${swap_percentage}%)"

            echo "当前虚拟内存: $swap_info"

            read -p "是否调整大小?(Y/N): " choice

            case "$choice" in
              [Yy])
                # 输入新的虚拟内存大小
                read -p "请输入虚拟内存大小MB: " new_swap

                # 获取当前系统中所有的 swap 分区
                swap_partitions=$(grep -E '^/dev/' /proc/swaps | awk '{print $1}')

                # 遍历并删除所有的 swap 分区
                for partition in $swap_partitions; do
                  swapoff "$partition"
                  wipefs -a "$partition"  # 清除文件系统标识符
                  mkswap -f "$partition"
                  echo "已删除并重新创建 swap 分区: $partition"
                done

                # 确保 /swapfile 不再被使用
                swapoff /swapfile

                # 删除旧的 /swapfile
                rm -f /swapfile

                # 创建新的 swap 分区
                dd if=/dev/zero of=/swapfile bs=1M count=$new_swap
                chmod 600 /swapfile
                mkswap /swapfile
                swapon /swapfile

                if [ -f /etc/alpine-release ]; then
                    echo "/swapfile swap swap defaults 0 0" >> /etc/fstab
                    echo "nohup swapon /swapfile" >> /etc/local.d/swap.start
                    chmod +x /etc/local.d/swap.start
                    rc-update add local
                else
                    echo "/swapfile swap swap defaults 0 0" >> /etc/fstab
                fi

                echo "虚拟内存大小已调整为${new_swap}MB"
                ;;
              [Nn])
                echo "已取消"
                ;;
              *)
                echo "无效的选择，请输入 Y 或 N。"
                ;;
            esac
    ;;

  7)
    clear
    mkdir -p /home/game
    docker cp steamcmd:/home/steam/Steam/steamapps/common/PalServer/Pal/Saved/ /home/game/palworld/ > /dev/null 2>&1
    cd /home/game && tar czvf palworld_$(date +"%Y%m%d%H%M%S").tar.gz palworld > /dev/null 2>&1
    rm -rf /home/game/palworld/
    echo -e "\033[0;32m游戏存档已导出存放在: /home/game/\033[0m"
    ;;
  8)
    clear
    tmux kill-session -t my1
    docker exec -it steamcmd bash -c "rm -rf /home/steam/Steam/steamapps/common/PalServer/Pal/Saved/*"
    cd /home/game/ && ls -t /home/game/*.tar.gz | head -1 | xargs -I {} tar -xzf {}
    docker cp /home/game/palworld/Config steamcmd:/home/steam/Steam/steamapps/common/PalServer/Pal/Saved/Config > /dev/null 2>&1
    docker cp /home/game/palworld/ImGui steamcmd:/home/steam/Steam/steamapps/common/PalServer/Pal/Saved/ImGui > /dev/null 2>&1
    docker cp /home/game/palworld/SaveGames steamcmd:/home/steam/Steam/steamapps/common/PalServer/Pal/Saved/SaveGames > /dev/null 2>&1
    docker exec -it -u root steamcmd bash -c "chmod -R 777 /home/steam/Steam/steamapps/common/PalServer/Pal/Saved/"
    rm -rf /home/game/palworld/
    echo -e "\033[0;32m游戏存档已导入\033[0m"
    docker restart steamcmd > /dev/null 2>&1
    pal_start
    ;;

  9)
    clear
    echo "幻兽帕鲁游戏存档定时备份"
    echo "------------------------"
    echo "1. 每周备份       2. 每天备份       3. 每小时备份"
    echo "------------------------"
    read -p "请输入你的选择: " dingshi
    case $dingshi in
        1)
            pal_backup
            (crontab -l ; echo "0 0 * * 1 ./pal_backup.sh") | crontab - > /dev/null 2>&1
            echo "每周一备份，已设置"

            ;;
        2)
            pal_backup
            (crontab -l ; echo "0 3 * * * ./pal_backup.sh") | crontab - > /dev/null 2>&1
            echo "每天凌晨3点备份，已设置"

            ;;
        3)
            pal_backup
            (crontab -l ; echo "0 * * * * ./pal_backup.sh") | crontab - > /dev/null 2>&1
            echo "每小时整点备份，已设置"

            ;;
        *)
            echo "已取消"
            ;;
    esac
    ;;

  10)
    clear
    tmux kill-session -t my1
    cd ~ && curl -sS -O https://kejilion.pro/PalWorldSettings.ini

    echo "配置游戏参数"
    echo "------------------------"
    read -p "设置加入的密码（回车默认无密码）: " server_password
    read -p "设置游戏难度: （1. 简单    2. 普通    3. 困难）:" Difficulty
      case $Difficulty in
        1)
            Difficulty=1
            ;;

        2)
            Difficulty=2
            ;;
        3)
            Difficulty=3
            ;;
        *)
            echo "-默认设置为普通难度"
            Difficulty=2
            ;;
      esac

    read -p "经验值倍率: （回车默认1倍）:" exp_rate
      ExpRate=${exp_rate:-1}
    read -p "死亡后掉落设置: （1. 掉落    2. 不掉落）:" DeathPenalty
      case $DeathPenalty in
        1)
            DeathPenalty=All
            ;;

        2)
            DeathPenalty=None
            ;;
        *)
            DeathPenalty=All
            echo "-默认设置为掉落"
            ;;
      esac

    read -p "设置pvp模式: （1. 开启    2. 关闭）:" pal_pvp

      case $pal_pvp in
        1)
            pal_pvp=True
            ;;
        2)
            pal_pvp=False
            ;;
        *)
            pal_pvp=False
            echo "-默认关闭pvp模式"
            ;;
      esac

    # 更新配置文件
    sed -i "s/ServerPassword=\"\"/ServerPassword=\"$server_password\"/" ~/PalWorldSettings.ini
    sed -i "s/Difficulty=2/Difficulty=$Difficulty/" ~/PalWorldSettings.ini
    sed -i "s/ExpRate=1.000000/ExpRate=$ExpRate/" ~/PalWorldSettings.ini
    sed -i "s/DeathPenalty=All/DeathPenalty=$DeathPenalty/" ~/PalWorldSettings.ini
    sed -i "s/bEnablePlayerToPlayerDamage=False/bEnablePlayerToPlayerDamage=$pal_pvp/" ~/PalWorldSettings.ini
    sed -i "s/bIsPvP=False/bIsPvP=$pal_pvp/" ~/PalWorldSettings.ini
    echo "------------------------"
    echo "配置文件已更新"

    docker exec -it steamcmd bash -c "rm -f /home/steam/Steam/steamapps/common/PalServer/Pal/Saved/Config/LinuxServer/PalWorldSettings.ini"
    docker cp ~/PalWorldSettings.ini steamcmd:/home/steam/Steam/steamapps/common/PalServer/Pal/Saved/Config/LinuxServer/ > /dev/null 2>&1
    docker exec -it -u root steamcmd bash -c "chmod -R 777 /home/steam/Steam/steamapps/common/PalServer/Pal/Saved/"
    rm -f ~/PalWorldSettings.ini
    echo -e "\033[0;32m游戏配置已导入\033[0m"
    docker restart steamcmd > /dev/null 2>&1
    pal_start
    ;;


  11)
    clear
    tmux kill-session -t my1
    docker restart steamcmd > /dev/null 2>&1
    docker exec -it steamcmd bash -c "/home/steam/steamcmd/steamcmd.sh +login anonymous +app_update 2394010 validate +quit"
    clear
    echo -e "\033[0;32m幻兽帕鲁已更新\033[0m"
    pal_start
    ;;

  12)
    clear
    docker rm -f steamcmd
    docker rmi -f cm2network/steamcmd
    ;;

  k)
    cd ~
    curl -sS -O https://kejilion.pro/kejilion.sh && chmod +x kejilion.sh && ./kejilion.sh
    exit
    ;;

  00)
    cd ~
    curl -sS -O https://kejilion.pro/pal_log.sh && chmod +x pal_log.sh && ./pal_log.sh
    rm pal_log.sh
    echo ""
    curl -sS -O https://kejilion.pro/palworld.sh && chmod +x palworld.sh
    echo "脚本已更新到最新版本！"
    break_end
    palworld
    ;;


  0)
    clear
    exit
    ;;

  *)
    echo "无效的输入!"
    ;;
esac
    break_end
done