#!/bin/bash

# these are for non-spamc setups
SPAM_REPORT="spamassassin --report"
HAM_REPORT="spamassassin --revoke"
SPAM_CHECK="spamassassin -e"

# these are for spamc
SPAM_REPORT="spamc --learntype=spam -s $((100 * 1024 * 1024))"
HAM_REPORT="spamc --learntype=ham -s $((100 * 1024 * 1024))"
SPAM_CHECK="spamc -s $((100 * 1024 * 1024)) -c"

# needs rules directories
NEEDS_SPAM_RULES="$(echo ~/Maildir/spam/needs_rules)"
NEEDS_HAM_RULES="$(echo ~/Maildir/spam/needs_ham_rules)"

if [ "$(basename $0)" = "learn_from_spam" ]; then
    for file in "$@"; do
        if ! formail -c < "$file"|grep -q '^List-Id'; then
            $SPAM_REPORT < "$file" >/dev/null 2>&1;
        else
            spamasssin --local --report < "$file" >/dev/null 2>&1;
        fi
        formail -c < "$file" | grep -e '^From ' -e 'From: ' | spamassassin --add-to-blacklist >/dev/null 2>&1;
        formail -c < "$file" | grep -e '^From ' -e 'From: ' | spamassassin --remove-from-whitelist >/dev/null 2>&1;
        # check to see if it's still ham
        if $SPAM_CHECK < "$file" >/dev/null 2>&1; then
            mkdir -p "${NEEDS_SPAM_RULES}/cur";
            TMP="$(mktemp -d)"
            trap "rm -f '$TMP'/file; rmdir '$TMP'" EXIT
            # this message is still not spam; may need custom rules
            spamc -s $((100 * 1024 * 1024)) < "$file" > "$TMP/file" 2>/dev/null;
            mv "$TMP/file" "${NEEDS_SPAM_RULES}/cur/$(basename "$file")";
            trap - EXIT;
            rmdir "$TMP";
        elif [ "$(pwd)" = "${NEEDS_SPAM_RULES}" ]; then
            rm -f "$file";
        fi;
    done;
elif [ "$(basename $0)" = "learn_from_ham" ]; then
    for file in "$@"; do
        $HAM_REPORT < "$file" >/dev/null 2>&1;
        formail -c < "$file" |grep -e '^From ' -e 'From: ' -e 'To: ' -e 'Cc: '|spamassassin --add-to-whitelist >/dev/null 2>&1;
        if ! $SPAM_CHECK < "$file" >/dev/null 2>&1; then
            mkdir -p "${NEEDS_HAM_RULES}/cur";
            TMP="$(mktemp -d)"
            trap "rm -f '$TMP'/file; rmdir '$TMP'" EXIT
            spamc -s $((100 * 1024 * 1024)) < "$file" > "$TMP/file" 2>/dev/null;
            mv "$TMP/file" "${NEEDS_HAM_RULES}/cur/$(basename "$file")";
            trap - EXIT;
            rmdir "$TMP";
        elif [ "$(pwd)" = "${NEEDS_HAM_RULES}" ]; then
            rm -f "$file";
        fi;
    done;
else
    "Called in a way this script cannot recognize";
fi;