#!/usr/bin/perl
#
# Copyright (c) 1994-2008 Carnegie Mellon University.  All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in
#    the documentation and/or other materials provided with the
#    distribution.
#
# 3. The name "Carnegie Mellon University" must not be used to
#    endorse or promote products derived from this software without
#    prior written permission. For permission or any legal
#    details, please contact
#      Carnegie Mellon University
#      Center for Technology Transfer and Enterprise Creation
#      4615 Forbes Avenue
#      Suite 302
#      Pittsburgh, PA  15213
#      (412) 268-7393, fax: (412) 268-7395
#      innovation@andrew.cmu.edu
#
# 4. Redistributions of any form whatsoever must retain the following
#    acknowledgment:
#    "This product includes software developed by Computing Services
#     at Carnegie Mellon University (http://www.cmu.edu/computing/)."
#
# CARNEGIE MELLON UNIVERSITY DISCLAIMS ALL WARRANTIES WITH REGARD TO
# THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS, IN NO EVENT SHALL CARNEGIE MELLON UNIVERSITY BE LIABLE
# FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN
# AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
# OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

require 5;

#use strict; # XXX does not compile with strict :(
use warnings;

$| = 1;

die "must not run as root" if ($< == 0);

if(@ARGV < 1) {
    print "usage: masssievec <path to sievec> [imapd.conf]\n";
    exit;
}

$SIEVEC = shift @ARGV;
$imapdconf = shift @ARGV;
if(!defined($imapdconf)) {
    $imapdconf = "/etc/imapd.conf";
}

$sievedir = "/var/sieve";

if(! -x $SIEVEC) {
    print "$SIEVEC is not executable\n";
    exit;
}

if($SIEVEC !~ /^\//) {
    print "$SIEVEC is not an absolute path\n";
    exit;
}

sub read_conf {
    my $file = shift;

    open CONF, $file or die "can't open $file";
    while (<CONF>) {
        if (/^#/) {
            next;
        }
        if (/\@include:\s+(.*)$/) {
            push @configs, $1;
        }
        if (/^sieveusehomedir:\s+(1|t|yes|on)/) {
            print "you are storing sieve scripts in user's home directories, this script cannot deal with that\n";
            exit;
        }
        if (/^sievedir:\s+(.*)$/) {
            $sievedir = $1;
        }
    }
    close CONF;
}

push @configs, $imapdconf;

while ($conf = shift @configs) {
    read_conf($conf);
}

print "you are using $sievedir as your sieve directory.\n";

opendir TOP, $sievedir;
while (defined($s = readdir TOP)) {
    next if ($s eq "." || $s eq "..");
    chdir $sievedir . "/$s";
    opendir THISONE, ".";

    while(defined($t = readdir THISONE)) {
        next if ($t eq "." || $t eq "..");
        print "processing user $t\n";
        chdir $t;

        opendir USER, ".";
        while(defined($u = readdir USER)) {
            next if ($u eq "." || $u eq "..");
            if($u eq "default" && -l $u) {
                # special case
                $dest = readlink $u;
                next unless($dest =~ m/\.script$/);
                $dest =~ s/\.script$//;
                symlink "$dest.bc", "defaultbc" || warn "can't symlink $dest.bc to defaultbc: $!";
                unlink ("default");
            } elsif ($u eq "default.bc" && -l $u) {
                # slightly different upgrade format
                $dest = readlink $u;
                next unless($dest =~ m/\.bc$/ && $dest ne "default.bc");
                symlink "$dest", "defaultbc" || warn "can't symlink $dest to defaultbc: $!";
                unlink ("default.bc");
            } elsif ($u eq "default" || $u eq "default.bc") {
                warn "$u is not a symlink";
            } else {
                next unless($u =~ m/\.script$/);
                $out = $u;
                $out =~ s/\.script$//;

        FORK: {
                if($pid = fork()) {
                    #parent, do nothing
                    waitpid $pid, 0;
                } elsif (defined $pid) {
                    # child
                    exec $SIEVEC, $u, "$out.bc";
                    die "shouldn't be here";
                } elsif ($! =~ /No more process/) {
                    # EAGAIN
                    sleep 5; redo FORK;
                } else {
                    die "cant fork: $!";
                }
              }

                $rc = $? & 0xff00;
                $rc >>= 8;
                if($rc) {
                    print "got error compiling $u.\n";
                }
            }
        }
        chdir $sievedir . "/$s";
    }
    closedir THISONE;
}
closedir TOP
