# $Id$

$VERSION = "1.8";
%IRSSI = (
    authors     => "Matti Hiljanen",
    contact     => "qvr\@staff.peliportti.net",
    name        => "mail-maildir",
    description => "Mail counter statusbar item with maildir support",
    license     => "Public Domain",
    url         => "http://matin.maapallo.org/softa/irssi",
);

# Mail counter statusbar item
# for irssi 0.7.99 by Timo Sirainen
#
# Maildir support added by Matti Hiljanen
# 
#  /SET maildir_mode - ON/OFF 
#  /SET mail_file - specifies mbox file/Maildir location
#  /SET mail_refresh_time - in seconds, how often to check for new mail

use Irssi::TextUI;

my $maildirmode = 0; # maildir=1, file(spools)=0

my $last_refresh_time, $refresh_tag;

# for mbox caching
my $last_size, $last_mtime, $last_mailcount, $last_mode;

sub mbox_count {
  my $mailfile = shift;
  my $count;
  my $maildirmode=Irssi::settings_get_bool('maildir_mode'); 
  if (!$maildirmode) {
     my @stat = stat($mailfile);
     my $size = $stat[7];
     my $mtime = $stat[9];

     # if the file hasn't changed, get the count from cache
     return $last_mailcount if ($last_size == $size && $last_mtime == $mtime);
     $last_size = $size;
     $last_mtime = $mtime;

     my $f = gensym;
     return 0 if (!open($f, $mailfile));

     while (<$f>) {
        $count++ if (/^From /);
        $count-- if (/^Subject: .*FOLDER INTERNAL DATA/);
     }
     close($f);
     $last_mailcount = $count;
  } else {
     opendir(DIR, "$mailfile/cur") or return 0;
     while (defined(my $file = readdir(DIR))) {
        next if $file =~ /S/ || $file =~ /^(.|..)$/;
        $count++;
     }
     closedir(DIR);

     opendir(DIR, "$mailfile/new") or return 0;
     while (defined(my $file = readdir(DIR))) {
        next if $file =~ /^(.|..)$/;
        $count++;
     }
     closedir(DIR);
  }
  return $count;
}

sub mail {
  my ($item, $get_size_only) = @_;

  $count = mbox_count(Irssi::settings_get_str('mail_file'));
  if ($count == 0) {
    # no mail - don't print the [Mail: ] at all
    if ($get_size_only) {
      $item->{min_size} = $item->{max_size} = 0;
    }
  } else {
    $item->default_handler($get_size_only, undef, $count, 1);
  }
}

sub refresh_mail {
  Irssi::statusbar_items_redraw('mail');
}

sub read_settings {
  my $time = Irssi::settings_get_int('mail_refresh_time');
  my $mode = Irssi::settings_get_bool('maildir_mode');
  unless ($time == $last_refresh_time) {
     $last_refresh_time = $time;
     Irssi::timeout_remove($refresh_tag) if ($refresh_tag);
     $refresh_tag = Irssi::timeout_add($time*1000, 'refresh_mail', undef);
  }
  return if ($mode == $last_mode);
  $last_mode = $mode;
  if (!$mode) {
     Irssi::settings_set_str('mail_file', "$ENV{'MAIL'}");
  } else {
     Irssi::settings_set_str('mail_file', "$ENV{'HOME'}/Maildir");
  }
  refresh_mail;
}

if (!$maildirmode) {
   Irssi::settings_add_str('misc', 'mail_file', $ENV{'MAIL'});
} else {
   Irssi::settings_add_str('misc', 'mail_file', "$ENV{'HOME'}/Maildir");
}
Irssi::settings_add_int('misc', 'mail_refresh_time', 60);
Irssi::settings_add_bool('misc', 'maildir_mode', "$maildirmode");

Irssi::statusbar_item_register('mail', '{sb Mail: $0-}', 'mail');

read_settings();
Irssi::signal_add('setup changed', 'read_settings');