#!/bin/bash

source /usr/local/lib/err

usage() {
  cat <<EOF
Usage: ${0##*/} [OPTIONS] push|pull HOST

Turn off mail receiving on OLD_HOST, run btrbk to move mail to NEW_HOST,
turn on mail receiving on NEW_HOST.  Assumes we want to move all
filesystems unless passing -o.

-i         Disallow incremental backup.
-o         Only btrbk /o, instead of all filesystems.
-h|--help  Print help and exit.

I used to adjust home network dns so NEW_HOST resolves locally if it is
on the local network, but its simpler just not to and just rely
on the internet. Email can wait.

Note: Uses GNU getopt options parsing style
EOF
  exit $1
}


restore_new_btrbk=false
restore_old_btrbk=false
err-cleanup() {
  if $restore_new_btrbk; then
    e WARNING: due to failure, btrbk.timer may need manual restoration:
    e $new_shell systemctl start btrbk.timer
  fi
  if $restore_old_btrbk; then
    e WARNING: due to failure, btrbk.timer may need manual restoration:
    e $old_shell systemctl start btrbk.timer
  fi
}

pre="${0##*/}:"
m() { printf "$pre %s\n"  "$*"; "$@"; }
e() { printf "$pre %s\n"  "$*"; }
err() { echo "$pre ERROR: $*" >&2; }

if [[ $EUID != 0 ]]; then
  err "requires running as root"
  exit 1
fi


##### begin command line parsing ########

force=false
mp_args="-m /o,/q,/a"
temp=$(getopt -l force,help ioh "$@") || usage 1
eval set -- "$temp"
while true; do
  case $1 in
    --force) force=true ;;
    -i) incremental_arg="-i" ;;
    -o) mp_args="-m /o" ;;
    -h|--help) usage ;;
    --) shift; break ;;
    *) echo "$0: Internal error! unexpected args: $*" ; exit 1 ;;
  esac
  shift
done


(( $# == 2 )) || usage 1

if [[ ! $HOSTNAME ]]; then
  err "\$HOSTNAME is unset"
  exit 1
fi

case $1 in
  push)
    old_host=$HOSTNAME
    old_hostname=$HOSTNAME
    new_host=$2
    bbk_args="-t $new_host"
    new_shell="ssh -F $HOME/.ssh/confighome root@$new_host"
    $new_shell -v hostname
    new_hostname=$($new_shell hostname)
    ;;
  pull)
    old_host=$2
    new_host=$HOSTNAME
    new_hostname=$HOSTNAME
    bbk_args="-s $old_host"
    old_shell="ssh -F $HOME/.ssh/confighome root@$old_host"
    # tests ssh connection
    if ! old_hostname=$($old_shell hostname); then
      echo "retrying failed $old_shell with -v"
      $old_shell -v hostname
      exit 1
    fi
    ;;
  *)
    err invalid first argument
    exit 1
    ;;
esac

source /a/bin/bash_unpublished/source-state

if [[ $old_hostname != "$MAIL_HOST" ]] && ! $force; then
  err "\$old_hostname($old_hostname) != \$MAIL_HOST($MAIL_HOST). Rerun with --force if you really want this."
  exit 1
fi

if [[ ! $new_host || ! $old_host ]]; then
  echo "$0: bad args. see script"
  exit 1
fi


########### end initial processing, begin actually modifying things ##########

if $new_shell systemctl is-active btrbk.timer; then
  m $new_shell systemctl stop btrbk.timer
  restore_new_btrbk=true
fi
if $old_shell systemctl is-active btrbk.timer; then
  m $old_shell systemctl stop btrbk.timer
  restore_old_btrbk=true
fi

btrbk_test="systemctl is-active btrbk.service"
active=true
while $active; do
  active=false
  for shell in "$new_shell" "$old_shell"; do
    e $shell $btrbk_test
    status=$($shell $btrbk_test) ||:
    case $status in
      inactive|failed) : ;;
      *)
        # This covers conditions like "activating", which still return 3 from
        # systemctl is-active.
        active=true
        e "btrbk active on shell:$shell, status:$status, sleeping 8 seconds"
        sleep 8
        break
        ;;
    esac
  done
done

# ensure these are unused before doing anything

e "On $new_host: umounting /m and /o, checking emacs"
$new_shell bash -s <<'EOF'
set -eE
if pgrep -G iank -u iank -f 'emacs --daemon' &>/dev/null; then
  export XDG_RUNTIME_DIR=/run/user/1000
  bufs="$(emacsclient --eval "$(cat /a/bin/ds/unsaved-buffers.el)"| sed '/^"nil"$/d;s/^"(/E: /;s/)"$//')"
  if [[ $bufs ]]; then
    echo "error: on $HOSTNAME, unsaved emacs files: $bufs" >&2
    exit 1
  fi
fi
for dir in m o; do
  if mountpoint -q /$dir; then
    echo On $new_host: umount /$dir
    umount /$dir
  fi
done
EOF

$old_shell bash -s <<'EOF'
if pgrep -G iank -u iank -f 'emacs --daemon' &>/dev/null; then
  export XDG_RUNTIME_DIR=/run/user/1000
  bufs="$(emacsclient --eval "$(cat /a/bin/ds/unsaved-buffers.el)"| sed '/^"nil"$/d;s/^"(/E: /;s/)"$//')"
  if [[ $bufs ]]; then
    echo "error: on $HOSTNAME, unsaved emacs files: $bufs" >&2
    exit 1
  fi
fi
EOF

# previously, I was checking to see if the new mail host
# is on my home network, then changing my home dns
# to resolve on the local network, so that I didnt
# have to send traffic out to the internet or rely
# on that. However, that breaks for a laptop that roams.
# So, we could have a cronjob that updates that dns,
# however, another solution is to just use ipv6,
# and I prefer that.
#
# TODO: enable ipv6 for email. exim config setting disables it.
# need to add vpn support. need to add firewall / routing.
# I think exim will try ipv6 first, so no need to disable
# ipv6 i think.


e Running initial btrbk
if ! m btrbk-run -v $bbk_args $incremental_arg $mp_args; then
  ret=$?
  err "failed initial btrbk"
  exit $ret
fi

if ! m $old_shell /a/exe/primary-setup $new_hostname; then
  ret=$?
  err "failed \$old_shell primary-setup \$new_hostname. fix and rerun switch-mail-host"
  exit $ret
fi

# Try to prevent emacs from saving stale data it has in memory to disk. eg: files, recentf list, etc.
# But if emacs ignores the signal, let it live.
m $new_shell killall -q emacs ||:

e Running main btrbk
m btrbk-run -v $bbk_args $incremental_arg -m /o || ret=$?
if (( ret )); then
  bang="$(printf "$(tput setaf 5)█$(tput sgr0)%.0s" 1 2 3 4 5 6 7)"
  e $bang failed btrbk of /o. restoring old host as primary
  m $old_shell /a/exe/primary-setup localhost
  exit $ret
fi

# once I accidentally accepted incoming mail on old host. I used this script to copy over that mail:
#
# die=false; for d in o.leaf.2021-05-29T10:02:08-0400/m/{4e,md,4e2}/{,l/}!(*myarchive)/new; do if $die; then break; fi; find $d -type f -mtime -5 | while read -r f; do dir="${f%new/*}"; dir="btrbk/o.20210530T000011-0400/${dir#*/}"; fname="${f##*/}"; [[ -e $dir/new/$fname || -e $dir/cur/$fname ]] && continue; if ! e cp -a $f /${dir#*/*/}new; then echo failed cp; die=true; break; fi ; done; done

# once I accidentally sent mail from non-main mail host. to copy into the main mail host's sent dir, cd into dir of non-mail mail host Sent/cur, then
#
# shopt -s nullglob; find . -type f -mtime -2 | while read -r f; do a=( /m/4e/Sent/cur/${f%,*}* ); if (( ${#a[@]} )); then e exists $a; else m cp -a $f /m/4e/Sent/cur; fi; done

if ! m $new_shell /a/exe/primary-setup localhost; then
  ret=$?
  err "failed final primary-setup, just fix and rerun: $new_shell /a/exe/primary-setup localhost"
  exit $ret
fi

m exit 0