Alink_body.php

<?php

class Alink {

	private static $attrs_ref = array( "href", "rel", "target", "class", "id", "content", "name", "itemprop" );
	private static $attrs_img_ref = array( "src", "rel", "title", "alt", "class", "id", "content", "itemprop" ); 

	private static $attrs_like = array( "data-" );
	private static $protocols = array( "https://", "http://", "ftp://" );
	
	/**
	 * @param $parser Parser
	 * @param $frame PPFrame
	 * @param $args array
	 * @return string
	 */
	public static function process_alink( &$parser, $frame, $args ) {

		$attrs = array();
		$text = "";
		
		// Default urlencode
		$urlencode = true;
		
		// Process page
		$noprefix = false;

		foreach ( $args as $arg ) {
			$arg_clean = trim( $frame->expand( $arg ) );
			$arg_proc = explode( "=", $arg_clean, 2 );
			
			if ( count( $arg_proc ) == 1 ){
				// If arg urlencode -> trigger
				if ( trim( $arg_proc[0] ) == "nourlencode" ) {
					$urlencode = false;
				} elseif ( trim( $arg_proc[0] ) == "noprefix" ) {
					$noprefix = true;
				} else {
					$text = trim( $arg_proc[0] );
				}
				
			} else {
			
				if ( in_array( trim( $arg_proc[0] ), self::$attrs_ref ) ) {
					$attrs[ trim( $arg_proc[0] ) ] = trim( $arg_proc[1] );
				}
				
				foreach ( self::$attrs_like as $attr_like ) {
					if ( strpos( trim( $arg_proc[0] ), $attr_like ) === 0 ) {
						$attrs[ trim( $arg_proc[0] ) ] = trim( $arg_proc[1] );
					}
				}
			}
		}

		// Code for dealing with internal - external
		$external = 0;  
		if ( isset( $attrs["href"] ) ) {
			foreach ( self::$protocols as $protocol ) {
				$detect = strpos( $attrs["href"], $protocol );
				if ( is_int( $detect ) ) {
						$external = 1;
				}
			}
		}

		// Saving link for modifications
		$link = null;
		
		if ( isset( $attrs["href"] ) ) {
			$link = $attrs["href"];
			
			if ( $external == 0 ) {
				
				#Handling internal links
				$anchor_detect = strpos( $attrs["href"], "#" );
				if ( ! is_int( $anchor_detect ) || $anchor_detect != 0 ) {
					
					global $wgArticlePath;
					$page = $attrs["href"];
					$page = str_replace( " ", "_", $page );
					$attrs["href"] = $wgArticlePath;
					
					if ( $urlencode ) {
						$page = urlencode( $page );
					}
					
					$attrs["href"] = str_replace( "$1", $page, $attrs["href"] );
				}
			}
		}
		
		// If no text, use href
		
		if ( $text == "" && ! is_null( $link ) ) {
			$text = $link;
			
			// No prefixed title
			if ( $noprefix ) {
				
				$title = Title::newFromText( $link );
				if ( $title ) {
					$text = $title->getText();
				}
				
			}
		}

		$tag = 	Html::element(
			'a',
				$attrs,
			$text
		);
		
		return $parser->insertStripItem( $tag, $parser->mStripState );
	}

	/**
	 * @param $parser Parser
	 * @param $frame PPFrame
	 * @param $args array
	 * @return string
	 */
	public static function process_aimg( &$parser, $frame, $args ) {

		// TODO: More control of img sources should be allowed
		
		$attrs = array();
		
		foreach ( $args as $arg ) {
			$arg_clean = trim( $frame->expand( $arg ) );
			$arg_proc = explode( "=", $arg_clean, 2 );
			
			if ( count( $arg_proc ) == 1 ){
			} else {
			
				if ( in_array( trim( $arg_proc[0] ), self::$attrs_img_ref ) ) {
					$attrs[ trim( $arg_proc[0] ) ] = trim( $arg_proc[1] );
				}
				
				foreach ( self::$attrs_like as $attr_like ) {
					if ( strpos( $arg_proc[0], $attr_like ) == 0 ) {
						$attrs[ trim( $arg_proc[0] ) ] = trim( $arg_proc[1] );
					}
				}
			}
		}
		
		// Code for dealing with internal - external
		$external = 0;  
		if ( isset( $attrs["src"] ) ) {
			foreach ( self::$protocols as $protocol ) {
				$detect = strpos( $attrs["src"], $protocol );
				if ( is_int( $detect ) ) {
						$external = 1;
				}
			}
		}

		if ( $external == 0 ) {
			if ( isset( $attrs["src"] ) ) {
				$page = $attrs["src"];
				$page = str_replace( " ", "_", $page );
				$file = wfFindFile( $page );
				if ( $file && $file->exists() ) {
					$attrs["src"] = $file->getUrl();
				}

			}
		}
		

		$tag = 	Html::element(
			'img',
				$attrs
		);
		
		return $parser->insertStripItem( $tag, $parser->mStripState );
	}

}