moodle/enrol/ajax.php

<?php
// This file is part of Moodle - http://moodle.org/
//
// Moodle 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 3 of the License, or
// (at your option) any later version.
//
// Moodle 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 Moodle.  If not, see <http://www.gnu.org/licenses/>.

/**
 * This file processes AJAX enrolment actions and returns JSON
 *
 * @package moodlecore
 * @copyright  2010 Sam Hemelryk
 * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
 */

define('AJAX_SCRIPT', true);

require('../config.php');
require_once("$CFG->dirroot/enrol/locallib.php");
require_once("$CFG->dirroot/enrol/renderer.php");
require_once("$CFG->dirroot/group/lib.php");

// Must have the sesskey
require_sesskey();
$id      = required_param('id', PARAM_INT); // course id
$action  = required_param('action', PARAM_ACTION);

$PAGE->set_url(new moodle_url('/enrol/ajax.php', array('id'=>$id, 'action'=>$action)));

$course = $DB->get_record('course', array('id'=>$id), '*', MUST_EXIST);
$context = get_context_instance(CONTEXT_COURSE, $course->id, MUST_EXIST);

if ($course->id == SITEID) {
    redirect(new moodle_url('/'));
}

require_login($course);
require_capability('moodle/course:enrolreview', $context);

$manager = new course_enrolment_manager($course);

$outcome = new stdClass;
$outcome->success = false;
$outcome->response = new stdClass;
$outcome->error = '';

if (!confirm_sesskey()) {
    $outcome->error = 'invalidsesskey';
    echo json_encode($outcome);
    die();
}

switch ($action) {
    case 'unenrol':

        $ue = $DB->get_record('user_enrolments', array('id'=>required_param('ue', PARAM_INT)), '*', MUST_EXIST);
        if ($manager->unenrol_user($ue)) {
            $outcome->success = true;
        } else {
            $outcome->error = 'unabletounenrol';
        }
        break;
    case 'unassign':
        $role = required_param('role', PARAM_INT);
        $user = required_param('user', PARAM_INT);
        if ($manager->unassign_role_from_user($user, $role)) {
            $outcome->success = true;
        } else {
            $outcome->error = 'unabletounassign';
        }
        break;

    case 'assign':

        $user = required_param('user', PARAM_INT);
        $user = $DB->get_record('user', array('id'=>$user), '*', MUST_EXIST);
        $roleid = required_param('roleid', PARAM_INT);

        if (!is_enrolled($context, $user)) {
            $outcome->error = 'mustbeenrolled';
            break; // no roles without enrolments here in this script
        }

        if ($manager->assign_role_to_user($roleid, $user->id)) {
            $outcome->success = true;
            $outcome->response->roleid = $roleid;
        } else {
            $outcome->error = 'unabletoassign';
        }
        break;

    case 'getassignable':
        $outcome->success = true;
        $outcome->response = $manager->get_assignable_roles();
        break;
    case 'getcohorts':
        require_capability('moodle/course:enrolconfig', $context);
        $outcome->success = true;
        $outcome->response = $manager->get_cohorts();
        break;
    case 'enrolcohort':
        require_capability('moodle/course:enrolconfig', $context);
        $roleid = required_param('roleid', PARAM_INT);
        $cohortid = required_param('cohortid', PARAM_INT);
        $outcome->success = $manager->enrol_cohort($cohortid, $roleid);
        break;
    case 'enrolcohortusers':
        require_capability('moodle/course:enrolconfig', $context);
        $roleid = required_param('roleid', PARAM_INT);
        $cohortid = required_param('cohortid', PARAM_INT);
        $result = $manager->enrol_cohort_users($cohortid, $roleid);
        if ($result !== false) {
            $outcome->success = true;
            $outcome->response->users = $result;
            $outcome->response->message = get_string('enrollednewusers', 'enrol', $result);
        }
        break;
    case 'searchusers':
        $enrolid = required_param('enrolid', PARAM_INT);
        $search  = optional_param('search', '', PARAM_CLEAN);
        $page = optional_param('page', 0, PARAM_INT);
        $outcome->response = $manager->get_potential_users($enrolid, $search, false, $page);
        foreach ($outcome->response['users'] as &$user) {
            $user->picture = $OUTPUT->user_picture($user);
            $user->fullname = fullname($user);
        }
        $outcome->success = true;

        break;

    case 'enrol':
        $enrolid = required_param('enrolid', PARAM_INT);
        $userid = required_param('userid', PARAM_INT);

        $roleid = optional_param('role', null, PARAM_INT);
        $duration = optional_param('duration', 0, PARAM_INT);
        $startdate = optional_param('startdate', 0, PARAM_INT);
        if (empty($roleid)) {
            $roleid = null;
        }

        switch($startdate) {
            case 2:
                $timestart = $course->startdate;
                break;
            case 3:
            default:
                $today = time();
                $today = make_timestamp(date('Y', $today), date('m', $today), date('d', $today), 0, 0, 0);
                $timestart = $today;
                break;
        }
        if ($duration <= 0) {
            $timestart = 0;
            $timeend = 0;
        } else {
            $timeend = $timestart + ($duration*24*60*60);
        }

        $user = $DB->get_record('user', array('id'=>$userid), '*', MUST_EXIST);
        $instances = $manager->get_enrolment_instances();
        $plugins = $manager->get_enrolment_plugins();
        if (!array_key_exists($enrolid, $instances)) {
            $outcome->error = 'invalidinstance';
            break;
        }
        $instance = $instances[$enrolid];
        $plugin = $plugins[$instance->enrol];
        try {
            $plugin->enrol_user($instance, $user->id, $roleid, $timestart, $timeend);
        } catch (Exception $e) {
            $outcome->error = 'unabletoenrol';
            break;
        }
        $outcome->success = true;
        break;

    default:
        $outcome->error = 'unknownaction';
        break;
}

echo json_encode($outcome);
die();