282 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			282 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * StatusNet - the distributed open-source microblogging tool
 | 
						|
 * Copyright (C) 2011, StatusNet, Inc.
 | 
						|
 *
 | 
						|
 * Restrict the email addresses in a domain to a select whitelist
 | 
						|
 *
 | 
						|
 * PHP version 5
 | 
						|
 *
 | 
						|
 * This program is free software: you can redistribute it and/or modify
 | 
						|
 * it under the terms of the GNU Affero General Public License as published by
 | 
						|
 * the Free Software Foundation, either version 3 of the License, or
 | 
						|
 * (at your option) any later version.
 | 
						|
 *
 | 
						|
 * This program 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 Affero General Public License for more details.
 | 
						|
 *
 | 
						|
 * You should have received a copy of the GNU Affero General Public License
 | 
						|
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 *
 | 
						|
 * @category  Cache
 | 
						|
 * @package   StatusNet
 | 
						|
 * @author    Evan Prodromou <evan@status.net>
 | 
						|
 * @author    Zach Copley <zach@status.net>
 | 
						|
 * @copyright 2011 StatusNet, Inc.
 | 
						|
 * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
 | 
						|
 * @link      http://status.net/
 | 
						|
 */
 | 
						|
 | 
						|
if (!defined('STATUSNET')) {
 | 
						|
    // This check helps protect against security problems;
 | 
						|
    // your code file can't be executed directly from the web.
 | 
						|
    exit(1);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Restrict the email addresses to a domain whitelist
 | 
						|
 *
 | 
						|
 * @category  General
 | 
						|
 * @package   StatusNet
 | 
						|
 * @author    Evan Prodromou <evan@status.net>
 | 
						|
 * @author    Zach Copley <zach@status.net>
 | 
						|
 * @copyright 2011 StatusNet, Inc.
 | 
						|
 * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
 | 
						|
 * @link      http://status.net/
 | 
						|
 */
 | 
						|
class DomainWhitelistPlugin extends Plugin
 | 
						|
{
 | 
						|
    /**
 | 
						|
     * Get the path to the plugin's installation directory. Used
 | 
						|
     * to link in js files and whatnot.
 | 
						|
     *
 | 
						|
     * @return String the absolute path
 | 
						|
     */
 | 
						|
    protected function getPath() {
 | 
						|
        return preg_replace('/^' . preg_quote(INSTALLDIR, '/') . '\//', '', dirname(__FILE__));
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Link in a JavaScript script for the whitelist invite form
 | 
						|
     *
 | 
						|
     * @param Action $action Action being shown
 | 
						|
     *
 | 
						|
     * @return boolean hook flag
 | 
						|
     */
 | 
						|
    function onEndShowStatusNetScripts($action) {
 | 
						|
        $name = $action->arg('action');
 | 
						|
        if ($name == 'invite') {
 | 
						|
            $action->script($this->getPath() . '/js/whitelistinvite.js');
 | 
						|
        }
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    function onRequireValidatedEmailPlugin_Override($user, &$knownGood)
 | 
						|
    {
 | 
						|
        $knownGood = (!empty($user->email) && $this->matchesWhitelist($user->email));
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    function onEndValidateUserEmail($user, $email, &$valid)
 | 
						|
    {
 | 
						|
        if ($valid) { // it's otherwise valid
 | 
						|
            if (!$this->matchesWhitelist($email)) {
 | 
						|
                $whitelist = $this->getWhitelist();
 | 
						|
                if (count($whitelist) == 1) {
 | 
						|
                    // TRANS: Client exception thrown when a given e-mailaddress is not in the domain whitelist.
 | 
						|
                    // TRANS: %s is a whitelisted e-mail domain.
 | 
						|
                    $message = sprintf(_m('Email address must be in this domain: %s.'),
 | 
						|
                                       $whitelist[0]);
 | 
						|
                } else {
 | 
						|
                    // TRANS: Client exception thrown when a given e-mailaddress is not in the domain whitelist.
 | 
						|
                    // TRANS: %s are whitelisted e-mail domains separated by comma's (localisable).
 | 
						|
                    $message = sprintf(_m('Email address must be in one of these domains: %s.'),
 | 
						|
                                       // TRANS: Separator for whitelisted domains.
 | 
						|
                                       implode(_m('SEPARATOR',', '), $whitelist));
 | 
						|
                }
 | 
						|
                throw new ClientException($message);
 | 
						|
            }
 | 
						|
        }
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    function onStartAddEmailAddress($user, $email)
 | 
						|
    {
 | 
						|
        if (!$this->matchesWhitelist($email)) {
 | 
						|
            // TRANS: Exception thrown when an e-mail address does not match the site's domain whitelist.
 | 
						|
            throw new Exception(_m('That email address is not allowed on this site.'));
 | 
						|
        }
 | 
						|
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    function onEndValidateEmailInvite($user, $email, &$valid)
 | 
						|
    {
 | 
						|
        if ($valid) {
 | 
						|
            $valid = $this->matchesWhitelist($email);
 | 
						|
        }
 | 
						|
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    function matchesWhitelist($email)
 | 
						|
    {
 | 
						|
        $whitelist = $this->getWhitelist();
 | 
						|
 | 
						|
        if (empty($whitelist) || empty($whitelist[0])) {
 | 
						|
            return true;
 | 
						|
        }
 | 
						|
 | 
						|
        $userDomain = $this->domainFromEmail($email);
 | 
						|
 | 
						|
        return in_array($userDomain, $whitelist);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Helper function to pull out a domain from
 | 
						|
     * an email address
 | 
						|
     *
 | 
						|
     * @param string $email and email address
 | 
						|
     * @return string the domain
 | 
						|
     */
 | 
						|
    function domainFromEmail($email)
 | 
						|
    {
 | 
						|
        $parts = explode('@', $email);
 | 
						|
        return strtolower(trim($parts[1]));
 | 
						|
    }
 | 
						|
 | 
						|
    function getWhitelist()
 | 
						|
    {
 | 
						|
        $whitelist = common_config('email', 'whitelist');
 | 
						|
 | 
						|
        if (is_array($whitelist)) {
 | 
						|
            return $this->sortWhiteList($whitelist);
 | 
						|
        } else {
 | 
						|
            return explode('|', $whitelist);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * This is a filter function passed in to array_filter()
 | 
						|
     * in order to strip out the user's domain, which will
 | 
						|
     * be re-inserted as the first element (see sortWhitelist()
 | 
						|
     * below).
 | 
						|
     *
 | 
						|
     * @param string $domain domain to check
 | 
						|
     * @return boolean whether to include the domain
 | 
						|
     */
 | 
						|
    function userDomainFilter($domain)
 | 
						|
    {
 | 
						|
        $user       = common_current_user();
 | 
						|
        $userDomain = $this->domainFromEmail($user->email);
 | 
						|
        if ($userDomain == $domain) {
 | 
						|
            return false;
 | 
						|
        }
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * This function sorts the whitelist alphabetically, and sets the
 | 
						|
     * current user's domain as the first element in the array of
 | 
						|
     * allowed domains. Mostly, this is for the JavaScript on the invite
 | 
						|
     * page--in the case of multiple allowed domains, it's nicer if the
 | 
						|
     * user's own domain is the first option, and this seemed like a good
 | 
						|
     * way to do it.
 | 
						|
     *
 | 
						|
     * @param array $whitelist whitelist of allowed email domains
 | 
						|
     * @return array an ordered or sorted version of the whitelist
 | 
						|
     */
 | 
						|
    function sortWhitelist($whitelist)
 | 
						|
    {
 | 
						|
        $whitelist = array_unique($whitelist);
 | 
						|
        natcasesort($whitelist);
 | 
						|
 | 
						|
        $user = common_current_user();
 | 
						|
 | 
						|
        if (!empty($user) && !empty($user->email)) {
 | 
						|
            $userDomain = $this->domainFromEmail($user->email);
 | 
						|
 | 
						|
            $orderedWhitelist = array_values(
 | 
						|
                array_filter(
 | 
						|
                    $whitelist,
 | 
						|
                    array($this, "userDomainFilter")
 | 
						|
                )
 | 
						|
            );
 | 
						|
 | 
						|
            if (in_array($userDomain, $whitelist)) {
 | 
						|
                array_unshift($orderedWhitelist, $userDomain);
 | 
						|
            }
 | 
						|
            return $orderedWhitelist;
 | 
						|
        }
 | 
						|
 | 
						|
        return $whitelist;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Show a fancier invite form when domains are restricted to the
 | 
						|
     * whitelist.
 | 
						|
     *
 | 
						|
     * @param action $action the invite action
 | 
						|
     * @return boolean hook value
 | 
						|
     */
 | 
						|
    function onStartShowInviteForm($action)
 | 
						|
    {
 | 
						|
        $this->showConfirmDialog($action);
 | 
						|
        $form = new WhitelistInviteForm($action, $this->getWhitelist());
 | 
						|
        $form->show();
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    function showConfirmDialog($action)
 | 
						|
    {
 | 
						|
        // For JQuery UI modal dialog
 | 
						|
        $action->elementStart(
 | 
						|
            'div',
 | 
						|
            // TRANS: Title for invitiation deletion dialog.
 | 
						|
            array('id' => 'confirm-dialog', 'title' => _m('Confirmation Required'))
 | 
						|
        );
 | 
						|
        // TRANS: Confirmation text for invitation deletion dialog.
 | 
						|
        $action->text(_m('Really delete this invitation?'));
 | 
						|
        $action->elementEnd('div');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * This is a bit of a hack. We take the values from the custom
 | 
						|
     * whitelist invite form and reformat them so they look like
 | 
						|
     * their coming from the the normal invite form.
 | 
						|
     *
 | 
						|
     * @param action &$action the invite action
 | 
						|
     * @return boolean hook value
 | 
						|
     */
 | 
						|
    function onStartSendInvitations(&$action)
 | 
						|
    {
 | 
						|
       $emails    = array();
 | 
						|
       $usernames = $action->arg('username');
 | 
						|
       $domains   = $action->arg('domain');
 | 
						|
 | 
						|
       for($i = 0; $i < count($usernames); $i++) {
 | 
						|
           if (!empty($usernames[$i])) {
 | 
						|
               $emails[] = $usernames[$i] . '@' . $domains[$i] . "\n";
 | 
						|
           }
 | 
						|
       }
 | 
						|
 | 
						|
       $action->args['addresses'] = implode($emails);
 | 
						|
 | 
						|
       return true;
 | 
						|
    }
 | 
						|
 | 
						|
    function onPluginVersion(&$versions)
 | 
						|
    {
 | 
						|
        $versions[] = array('name' => 'DomainWhitelist',
 | 
						|
                            'version' => GNUSOCIAL_VERSION,
 | 
						|
                            'author' => 'Evan Prodromou, Zach Copley',
 | 
						|
                            'homepage' => 'http://status.net/wiki/Plugin:DomainWhitelist',
 | 
						|
                            'rawdescription' =>
 | 
						|
                            // TRANS: Plugin description.
 | 
						|
                            _m('Restrict domains for email users.'));
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
}
 |