bin/retrain

   1 #!/bin/sh
   2 set -eu
   3
   4 NICE='/usr/bin/nice -20'
   5 NICE=''
   6
   7 RESUBMIT="$NICE $HOME/.etc/mailfilter/bin/resubmit"
   8
   9 PAUSETIME=10
  10
  11 HAM=0
  12 SPAM=0
  13
  14 maildir=
  15 files=
  16 for i in "$@"; do
  17   case "$i" in
  18     --ham) HAM=1;;
  19     --spam) SPAM=1;;
  20     *)
  21       if [ -f "$i" ] && [ -r "$i" ]; then
  22         files="$files $i"
  23       elif [ -d "$i"/cur ] && [ -d "$i"/new ] && [ -d "$i"/tmp ]; then
  24         if [ -n "$maildir" ]; then
  25           echo "E: Maildir already specified: $i" >&2
  26           exit 1
  27         fi
  28         maildir="$i"
  29       else
  30         echo "E: unknown argument: $i" >&2
  31         exit 255
  32       fi
  33       ;;
  34   esac
  35 done
  36
  37 ARG=
  38 case "$HAM/$SPAM" in
  39   0/0)
  40     echo "E: you need to specify either --ham or --spam." >&2
  41     exit 255
  42     ;;
  43   1/1)
  44     echo "E: you cannot specify both --ham and --spam." >&2
  45     exit 255
  46     ;;
  47   0/1) ARG=--spam;;
  48   1/0) ARG=--ham;;
  49 esac
  50
  51 if [ -n "$maildir" ]; then
  52   ME="${0##*/}"
  53   LOCKFILE="$maildir/.$ME.lock"
  54   trap "rm -f $LOCKFILE" 1 2 3 4 5 6 7 8 10 11 12 13 14 15
  55   if ! lockfile -0 -r0 -l 3600 "$LOCKFILE" 2>/dev/null; then
  56     echo "E: another $ME process is already running over that Maildir." >&2
  57     trap - 0 1 2 3 4 5 6 7 8 10 11 12 13 14 15
  58     exit 1
  59   fi
  60
  61   files="$files $(find $MAILDIR/cur $MAILDIR/new -type f)"
  62 fi
  63
  64 for msg in $files; do
  65   $NICE ${0%/*}/train $ARG "$msg" | $RESUBMIT || ret=$?
  66   case "${ret:-0}" in
  67     0|75) :;;
  68     *)
  69       echo "E: resubmission returned with exit code $ret" >&2
  70       continue
  71       ;;
  72   esac
  73   rm -f "$msg"
  74   sleep $PAUSETIME
  75 done
  76
  77 if [ -n "${LOCKFILE:-}" ]; then
  78   rm -f "$LOCKFILE"
  79   trap - 1 2 3 4 5 6 7 8 10 11 12 13 14 15
  80 fi
  81
  82 exit 0