#!/usr/bin/env sh

######################################################################
# @author      : swytch (adapted from Luke Smith - lukesmith.xyz)
# @file        : mailsync
# @license     : GPLv3
# @created     : Sunday Feb 14, 2021 12:46:24 CET
#
# @description : synchronize mail accounts
######################################################################

# First, get the right variables for the mbsync file, the pass archive, notmuch
# and the GPG home.  This is done by searching common profile files for variable
# assignments.

timestamp="${XDG_STATE_HOME:-$HOME/.local/state}/mailsynclastrun"

eval "$(grep -h -- \
	"^\s*\(export \)\?\(XDG_CACHE_HOME\|XDG_CONFIG_HOME\|XDG_DATA_HOME\)=" \
	"$HOME/.zshenv" 2>/dev/null)"

eval "$(grep -h -- \
	"^\s*\(export \)\?\(MBSYNCRC\|PASSWORD_STORE_DIR\|NOTMUCH_CONFIG\|GNUPGHOME\)=" \
	"$XDG_CONFIG_HOME/zsh/.zprofile" "$HOME/.pam_environment" 2>/dev/null)"

export GPG_TTY=$TTY
[ -r "$HOME/.dbus/Xdbus" ] && source "$HOME/.dbus/Xdbus"

# Config file location must be passed at execution, not as envrionment variable
[ -n "$MBSYNCRC" ] && alias mbsync="mbsync -c $MBSYNCRC" || MBSYNCRC="$HOME/.mbsyncrc"

pgrepoutput="$(pgrep -ax X\(\|org\|wayland\))"
displays="$(echo "$pgrepoutput" | grep -wo "[0-9]*:[0-9]\+" | sort -u)"
notify() {
        [ -n "$pgrepoutput" ] && for x in ${displays:-0:}; do
                export DISPLAY=$x
                notify-send "neomutt" " $2 new mail(s) in \`$1\` account."
        done;
}

# Check account for new mail. Notify if there is new content.
syncandnotify() {
    acc="$(echo "$account" | sed "s/.*\///")"
    mbsync "$acc"
	new="$(find "${XDG_DATA_HOME:-$HOME/.local/share}"/mail/$acc -path */new/* -type f -newer "$timestamp" | grep -iv -e "trash" -e "drafts" 2>/dev/null)"
    newcount=$(echo "$new" | sed '/^\s*$/d' | wc -l)
    if [ "$newcount" -gt "0" ]; then
        notify "$acc" "$newcount" &
        [ -n "$(pidof dwmblocks)" ] && kill -40 $(pidof dwmblocks)
    fi
}

accounts="$(awk '/^Channel/ {print $2}' "$MBSYNCRC")"

# Parallelize multiple accounts
for account in $accounts; do
    syncandnotify &
done

wait

notmuch new 2>/dev/null

#Create a touch file that indicates the time of the last run of mailsync
touch $timestamp