mirror of
https://github.com/CiscoDevNet/cml-community.git
synced 2025-10-24 08:33:45 +00:00
0683e4af4a
Add a script to be deployed on CML that allows labs marked with "(autostart)" in their descriptions to be started as soon as the CML server boots up.
125 lines
3.3 KiB
Bash
125 lines
3.3 KiB
Bash
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
ensure_dir() {
|
|
local install_dir="$1"
|
|
if ! test -d "$install_dir"; then
|
|
echo "Error: $install_dir isn't there"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
install_default() {
|
|
local install_dir="/etc/default"
|
|
ensure_dir $install_dir
|
|
cat >"$install_dir/cml-autostart" <<EOF
|
|
USERNAME="admin"
|
|
PASSWORD="password-that-needs-to-be-changed"
|
|
TITLE_REGEX="\(autostart\)"
|
|
EOF
|
|
chown root:root "$install_dir/cml-autostart"
|
|
chmod 0600 "$install_dir/cml-autostart"
|
|
}
|
|
|
|
install_script() {
|
|
local install_dir="/usr/local/bin"
|
|
ensure_dir $install_dir
|
|
cat >"$install_dir/cml-autostart.sh" <<EOF
|
|
#!/bin/bash
|
|
|
|
# set variables
|
|
HOST="ip6-localhost"
|
|
# the /etc/default/cml-autostart needs to define and these need to be present
|
|
# in the environment!
|
|
# - USERNAME
|
|
# - PASSWORD
|
|
# - TITLE_REGEX
|
|
|
|
# set -x
|
|
|
|
API="https://\${HOST}/api/v0"
|
|
|
|
# wait until the system is ready
|
|
until [ "true" = "\$(curl -skX GET \$API/system_information | jq -r .ready)" ]; do
|
|
echo "Waiting for controller to be ready..."
|
|
sleep 5
|
|
done
|
|
|
|
# get authentication token
|
|
TOKEN=\$(curl -skX POST "\${API}/authenticate" \\
|
|
-H "Content-Type: application/json" \\
|
|
-d '{"username":"'\${USERNAME}'","password":"'\${PASSWORD}'"}' |
|
|
jq -r .)
|
|
|
|
# get all labs
|
|
LABS=\$(curl -skX GET "\${API}/labs?show_all=true" \\
|
|
-H "Authorization: Bearer \${TOKEN}")
|
|
|
|
while read -r lab; do
|
|
labinfo=\$(curl -skX GET "\${API}/labs/\${lab}" \\
|
|
-H "Authorization: Bearer \${TOKEN}")
|
|
description=\$(echo \$labinfo | jq -r '.lab_description')
|
|
title=\$(echo \$labinfo | jq -r '.lab_title')
|
|
if [[ "\${description}" =~ \${TITLE_REGEX} ]]; then
|
|
curl -skX PUT "\${API}/labs/\${lab}/start" \\
|
|
-H "Authorization: Bearer \${TOKEN}"
|
|
echo "Lab '\${title}' (ID: \${lab}) start initiated"
|
|
fi
|
|
done <<<\$(echo \$LABS | jq -r '. | join("\n")')
|
|
exit
|
|
EOF
|
|
chown root:root "$install_dir/cml-autostart.sh"
|
|
chmod a+x "$install_dir/cml-autostart.sh"
|
|
}
|
|
|
|
install_service_unit() {
|
|
local install_dir="/etc/systemd/system"
|
|
ensure_dir $install_dir
|
|
cat >"$install_dir/cml-autostart.service" <<EOF
|
|
[Unit]
|
|
Description=Autostart CML Labs based on description text
|
|
After=network.target
|
|
After=virl2.target
|
|
|
|
[Service]
|
|
Type=oneshot
|
|
EnvironmentFile=/etc/default/cml-autostart
|
|
ExecStart=/usr/local/bin/cml-autostart.sh
|
|
User=virl2
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
EOF
|
|
chown root:root "$install_dir/cml-autostart.service"
|
|
systemctl daemon-reload
|
|
systemctl &>/dev/null enable cml-autostart.service
|
|
}
|
|
|
|
if [[ $EUID -ne 0 ]]; then
|
|
echo "This script must be run as root." >&2
|
|
exit 1
|
|
fi
|
|
|
|
echo -n "installing defaults"
|
|
install_default
|
|
echo -e "\t✅"
|
|
echo -n "installing script"
|
|
install_script
|
|
echo -e "\t✅"
|
|
echo -n "installing service unit"
|
|
install_service_unit
|
|
echo -e "\t✅"
|
|
|
|
cat <<EOF
|
|
*****************************************************
|
|
* ⚠️ IMPORTANT! ⚠️ *
|
|
* you need to ensure that you change the username *
|
|
* and password for a user of the system that can *
|
|
* start the labs in /etc/default/cml-autostart *
|
|
* *
|
|
* Add the string "(autostart)" to the description *
|
|
* of each lab that you wish to start automatically. *
|
|
*****************************************************
|
|
EOF
|