source: trunk/third/moira/regtape/grouper.pl @ 24319

#!/moira/bin/perl -Tw
# $Id: grouper.pl 3956 2010-01-05 20:56:56Z zacheiss $

die "Usage: $0 password\n" unless ($#ARGV == 0);
$whpassword = $ARGV[0];
$db = "";
$mrtest = "mrtest";
$logfile = "/moira/grouper.log";
$ENV{'PATH'} = "/moira/bin";
use DBI;

$warehouse = DBI->connect("dbi:Oracle:warehouse", "moira", $whpassword,
                          { RaiseError => 1 });
$moira = DBI->connect("dbi:Oracle:moira", "moira", "moira",
                      { RaiseError => 1});

# Get the current term
($term) = $warehouse->selectrow_array("SELECT term_code ".
                                    "FROM wareuser.whsis_academic_terms ".
                                    "WHERE is_current_term='Y'");
# Convert from "2000FA" to "FA00"
$term =~ s/\d\d(\d\d)(..)/$2$1/;

# Get list of current classes
$classes =
    $warehouse->selectcol_arrayref("SELECT UNIQUE master_subject ".
                                   "FROM wareuser.subject_enrollment_moira ".
                                   "WHERE term = " . $warehouse->quote($term));

# Get names of current Grouper lists
$sth = $moira->prepare("SELECT name FROM list");
$sth->execute;
while (($name) = $sth->fetchrow_array) {
    next if $name !~ /^(fa|sp|su|ja)\d\d-/;
    $lists{$name} = $name;
}

$lists{"registrar"} = "registrar";

# And MIT ID to username mappings
$sth = $moira->prepare("SELECT login, clearid FROM users ".
                       "WHERE status = 1 OR status = 2");
$sth->execute;
while (($user, $mitid) = $sth->fetchrow_array) {
    $users{$mitid} = $user;
}

($root_id) = $moira->selectrow_array("SELECT users_id FROM users ".
                                   "WHERE login = 'root'");

open(MRTEST, "|$mrtest >/dev/null 2>&1");
print MRTEST "connect $db\n";
print MRTEST "auth\n";
open(LOG, ">>$logfile");

# Create any lists that don't already exist in Moira
foreach $class (@$classes) {
     $base = "\L$term-$class";
     $staff = "$base-staff";

     # check_list(name, owner, export, desc)
     &check_list($staff, $staff, 0, 
                 "Teaching staff list for $class");
     &check_list("$base-reg", $staff, 0,
                 "*** DO NOT MODIFY *** Automatically-created registered students list for $class");
     &check_list("$base-others", $staff, 0,
                 "Non-registered students and miscellaneous people list for $class");
     &check_list($base, $staff, 1,
                 "*** DO NOT MODIFY *** Automatically-created participants list for $class");
     if (!$lists{$base}) {
         &add_member($staff, LIST, $base);
         &add_member("$base-reg", LIST, $base);
         &add_member("$base-others", LIST, $base);
     }
}

# Now fill in -reg lists
foreach $class (@$classes) {
    $changed = 0;
    $clist = "\L$term-$class-reg";

    # Get current list membership in Moira 
    %mstudents = (); 
    $sth = $moira->prepare("SELECT u.login FROM users u, imembers i, list l ".
                           "WHERE l.list_id = i.list_id AND i.member_id = ".
                           "u.users_id AND i.direct = 1 AND i.member_type = ".
                           "'USER' AND l.name = " . $moira->quote($clist)); 
    $sth->execute; 
    while (($login) = $sth->fetchrow_array) { 
        $mstudents{$login} = $login; 
    }

    $wstudents = $warehouse->selectcol_arrayref("SELECT UNIQUE mit_id ".
                             "FROM wareuser.subject_enrollment_moira ".
                             "WHERE term = " . $warehouse->quote($term) .
                             " AND master_subject = " . 
                             $warehouse->quote($class));

    foreach $mitid (@$wstudents) {
        $login = $users{$mitid};
        next if !$login;
        if (!$mstudents{$login}) {
            print LOG "Adding $login to $clist\n";
            &add_member($login, USER, $clist);
            $changed = 1;
        } else {
            delete $mstudents{$login};
        }
    }

    # Everyone in wstudents will have been removed from mstudents
    # now, so delete the remaining users since they don't belong
    foreach $login (keys(%mstudents)) {
        print LOG "Deleting $login from $clist\n";
        &del_member($login, USER, $clist);
        $changed = 1;
    }

    if ($changed) {
        $moira->do("UPDATE list SET modtime = SYSDATE, modby = $root_id, ".
                   "modwith = 'grouper' WHERE name = " . 
                   $moira->quote($clist));
    }
}

print MRTEST "quit\n";
close(MRTEST);
close(LOG);
$moira->disconnect;
$warehouse->disconnect;

exit 0;

sub check_list {
    my ( $name, $owner, $export, $desc ) = @_;
    if (!$lists{$name}) {
        print LOG "Creating $name\n";
        print MRTEST "qy alis $name 0 0 1 $export $export \"create unique GID\" 0 0 [NONE] LIST $owner NONE NONE \"$desc\"\n";
    }
}

sub add_member {
    my ( $user, $type, $list ) = @_;
    print MRTEST "qy amtl $list $type $user\n";
}

sub del_member {
    my ( $user, $type, $list ) = @_;
    print MRTEST "qy dmfl $list $type $user\n";
}