--- loncom/auth/lonroles.pm	2018/04/28 00:57:58	1.337
+++ loncom/auth/lonroles.pm	2018/05/28 23:26:12	1.338
@@ -1,7 +1,7 @@
 # The LearningOnline Network with CAPA
 # User Roles Screen
 #
-# $Id: lonroles.pm,v 1.337 2018/04/28 00:57:58 raeburn Exp $
+# $Id: lonroles.pm,v 1.338 2018/05/28 23:26:12 raeburn Exp $
 #
 # Copyright Michigan State University Board of Trustees
 #
@@ -145,6 +145,8 @@ use GDBM_File;
 use LONCAPA qw(:DEFAULT :match);
 use HTML::Entities;
 
+my $registered_cleanup;
+my $rosterupdates;
 
 sub redirect_user {
     my ($r,$title,$url,$msg) = @_;
@@ -228,6 +230,8 @@ sub handler {
         $update = $then;
     }
 
+    $registered_cleanup=0;
+    @{$rosterupdates}=();
     &Apache::loncommon::get_unprocessed_cgi($ENV{'QUERY_STRING'});
 
 # -------------------------------------------------- Check if setting hot list 
@@ -649,6 +653,10 @@ ENDENTERKEY
                                                                                        $cdom.'_'.$cnum);
                                     }
                                 }
+                                if (($env{'request.lti.login'}) &&
+                                    ($env{'request.lti.rosterid'} || $env{'request.lti.passbackid'})) {
+                                    &process_lti($r,$cdom,$cnum);
+                                }
 				$r->internal_redirect($dest);
 			    }
 			    return OK;
@@ -670,6 +678,10 @@ ENDENTERKEY
 			    if (($ferr) && ($tadv)) {
 				&error_page($r,$ferr,$furl);
 			    } else {
+                                if (($env{'request.lti.login'}) &&
+                                    ($env{'request.lti.rosterid'} || $env{'request.lti.passbackid'})) {
+                                    &process_lti($r,$cdom,$cnum);
+                                }
 				# Check to see if the user is a CC entering a course 
 				# for the first time
 				if ((($role eq 'cc') || ($role eq 'co')) 
@@ -3094,6 +3106,84 @@ sub get_queued {
            $output.'</fieldset></div><br clear="all" />';
 }
 
+sub process_lti {
+    my ($r,$cdom,$cnum) = @_;
+    my %lti = &Apache::lonnet::get_domain_lti($cdom,'provider');
+    my $uriscope = &LONCAPA::ltiutils::lti_provider_scope($env{'request.lti.uri'},
+                                                          $cdom,$cnum);
+    my $lonhost = $r->dir_config('lonHostID');
+    my $internet_names = &Apache::lonnet::get_internet_names($lonhost);
+    if ($env{'request.lti.rosterid'} &&
+        $env{'request.lti.rosterurl'}) {
+        if (ref($lti{$env{'request.lti.login'}}) eq 'HASH') {
+            if ($lti{$env{'request.lti.login'}}{'roster'}) {
+                my @lcroles = ('in','ta','ep','st');
+                my @possibleroles;
+                foreach my $role (@lcroles) {
+                    if (&Apache::lonnet::allowed('c'.$role,"$cdom/$cnum")) {
+                        push(@possibleroles,$role);
+                    }
+                }
+                my $owner = $env{'course.'.$cdom.'_'.$cnum.'.internal.courseowner'};
+                if ($owner eq $env{'user.name'}.':'.$env{'user.domain'}) {
+                    my $crstype = &Apache::loncommon::course_type($cdom.'_'.$cnum);
+                    if ($crstype eq 'Community') {
+                        unshift(@possibleroles,'co');
+                    } else {
+                        unshift(@possibleroles,'cc');
+                    }
+                }
+                if (@possibleroles) {
+                    push(@{$rosterupdates},{cid        => $cdom.'_'.$cnum,
+                                            lti        => $env{'request.lti.login'},
+                                            ltiref     => $lti{$env{'request.lti.login'}},
+                                            id         => $env{'request.lti.rosterid'},
+                                            url        => $env{'request.lti.rosterurl'},
+                                            sourcecrs  => $env{'request.lti.sourcecrs'},
+                                            uriscope   => $uriscope,
+                                            possroles  => \@possibleroles,
+                                            intdoms    => $internet_names,
+                                           });
+                    unless ($registered_cleanup) {
+                        my $handlers = $r->get_handlers('PerlCleanupHandler');
+                        $r->set_handlers('PerlCleanupHandler' =>
+                                         [\&ltienroll,@{$handlers}]);
+                        $registered_cleanup=1;
+                    }
+                }
+            }
+        }
+    }
+    if ($env{'request.lti.passbackid'} &&
+        $env{'request.lti.passbackurl'}) {
+        if (ref($lti{$env{'request.lti.login'}}) eq 'HASH') {
+            if ($lti{$env{'request.lti.login'}}{'passback'}) {
+                my ($pbnum,$error) =
+                    &LONCAPA::ltiutils::store_passbackurl($env{'request.lti.login'},
+                                                          $env{'request.lti.passbackurl'},
+                                                          $cdom,$cnum);
+                if ($pbnum eq '') {
+                    $pbnum = $env{'request.lti.passbackurl'};
+                }
+                &Apache::lonnet::put('nohist_'.$cdom.'_'.$cnum.'_passback',
+                                     {"$uriscope\0$env{'request.lti.sourcecrs'}\0$env{'request.lti.login'}" =>
+                                     "$pbnum\0$env{'request.lti.passbackid'}"});
+            }
+        }
+    }
+    return;
+}
+
+sub ltienroll {
+    if (ref($rosterupdates) eq 'ARRAY') {
+        foreach my $item (@{$rosterupdates}) {
+            if (ref($item) eq 'HASH') {
+                &LONCAPA::ltiutils::batchaddroster($item);
+            }
+        }
+    }
+}
+
 1;
 __END__