#!/bin/bash

while true; do

   # 提示用户确认，并提供默认选项

   read -e -p "此操作将会删除用户及其所有配置([y]/n): " -i "y" confirm

   # 标准化输入为小写

   confirm=${confirm,,}  # 将输入转换为小写

   # 检查用户输入是否为 'yes' 或 'no'

   if [[ "$confirm" == "y" ]]; then

       echo "继续操作。"

       break  # 退出循环，继续执行后续代码

   elif [[ "$confirm" == "n" ]]; then

       echo "操作已取消。"

       exit 0

   else

       echo "无效输入，请输入y或n。"

   fi

done

# 删除现有用户（如果存在），并忽略不存在的错误

userdel -f -r pub 2>/dev/null

# 添加新用户

useradd -m pub || { echo "无法创建用户 'pub'"; exit 1; }

# 随机生成长密码并设置

new_password=$(tr -dc A-Za-z0-9 </dev/urandom | head -c 64)  # 生成 64 位随机密码

echo "pub:$new_password" | chpasswd || { echo "设置密码失败"; exit 1; }

echo "为用户pub设置的新密码是：$new_password"  # 显示密码

# 尝试从文件读取 SSH 公钥

if [ -f "LAN-ssh_pub" ]; then

   ssh_pub_key=$(<LAN-ssh_pub)

   echo "从文件读取SSH公钥成功。"

else

   read -p "请输入LAN的ssh公钥：" ssh_pub_key

fi

# 创建.ssh 目录并设置公钥

mkdir -p /home/pub/.ssh || { echo "无法创建SSH目录"; exit 1; }

echo "$ssh_pub_key" > /home/pub/.ssh/authorized_keys || { echo "无法写入公钥"; exit 1; }

chown -R pub:pub /home/pub/.ssh

# 创建和设置 ssh 登录脚本

cat > /home/pub/ssh_for_pub.sh << 'EOF'

#!/bin/bash

m=$(whoami)

if [[ -n $SSH_CONNECTION || $- == *i* ]]; then

   # 显示版权和开发者信息

   printf "\e[1;34m本脚本由 @joytion 开发。\e[0m\n"

   printf "\e[1;34m版权所有 © 2024 Joytion. 保留所有权利。\e[0m\n"

   echo -e "\e[01;33m** 您是特殊用户 'pub'，将启用定制的SSH登录模式 **\e[0m"

   # 请求用户输入SSH登录的用户名，并检查是否通过Ctrl+D中断

   read -p "请输入自己在内网的用户名: " ssh_user

   if [[ $? -ne 0 ]]; then

       echo -e "\e[1;31m输入中断，操作已取消。\e[0m"

       exit 1

   fi

   # 使用正则表达式清洗和验证用户名

   if [[ ! "$ssh_user" =~ ^[a-zA-Z0-9_-]{1,32}$ ]]; then

       echo -e "\e[1;31m无效的用户名。请使用1到32个字母、数字、下划线或破折号。\e[0m"

       exit 1

   fi

   # 通知用户即将尝试的操作

   echo -e "\e[1;32m正在尝试使用 SSH 通过 ${ssh_user} 登录内网...\e[0m"

   # 执行SSH命令，并检查是否成功

   ssh -p xxxx "${ssh_user}"@localhost

   if [[ $? -ne 0 ]]; then

       echo -e "\e[1;31mSSH 登录失败或被中断。\e[0m"

       exit 1

   fi

else

   echo -e "\e[1;31m本脚本仅限于SSH会话中使用。\e[0m"

fi

exit 0

EOF

chmod +x /home/pub/ssh_for_pub.sh || { echo "无法设置脚本权限"; exit 1; }

chown pub:pub /home/pub/ssh_for_pub.sh || { echo "无法设置脚本所有权"; exit 1; }

echo "pub用户创建成功！"