#!/bin/sh
APP_ARG_KEYSDIR="/etc/tmate-ssh-server/keys"
CHANGED=0
gen_key() {
  keytype=$1
  ks="${keytype}_"
  key="$APP_ARG_KEYSDIR/ssh_host_${ks}key"
  if [ ! -e "${key}" ] ; then
    CHANGED=1
    ssh-keygen -t ${keytype} -f "${key}" -N ''
    return $?
  fi
}

gen_key dsa && \
gen_key rsa && \
gen_key ecdsa || \
  exit 1

[ "$CHANGED" == 1 ] && chmod 400 $APP_ARG_KEYSDIR/*key

exit 0
