#!/usr/bin/env python
# *public domain*

import sys, os, time

INTERVAL=60

class ScreenNotifier:
  statusline = "%{= W}%m/%d %D %c %{= R}@@MAIL@@%{= W} %-Lw%{= BW}%50>%n%f* %t%{-}%+Lw%<"
  
  def __init__(self):
    self.mails = -1
    self.notify(0)
    return

  def notify(self, n):
    if self.mails == n:
      return
    if n:
      s = ScreenNotifier.statusline.replace('@@MAIL@@', '<%d>' % n);
    else:
      s = ScreenNotifier.statusline.replace('@@MAIL@@', '');
    os.system("screen -X hardstatus string '%s'" % s);
    self.mails = n
    return

# main
if __name__ == "__main__":
  dir = sys.argv[1]+"/new"
  scr = ScreenNotifier()
  while 1:
    scr.notify(len(os.listdir(dir)))
    time.sleep(INTERVAL)