#!/bin/bash

rsyncOptions='-vxcaHAXF'
rsync --help | \
  grep -q -- --may-modify-others && \
  rsyncOptions="${rsyncOptions} --may-modify-others"

[ -r "#ETCDIR#/backup.conf" ] && \
  . "#ETCDIR#/backup.conf"

usage()
{
  >&2 echo \
'Usage:  backup /tmp/pidFile /path/to/destination/ user@source:path [proxy_user@ssh_host]
Backup files from remote with rsync, possibly via SSH-tunnel.

With no arguments, information is expected to be in array $backups in #ETCDIR#/backup.conf with name of executable (e.g. a symlink) as key.
With one argument, information is expected to be in array ${backups[$1]} in #ETCDIR#/backup.conf.

Options:
  /tmp/pidFile           location of file to store PID in
  /path/to/destination   location to store backups in
  user@source:path       remote data to back up
  proxy_user@ssh_host    ssh login to proxy node (optional)
#HELPTEXT#              #'
  [ -z "$1" ] && exit 1
  exit $1
}

extract_ssh_host() {
  {
    printf '%s\n' "$1"
    if [ -f ~/.ssh/config ]; then
      sed '
        /^Host\s\+'"$1"'$/,/^Host\s/ {
          s/^\s*Hostname\s\+//
          t
        }
        d
      ' ~/.ssh/config
    fi
  } \
    | tail -n1
}

extract_ssh_ip_protocols() {
  getent ahosts "$(extract_ssh_host "$1")" \
    | sed '
      s/^\([0-9]\+\.\)\{3\}[0-9]\+\s\+STREAM\(\s.*\)\?$/4/
      t
      s/^[0-9a-f][0-9a-f:]\+\s\+STREAM\(\s.*\)\?$/6/
      t
      d
    ' \
    | sort -u \
    | if [ -f ~/.ssh/config ]; then
      grep -vxF "$(
        sed '
          /^Host\s\+'"$1"'$/,/^Host\s/ {
            s/^\s*AddressFamily\s\+inet\s*$/6/
            t
            s/^\s*AddressFamily\s\+inet6\s*$/4/
            t
          }
          d
        ' ~/.ssh/config
      )"
    else
      cat
    fi
}

if [ $# -eq 1 ]; then
  if [ "$1" == "--help" ]; then
    usage 0
  elif [ "$1" == "--version" ]; then
    echo '#VERSION#'
    exit
  fi
fi

seldom=false
if [ $# -eq 0 ] || [ $# -eq 1 ]; then
  if [ $# -eq 0 ]; then
    backupID="$(basename $0)"
  else
    backupID="$1"
  fi
  [ -z "${backups[${backupID}]}" ] && usage
  set /tmp/${backupID}.pid ${backups[${backupID}]}
  if printf '%s\n' "${seldomBackups[@]}" | \
    grep -qxF "${backupID}"; then
    seldom=true
  fi
fi

Basis="$2"
pidFile="$1"
QuellIP=$(echo "$3" | sed "s|^[a-zA-Z]*://||; s|^[a-zA-Z]*@||; s|:\?/.*$||")

if [ "$#" -eq 3 ]; then
  Quelle="$3"
  ipVer=$(
    extract_ssh_ip_protocols "${QuellIP}" \
      | sort -n \
      | tail -n1
  )
elif [ "$#" -eq 4 ]; then
  sshHopp="$4"
  lokPort=$[$RANDOM/2+8000]
  HoppIP="${sshHopp#*@}"
  ipVer=$(
    {
      extract_ssh_ip_protocols "${QuellIP}"
      extract_ssh_ip_protocols "${HoppIP}"
    } \
      | sort -n \
      | uniq -d \
      | tail -n1
  )
  if [ ${ipVer} -eq 4 ]; then
    localAddress='127.0.0.1'
  elif [ ${ipVer} -eq 6 ]; then
    localAddress='[::1]'
  else
    >&2 echo 'ip version of hopp and target must be the same'
    exit 1
  fi
  rsyncShell="-e ssh -${ipVer} -p${lokPort} -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null"
  tunnelBefehl="ssh -${ipVer} -t -t -L${localAddress}:${lokPort}:${QuellIP}:22 ${sshHopp}"
  Quelle="$(echo "$3" | sed "s|${QuellIP}|${localAddress}|")"
else
  usage
fi

rsyncOptions="${rsyncOptions}${ipVer}"

if [ ! -d ${Basis} ]; then
  for neededMount in "${neededMounts[@]}"; do
    if ! mountpoint -q "${neededMount}"; then
      exit 11
    fi
  done
  exit 2
fi

neues_Datum="${Basis}/$(date "+%Y_%m_%d")"
neues="${Basis}/aktuell"
linkdests=""
for s in $(ls -1 "${Basis}" | sort -r | grep -vxF -m 20 aktuell ); do
  linkdests="${linkdests} --link-dest ${Basis}/${s}"
done

if [ ! "$(whoami)" == "root" ]; then
  echo "I need to be root."
  exit 3
fi

[ -w "${Basis}" ] || exit 11
[ -e "${neues_Datum}" ] && exit 4
if ${seldom}; then
  for date_diff in $(seq ${seldomness}); do
    if [ -e "${Basis}/$(date '+%Y_%m_%d' -d@$(($(date '+%s')-24*60*60*date_diff)))" ]; then
      exit 4
    fi
  done
fi
[ -s "${pidFile}" ] && kill -0 $(cat "${pidFile}") 2>/dev/null && exit 5

echo $$ > "${pidFile}"

if [ -n "${QuellIP}" ]; then
  if ! ping -c1 "$(extract_ssh_host "${QuellIP}")" >/dev/null; then
    rm "${pidFile}"
    exit 11
  fi
fi
if [ -n "${HoppIP}" ]; then
  if ! ping -c1 "$(extract_ssh_host "${HoppIP}")" >/dev/null; then
    rm "${pidFile}"
    exit 11
  fi
fi

if [ -n "${tunnelBefehl}" ]; then
  ${tunnelBefehl} &
  backgroundPid=$!
  sleep 4
fi

for toExclude in "${excludes[@]}"; do
  excludeArgs="${excludeArgs} --exclude ${toExclude}"
done

if [ ! -e "${neues}/wip" ]; then
  mkdir -p "${neues}/wip"
fi
chmod 750 "${neues}"{,/wip}
chown root:root "${neues}"{,/wip}
if [ -z "${rsyncShell}" ]; then
  rsync ${rsyncOptions} \
    ${linkdests} \
    ${excludeArgs} \
    ${Quelle} "${neues}/wip/"
  sleep 1
  rsync ${Quelle}
else
  rsync "${rsyncShell}" \
    ${rsyncOptions} \
    ${linkdests} \
    ${excludeArgs} \
    ${Quelle} "${neues}/wip/"
  sleep 1
  rsync "${rsyncShell}" ${Quelle}
fi
erg=$?

[ -n "${backgroundPid}" ] && kill "${backgroundPid}"

if [ ${erg} -eq 0 ] || [ ${erg} -eq 24 ]; then
  chmod o-rwx "${neues}/wip"
  neueres_Datum="${Basis}/$(date "+%Y_%m_%d")"
  if [ ! -e "${neueres_Datum}" ]; then
    neues_Datum="${neueres_Datum}"
  fi
  mv "${neues}/wip" "${neues_Datum}"
  rmdir "${neues}"
  rm "${pidFile}"
else
  rm "${pidFile}"
  exit 11
fi