mirror of
https://github.com/kuaifan/dootask.git
synced 2025-12-10 18:02:55 +00:00
100 lines
3.4 KiB
Bash
100 lines
3.4 KiB
Bash
#!/bin/sh
|
||
#
|
||
# 重置用户密码脚本
|
||
#
|
||
# 使用方法:
|
||
# ./repassword.sh [账号标识符] [自定义密码]
|
||
#
|
||
# 参数说明:
|
||
# [账号标识符]: 可选，可以是用户ID(纯数字)或邮箱地址。不提供时默认为第一个管理员用户
|
||
# [自定义密码]: 可选，指定要设置的新密码。不提供时会自动生成随机密码
|
||
#
|
||
# 使用示例:
|
||
# ./repassword.sh # 重置第一个管理员用户密码(随机生成)
|
||
# ./repassword.sh 123 # 重置ID=123的用户密码(随机生成)
|
||
# ./repassword.sh user@example.com # 重置邮箱为user@example.com的用户密码(随机生成)
|
||
# ./repassword.sh 123 newpass # 重置ID=123的用户密码为"newpass"
|
||
# ./repassword.sh user@example.com newpass # 重置邮箱为user@example.com的用户密码为"newpass"
|
||
#
|
||
|
||
account_identifier=$1
|
||
custom_password=$2
|
||
|
||
GreenBG="\033[42;37m"
|
||
RedBG="\033[41;37m"
|
||
Font="\033[0m"
|
||
|
||
# 生成随机密码
|
||
new_encrypt=$(date +%s%N | md5sum | awk '{print $1}' | cut -c 1-6)
|
||
if [ -z "$custom_password" ]; then
|
||
new_password=$(date +%s%N | md5sum | awk '{print $1}' | cut -c 1-16)
|
||
else
|
||
new_password=$custom_password
|
||
fi
|
||
md5_password=$(echo -n `echo -n $new_password | md5sum | awk '{print $1}'`$new_encrypt | md5sum | awk '{print $1}')
|
||
|
||
# 构建查询条件
|
||
if [ -z "$account_identifier" ]; then
|
||
# 默认查询第一个管理员
|
||
admin_query=$(echo "SELECT userid FROM ${MYSQL_PREFIX}users WHERE identity LIKE '%,admin,%' ORDER BY userid LIMIT 1;" | mysql -u$MYSQL_USER -p$MYSQL_PASSWORD $MYSQL_DATABASE)
|
||
identifier_value=$(echo "$admin_query" | sed -n '2p')
|
||
|
||
if [ -z "$identifier_value" ]; then
|
||
echo "${RedBG}错误：未找到管理员用户！${Font}"
|
||
exit 1
|
||
fi
|
||
|
||
where_field="userid"
|
||
identifier_type="管理员ID"
|
||
else
|
||
# 检查是否为纯数字（ID）
|
||
# 使用更兼容的 shell 语法检查是否为纯数字
|
||
case "$account_identifier" in
|
||
''|*[!0-9]*)
|
||
# 非纯数字，视为邮箱
|
||
where_field="email"
|
||
identifier_type="邮箱"
|
||
identifier_value="$account_identifier"
|
||
;;
|
||
*)
|
||
# 纯数字，视为ID
|
||
where_field="userid"
|
||
identifier_type="ID"
|
||
identifier_value="$account_identifier"
|
||
;;
|
||
esac
|
||
fi
|
||
|
||
# 构建 WHERE 条件（为邮箱添加引号）
|
||
if [ "$where_field" = "email" ]; then
|
||
where_condition="where $where_field='$identifier_value'"
|
||
else
|
||
where_condition="where $where_field=$identifier_value"
|
||
fi
|
||
|
||
# 查询用户信息
|
||
content=$(echo "select userid,email from ${MYSQL_PREFIX}users $where_condition;" | mysql -u$MYSQL_USER -p$MYSQL_PASSWORD $MYSQL_DATABASE)
|
||
|
||
# 提取用户ID和邮箱
|
||
user_id=$(echo "$content" | sed -n '2p' | awk '{print $1}')
|
||
account=$(echo "$content" | sed -n '2p' | awk '{print $2}')
|
||
|
||
if [ -z "$account" ]; then
|
||
echo "${RedBG}错误：${identifier_type} '${identifier_value}' 的账号不存在！${Font}"
|
||
exit 1
|
||
fi
|
||
|
||
# 更新密码
|
||
mysql -u$MYSQL_USER -p$MYSQL_PASSWORD $MYSQL_DATABASE <<EOF
|
||
update ${MYSQL_PREFIX}users set encrypt='${new_encrypt}',password='${md5_password}' $where_condition;
|
||
EOF
|
||
|
||
# 只在 identifier_type="ID" 时才输出ID
|
||
if [ "$identifier_type" = "ID" ]; then
|
||
echo "ID: ${GreenBG}${user_id}${Font}"
|
||
fi
|
||
|
||
# 输出邮箱和密码
|
||
echo "邮箱: ${GreenBG}${account}${Font}"
|
||
echo "密码: ${GreenBG}${new_password}${Font}"
|