# The LearningOnline Network
# Starts a user off based of an existing token.
#
# $Id: migrateuser.pm,v 1.34 2018/07/04 16:58:19 raeburn Exp $
#
# Copyright Michigan State University Board of Trustees
#
# This file is part of the LearningOnline Network with CAPA (LON-CAPA).
#
# LON-CAPA is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# LON-CAPA is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with LON-CAPA; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
# /home/httpd/html/adm/gpl.txt
#
# http://www.lon-capa.org/
#

package Apache::migrateuser;

use strict;
use LONCAPA qw(:DEFAULT :match);
use Apache::Constants qw(:common :http :methods);
use Apache::lonauth;
use Apache::lonnet;
use Apache::lonlocal;
use Apache::lonlogin();
use Apache::ltiauth;
use CGI::Cookie;

sub goto_login {
    my ($r,$domain,$data) = @_;
    if ((ref($data) eq 'HASH') && ($data->{'lti.login'})) {
        &Apache::ltiauth::invalid_request($r,'22');
    } else {
        &Apache::loncommon::content_type($r,'text/html');
        $r->send_http_header;
        my $url = '/adm/login';
        if ($domain) {
            $url .= '?domain='.$domain;
        }
        $r->print(&Apache::loncommon::start_page('Going to login',undef,
					         {'redirect' => [0,$url],}).
	          '<h1>'.&mt('One moment please...').'</h1>'.
	          '<p>'.&mt('Transferring to login page.').'</p>'.
	          &Apache::loncommon::end_page());
    }
    return OK;
}

sub sso_check {
    my ($data) = @_;
    my %extra_env;
    if (ref($data) eq 'HASH') {
        if ($data->{'sso.login'}) {
            $extra_env{'request.sso.login'} = $data->{'sso.login'};
        }
        if ($data->{'sso.reloginserver'}) {
            $extra_env{'request.sso.reloginserver'} = 
                $data->{'sso.reloginserver'};
        }
    }
    return \%extra_env;
}

sub lti_check {
    my ($data) = @_;
    my %lti_env;
    if (ref($data) eq 'HASH') {
        if ($data->{'lti.login'}) {
            $lti_env{'request.lti.login'} = $data->{'lti.login'};
            if ($data->{'lti.reqcrs'}) {
                $lti_env{'request.lti.reqcrs'} = $data->{'lti.reqcrs'};
            }
            if ($data->{'lti.reqrole'}) {
                $lti_env{'request.lti.reqrole'} = $data->{'lti.reqrole'};
            }
            if ($data->{'lti.selfenrollrole'}) {
                $lti_env{'request.lti.selfenrollrole'} = $data->{'lti.selfenrollrole'};
            }
            if ($data->{'lti.uri'}) {
                $lti_env{'request.lti.uri'} = $data->{'lti.uri'};
            }
            if ($data->{'lti.target'}) {
                $lti_env{'request.lti.target'} = $data->{'lti.target'};
            }
            if ($data->{'lti.sourcecrs'}) {
                $lti_env{'request.lti.sourcecrs'} = $data->{'lti.sourcecrs'};
            }
        }
        if ($data->{'lti.passbackid'}) {
            $lti_env{'request.lti.passbackid'} = $data->{'lti.passbackid'};
        }
        if ($data->{'lti.passbackurl'}) {
            $lti_env{'request.lti.passbackurl'} = $data->{'lti.passbackurl'};
        }
        if ($data->{'lti.rosterid'}) {
            $lti_env{'request.lti.rosterid'} = $data->{'lti.rosterid'};
        }
        if ($data->{'lti.rosterurl'}) {
            $lti_env{'request.lti.rosterurl'} = $data->{'lti.rosterurl'};
        }
    }
    return \%lti_env;
}

sub ip_changed {
    my ($r,$udom,$camefrom,$dataref) = @_;
    &Apache::loncommon::content_type($r,'text/html');
    $r->send_http_header;
    if (ref($dataref) eq 'HASH') {
        my $title = 'LON-CAPA Session redirected';
        my $message = &mt('Your internet address has changed since you logged in.');  
        my $rule_in_effect;
        if ($dataref->{'balancer'}) {
            my $baldom = &Apache::lonnet::host_domain($camefrom);
            my $balprimaryid = &Apache::lonnet::domain($baldom,'primary');
            my $balintdom = &Apache::lonnet::internet_dom($balprimaryid);
            my $uprimaryid = &Apache::lonnet::domain($udom,'primary'); 
            my $uintdom = &Apache::lonnet::internet_dom($uprimaryid);
            my $dom_in_use;
            if (($uintdom ne '') && ($uintdom eq $balintdom)) {
                $dom_in_use = $udom;
            } else {
                $dom_in_use = $baldom;
            }
            my ($result,$cached)=&Apache::lonnet::is_cached_new('loadbalancing',$dom_in_use);
            unless (defined($cached)) {
                my $cachetime = 60*60*24; 
                my %domconfig =
                    &Apache::lonnet::get_dom('configuration',['loadbalancing'],$dom_in_use);
                if (ref($domconfig{'loadbalancing'}) eq 'HASH') {
                    $result = &Apache::lonnet::do_cache_new('loadbalancing',$dom_in_use,
                                                            $domconfig{'loadbalancing'},$cachetime);
                }
            }
            if (ref($result) eq 'HASH') {
                (undef,my $currtargets,my $currrules) =
                    &Apache::lonnet::check_balancer_result($result,$dataref->{'server'});
                if (ref($currrules) eq 'HASH') {
                    if ($dataref->{'sso.login'}) {
                        if ($currrules->{'_LC_ipchangesso'} ne '') {
                            $rule_in_effect = $currrules->{'_LC_ipchangesso'};
                        }
                    } else {
                        if ($currrules->{'_LC_ipchange'} ne '') {
                            $rule_in_effect = $currrules->{'_LC_ipchange'};
                        }
                    }
                }
            }
        }
        my $url;
        my $lonhost= $r->dir_config('lonHostID');
        my $switchto = $lonhost;
        if ($rule_in_effect ne 'offloadedto') {
            my $hosthere;
            my @ids=&Apache::lonnet::current_machine_ids();
            unless ($rule_in_effect eq 'balancer') {
                if (grep(/^\Q$rule_in_effect\E$/,@ids)) {
                    $hosthere = 1;
                }
            }
            unless ($hosthere) {
                if ($dataref->{'role'}) {
                    my ($adom,$aname);
                    if ($dataref->{'role'} =~ m{^au\./($match_domain)/$}) {
                        $adom = $1;
                        $aname = $dataref->{'username'};
                    } elsif ($dataref->{'role'} =~ m{^(?:ca|aa)\./($match_domain)/($match_username)$}) {
                        $adom = $1;
                        $aname = $2;
                    }
                    if ($adom ne '' && $aname ne '') {
                        my $ahome = &Apache::lonnet::homeserver($aname,$adom);
                        unless ($ahome eq 'no_host') {
                            if ($ahome && grep(/^\Q$ahome\E$/,@ids)) {
                                $hosthere = 1;
                            }
                        }
                    }
                }
            }
            unless ($hosthere) {
                my $hostname;
                if ($rule_in_effect eq 'balancer') {
                    $hostname = &Apache::lonnet::hostname($dataref->{'server'});
                    if ($hostname) {
                        $switchto = $dataref->{'server'};
                    }
                } else {
                    $hostname = &Apache::lonnet::hostname($rule_in_effect);
                    if ($hostname) {
                        $switchto = $rule_in_effect;
                    }
                }
                if ($hostname) {
                    my $protocol = $Apache::lonnet::protocol{$switchto};
                    $protocol = 'http' if ($protocol ne 'https');
                    $url = $protocol.'://'.$hostname;
                    if ($rule_in_effect eq 'balancer') {
                        $message .= '<br />'.
                                    &mt('As a result, your LON-CAPA session is being redirected to the server where you originally logged in.');
                    } else {
                        $message .= '<br />'.
                                    &mt('As a result, your LON-CAPA session is being redirected.');
                    }
                }
            }
        }
        if ($dataref->{'sso.login'}) {
            $url .= '/adm/roles';
        } else {
            $url .= '/adm/login';
            if ($udom) {
                $url .= '?domain='.$udom;
            }
            $message .= '<br />'.&mt('You will need to provide your password one more time.');
        }
        my %info= (
                    'domain'          => $udom,
                    'username'        => $dataref->{'username'},
                    'role'            => $dataref->{'role'},
                    'sessionserver'   => $lonhost,
                  );
        if ($dataref->{'origurl'}) {
            $info{'origurl'} = $dataref->{'origurl'};
        }
        if ($dataref->{'symb'}) {
            $info{'symb'} = $dataref->{'symb'};
        }
        my $iptoken = &Apache::lonnet::tmpput(\%info,$switchto);
        unless ($iptoken eq 'conlost') {
            $url .= ($url =~ /\?/) ? '&' : '?';
            $url .= 'iptoken='.$iptoken;
        }
        $r->print(&Apache::loncommon::start_page($title,undef,
                                                 {'redirect' =>
                                                  [2,$url],}).
                  '<h1>'.&mt('One moment please...').'</h1>'.
                  '<p class="LC_warning">'.$message.'</p>'.
                  &Apache::loncommon::end_page());
    } else {
        return &goto_login($r);
    }
    return OK;
}

sub logout {
    my ($r,$handle,$data,$lti_env) = @_;
    my $lonidsdir=$r->dir_config('lonIDsDir');
    if (unlink("$lonidsdir/$handle.id")) {
        if (($env{'user.linkedenv'} =~ /^[a-f0-9]+_linked$/) &&
            (-l "$lonidsdir/$env{'user.linkedenv'}.id") &&
            (readlink("$lonidsdir/$env{'user.linkedenv'}.id") eq "$lonidsdir/$handle.id")) {
            unlink("$lonidsdir/$env{'user.linkedenv'}.id");
        }
    }
    my %temp=('logout' => time);
    &Apache::lonnet::put('email_status',\%temp);
    &Apache::lonnet::log($env{'user.domain'},
                         $env{'user.name'},
                         $env{'user.home'},
                         "Logout $ENV{'REMOTE_ADDR'}");

    &Apache::loncommon::content_type($r,'text/html');

    #expire the cookies
    my %cookies=CGI::Cookie->parse($r->header_in('Cookie'));
    foreach my $name (keys(%cookies)) {
        next unless ($name =~ /^lon(|S|Link|Pub)ID$/);
        my $c = new CGI::Cookie(-name    => $name,
                                -value   => '',
                                -expires => '-10y',);
        $r->headers_out->add('Set-cookie' => $c);
    }
    my (%info,%user_info,%lti_info);
    if (ref($lti_env) eq 'HASH') {
        %lti_info = %{$lti_env};
    }
    my $lonhost = $r->dir_config('lonHostID');
    if (ref($data) eq 'HASH') {
        %user_info=('ip'       => $ENV{'REMOTE_ADDR'},
                    'domain'   => $data->{'domain'},
                    'username' => $data->{'username'},
                    'role'     => $data->{'role'},
                    'origurl'  => $data->{'origurl'},
                    'symb'     => $data->{'symb'},
                    'server'   => $lonhost);
    }
    %info = (%user_info,%lti_info);
    my $token = &Apache::lonnet::tmpput(\%info,$lonhost);
    my $url = '/adm/migrateuser?token='.$token;
    $r->send_http_header;
    $r->print(
        &Apache::loncommon::start_page('Updating Session ...',undef,
                                       {'redirect'       => [0.1,$url],
                                        'only_body'      => 1,}).
        &Apache::loncommon::end_page());
    $r->register_cleanup(\&flush_course_logs);
    return;
}

sub flush_course_logs {
    &Apache::lonnet::flushcourselogs();
    return OK;
}

sub handler {
    my ($r) = @_;
    
    &Apache::loncommon::get_unprocessed_cgi($ENV{'QUERY_STRING'},['token']);
    my %data =   &Apache::lonnet::tmpget($env{'form.token'});
    if (keys(%data) == 0) {
        return &goto_login($r);
    }
    my $delete = &Apache::lonnet::tmpdel($env{'form.token'});

    &Apache::lonlocal::get_language_handle($r);

    if ($delete ne 'ok') {
	return &goto_login($r,undef,\%data);
    }

    if (!defined($data{'username'}) || !defined($data{'domain'})) {
        return &goto_login($r,undef,\%data);
    }
    if ($data{'ip'} ne $ENV{'REMOTE_ADDR'}) {
        &Apache::lonnet::logthis('IP change when session migration requested -- was: '.
                 $data{'ip'}.'; now: '.$ENV{'REMOTE_ADDR'}.' for '.$data{'username'}.':'.$data{'domain'});
	return &ip_changed($r,$data{'domain'},$data{'server'},\%data);
    }

    &Apache::lonnet::logthis("Allowing access for $data{'username'}:$data{'domain'} to $data{'role'}");
    my $home=&Apache::lonnet::homeserver($data{'username'},$data{'domain'});
    my $udom;
    if (&Apache::lonnet::domain($data{'domain'})) {
        $udom=$data{'domain'};
    }
    if ($home =~ /(con_lost|no_such_host)/) { return &goto_login($r,$udom,\%data); }

    my $sso_env = &sso_check(\%data);
    my $lti_env = &lti_check(\%data);

    my $extra_env;
    if ((ref($sso_env) eq 'HASH') && (keys(%{$sso_env}))) {
        $extra_env = $sso_env;
    } elsif ((ref($lti_env) eq 'HASH') && (keys(%{$lti_env}))) {
        $extra_env = $lti_env;
    }

    my %form;
    if ($data{'symb'} ne '') {
        $form{'symb'} = $data{'symb'};
    }
    if ($data{'iptoken'} ne '') {
        $form{'iptoken'} = $data{'iptoken'};
    }
    if ($data{'noloadbalance'} ne '') {
        $form{'noloadbalance'} = $data{'noloadbalance'};
    }

    if (!$data{'role'}) {
	my $handle = &Apache::lonnet::check_for_valid_session($r);
	if ($handle) {
	    &Apache::lonnet::transfer_profile_to_env($r->dir_config('lonIDsDir'),
						     $handle);
            if ($data{'lti.login'}) {
                my $needslogout;
                if ($env{'request.lti.login'}) {
                    if (($env{'user.name'} ne $data{'username'}) ||
                        ($env{'user.domain'} ne $data{'domain'})) {
                        $needslogout = 1;
                    }
                } else {
                    $needslogout = 1;
                }
# If access is via LTI, and user already has a non-LTI session cookie 
# (and session) or has an LTI session cookie for a different username,
# logout the existing session, and start a new one
                if ($needslogout) {
                    &logout($r,$handle,\%data,$lti_env);
                } elsif (($data{'lti.reqcrs'}) && ($data{'lti.reqrole'} eq 'cc')) {
                    $form{'lti.reqcrs'} = $data{'lti.reqcrs'};
                    $form{'lti.reqrole'} = $data{'lti.reqrole'};
                    $form{'lti.sourcecrs'} = $data{'lti.sourcecrs'};
                    $form{'lti.uri'} = $data{'lti.uri'};
                    if ($data{'lti.passbackid'}) {
                        $form{'lti.passbackid'} = $data{'lti.passbackid'};
                    }
                    if ($data{'lti.passbackurl'}) {
                        $form{'lti.passbackurl'} = $data{'lti.passbackurl'};
                    }
                    if ($data{'lti.rosterid'}) {
                        $form{'lti.rosterid'} = $data{'lti.rosterid'};
                    }
                    if ($data{'lti.rosterurl'}) {
                        $form{'lti.rosterurl'} = $data{'lti.rosterurl'};
                    }
                    if ($data{'lti.target'}) {
                        $form{'lti.target'} = $data{'lti.target'};
                    }
                    &Apache::loncommon::content_type($r,'text/html');
                    $r->send_http_header;
                    &Apache::ltiauth::lti_reqcrs($r,$data{'domain'},\%form,$data{'username'},$data{'domain'});
                } else {
                    if (ref($lti_env) eq 'HASH') {
                        delete($lti_env->{'reqcrs'});
                        delete($lti_env->{'reqrole'});
                        delete($lti_env->{'selfenrollrole'});
                    }
                    if ($data{'lti.selfenrollrole'}) {
                        if (&Apache::ltiauth::lti_enroll($data{'username'},$data{'domain'},
                                                         $data{'lti.selfenrollrole'}) eq 'ok') {
                            my $url = '/adm/roles?selectrole=1&'.
                                      &escape($data{'lti.selfenrollrole'}).'=1';
                            if ($data{'origurl'} =~ m{/default_\d+\.sequence$}) {
                                $url .= '&orgurl='.$data{'origurl'}.'&navmap=1';
                            } elsif ($data{'origurl'} ne '') {
                                $url .= '&orgurl='.$data{'origurl'};
                            }
                            if (ref($lti_env) eq 'HASH') {
                                &Apache::lonnet::appenv($lti_env);
                            }
                            $r->internal_redirect($url);
                        } else {
                            &Apache::ltiauth::invalid_request($r,23);
                        }
                    } elsif ($data{'origurl'} ne '') {
                        my $url = $data{'origurl'};
                        if ($url =~ m{/default_\d+\.sequence$}) {
                            $url .= (($url =~/\?/)?'&':'?').'navmap=1';
                        }
                        if (ref($lti_env) eq 'HASH') {
                            &Apache::lonnet::appenv($lti_env);
                        }
                        $r->internal_redirect($url);
                    } else {
                        if (ref($lti_env) eq 'HASH') {
                            &Apache::lonnet::appenv($lti_env);
                        }
                    }
                }
            } elsif ($data{'origurl'} ne '') {
                $r->internal_redirect($data{'origurl'});
            } elsif ($env{'request.course.id'}) {
                $r->internal_redirect('/adm/navmaps');
	    } else {
		$r->internal_redirect('/adm/roles');
	    }
	} else {
            my $desturl = '/adm/roles';
            if ($data{'origurl'} ne '') {
                $desturl = $data{'origurl'};
                if ($data{'lti.login'}) {
                    $desturl = $data{'origurl'};
                    if ($desturl =~ m{/default_\d+\.sequence$}) {
                        $desturl .= (($desturl =~/\?/)?'&':'?').'navmap=1';
                    }
                }
            }
            my $skipcritical;
            if (($data{'lti.login'}) && ($data{'lti.reqcrs'}) &&
                ($data{'lti.reqrole'} eq 'cc')) {
                $skipcritical = 1;
            }
	    &Apache::lonauth::success($r,$data{'username'},$data{'domain'},
				      $home,$desturl,$extra_env,\%form,$skipcritical);
	}
	return OK;
    }

    my $next_url='/adm/roles?selectrole=1&'.&escape($data{'role'}).'=1';
    if ($data{'origurl'} ne '') {
        $next_url .= '&orgurl='.&escape($data{'origurl'});
        if ($data{'lti.login'}) {
            if (($data{'origurl'} =~ m{/default_\d+\.sequence$}) ||
                ($data{'origurl'} =~ m{^/res/.+\.sequence$})) {
                $next_url .= '&navmap=1';
            }
        }
    }
    &Apache::lonauth::success($r,$data{'username'},$data{'domain'},$home,
			      $next_url,$extra_env,\%form);
    return OK;
}

1;
__END__