bin/manage-slaves new

1 files changed, 123 insertions, 0 deletions

diff --git a/bin/manage-slaves b/bin/manage-slaves
new file mode 100755
index 0000000..cfa8c4d
--- /dev/null
+++ b/bin/manage-slaves
@@ -0,0 +1,123 @@
+#!/bin/sh
+
+# shellcheck source=../lib/load-configuration
+. "${0%/*}/../lib/load-configuration"
+
+# shellcheck disable=SC2016
+usage() {
+  >&2 echo ''
+  >&2 echo 'manage-slaves $action [parameters]: manage the list of build slaves'
+  >&2 echo ''
+  >&2 echo 'possible actions:'
+  >&2 echo '  add $name $owner $ssh-key-fingerprint'
+  >&2 echo '  disable $name'
+  >&2 echo '  enable $name'
+  exit 1
+}
+
+if [ $# -eq 0 ]; then
+  usage
+fi
+
+if errors=$(
+  printf '%s\n' "$@" | \
+    grep -vx '[-0-9a-zA-Z=+/]\+'
+); then
+  >&2 echo 'The following parameters contain invalid characters:'
+  printf '%s\n' "${errors}" | \
+    sed '
+      s/^/"/
+      s/$/"/
+    ' >&2
+  usage
+fi
+
+case "$1" in
+  'add')
+    shift
+    if [ $# -ne 3 ]; then
+      >&2 printf '"add" expects 3 parameters, %s were given\n' "$#"
+      usage
+    fi
+    person_id=$(
+      # shellcheck disable=SC2016
+      {
+        printf 'SELECT `persons`.`id`'
+        printf ' FROM `persons`'
+        printf ' WHERE `persons`.`name`="%s"' \
+          "${2}"
+      } | \
+        mysql_run_query
+    )
+    if [ -z "${person_id}" ]; then
+      >&2 printf 'Cannot find person "%s" in the database.\n' \
+        "${2}"
+      usage
+    fi
+    duplicate=$(
+      # shellcheck disable=SC2016
+      {
+        printf 'SELECT CONCAT('
+          printf '"ssh-key: ",'
+          printf '`ssh_keys`.`fingerprint`'
+        printf ')'
+        printf ' FROM `ssh_keys`'
+        printf ' WHERE `ssh_keys`.`fingerprint`="%s";\n' \
+          "${3}"
+        printf 'SELECT CONCAT('
+          printf '"build-slave: ",'
+          printf '`build_slaves`.`name`'
+        printf ')'
+        printf ' FROM `build_slaves`'
+        printf ' WHERE `build_slaves`.`name`="%s";\n' \
+          "${1}"
+      } | \
+        mysql_run_query
+    )
+    if [ -n "${duplicate}" ]; then
+      >&2 printf 'Some entry already existed in the database:\n%s\n' \
+        "${duplicate}"
+      usage
+    fi
+    # shellcheck disable=SC2016
+    {
+      printf 'INSERT INTO `ssh_keys`(`owner`,`fingerprint`)'
+      printf ' VALUES'
+      printf ' (%s,"%s");\n' \
+        "${person_id}" \
+        "${3}"
+      printf 'INSERT INTO `build_slaves`('
+        printf '`name`,'
+        printf '`ssh_key`,'
+        printf '`is_sane`,'
+        printf '`access_allowed`'
+      printf ') VALUES'
+      printf '("%s",LAST_INSERT_ID(),1,1);\n' \
+        "${1}"
+    } | \
+      mysql_run_query
+    printf 'command="%s/bin/slave-build-connect %s" ssh-rsa %s %s@%s\n' \
+      "${base_dir}" \
+      "${1}" \
+      "${3}" \
+      "${2}" \
+      "${1}" >> \
+      ~/".ssh/authorized_keys"
+  ;;
+  'disable'|'enable')
+    action="$1"
+    shift
+    if [ $# -ne 1 ]; then
+      >&2 printf '"%s" expects one parameter, %s were given\n' \
+        "${action}" \
+        "$#"
+      usage
+    fi
+    >&2 echo 'not yet implemented'
+    exit 1
+  ;;
+  *)
+    >&2 printf 'unknown action "%s"\n' "$1"
+    usage
+  ;;
+esac