Output individual MHC files if a directory is specified
[tpope-extra.git] / perl / sctweb
1 #!/usr/bin/perl -w
2 # $Id$
3 # -*- perl -*- vim: ft=perl sw=4 sts=4
4
5 # Brief usage instructions:
6 # Create a ~/.sctwebrc that has SID=yourssn and PIN=yourpin
7 # You'll need to change the url below if you go anywhere but Northwest Vista.
8
9 use strict;
10 use LWP::UserAgent;
11 use HTML::TableExtract;
12 #use Time::Local;
13 use Date::Calc::Object qw(Day_of_Week Decode_Day_of_Week Delta_Days);
14 use Date::Calendar::Profiles qw($Profiles);
15 use Date::Calendar::Year;
16 use vars qw($ua %opts);
17
18 my ($ts, $row, $response);
19 my $url="https://sctweb.accd.edu/ia-bin4/tsrvweb.exe?WID=W&tserve_tip_read_destroy&tserve_host_code=4&tserve_tiphost_code=0";
20
21 if (-r $ENV{HOME} . "/.sctwebrc") {
22     open CONFIG, $ENV{HOME} . "/.sctwebrc";
23     while(<CONFIG>) {
24         s/\#.*//;
25         next unless m/^([^=]*)=(.*)/;
26         $opts{$1}=$2;
27     }
28     close CONFIG;
29 }
30
31 $ua = LWP::UserAgent->new;
32 $ua->timeout(10);
33 $ua->env_proxy;
34 $ua->cookie_jar( {} );
35 $ua->get("$url&tserve_tip_write=||WID&tserve_transconfig=astulog.cfg") or die "$!";
36 $response = $ua->post("$url&tserve_trans_config=astulog.cfg", { tserve_tip_write => "%7C%7CWID%7CSID%7CPIN", SID => $opts{SID}, PIN => $opts{PIN} }) or die "$!";
37
38 $response = $ua->get("$url&tserve_tip_write=||WID|SID|PIN|Term&tserve_trans_config=" . ("regterm.cfg"||"rgrdterm.cfg")); # Valid terms
39 die $response->status_line unless $response->is_success;
40 my @terms = grep {s/^<option value="([^"]*)">.*/$1/} (split( /\r\n/, $response->content));
41 die "Site down. Try again later.\n" unless (@terms);
42
43 sub generate_id {
44     my ($section, $number) = split("-", shift);
45     my $id = 0;
46     foreach (split //, $section) {$id=26*$id+(ord($_)-1)%32;}
47     $id=10000*$id+$number;
48     return $id;
49 }
50
51 sub off_for_holidays {
52     my %class = @_;
53     $class{'duration'} =~ /(\d\d\d\d)(\d\d)(\d\d)-(\d\d\d\d)(\d\d)(\d\d)/;
54     my $days = Delta_Days($1,$2,$3,$4,$5,$6);
55     my $firstday = Date::Calc->new($1,$2,$3);
56     my $lastday = Date::Calc->new($4,$5,$6);
57     my $year = Date::Calendar::Year->new($1, $Profiles->{'US-TX'});
58     my @holidays=();
59     my @days=();
60     foreach(split(" ",$class{days})) {
61         $days[Decode_Day_of_Week($_)] = 1;
62     }
63     foreach ($year->search("")) {
64         next unless ($year->is_full($_) && $_>=$firstday && $_<=$lastday);
65         if (defined($days[Day_of_Week($_->date)])) {
66             push @holidays, $_;
67         }
68     }
69     return wantarray ? @holidays : "@holidays";
70 }
71
72 sub off_for_exams {
73     my %class = @_;
74     my $time = $class{'begin'} . "-" . $class{'end'};
75     $class{'duration'} =~ /\d{8}-(\d\d\d\d)(\d\d)(\d\d)/;
76     my $lastday = Date::Calc->new($1,$2,$3);
77     return wantarray ? () : undef unless (Day_of_Week($lastday->date) == 7);
78     my $beginning="(08:00-09:15|11:00-12:15|14:00-15:15|17:00-18:15|17:30-18:45|20:00-21:15|20:30-21:45)";
79     my $ending   ="(06:30-07:45|09:30-10:45|12:30-13:45|15:30-16:45|18:30-19:45|19:00-20:15)";
80     my $days = join(" ",map {Decode_Day_of_Week($_)} split(/ /, $class{'days'}));
81     if($days eq "1 3") {
82         if($time =~ /$beginning/) {
83             return $lastday-4;
84         } elsif($time =~ /$ending/) {
85             return $lastday-6;
86         }
87     } elsif($days eq "2 4") {
88         if($time =~ /$beginning/) {
89             return $lastday-3;
90         } elsif($time =~ /$ending/) {
91             return $lastday-5;
92         }
93     } else {
94         return wantarray ? () : "";
95     }
96 }
97
98 sub capitalize {
99     local $_ = shift;
100     s/\b([A-Z])([A-Z]*)\b/$1\L$2/g;
101     s/\b(I)(i*)\b/$1\U$2/g;
102     s/\bUs\b/US/g;
103     s/ (And|For|Of|Or|The|To|With) / \l$1 /g;
104     s/\b(Ma?c)([a-z])/$1\u$2/g;
105     s/\b(Tcp\/Ip|Pc)\b/\U$&/g;
106     s/\bThru\b/Through/g;
107     s/\bAcct\b/Accounting/g;
108     s/\bAmer\b/American/g;
109     s/\bChem\b/Chemistry/g;
110     s/\bComp\b/Composition/g;
111     s/\bFed\b/Federal/g;
112     s/\bGen\b/General/g;
113     #s/\bGovt\b/Government/g;
114     s/\bLit\b/Literature/g;
115     s/\bPrin\b/Principles/g;
116     return $_;
117 }
118
119 sub get_schedule_terms {
120     my @terms;
121     my @localtime=localtime();
122     if ($localtime[4]<3) { # Through Mar 31
123         @terms=((1900+$localtime[5])*10+2);
124     } elsif ($localtime[4]<5) { # Through May 31
125         @terms=((1900+$localtime[5])*10+2,(1900+$localtime[5])*10+3);
126     } elsif ($localtime[4]<7) { # through July 31
127         @terms=((1900+$localtime[5])*10+3,(1900+$localtime[5])*10+4);
128     } elsif ($localtime[4]<8) { # through Aug 31
129         @terms=((1900+$localtime[5])*10+4,(1900+$localtime[5])*10+1);
130     } elsif ($localtime[4]<10) { # through Oct 31
131         @terms=((1900+$localtime[5])*10+1);
132     } else {
133         @terms=((1900+$localtime[5])*10+1,(1901+$localtime[5])*10+2);
134     }
135     return @terms;
136 }
137
138 sub get_schedule {
139     my @readheaders = ("Section ID/Title", "Session", "Instructor", "Credits", "CallNumber","GradeType", "Days", "Time", "Start/EndDates", "Site/Building/Room", "Message");
140     my $te = new HTML::TableExtract( headers => [ @readheaders ] );
141     my (@schedule, @terms, $classid, $begin, $end);
142     @terms = get_schedule_terms();
143     foreach(@_ || @terms) {
144         $response = $ua->get("$url&tserve_tip_write=||WID|SID|PIN|Term&tserve_trans_config=rstusch.cfg&Term=".($_) );
145         die $response->status_line unless $response->is_success;
146         $te->parse($response->content);
147     }
148     foreach my $ts ($te->table_states) {
149         foreach my $row ($ts->rows) {
150             map { s/\xa0//g; $_} @$row;
151             $row->[0] =~ s/.*launchWebCT\("([^"]*)"\).*/$1/s;
152             $row->[0] =~ s/(.*) <BR>(.*)/capitalize($2)/eg;
153             $classid=$1;
154             $row->[2] =~ s/([A-Z]r?)$/$1./;
155             ($begin, $end) = split("-", $row->[7]);
156             if($row->[7] =~ /^(\d\d):\d\d-(\d\d):\d\dPM$/) {
157                 $begin =~ s/^(\d\d)/$1+12/e
158                 if($1 < $2);
159                 $end =~ s/^(\d\d)/$1+12/e;
160                 $end =~ s/PM$//;
161             } else { $end =~ s/AM$//; }
162             $row->[9] =~ s/ON CAMPUS\n//g;
163             $row->[9] =~ s/\n/ /g;
164             push @schedule, {
165                 id => $classid,
166                 title => $row->[0],
167                 session => $row->[1],
168                 instructor => capitalize($row->[2]),
169                 credits => $row->[3],
170                 callnumber => $row->[4],
171                 gradetype => $row->[5],
172                 days => $row->[6],
173                 'time' => $row->[7],
174                 begin => $begin,
175                 end => $end || undef,
176                 duration => $row->[8],
177                 location => $row->[9],
178                 message => $row->[10],
179             };
180         }
181     }
182     return @schedule;
183 }
184
185 sub get_mhc_header {
186 return (
187 "X-SC-Subject: New Years Day\nX-SC-Category: Holiday\nX-SC-Cond: 1 Jan\nX-SC-Record-Id: <New_Years_Day\@from.sctweb>\n",
188 "X-SC-Subject: Martin Luther King, Jr. Day\nX-SC-Category: Holiday\nX-SC-Cond: 3rd Mon Jan\nX-SC-Duration: 19861102-\nX-SC-Record-Id: <Martin_Luther_King_Jr_Day\@from.sctweb>\n",
189 "X-SC-Subject: Presidents Day\nX-SC-Category: Holiday\nX-SC-Cond: 3rd Mon Feb\nX-SC-Duration: 19710515-\nX-SC-Record-Id: <Presidents_Day\@from.sctweb>\n",
190 "X-SC-Subject: Memorial Day\nX-SC-Category: Holiday\nX-SC-Cond: Last Mon May\nX-SC-Duration: 19710531-\nX-SC-Record-Id: <Memorial_Day\@from.sctweb>\n",
191 "X-SC-Subject: Independence Day\nX-SC-Category: Holiday\nX-SC-Cond: 4 Jul\nX-SC-Duration: 17760704-\nX-SC-Record-Id: <Independence_Day\@from.sctweb>\n",
192 "X-SC-Subject: Labor Day\nX-SC-Category: Holiday\nX-SC-Cond: 1st Mon Sep\nX-SC-Duration: 18840901-\nX-SC-Record-Id: <Labor_Day\@from.sctweb>\n",
193 "X-SC-Subject: Columbus Day\nX-SC-Category: Holiday\nX-SC-Cond: 2nd Mon Oct\nX-SC-Duration: 19711011-\nX-SC-Record-Id: <Columbus_Day\@from.sctweb>\n",
194 "X-SC-Subject: Veterans Day\nX-SC-Category: Holiday\nX-SC-Cond: 11 Nov\nX-SC-Duration: 19261111-\nX-SC-Record-Id: <Veterans_Day\@from.sctweb>\n",
195 "X-SC-Subject: Thanksgiving\nX-SC-Category: Holiday\nX-SC-Cond: 4th Thu Nov\nX-SC-Record-Id: <Thanksgiving\@from.sctweb>\n",
196 "X-SC-Subject: Christmas\nX-SC-Category: Holiday\nX-SC-Cond: 25 Dec\nX-SC-Record-Id: <Christmas\@from.sctweb>\n",
197 );
198 }
199
200 sub print_mhc_header {
201 my $header .= join("\n", get_mhc_header) . "\n";
202     if(defined($_[0])) {
203         open SCHEDULE, (">" . $_[0]);
204         print SCHEDULE $header;
205         close SCHEDULE;
206     } else {
207         print $header;
208     }
209 }
210
211 sub do_mhc_schedule {
212     $| = 1;
213     my %days = (M => "Mon", T => "Tue", W => "Wed", R => "Thu", F => "Fri", S => "Sat", U => "SU");
214     my (@terms, $file, $current, @mhc, @schedule);
215     $file = shift unless (!defined $_[0] or $_[0] =~ /^\d/);
216     @schedule = get_schedule(@_);
217     @mhc = get_mhc_header;
218     #print_mhc_header($file);
219     foreach my $row (@schedule) {
220         map {s/\n/-/g if defined; $_} %$row;
221         my $id=generate_id($row->{'id'});
222         $row->{'days'} =~ s/([MTWRFS])/ $days{$1}/g;
223         $row->{'days'} =~ s/^ //;
224         $row->{'duration'} =~ s/(\d\d)-(\d\d)-(\d\d)/20$3$1$2/g;
225         my @day = map { "!" . $_ } (off_for_holidays(%$row),off_for_exams(%$row));
226         $current = "";
227         #print "# $id\n";
228         $current .= "X-SC-Subject: " . $row->{'title'} . "\n";
229         $current .= "X-SC-Location: " . $row->{'location'} . "\n";
230         $current .= "X-SC-Category: School\n";
231         $current .= "X-SC-Cond: " . $row->{'days'} . "\n";
232         $current .= "X-SC-Time: ".$row->{'begin'}."-".$row->{'end'}."\n";
233         $current .= "X-SC-Duration: " . $row->{'duration'} . "\n";
234         $current .= "X-SC-Day: @day\n" if(exists $day[0]);
235         $current .= "X-SC-Alarm: 20 minutes\n";
236         $current .= "X-SC-Record-Id: <".$row->{'id'}."\@from.sctweb>\n";
237         #print "\n$current";
238         push @mhc, $current;
239     }
240     if(defined($file) && (-d $file)) {
241         my @lines;
242         foreach my $name (<$file/[1-9]*>) {
243             next unless $name =~ /^$file\/[1-9][0-9]*$/;
244             open FH, $name;
245             @lines = <FH>;
246             close FH;
247             foreach (@lines) {
248                 unlink $name if /^X-SC-Record-Id: <.*\@from.sctweb>/;
249             }
250         }
251         my $i=0;
252         foreach (@mhc) {
253             while(-f ++$i) {}
254             open FH, ">$file/$i" or die $!;
255             print FH $_;
256             close FH;
257         }
258     } else {
259         open(STDOUT, ">>" . $file) if(defined($file));
260         print "# MHC school schedule\n# Autogenerated by sctweb ".localtime()."\n\n";
261         print join("\n", @mhc);
262     }
263 }
264
265 sub do_vcalendar_schedule {
266     $| = 1;
267     my %days = (M => "MO", T => "TU", W => "WE", R => "TH", F => "FR", S => "SA", U => "SU");
268     my $file = shift if (defined $_[0] and $_[0] !~ /^\d/);
269     my @schedule = get_schedule(@_);
270     open(STDOUT, ">>" . $file) if(defined($file) && (! -d $file));
271     open(STDOUT, ">/dev/null") if(defined($file) && (-d $file));
272     print "BEGIN:VCALENDAR\r\nVERSION:1.0\r\n";
273     foreach my $row (@schedule) {
274         map { s/\n/-/g; $_} %$row;
275         $row->{'days'} =~ s/([MTWRFS])/ $days{$1}/g;
276         $row->{'days'} =~ s/^ //;
277         $row->{'duration'} =~ s/(\d\d)-(\d\d)-(\d\d)/20$3$1$2/g;
278         my @day = (off_for_holidays(%$row),off_for_exams(%$row));
279         my $day = "";
280         if(exists($day[0])) {
281             $day = join(";", @day);
282             $day =~ s/\b(\d{8})\b/$1T000000/g;
283         }
284         my ($starttime, $stoptime)=($row->{'begin'}, $row->{'end'});
285         my ($startdate, $stopdate)=split(/-/, $row->{'duration'});
286         $starttime =~ s/://;
287         $stoptime =~ s/://;
288         if(defined($file) && (-d $file)) {
289             open FH, ">$file/" . $row->{'id'} . ".vcs" or die "$!";
290             select FH;
291             print "BEGIN:VCALENDAR\r\nVERSION:1.0\r\n";
292         }
293         print "BEGIN:VEVENT\r\n";
294         print "SUMMARY:", $row->{'title'}, "\r\n";
295         print "DESCRIPTION:", $row->{'id'}, "\r\n";
296         print "LOCATION:", $row->{'location'}, "\r\n";
297         print "CATEGORIES:Education\r\n";
298         print "DTSTART:", $startdate."T".$starttime, "00\r\n";
299         print "DTEND:", $startdate."T".$stoptime, "00\r\n";
300         print "RRULE:W1 ", $row->{'days'} . " $stopdate", "T000000\r\n";
301         print("EXDATE:$day\r\n") if($day);
302         print "ATTENDEE;ROLE=OWNER;STATUS=CONFIRMED:", $opts{NAME}, "\r\n" if(defined($opts{NAME}));
303         print "END:VEVENT\r\n";
304         if(defined($file) && (-d $file)) {
305             print "END:VCALENDAR\r\n";
306             close FH;
307             select STDOUT;
308         }
309     }
310     print "END:VCALENDAR\r\n";
311 }
312
313 sub do_html_schedule {
314     my @showheaders = ("Section ID/Title", "Instructor", "Days", "Time", "Duration", "Location");
315     my @schedule = get_schedule(@_);
316     print '<table cellpadding="3" cellspacing="0">'."\n<tr><th>";
317     print join("</th><th>",@showheaders);
318     print "</th></tr>\n";
319         foreach my $row (@schedule) {
320             map { s/\n/<br \/>/g; $_} %$row;
321             $row->{'time'} =~ s/-/-<wbr \/>/;
322             print '<tr>';
323             print "<td>" . $row->{'id'} . "<br />" . $row->{'title'} . "</td>";
324             print "<td>" . $row->{'instructor'} . "</td>";
325             print "<td>" . $row->{'days'} . "</td>";
326             print "<td>" . $row->{'time'} . "</td>";
327             print "<td>" . $row->{'duration'} . "</td>";
328             print "<td>" . $row->{'location'} . "</td>";
329             print "</tr>\n";
330         }
331     print "</table>\n";
332 }
333
334 sub do_html_grades {
335     my @readheaders = ("Section ID", "Course Title", "Grade", "Earned  Hours", "Quality  Hours", "Quality  Points", "GPA");
336     my @showheaders = ("Section ID", "Course Title", "Grade", "Earned<br />Hours", "Quality<br />Hours", "Quality<br />Points", "GPA");
337     my ($row, $lastrow);
338     my $te = new HTML::TableExtract( headers => [ @readheaders ] );
339     my $response = $ua->get("$url&tserve_tip_write=||WID|SID|PIN|Term&tserve_trans_config=rgrdterm.cfg"); # Valid grading terms
340     die $response->status_line unless $response->is_success;
341     my @terms = grep {s/^<option value="([^"]*)">.*/$1/} (split( /\r\n/, $response->content));
342     foreach(@ARGV || reverse @terms) {
343         $response = $ua->get("$url&tserve_tip_write=||WID|SID|PIN|Term&tserve_trans_config=rgrades.cfg&Term=".($_) );
344         die $response->status_line unless $response->is_success;
345         $te->parse($response->content);
346     }
347     print '<table cellpadding="3" cellspacing="0">'."\n<tr><th>";
348     print join("</th><th>",@showheaders);
349     print "</th></tr>\n";
350     foreach $ts ($te->table_states) {
351         foreach $row ($ts->rows) {
352             map { s/\xa0//g; s/\n/<br \/>/g; $_} @$row;
353             $row->[1] = capitalize($row->[1]);
354             $lastrow=$row;
355             next if $row->[0] =~ /:/;
356             print '<tr><td>';
357             print join('</td><td>', @$row);
358             print "</td></tr>\n";
359         }
360     }
361     print '<tr><td colspan="2">', capitalize($lastrow->[0]);
362     shift @$lastrow; shift @$lastrow;
363     print "</td><td>";
364     print join('</td><td>', @$lastrow);
365     print "</td></tr>\n";
366     print "</table>\n";
367 }
368
369 sub do_transcripts {
370     $response = $ua->get("$url&tserve_tip_write=||WID|SID|PIN&tserve_trans_config=rtranscr.cfg&CareerReqNum=1");
371     foreach (split /\n/, $response->content) {
372         next unless s/^\&nbsp;|<pre>// || s/\&nbsp;/ /;
373         s/<\/?[Hh]\d>//g;
374         print "$_\n";
375     }
376 }
377
378 my $arg = "";
379 $arg = shift if (defined($ARGV[0]) && $ARGV[0] =~ /^-\w$/);
380
381 if ($arg eq "-g") {
382     do_html_grades(@ARGV);
383 } elsif ($arg eq "-s") {
384     do_html_schedule(@ARGV);
385 } elsif ($arg eq "-m") {
386     do_mhc_schedule(@ARGV);
387 } elsif ($arg eq "-v") {
388     do_vcalendar_schedule(@ARGV);
389 } elsif ($arg eq "-t") {
390     do_transcripts;
391 }
392
393 $ua->get("$url&tserve_trans_config=alogout.cfg"); # Logout