SMWNSProtect_body.php

<?php
/*
 * SMWFileProtect Class
 *
 * Copyright (C) 2011-2015  Toni Hermoso Pulido <toniher@cau.cat>
 *
 * This program 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.
 *
 * 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 General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */


class SMWNSProtect {
	private $dbr;
	private $db_page, $db_page_links;


	public function __construct() {
		$this->dbr = wfGetDB( DB_SLAVE );
	}
	
	/**
	 * The main function call to display the backward navigation element for
	 * the page  $pageid
	 */
	public function executeNSRefer( $title, $user ) {
		
		global $wgContLang;

		#Get PageID
		$pageid =$title->getArticleID();
		
		// First we check
		global $SMWFileProtectReferNS;
		if ( ! $SMWFileProtectReferNS ) {
			return true;
		}

		// Allow to group
		if ( $this->groupCheck($user) ) {
			return true;
		}

		#Get list of linked page namespaces
		$listReferer = array();
		$listReferer=$this->loadListRefererNS($pageid);

		// Namespace permissions Lockdown
		global $wgNamespacePermissionLockdown;

		if ( !isset( $wgNamespacePermissionLockdown ) ) {
			// If no Lockdown, true
			return true;
		}

		//Count linked pages
		$nbReferer= count( $listReferer );

		if ( $nbReferer > 0 ) {
			
			foreach ( $listReferer as $NSReferer ) {
				// Get namespace
				if ( array_key_exists( $NSReferer, $wgNamespacePermissionLockdown ) ) {
					if ( array_key_exists( "read", $wgNamespacePermissionLockdown[$NSReferer] ) ) {
						$detect = $this->groupDetect( $wgNamespacePermissionLockdown[$NSReferer]["read"], $user->getEffectiveGroups() );
						if ( $detect == 0 ) {
							return false;
						}
					} else {
						if ( array_key_exists( "*", $wgNamespacePermissionLockdown[$NSReferer] ) ) {
							$detect = $this->groupDetect( $wgNamespacePermissionLockdown[$NSReferer]["*"], $user->getEffectiveGroups() );
							if ( $detect == 0 ) {
								return false;
							}
						}
					}
				}
			}

			return true;

		} else {
			return true;
		}


  	}
  	
	/**
	 * Load the referers list for the article $pageid
	 */
	 
	private function loadListRefererNS( $pageid ) {
	
		if (! is_numeric($pageid)) return array();
		//$SQL2 = "select g.il_from from ".$this->db_image_links." g, ".$this->db_page." p where g.il_to=p.page_title and p.page_id=?";

		$table = array( 'imagelinks', 'page' );
		$vars = array( 'il_from' );
		$conds = array( 'il_to=page_title', 'page_id='.$pageid );
		$options = array();
		$condoptions = array();

		$result = $this->dbr->select( $table, $vars, $conds, 'SMWFileProtect::loadListReferer', $options, $condoptions );

		// $tbs=$this->dbr->safeQuery($SQL2,$pageid);
	
		$listReferer=array();
		$i=0;
		
		foreach ( $result as $row ) {
			$title = Title::newFromId( $row->il_from );
			if ( ( get_class( $title )=="Title" ) && ( $title->exists() ) ) {
				$listReferer[$i]=$title->getNamespace();
				$i++;
			}
		}

		return( array_unique( $listReferer ) );
	}

	private function groupDetect( $lckgrps, $usergrps ) {

		$detect = 0;

		if ( in_array( "*", $lckgrps ) ) {
			$detect = 1;
		} else {
			foreach ( $usergrps as $usergrp ) {
				if ( in_array( $usergrp, $lckgrps ) ) {
					$detect = 1;
				}
			}

		}

		return $detect;
	}

	private function groupCheck( $user ) {

		global $SMWFileProtectRights;
		foreach ( $SMWFileProtectRights as $grp ) {
			if( in_array( $grp,$user->getEffectiveGroups()) ) {
				return true;
			}
		}
	}

}