#!/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";
}