83 lines
2.9 KiB
Bash
Executable File
83 lines
2.9 KiB
Bash
Executable File
#!/bin/sh
|
||
# Purpose {{{
|
||
## This script will :
|
||
### Try to create the workdir (for Compute Cluster) for the user passed in argument.
|
||
### Set the correct permissions on this directory according to user's uid/gid.
|
||
# }}}
|
||
# Vars {{{
|
||
DEBUG=1
|
||
|
||
username="${1}"
|
||
|
||
workdir_basepath="/mnt/work"
|
||
user_id="$(id -u "${username}" 2>/dev/null)"
|
||
user_primary_group="$(id -g "${username}" 2>/dev/null)"
|
||
user_workdir_path="${workdir_basepath}/${username}"
|
||
# }}}
|
||
|
||
# Ensure to get one argument {{{
|
||
if [ "${#}" -eq 1 ]; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : arg check — ${0} get one argument: ${1}."
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : arg check — ${0} should get only one argument."
|
||
exit 1
|
||
fi
|
||
# }}}
|
||
# Check user exist {{{
|
||
if [ "${user_id}" ]; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : user check — ${username} exists."
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : user check — ${username} doesn't exists."
|
||
exit 1
|
||
fi
|
||
|
||
# }}}
|
||
# Check if workdir already exist {{{
|
||
if [ $(ls -1d "${user_workdir_path}" 2> /dev/null) ]; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : workdir check — ${user_workdir_path} already exists. Abord this script."
|
||
exit 1
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : workdir check — ${user_workdir_path} doesn't exists. Continue."
|
||
fi
|
||
|
||
# }}}
|
||
# Check group exist {{{
|
||
if [ "${user_primary_group}" ]; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : group check — group ${user_primary_group} exists."
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : group check — ${username}'s primary doesn't seems available on this system."
|
||
exit 1
|
||
fi
|
||
|
||
# }}}
|
||
# Check work base path is available {{{
|
||
if [ $(ls -1 -- "${workdir_basepath}" 2> /dev/null| wc -l) -gt 5 ]; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : work_base check — ${workdir_basepath} seems available."
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : work_base check — ${workdir_basepath} doesn't seems available."
|
||
exit 1
|
||
fi
|
||
|
||
# }}}
|
||
# Try to create user's workdir {{{
|
||
command mkdir -p -- "${user_workdir_path}"
|
||
if [ $(ls -1d "${user_workdir_path}" 2> /dev/null) ]; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : workdir creation — ${user_workdir_path} ok."
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : workdir creation — ${user_workdir_path} error."
|
||
exit 1
|
||
fi
|
||
|
||
# }}}
|
||
# Try to set permissions on user's workdir {{{
|
||
if command chown "${user_id}:${user_primary_group}" -R -- "${user_workdir_path}"; then
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : workdir permissions — set permission for ${user_id}:${user_primary_group} ok."
|
||
else
|
||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6s\e[m\n' "DEBUG : workdir permissions — set permission for ${user_id}:${user_primary_group} error."
|
||
exit 1
|
||
fi
|
||
|
||
# }}}
|
||
|
||
exit 0
|