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 # needs rules directories
  14 NEEDS_SPAM_RULES=~/Maildir/spam/needs_rules/cur/
  15 NEEDS_HAM_RULES=~/Maildir/spam/needs_ham_rules/cur/
  16
  17 if [ "$(basename $0)" = "learn_from_spam" ]; then
  18     for file in "$@"; do
  19         if ! formail -c < "$file"|grep -q '^List-Id'; then
  20             $SPAM_REPORT < "$file" >/dev/null 2>&1;
  21         fi;
  22         formail -c < "$file" | grep -e '^From ' -e 'From: ' | spamassassin --add-to-blacklist >/dev/null 2>&1;
  23         formail -c < "$file" | grep -e '^From ' -e 'From: ' | spamassassin --remove-from-whitelist >/dev/null 2>&1;
  24         # check to see if it's still spam
  25         if ! $SPAM_CHECK < "$file" >/dev/null 2>&1; then
  26             TMP="$(mktemp -d)"
  27             trap "rm -f '$TMP'/file; rmdir '$TMP'" EXIT
  28             # this message is still not spam; may need custom rules
  29             spamc < "$file" > "$TMP/file" 2>/dev/null;
  30             mv "$TMP/file" "${NEEDS_SPAM_RULES}$(basename "$file")";
  31             trap - EXIT;
  32             rmdir "$TMP";
  33         elif [ "$file" = "${NEEDS_SPAM_RULES}$(basename "$file")" ]; then
  34             rm -f "$file";
  35         fi;
  36     done;
  37 elif [ "$(basename $0)" = "learn_from_ham" ]; then
  38     for file in "$@"; do
  39         $HAM_REPORT < "$file" >/dev/null 2>&1;
  40         formail -c < "$file" |grep -e '^From ' -e 'From: ' -e 'To: ' -e 'Cc: '|spamassassin --add-to-whitelist >/dev/null 2>&1;
  41         if $SPAM_CHECK < "$file" >/dev/null 2>&1; then
  42             mkdir -p ~/Maildir/spam/needs_ham_rules/cur;
  43             TMP="$(mktemp -d)"
  44             trap "rm -f '$TMP'/file; rmdir '$TMP'" EXIT
  45             spamc < "$file" > "$TMP/file" 2>/dev/null;
  46             mv "$TMP/file" "${NEEDS_HAM_RULES}$(basename "$file")";
  47             trap - EXIT;
  48             rmdir "$TMP";
  49         elif [ "$file" = "${NEEDS_HAM_RULES}$(basename "$file")" ]; then
  50             rm -f "$file";
  51         fi;
  52     done;
  53 else
  54     "Called in a way this script cannot recognize";
  55 fi;