Extension:AuthorRestriction

<?php
/*
 * @author Jean-Lou Dupont -- www.bluecortex.com
 * @package MediaWiki
 * @subpackage Extensions
 * 
 * <b>Purpose:</b>  This extension adds a 'read' restriction to protected article.
 * Only the users with the 'author' permission can 'read' the protected article.
 * The extension adds functionality to the existing 'protect' function of MW. 
 *
 * Installation:
 * include("extensions/AuthorRestriction.php");
 *
 * HISTORY:
 * V1.1   
 *  - Corrected shortcoming when dealing with non-'read' actions.
 * v1.2
 *  - Changed loading order of extension in order to integrate better
 *    with other Namespace Permission type extensions.
 * -- Moved to SVN management
 * v2.0 - Integration with ArticleEx to get rid of patch in Article.php
 *        (almost complete re-write)
 */
 
$wgExtensionFunctions[] = 'AuthorRestrictionSetup';
global $wgHooks;
$wgHooks['SpecialVersionExtensionTypes'][] = 'AuthorRestrictionSpecialPage' ;
 
function AuthorRestrictionSetup()
{
        global $wgMessageCache, $wgRestrictionTypes, $wgRestrictionLevels, $wgHooks ;
 
        $wgRestrictionTypes[] =      'read';
        $wgMessageCache->addMessage( 'restriction-read' ,    'Read' );
        $wgMessageCache->addMessage( 'protect-level-author', 'Authors Only' );
        $wgRestrictionLevels[] =     'author';
 
        global $wgHooks;
        $wgHooks['ArticleViewExBegin'][] = 'AuthorRestrictionUserCan';
 
        global $wgExtensionCredits;
        $wgExtensionCredits['other'][] = array(
                'name'    => "AuthorRestriction",
                'version' => 'v2.0 $LastChangedRevision: 63 $',
                'author'  => 'Jean-Lou Dupont [http://www.bluecortex.com]',
                'description' => 'ArticleEx extension status: '
        );
}
 
function AuthorRestrictionUserCan( &$article )
{
  global $action, $wgUser;
 
  # if the action is not related to a 'view' (i.e. 'read') request, get out.
  if ($action != 'view')
   return true;  #don't stop processing the hook chain
 
  // Load any restriction associated with the 'read' right
  $r = $article->mTitle->getRestrictions('read');
 
  // If 'author' restriction is active, then check for 'author' right
  if ( in_array('author', $r) === true ) 
  {
    // Does the user belongs in the 'author' group?
        #if ( !in_array('author', $user->getGroups()) )
        if ( !in_array('author', $wgUser->getGroups()) )
        {
                global $wgOut;
                $wgOut->setPageTitle( wfMsg( 'badaccess' ) );
                $wgOut->addWikiText( wfMsg( 'badaccess-group0' ) );
                $wgOut->output();
                exit;
        } 
  }
  return true; # don't stop processing hook chain.
}
function AuthorRestrictionSpecialPage()
{
        global $wgExtensionCredits;
 
        if (class_exists('ArticleExClass'))
                $result = 'Found -- Author Restriction extension operational!';
        else
                $result = '<b>not found -- Author Restriction extension not operational!</b>';
 
        foreach ( $wgExtensionCredits['other'] as $index => &$el )
        {
                if ($el['name']=='AuthorRestriction')
                        $el['description'].=$result;
        }
}