forked from redhat-openstack/khaleesi
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
executable file
·152 lines (121 loc) · 4.19 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
#! /usr/bin/env bash
# vim: ts=4:sw=4:et
set -e -u
declare -r SCRIPT_DIR=$(cd $(dirname "$0") && pwd)
cd $SCRIPT_DIR
source tools/lib/functions.bash
print_usage() {
read -r -d '' help <<-EOF_HELP || true
Usage:
$( basename $0) --use <settings> <playbook>
$( basename $0) -h|--help
Options:
--use <settings> All settings needed by khaleesi should be passed to
in yaml format. Will fail without this argument
-h|--help show this help
--tags TAGS,... execute only TAGS
--skip-tags TAGS,...
--inventory FILE Specify an alternate inventory file than local_hosts
--no-logs do not collect logs after running playbook
--take-snapshot take snapshot of the VM after running playbook
--silent show only minimal details, skips showing files
--verbose show a lot of output, enables -vvvv on ansible playbook
--dry-run Only print the commands that would be executed
Creating settings file:
use khaleesi-settings utility to generate settings file
EOF_HELP
echo -e "$help"
return 0
}
usage() {
local exit_val=${1:-1}
if [[ $exit_val -eq 0 ]]; then
print_usage
else
print_usage >&2
fi
# use stdout if exit value is 0 else stderr
exit $exit_val
}
parse_args() {
### while there are args parse them
while [[ -n "${1+xxx}" ]]; do
case $1 in
-h|--help) SHOW_USAGE=true; break ;; # exit the loop
-q|--silent) SILENT=true; shift ;;
--verbose) KHALEESI_VERBOSE=true; KHALEESI_SSH_VERBOSE=true; shift ;;
--use) SETTINGS_FILE=$2; shift 2 ;;
--no-logs) COLLECT_LOGS=false; shift 1 ;;
--take-snapshot) TAKE_SNAPSHOT=true; shift ;;
--inventory) INVENTORY_FILE=$2; shift 2 ;;
--dry-run) DRY_RUN=true; shift ;;
*.yml) PLAYBOOK=$1; shift ;;
*) ARGS_FOR_ANSIBLE+=" $1"; shift ;; # pass anything that run.sh
# doesn't need to ansible
esac
done
return 0
}
collect_logs() {
mkdir -p collected_files
ansible_playbook playbooks/collect_logs.yml $ARGS_FOR_ANSIBLE \
> collected_files/collect_logs.txt 2>&1
}
take_snapshot() {
ansible_playbook playbooks/snapshot.yml --tags=snapshot
}
ansible_playbook() {
local playbook=$1; shift
local cmdline="ansible-playbook -i ${INVENTORY_FILE} --extra-vars @${SETTINGS_FILE}"
if $KHALEESI_VERBOSE || $KHALEESI_SSH_VERBOSE; then
cmdline+=" -v"
$KHALEESI_SSH_VERBOSE && cmdline+="vvv"
fi
export ANSIBLE_FORCE_COLOR=true
execute $cmdline $playbook "$@"
}
validate_args() {
# usage
$SHOW_USAGE && usage 0 # if --help show usage and exit 0
# validation
[ -z ${SETTINGS_FILE+xxx} ] && {
log_error "No ${RED}settings${NORMAL} file passed: \
did you forget ${BLUE}${BOLD}--use$NORMAL ?"
usage # exit 1 if no settings file
}
[ -z ${PLAYBOOK+xxx} ] && {
log_error "No playbook file passed: \
no $BLUE${BOLD}*.yml$NORMAL found in args "
usage
}
$DEPRECATED_OPTIONS_USED && {
log_warning "Deprecated options used: ${RED}Ignoring $NORMAL"
echo -e $BOLD$RED------------------------------------------ $NORMAL
print_usage
echo -e '-----------------------------------------------------------\n'
}
return 0
}
main() {
# set global defaults
KHALEESI_VERBOSE=${KHALEESI_VERBOSE:-false}
KHALEESI_SSH_VERBOSE=${KHALEESI_SSH_VERBOSE:-false}
INVENTORY_FILE=${INVENTORY_FILE:-local_hosts}
SHOW_USAGE=false
COLLECT_LOGS=true
TAKE_SNAPSHOT=false
ARGS_FOR_ANSIBLE=""
SILENT=false
DRY_RUN=false
DEPRECATED_OPTIONS_USED=false
parse_args "$@"
validate_args
$SILENT || on_exit init.print_result
$SILENT || cat_file $SETTINGS_FILE
# collect logs only if the settings are proper
$COLLECT_LOGS && on_exit collect_logs
$TAKE_SNAPSHOT && on_exit take_snapshot
echo -e "\nPlaybook: $PLAYBOOK"
ansible_playbook $PLAYBOOK $ARGS_FOR_ANSIBLE
}
main "$@"