rte.sh

#!/usr/bin/env bash

# set xdg_variables
APP="rte"

CONFIG_DIR="${XDG_CONFIG_HOME:-$HOME/.config}/$APP"
DATA_DIR="${XDG_DATA_HOME:-$HOME/.local/share}/$APP"

if [ "$1" == "fda" ]; then
  # just exit, hopefully agent will get populated in full disk access setting
  exit 0
fi

if [ "$1" == "init" ]; then
  # handling of client initialization
  if [ "$2" == "--force" ]; then
    # force remove old installs
    rm -fr "$CONFIG_DIR" "$DATA_DIR"
    shift
  fi
  # check if CONFIiG_DIR already exists
  if [ -d "$CONFIG_DIR" ]; then
    echo "$CONFIG_DIR already exists, start with"
    echo "  $0 init --force <url>"
    echo "to reconfigure."
    exit 1
  fi

  mkdir -p "$CONFIG_DIR" "$DATA_DIR"
  chmod 700 "$CONFIG_DIR" "$DATA_DIR"

  echo "downloading defaults"
  if ! rsync --quiet --timeout=5 "$2/defaults" "$CONFIG_DIR" && [ ! -f "$CONFIG_DIR/defaults" ]; then
    echo "can't reach $2 (or no config present), removing $CONFIG_DIR and giving up"
    rm -fr "$CONFIG_DIR" "$DATA_DIR"
    exit 2
  fi

  # initial git pull
  echo "getting the newest version from github and place it in $DATA_DIR"
  cd "$DATA_DIR" || exit 1
  git clone https://github.com/uwe-schwarz/rte.git 2>&1 | sed 's/^/  /'

  # got main config files, get interval and create timers
  source "$CONFIG_DIR/defaults"

  [ "$config" ] || exit 1

  echo "creating timers with interval=$interval"
  case $(uname) in
    Linux)
      echo "operating system is Linux, using a systemd user timer"
      echo "this creates a timer in ~/.config/systemd/user, maybe this is the wrong directory"
      mkdir -p ~/.config/systemd/user
      cat > ~/.config/systemd/user/rte.service << EOF
[Unit]
Description=rte
After=network.target
ConditionPathExists=$CONFIG_DIR

[Service]
Type=oneshot
ExecStart=$DATA_DIR/rte/rte.sh

[Install]
WantedBy=default.target
EOF
      cat > ~/.config/systemd/user/rte.timer << EOF
[Unit]
Description=Run "rte" every $interval seconds

[Timer]
OnBootSec=30min
OnUnitActiveSec=$interval

[Install]
WantedBy=timers.target
EOF
      systemctl --user daemon-reload
      systemctl --user enable --now "rte.timer"
      echo "systemd timer enabled."
      ;;
    Darwin)
      echo "operating system is macOS, using a fake application and launchd timer"
      mkdir -p ~/Applications
      osacompile -o ~/Applications/rte.app -e 'do shell script "'"$DATA_DIR"'/rte/rte.sh"'
      mv ~/Applications/rte.app/Contents/Info.plist ~/Applications/rte.app/Contents/Info.plist.bak
      awk '/^<\/dict>/{print "<key>LSUIElement</key>"; print "<string>1</string>"}1' ~/Applications/rte.app/Contents/Info.plist.bak > ~/Applications/rte.app/Contents/Info.plist
      open -g -j -a ~/Applications/rte.app --args fda
      open "x-apple.systempreferences:com.apple.preference.security?Privacy_AllFiles"
      echo "give full disk access or the needed rights to ~/Applications/rte.app/Contents/MacOS/applet."
      echo "it's probably best to give fda rights, otherwise the errors will get fuzzy do debug later."
      echo "if there is no applet in the list, add ~/Applications/rte.app/Contents/MacOS/applet"
      read -r -p "press enter if finished, last chance to ^c to cancel"
      echo "making LaunchAgent"
      mkdir -p ~/Library/LaunchAgents
      cat > ~/Library/LaunchAgents/com.github.uwe-schwarz.rte.plist << EOF
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN"
  "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
    <key>Label</key>
    <string>com.github.uwe-schwarz.rte</string>
    <key>ProgramArguments</key>
    <array>
      <string>open</string>
      <string>-g</string>
      <string>-j</string>
      <string>-a</string>
      <string>$HOME/Applications/rte.app</string>
    </array>
    <key>StartInterval</key> 
    <integer>$interval</integer>
</dict>
</plist>
EOF
      launchctl unload ~/Library/LaunchAgents/com.github.uwe-schwarz.rte.plist >/dev/null 2>/dev/null
      launchctl load ~/Library/LaunchAgents/com.github.uwe-schwarz.rte.plist

      echo "don't remove ~/Applications/rte.app"
      ;;
    *)
      echo "unknown system, please create timer yourself: start $DATA_DIR/rte/rte.sh every $interval seconds."
      exit 3
      ;;
  esac

  exit 0
fi

# are there other arguments?
if [ $# -ne 0 ]; then
  echo "start like this for first invocation:"
  echo "$0 init [--force] <url>"
fi

# normal operation

# defaults, overwrite with config
interval=10800
battery=no
caffeinate=yes
verbose=2
notify=$DATA_DIR/rte/notify.sh
notify_arg=topic_$h
notify_when=onerror
notify_prefix="rte $h:"

# if we need a timezone it should be UTC
export TZ=C

# read defaults
cd "$CONFIG_DIR" || exit 1
source "$CONFIG_DIR/defaults"

# check if gum log is available
if command -v ts >/dev/null; then
  logger=("ts" "%FT%H:%M:%.S")
elif command -v /opt/homebrew/bin/ts >/dev/null; then
  # set homebrew path
  export PATH="/opt/homebrew/bin:$PATH"
  logger=("/opt/homebrew/bin/ts" "%FT%H:%M:%.S")
else
  logger=("cat")
fi

# check for battery power
if [ "$battery" == "no" ]; then
  if command -v pmset >/dev/null 2>/dev/null; then
    pmset -g ps | grep -q "AC" || exit 0
  fi
fi

# hostname, only till first "."
h="$(hostname | cut -d. -f1)"; export h

# start time
t="$(date -u +%F-%H.%M.%S)"

# redownload config
if ! rsync --quiet --timeout=5 "$config/defaults" "$CONFIG_DIR"; then
  # config host not reachable, quit quietly
  exit 0
fi

# variables again
source "$CONFIG_DIR/defaults"

# TODO interval changed handling

# local logfile and basic redirection
logfile="$(mktemp)"
exec >"$logfile" 2>&1 </dev/null

# make directory for logfile upload
ssh "${config/:*}" "mkdir -p \"${config/*:}/logs/$h\""

# first thing update and restart if needed
cd "$DATA_DIR/rte" || exit 1
git remote update >/dev/null 2>/dev/null
if git status -uno | grep -q "Your branch is behind"; then
  # there are updates, pull and restart
  git status
  git pull
  if [[ -f "$logfile" ]]; then 
    rsync --quiet --timeout=10 "$logfile" "$config/logs/$h/$t.update.log"
    rm -f "$logfile"
  fi
  # restart after upgrade
  exec $0
fi

# no more updates, all base config is ok, redirects done. Starting execution

# create temp file and download hostname-folder
EXEC_DIR="$(mktemp -d)"; export EXEC_DIR
export DATA_DIR
export verbose

rsync -r "$config/$h/" "$EXEC_DIR"

# create $log as file for logging and $result as file for notify results
log="$(mktemp)"; export log
result="$(mktemp)"; export result

# keep track of time
secstart="$(date +%s)"

# caffeinate
if [ "$caffeinate" = yes ] && [ "$(uname)" = Darwin ]; then
  coffee=("caffeinate")
else
  unset coffee
fi

# init run
cd "$EXEC_DIR" || exit 1
run_exit_code=0
daily_exit_code=0
weekly_exit_code=0

# exec run if it exists, else run default script
if [ -f "$EXEC_DIR/run" ]; then
  chmod +x "$EXEC_DIR/run"

  # run hostname/run command and keep time
  echo "starting run" | "${logger[@]}"

  "${coffee[@]}" "$EXEC_DIR/run" | "${logger[@]}"
  run_exit_code=${PIPESTATUS[0]}
else
  # no run script, so start default script
  echo "$h/run not found, giving up." | "${logger[@]}"
fi
# get everything logged via file in the right place
cat "$log" | "${logger[@]}"

# daily
if [ -f "$EXEC_DIR/daily" ]; then
  touch "$CONFIG_DIR/last-day"
  if [ "$(date -u +%j)" != "$(< "$CONFIG_DIR/last-day")" ]; then
    # update day
    date +%j > "$CONFIG_DIR/last-day"
    # run daily
    "${coffee[@]}" "$EXEC_DIR/daily" | "${logger[@]}"
    daily_exit_code=${PIPESTATUS[0]}
  else
    echo "daily already ran today" | "${logger[@]}"
  fi
fi
# get everything logged via file in the right place
cat "$log" | "${logger[@]}"

# weekly
if [ -f "$EXEC_DIR/weekly" ]; then
  touch "$CONFIG_DIR/last-week"
  if [ "$(date -u +%W)" != "$(< "$CONFIG_DIR/last-week")" ]; then
    # update week
    date +%W > "$CONFIG_DIR/last-week"
    # run weekly
    "${coffee[@]}" "$EXEC_DIR/weekly" | "${logger[@]}"
    weekly_exit_code=${PIPESTATUS[0]}
  else
    echo "weekly already ran this week" | "${logger[@]}"
  fi
fi
# get everything logged via file in the right place
cat "$log" | "${logger[@]}"

# notify
exit_code=$((run_exit_code+daily_exit_code+weekly_exit_code))

# try to figure out if we should notify
notify_run=0
if [ "$notify_when" = "always" ]; then
  notify_run=1
elif [ "$notify_when" = "onerror" ] && [ $exit_code -ne 0 ]; then
  notify_run=1
fi

if [ -x "$notify" ] && [ $notify_run -eq 1 ]; then
  if [ -s "$result" ]; then
    echo "notify $notify_arg ${notify_prefix}${notify_prefix:+ } \\" | "${logger[@]}"
    pr -to 2 "$result" | "${logger[@]}"
    echo "exit codes: run $run_exit_code, daily $daily_exit_code, weekly $weekly_exit_code" | "${logger[@]}"
    "$notify" "$notify_arg" "${notify_prefix}${notify_prefix:+ }$(<"$result")"$'\n'"exit codes: run $run_exit_code, daily $daily_exit_code, weekly $weekly_exit_code"
  else
    if [ "$run_exit_code" -ne 0 ]; then echo "run $run_exit_code" >> "$result"; fi
    if [ "$daily_exit_code" -ne 0 ]; then echo "daily $daily_exit_code" >> "$result"; fi
    if [ "$weekly_exit_code" -ne 0 ]; then echo "weekly $weekly_exit_code" >> "$result"; fi
    echo "notify $notify_arg ${notify_prefix}${notify_prefix:+ }rte run on $h completed with exit codes:" | "${logger[@]}"
    pr -to 2 "$result" | "${logger[@]}"
    "$notify" "$notify_arg" "${notify_prefix}${notify_prefix:+ }rte run on $h completed with exit codes:"$'\n'"$(<"$result")"
  fi
fi

# conclude
secstop="$(date +%s)"
echo "completed in $((secstop-secstart)) seconds, exit code $exit_code" | "${logger[@]}"

# upload logs and exit
rsync --quiet --timeout=10 "$logfile" "$config/logs/$h/$t.log"

# remove temp files and directory, and log file
rm -f "$log" "$result" "$logfile"
rm -fr "$EXEC_DIR"
exit 0