136 lines
3.5 KiB
Bash
136 lines
3.5 KiB
Bash
#!/bin/bash
|
|
# Ver: 1.1 by Endial Fang (endial@126.com)
|
|
#
|
|
# 文件管理函数库
|
|
|
|
# 加载依赖项
|
|
. /usr/local/scripts/liblog.sh # 日志输出函数库
|
|
|
|
# 函数列表
|
|
|
|
# 确保指定的 文件/路径 所属权为指定的 用户/组
|
|
# 参数:
|
|
# $1 - 文件路径
|
|
# $2 - 用户
|
|
ensure_owned_by() {
|
|
local path="${1:?path is missing}"
|
|
local owner="${2:?owner is missing}"
|
|
|
|
chown "$owner":"$owner" "$path"
|
|
}
|
|
|
|
# 检测目录是否存在，如果不存在则创建，同时修改为指定的用户
|
|
# 参数:
|
|
# $1 - 目录路径
|
|
# $2 - 用户
|
|
ensure_dir_exists() {
|
|
local dir="${1:?directory is missing}"
|
|
local owner="${2:-}"
|
|
|
|
mkdir -p "${dir}"
|
|
if [[ -n $owner ]]; then
|
|
ensure_owned_by "$dir" "$owner"
|
|
fi
|
|
}
|
|
|
|
# 检测目录是否存在或为空
|
|
# 参数:
|
|
# $1 - 目录路径
|
|
is_dir_empty() {
|
|
local dir="${1:?missing directory}"
|
|
|
|
if [[ ! -e "$dir" ]] || [[ -z "$(ls -A "$dir")" ]]; then
|
|
true
|
|
else
|
|
false
|
|
fi
|
|
}
|
|
|
|
# 检测指定的路径当前用户是否可写入
|
|
# 参数:
|
|
# $1 - 文件或路径
|
|
# 返回值:
|
|
# true / false
|
|
is_writable() {
|
|
local file="${1:?missing file}"
|
|
local dir
|
|
dir="$(dirname "$file")"
|
|
|
|
if [[ ( -f "$file" && -w "$file" ) || ( ! -f "$file" && -d "$dir" && -w "$dir" ) ]]; then
|
|
true
|
|
else
|
|
false
|
|
fi
|
|
}
|
|
|
|
# 循环设置目录中子目录及文件权限
|
|
# 参数:
|
|
# $1 - paths (as a string).
|
|
# Flags:
|
|
# -f|--file-mode - 文件权限模式
|
|
# -d|--dir-mode - 目录权限模式
|
|
# -u|--user - 用户
|
|
# -g|--group - 用户组
|
|
configure_permissions_ownership() {
|
|
local -r paths="${1:?paths is missing}"
|
|
local dir_mode=""
|
|
local file_mode=""
|
|
local user=""
|
|
local group=""
|
|
|
|
# Validate arguments
|
|
shift 1
|
|
while [ "$#" -gt 0 ]; do
|
|
case "$1" in
|
|
-f|--file-mode)
|
|
shift
|
|
file_mode="${1:?missing mode for files}"
|
|
;;
|
|
-d|--dir-mode)
|
|
shift
|
|
dir_mode="${1:?missing mode for directories}"
|
|
;;
|
|
-u|--user)
|
|
shift
|
|
user="${1:?missing user}"
|
|
;;
|
|
-g|--group)
|
|
shift
|
|
group="${1:?missing group}"
|
|
;;
|
|
*)
|
|
LOG_E "Invalid command line flag $1" >&2
|
|
return 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
read -r -a filepaths <<< "$paths"
|
|
for p in "${filepaths[@]}"; do
|
|
if [[ -e "$p" ]]; then
|
|
LOG_D "Check $p"
|
|
if [[ -n ${dir_mode} ]]; then
|
|
LOG_D "Change permissions to ${dir_mode} of directories in $p"
|
|
find -L "$p" -type d -exec chmod "$dir_mode" {} \;
|
|
fi
|
|
if [[ -n ${file_mode} ]]; then
|
|
LOG_D "Change permissions to ${file_mode} of files in $p"
|
|
find -L "$p" -type f -exec chmod "$file_mode" {} \;
|
|
fi
|
|
if [[ -n $user ]] && [[ -n ${group} ]]; then
|
|
LOG_D "Change ownership to ${user}:${group} of files and directories in $p"
|
|
chown -LR "$user":"$group" "$p"
|
|
elif [[ -n $user ]] && [[ -z $group ]]; then
|
|
LOG_D "Change user to ${user} of files and directories in $p"
|
|
chown -LR "$user" "$p"
|
|
elif [[ -z $user ]] && [[ -n $group ]]; then
|
|
LOG_D "Change group to ${group} of files and directories in $p"
|
|
chgrp -LR "$group" "$p"
|
|
fi
|
|
else
|
|
LOG_E "$p does not exist"
|
|
fi
|
|
done
|
|
}
|