Bash用户管理脚本优化及错误提示改进求助

Hi there! 我帮你梳理下脚本里的问题，同时给出优化方案和实践建议：

问题描述 #

作为Bash脚本新手，我开发了一款Linux用户增删脚本，核心功能如下：

• -a <用户名>：新增用户，可选搭配-p <密码>设置密码、-s <Shell>指定登录Shell（默认/bin/bash）
• -d <用户名>：删除用户
• -h：查看帮助文档

基础的增删功能可以正常运行，但存在两个问题：

1. 如果未输入-a标志，却直接使用-p或-s参数，脚本会直接退出，没有明确的错误提示
2. 希望得到脚本的整体优化建议，以及指出其中的不良编程实践

现有脚本 #

#!/bin/bash
## checking if the user is privileged or not
if [[ $EUID != 0 ]]
then
 echo "Script has to be ran as root or sudo"
 echo "Aborting"
 exit 101
fi
## creating help functions
function usage() {
 echo "usage: ${0} -a <user> -p <password> -s <shell> | ${0} -d <user> | ${0} -h"
}
function help() {
 echo "$0 - Script to add of remove users"
 echo "-a - Add a new user"
 echo " -p - Set password while creating user if not mentioned will not set any password by default"
 echo " -s - Set a shell for the user default is /bin/bash if none specified"
 echo "-a - Remove a user" # 这里写错了，应该是-d
 echo "-h - Print this help text"
}
if [[ "$#" -lt "1" ]]; then
 echo "Argument has to be provided see $0 -h"
fi
shell=/bin/bash
password=$(openssl rand -base64 32)
while getopts :a:d:h opt; do
 case $opt in
 a) user=$OPTARG
 while getopts :p:s: test do # 嵌套getopts是错误用法，会导致参数解析混乱
 case $test in
 p) password=$OPTARG;;
 s) shell=$OPTARG;;
 /?) echo "The provided flag is not identified see $0 -h"
 exit;;
 :) echo "$OPTARG requires arguments see $0 -h"
 exit;;
 esac
 done
 if [[ "$1" != "-a" ]]
 then
 echo "You have to specify username using -a flag see $0 -h"
 fi
 useradd -m $user -s $shell
 echo "$user":"$password" | chpasswd
 echo "The password for the $user is $password";;
 d) userdel -f $OPTARG
 if [[ $? == 0 ]]
 then
 echo "user has been removed"
 else
 echo "There was some error removing the user"
 fi;;
 h) help
 exit;;
 /?) echo "$OPTARG option not valid";;
 :) echo "$OPTARG requires argument";;
 esac
done

核心问题分析 #

1. 嵌套getopts的错误用法：你在-a的case里嵌套了另一个getopts，这会导致参数解析逻辑混乱。getopts应该一次性处理所有命令行参数，而不是分段处理，这也是单独使用-p/-s时脚本无提示退出的根本原因。
2. Help函数错误：帮助文本里把删除用户的标志写成了-a，应该是-d，会误导用户。
3. 变量未加引号：比如useradd -m $user -s $shell，如果用户名或Shell路径包含空格（虽然不推荐，但要兼容），会导致命令执行失败。
4. 默认密码生成时机不当：脚本一开始就生成了默认密码，即使用户执行的是删除或查看帮助操作，完全没必要，应该放到新增用户的逻辑里。
5. 错误提示模糊：参数错误时没有明确指出具体的无效参数，用户很难快速定位问题。
6. 未处理剩余无效参数：如果用户输入了未定义的参数（比如-x），脚本不会给出提示，直接忽略。

优化后的脚本 #

#!/bin/bash
set -euo pipefail # 开启严格模式：遇到错误退出、未定义变量报错、管道失败则脚本退出

# 检查权限
if [[ $EUID -ne 0 ]]; then
    echo "错误：必须以root或sudo身份运行此脚本" >&2 # 错误信息输出到stderr
    exit 101
fi

# 帮助信息函数
show_help() {
    cat << EOF
$0 - Linux用户增删管理脚本

用法：
  $0 -a <用户名> [-p <密码>] [-s <Shell路径>]
  $0 -d <用户名>
  $0 -h

选项：
  -a <用户名>   新增用户
    -p <密码>   为新增用户设置密码（可选，默认生成随机密码）
    -s <Shell>  指定用户登录Shell（可选，默认/bin/bash）
  -d <用户名>   删除用户（添加-r参数可同时删除家目录，当前未启用）
  -h            显示此帮助信息
EOF
}

# 初始化变量
user=""
shell="/bin/bash"
password=""
action=""

# 解析命令行参数
while getopts "a:d:hp:s:" opt; do
    case $opt in
        a)
            action="add"
            user="$OPTARG"
            ;;
        d)
            action="delete"
            user="$OPTARG"
            ;;
        h)
            show_help
            exit 0
            ;;
        p)
            password="$OPTARG"
            ;;
        s)
            # 验证Shell路径是否存在
            if [[ ! -f "$shell" ]]; then
                echo "错误：指定的Shell路径 $shell 不存在" >&2
                exit 1
            fi
            shell="$OPTARG"
            ;;
        \?)
            echo "错误：无效选项 -$OPTARG" >&2
            show_help >&2
            exit 1
            ;;
        :)
            echo "错误：选项 -$OPTARG 需要参数" >&2
            show_help >&2
            exit 1
            ;;
    esac
done

# 检查必要参数
if [[ -z "$action" ]]; then
    echo "错误：必须指定操作类型（-a 新增或 -d 删除）" >&2
    show_help >&2
    exit 1
fi

if [[ -z "$user" ]]; then
    echo "错误：必须指定用户名" >&2
    show_help >&2
    exit 1
fi

# 执行新增用户操作
if [[ "$action" == "add" ]]; then
    # 生成随机密码（如果用户未指定）
    if [[ -z "$password" ]]; then
        password=$(openssl rand -base64 32)
    fi

    # 验证用户名是否符合规范（简单检查：不能以数字开头，只能包含字母、数字、下划线、减号）
    if [[ ! "$user" =~ ^[a-zA-Z_][a-zA-Z0-9_-]*$ ]]; then
        echo "错误：用户名 $user 不符合Linux规范" >&2
        exit 1
    fi

    # 创建用户
    if useradd -m -s "$shell" "$user"; then
        # 设置密码
        echo "$user:$password" | chpasswd
        echo "成功创建用户 $user"
        echo "用户密码：$password"
    else
        echo "错误：创建用户 $user 失败" >&2
        exit 1
    fi
fi

# 执行删除用户操作
if [[ "$action" == "delete" ]]; then
    # 检查用户是否存在
    if ! id "$user" &>/dev/null; then
        echo "错误：用户 $user 不存在" >&2
        exit 1
    fi

    # 删除用户（如果需要删除家目录，把userdel改为userdel -r）
    if userdel "$user"; then
        echo "成功删除用户 $user"
    else
        echo "错误：删除用户 $user 失败" >&2
        exit 1
    fi
fi

# 处理剩余的无效参数
shift $((OPTIND -1))
if [[ $# -gt 0 ]]; then
    echo "错误：无效的参数 $*" >&2
    show_help >&2
    exit 1
fi

不良编程实践指正与优化建议 #

1. 避免嵌套getopts：getopts是用于一次性解析所有命令行参数的工具，嵌套使用会破坏参数解析的逻辑，导致异常行为。
2. 开启严格模式：添加set -euo pipefail可以让脚本在遇到错误、未定义变量或管道失败时立即退出，避免隐藏的逻辑错误。
3. 变量始终加引号：所有变量引用都用双引号包裹（比如"$user"），防止空格或特殊字符导致的命令解析错误。
4. 错误信息输出到stderr：用>&2将错误信息输出到标准错误流，而不是标准输出，符合UNIX命令的规范，方便用户区分正常输出和错误信息。
5. 验证输入合法性：添加用户名格式检查、Shell路径存在性检查、用户是否存在的检查，避免执行无效操作。
6. 优化帮助信息：使用cat << EOF编写多行帮助文本，更易读且易维护，同时修正了原脚本中的错误描述。
7. 合理的退出码：使用标准的退出码（比如1表示一般错误，101表示权限不足），方便脚本被其他工具调用时判断执行结果。
8. 避免过度使用-f强制选项：原脚本中userdel -f会强制删除用户，即使用户当前处于登录状态，可能导致数据丢失或系统异常，建议先检查用户是否在线，再执行删除操作。
9. 延迟变量初始化：默认密码只在需要新增用户且未指定密码时生成，避免不必要的计算。

内容的提问来源于stack exchange，提问作者Abhijith