learn_from_spam

   1 #!/bin/bash
   2
   3 # these are for non-spamc setups
   4 SPAM_REPORT="spamassassin --report"
   5 HAM_REPORT="spamassassin --revoke"
   6 SPAM_CHECK="spamassassin -e"
   7
   8 # these are for spamc
   9 SPAM_REPORT="spamc --learntype=spam"
  10 HAM_REPORT="spamc --learntype=ham"
  11 SPAM_CHECK="spamc -c"
  12
  13 if [ "$(basename $0)" = "learn_from_spam" ]; then
  14     for file in "$@"; do
  15         if ! formail -c < "$file"|grep -q '^List-Id'; then
  16             $SPAM_REPORT < "$file" >/dev/null 2>&1;
  17         fi;
  18         formail -c < "$file" | grep -e '^From ' -e 'From: ' | spamassassin --add-to-blacklist >/dev/null 2>&1;
  19         formail -c < "$file" | grep -e '^From ' -e 'From: ' | spamassassin --remove-from-whitelist >/dev/null 2>&1;
  20         # check to see if it's still spam
  21         if ! $SPAM_CHECK < "$file" >/dev/null 2>&1; then
  22             TMP="$(mktemp -d)"
  23             trap "rm -f '$TMP'/file; rmdir '$TMP'" EXIT
  24             # this message is still not spam; may need custom rules
  25             spamc < "$file" > "$TMP/file" 2>/dev/null;
  26             mv "$TMP/file" ~/Maildir/spam/needs_rules/cur/"$(basename "$file")";
  27             trap - EXIT;
  28             rmdir "$TMP";
  29         fi;
  30     done;
  31 elif [ "$(basename $0)" = "learn_from_ham" ]; then
  32     for file in "$@"; do
  33         $HAM_REPORT < "$file" >/dev/null 2>&1;
  34         formail -c < "$file" |grep -e '^From ' -e 'From: ' -e 'To: ' -e 'Cc: '|spamassassin --add-to-whitelist >/dev/null 2>&1;
  35         if $SPAM_CHECK < "$file" >/dev/null 2>&1; then
  36             mkdir -p ~/Maildir/spam/needs_ham_rules/cur;
  37             TMP="$(mktemp -d)"
  38             trap "rm -f '$TMP'/file; rmdir '$TMP'" EXIT
  39             spamc < "$file" > "$TMP/file" 2>/dev/null;
  40             mv "$TMP/file" ~/Maildir/spam/needs_ham_rules/cur/"$(basename "$file")";
  41             trap - EXIT;
  42             rmdir "$TMP";
  43         fi;
  44     done;
  45 else
  46     "Called in a way this script cannot recognize";
  47 fi;