mirror of
https://github.com/kforney/pentest-distro-builder.git
synced 2024-11-01 08:59:11 -06:00
93 lines
2.6 KiB
Text
93 lines
2.6 KiB
Text
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
set -x
|
||
|
|
|
||
|
|
CURRENT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
|
||
|
|
|
||
|
|
source "$CURRENT_DIR/scripts/helpers.sh"
|
||
|
|
source "$CURRENT_DIR/scripts/variables.sh"
|
||
|
|
source "$CURRENT_DIR/scripts/shared.sh"
|
||
|
|
|
||
|
|
save_command_interpolation="#($CURRENT_DIR/scripts/continuum_save.sh)"
|
||
|
|
|
||
|
|
supported_tmux_version_ok() {
|
||
|
|
"$CURRENT_DIR/scripts/check_tmux_version.sh" "$SUPPORTED_VERSION"
|
||
|
|
}
|
||
|
|
|
||
|
|
handle_tmux_automatic_start() {
|
||
|
|
"$CURRENT_DIR/scripts/handle_tmux_automatic_start.sh"
|
||
|
|
}
|
||
|
|
|
||
|
|
another_tmux_server_running() {
|
||
|
|
if just_started_tmux_server; then
|
||
|
|
another_tmux_server_running_on_startup
|
||
|
|
else
|
||
|
|
# script loaded after tmux server start can have multiple clients attached
|
||
|
|
[ "$(number_tmux_processes_except_current_server)" -gt "$(number_current_server_client_processes)" ]
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
delay_saving_environment_on_first_plugin_load() {
|
||
|
|
if [ -z "$(get_tmux_option "$last_auto_save_option" "")" ]; then
|
||
|
|
# last save option not set, this is first time plugin load
|
||
|
|
set_last_save_timestamp
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
add_resurrect_save_interpolation() {
|
||
|
|
local status_right_value="$(get_tmux_option "status-right" "")"
|
||
|
|
# check interpolation not already added
|
||
|
|
if ! [[ "$status_right_value" == *"$save_command_interpolation"* ]]; then
|
||
|
|
local new_value="${save_command_interpolation}${status_right_value}"
|
||
|
|
set_tmux_option "status-right" "$new_value"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
number_of_sessions() {
|
||
|
|
tmux list-sessions |
|
||
|
|
wc -l |
|
||
|
|
sed "s/ //g"
|
||
|
|
}
|
||
|
|
|
||
|
|
# when tmux server is first started, number of sessions is 0
|
||
|
|
just_started_tmux_server() {
|
||
|
|
[ "$(number_of_sessions)" -eq 0 ]
|
||
|
|
}
|
||
|
|
|
||
|
|
start_auto_restore_in_background() {
|
||
|
|
"$CURRENT_DIR/scripts/continuum_restore.sh" &
|
||
|
|
}
|
||
|
|
|
||
|
|
update_tmux_option() {
|
||
|
|
local option="$1"
|
||
|
|
local option_value="$(get_tmux_option "$option")"
|
||
|
|
# replace interpolation string with a script to execute
|
||
|
|
local new_option_value="${option_value/$status_interpolation_string/$status_script}"
|
||
|
|
set_tmux_option "$option" "$new_option_value"
|
||
|
|
}
|
||
|
|
|
||
|
|
main() {
|
||
|
|
if supported_tmux_version_ok; then
|
||
|
|
handle_tmux_automatic_start
|
||
|
|
|
||
|
|
# Advanced edge case handling: start auto-saving only if this is the
|
||
|
|
# only tmux server. We don't want saved files from more environments to
|
||
|
|
# overwrite each other.
|
||
|
|
if ! another_tmux_server_running; then
|
||
|
|
# give user a chance to restore previously saved session
|
||
|
|
delay_saving_environment_on_first_plugin_load
|
||
|
|
add_resurrect_save_interpolation
|
||
|
|
fi
|
||
|
|
|
||
|
|
if just_started_tmux_server; then
|
||
|
|
start_auto_restore_in_background
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Put "#{continuum_status}" interpolation in status-right or
|
||
|
|
# status-left tmux option to get current tmux continuum status.
|
||
|
|
update_tmux_option "status-right"
|
||
|
|
update_tmux_option "status-left"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
main
|