#!/bin/bash

# update-all version #VERSION#
# updates all clients defined in #ETCDIR#/update-all.conf via local versions of update-me

. #ETCDIR#/update-all.conf

if [ "$#" -eq 2 ]; then
  system="${systems["$1"]}"
  if [ "${system}" == "localhost" ]
  then
    cmd=""
  else
    cmd="ssh -t -t ${system}"
  fi
  cmd="${cmd} update-me"
  ${cmd}
  err=$?
  if [ "${err}" -ne 0 ] \
  && [ -d "$2" ]; then
    printf '%s\n' "${err}" > "$2/$1"
  fi
  exit ${err}
fi

if [ "$#" -ne 0 ]; then
  >&2 printf 'usage:\n'
  >&2 printf '  %s\n' "${0##*/}"
  >&2 printf '  %s $id $tmpdir\n' "${0##*/}"
  exit 1
fi

tmp_dir=$(mktemp -d)
trap 'rm -rf --one-file-system "${tmp_dir}"' EXIT

pids=()

for system in "${!systems[@]}"; do
  if [ -z "${DISPLAY}" ]; then
    screen -S update-all -d -m "${0}" "${system}" "${tmp_dir}"
  else
    urxvt -title "${systems["${system}"]}: update-me" -e "${0}" "${system}" "${tmp_dir}" &
    pids+=($!)
  fi
done

if [ -z "${DISPLAY}" ]; then
  pids=(
    $(
      screen -list \
      | sed '
        s/^\s\+\([0-9]\+\)\.update-all\s.*$/\1/
        t
        d
      '
    )
  )
fi

while kill -0 "${pids[@]}"; do
  sleep 1
done

ls "${tmp_dir}" \
| while read -r system; do
  printf 'update-me on %s ("%s") failed\n' \
    "${system}" \
    "${systems["${system}"]}"
done

exit $(
  ls "${tmp_dir}" \
  | wc -l
)