Extension:C Style Wiki Includes

<?php
# 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, 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 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.,
# 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
# http://www.gnu.org/copyleft/gpl.html
 
/**
 * @author Daniel Friesen <dan_the_man@telus.net>
 * @license http://www.gnu.org/copyleft/gpl.html GNU General Public License 2.0 or later
 */
 
if( !defined( 'MEDIAWIKI' ) ) die( "This is an extension to the MediaWiki package and cannot be run standalone." );
 
// Two new permissions
$wgAvailableRights[] = 'cwikiinclude';
 
$wgExtensionFunctions[] = 'efCWikiIncludeSetup';
 
// Register hooks
$wgHooks['ParserAfterTidy'][] = 'efCWikiAfterParse';
$wgHooks['EditFilter'][] = 'efCheckCWikiInclude';
$wgHooks['ParserBeforeStrip'][] = 'efCWikiBeforeParse';
 
$wgExtensionCredits['parserhook'][] = array(
        'name' => 'C Style Wiki Includes',
        'url' => 'http://www.mediawiki.org/wiki/Extension:C_Style_Wiki_Includes',
        'author' => '[http://central.wiki-tools.com/wiki/User:Dantman Daniel Friesen]',
        'description' => 'C Style WikiText and HTML includes.',
);
 
$egCWikiInclude_limit = 3;
$egCWikiWikiText_html = "/^#include\s*\"(https?:\/\/[^<>\"']*?)\"(;?)$/sim";
$egCWikiWikiText_text = "/^#include\s*{(https?:\/\/[^<>\"']*?)}(;?)$/sim";
$egCWikiWikiText_wiki = "/^#include\s*<(https?:\/\/[^<>\"']*?)>(;?)$/sim";
$egCWikiInclude_html = "/<tt>#include\s*&quot;(https?:\/\/[^<>\"']*?)&quot;(;?)<\/tt><br \/>/sim";
$egCWikiInclude_text = "/<tt>#include\s*{(https?:\/\/[^<>\"']*?)}(;?)<\/tt>/sim";
$egCWikiInclude_wiki = "/<tt><nowiki>#include\s*<(https?:\/\/[^<>\"']*?)>(;?)<\/nowiki><\/tt><br \/>/sim";
function efCWikiIncludeSetup() {
        global $wgMessageCache;
        $wgMessageCache->addMessages(
                array(
                        'cwikiinclude_add_html' => 
                                'You tried to add a C Style HTML Include',
                        'cwikiinclude_modify_html' => 
                                'You tried to modify a C Style HTML Include',
                        'cwikiinclude_forbidden' =>
                                'Forbidden',
                )
        );
}
 
 
function efCWikiBeforeParse( &$parser, &$text, &$stripState ) {
        global $egCWikiWikiText_html, $egCWikiWikiText_text, $egCWikiWikiText_wiki, $egCWikiInclude_wiki, $egCWikiInclude_limit;
 
        /* Escape The Includes */
        $text = preg_replace( array(
                $egCWikiWikiText_html,
                $egCWikiWikiText_text,
                $egCWikiWikiText_wiki
        ), array(
                "<tt><nowiki>#include \"\\1\"\\2</nowiki></tt><br />",
                "<tt><nowiki>#include {\\1}\\2</nowiki></tt><br />",
                "<tt><nowiki>#include <\\1>\\2</nowiki></tt><br />"
        ), $text );
 
        /* Do the WikiText include */
        $text = preg_replace( $egCWikiInclude_wiki.'e', 'efCWikiIncludeCallback( "\\0", false, "\\1" )', $text, $egCWikiInclude_limit );
 
        return true;
}
function efCWikiAfterParse( &$parser, &$text ) { 
        global $egCWikiInclude_limit, $egCWikiInclude_html, $egCWikiInclude_text;
 
        /* Do the HTML include */
        $text = preg_replace( array(
                $egCWikiInclude_html.'e',
                $egCWikiInclude_text.'e'
        ), array(
                'efCWikiIncludeCallback( "\\0", false, html_entity_decode( "\\1" , ENT_QUOTES ) )',
                'efCWikiIncludeCallback( "\\0", true, html_entity_decode( "\\1" , ENT_QUOTES ) )'
        ), $text, $egCWikiInclude_limit );
 
        return true;
}
 
function efCWikiIncludeCallback( $original, $escapeHTML, $url ) {
        global $wgMemc, $wgUser, $wgOut, $wgTitle;
 
        $key = wfMemcKey( 'include', $url );
        $content = $wgMemc->get( $key );
        if( $content == "" ) {
                $content = Http::get( $url );
                if( $content ) {
                        $wgMemc->set( $key, $content, time() + 86400 );
                }
        }
        if( $escapeHTML ) {
                $text = htmlspecialchars( $text );
        }
        if( $content === false ) {
                $content = $original;
        }
        return $content;
}
 
function efCWikiDeleteCallback( $url ) {
        global $wgMemc;
        $key = wfMemcKey( 'include', $url );
        $wgMemc->delete( $key );
        return "";
}
 
/**
 * @todo Document
 * @param $editpage
 * @param $textbox1
 * @param $section
 */
function efCheckCWikiInclude( $editpage, $textbox1, $section )  {
        global $egCWikiInclude_limit, $egCWikiWikiText_html, $egCWikiWikiText_text, $egCWikiWikiText_wiki;
        # check for partial protection 
        global $wgUser,$wgParser;
 
        # Purge memcached for a URL when attempting a save
        preg_replace( array(
                $egCWikiWikiText_html.'e',
                $egCWikiWikiText_text.'e',
                $egCWikiWikiText_wiki.'e'
        ), array_fill ( 0, 3,
                'efCWikiDeleteCallback( html_entity_decode( "\\1" , ENT_QUOTES ) )'
        ), $text, $egCWikiInclude_limit );
 
        if ( !$wgUser->isAllowed( 'cwikiinclude_html' ) ) {
                $modifyProtect = false; 
                $text1 = $editpage->mArticle->getContent(true);
 
                if( $section != '' ) {
                        if( $section == 'new' ) {
                                $text1 = "";
                        } else {
                                $text1 = $wgParser->getSection( $text1, $section );
                        }
                }
 
                $text2 = $textbox1;
 
                preg_match_all( $egCWikiWikiText_html, $text1, $list1, PREG_SET_ORDER );
                preg_match_all( $egCWikiWikiText_html, $text2, $list2, PREG_SET_ORDER );
 
                if( count($list1) < count($list2)) { 
                        $msg = wfMsg( 'cwikiinclude_add_html'); 
                        $modifyProtect = true; 
                }
                else for ( $i=0 ; $i < count( $list1 ); $i++ ) {
                        if( $list1[$i][0] != $list2[$i][0]) { 
                                $msg = wfMsg( 'cwikiinclude_modify_html' );
                                $modifyProtect = true; 
                                break;
                        }
                }
 
                if( $modifyProtect ) {
                        global $wgOut;
                        $wgOut->setPageTitle( wfMsg( 'cwikiinclude_forbidden' ) );
                        $wgOut->addWikiText($msg);
                        return false;
                }
        }
        return true;
}