Refactor script-user-reset-x2go
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
#!/usr/bin/env bash
|
||||
# This script will add scripts-* to the PATH and the manual to each user's desktop
|
||||
# Copyright 2021-2023 Bryan C. Roessler
|
||||
# Copyright 2021-2024 Bryan C. Roessler
|
||||
|
||||
parent="${BASH_SOURCE[0]}"
|
||||
parent=${parent%/*}
|
||||
@@ -17,13 +17,11 @@ fi
|
||||
|
||||
((reload)) && [[ -f $parent/script-functions ]] && . "$parent"/script-functions
|
||||
|
||||
is_root
|
||||
|
||||
target=/usr/local/bin
|
||||
for script in script-*; do
|
||||
echo "Installing $script to $target"
|
||||
echo "Linking $script to $target"
|
||||
[[ $script == "script-functions" ]] && install -m 644 "$script" "$target"
|
||||
cp -u "$script" "$target/"
|
||||
sudo ln -s "$script" "$target/"
|
||||
done
|
||||
|
||||
# Install manual link
|
||||
|
||||
@@ -1,32 +0,0 @@
|
||||
#!/usr/bin/env bash
|
||||
# Generate a new QHTCP experiment directory
|
||||
# Copyright 2024 Bryan C. Roessler
|
||||
|
||||
echo "This script supports one optional argument, a project name"
|
||||
|
||||
PROJECTS_DIR="/mnt/data/StudiesQHTCP"
|
||||
TEMPLATE_DIR="/mnt/data/StudiesQHTCP/_TEMPLATE_2copy_rename_4every_new_QHTCPstudy_23_1001"
|
||||
PROJECT_PREFIX="$(whoami)-$(date +%y-%m-%d)"
|
||||
|
||||
ask_project_name() { read -r -p "Enter a new project name: " PROJECT_NAME; }
|
||||
|
||||
if [[ $# == 1 ]]; then
|
||||
PROJECT_NAME="$1"
|
||||
else
|
||||
ask_project_name
|
||||
fi
|
||||
|
||||
PROJECT_DIR="$PROJECTS_DIR/$PROJECT_PREFIX-$PROJECT_NAME"
|
||||
|
||||
while [[ -d $PROJECT_DIR ]]; do
|
||||
echo "A project already exists at $PROJECT_DIR"
|
||||
ask_project_name
|
||||
PROJECT_DIR="$PROJECTS_DIR/$PROJECT_PREFIX-$PROJECT_NAME"
|
||||
done
|
||||
|
||||
if mkdir "$PROJECT_DIR" &&
|
||||
cp -a "$TEMPLATE_DIR"/* "$PROJECT_DIR"; then
|
||||
echo "New project created at $PROJECT_DIR"
|
||||
fi
|
||||
|
||||
|
||||
@@ -3,29 +3,20 @@
|
||||
# Use --all to reset all user X2Go sessions
|
||||
# Copyright Bryan C. Roessler
|
||||
|
||||
parent="${BASH_SOURCE[0]}"
|
||||
parent=${parent%/*}
|
||||
parent="${BASH_SOURCE[0]%/*}"
|
||||
[[ -f $parent/script-functions ]] && . "$parent/script-functions" || exit 1
|
||||
|
||||
[[ -f $parent/script-functions ]] && . "$parent"/script-functions || exit 1
|
||||
echo "Usage: $0 [username|--all]"
|
||||
|
||||
echo "This script supports one optional argument, a username (or --all for all users)"
|
||||
|
||||
USER_MODE=0
|
||||
USER_LIST=()
|
||||
if [[ $EUID -gt 0 ]]; then
|
||||
echo "This script should normally be run as root (with sudo)"
|
||||
echo "Attempting to run in user mode"
|
||||
USER_MODE=1
|
||||
USER_LIST+=($(whoami))
|
||||
else
|
||||
if [[ $# -eq 1 ]]; then
|
||||
if [[ $1 == '--all' ]]; then
|
||||
for i in /home/*; do
|
||||
USER_LIST+=("${i##*/}")
|
||||
done
|
||||
echo "Run as root for all users. Running in user mode for: $(whoami)"
|
||||
USER_LIST+=("$(whoami)")
|
||||
else
|
||||
if [[ $1 == "--all" ]]; then
|
||||
for d in /home/*; do [[ -d $d ]] && USER_LIST+=("${d##*/}"); done
|
||||
elif [[ -n $1 ]]; then
|
||||
USER_LIST+=("$1")
|
||||
fi
|
||||
else
|
||||
prompt user
|
||||
USER_LIST+=("$user")
|
||||
@@ -34,26 +25,27 @@ else
|
||||
fi
|
||||
|
||||
for user in "${USER_LIST[@]}"; do
|
||||
# Clean local user cache
|
||||
# Remove X2Go cache files
|
||||
shopt -s nullglob
|
||||
caches=(/home/"$user"/.x2go/C-"$user"-* /home/"$user"/.xsession-x2go-*)
|
||||
shopt -u nullglob
|
||||
if [[ ${#caches} -gt 0 ]]; then
|
||||
ask_ok "Remove X2Go cache files for user $user?" &&
|
||||
rm -rf /home/"$user"/.x2go/C-"$user"-* &&
|
||||
echo "Removed: ${caches[*]} for user $user"
|
||||
if (( ${#caches[@]} )); then
|
||||
ask_ok "Remove X2Go cache files for $user?" &&
|
||||
rm -rf "${caches[@]}" &&
|
||||
echo "Removed: ${caches[*]} for $user"
|
||||
fi
|
||||
# Clean X2Go sessions
|
||||
if (( USER_MODE )); then
|
||||
|
||||
# Terminate X2Go sessions
|
||||
if [[ $EUID -gt 0 ]]; then
|
||||
mapfile -t sessions < <(x2golistsessions | grep "$user" | cut -f2 -d'|')
|
||||
else
|
||||
mapfile -t sessions < <(x2golistsessions_root | grep "$user" | cut -f2 -d'|')
|
||||
fi
|
||||
if [[ ${#sessions} -gt 0 ]]; then
|
||||
ask_ok "Terminate X2Go sessions for user $user?" &&
|
||||
if (( ${#sessions[@]} )); then
|
||||
ask_ok "Terminate X2Go sessions for $user?" &&
|
||||
for session in "${sessions[@]}"; do
|
||||
x2goterminate-session "$session" &&
|
||||
echo "Terminated: $session for user $user"
|
||||
echo "Terminated: $session for $user"
|
||||
done
|
||||
fi
|
||||
done
|
||||
|
||||
Reference in New Issue
Block a user