Current File : /home/jvzmxxx/wiki1/extensions/Flow/includes/Import/TemplateHelper.php

<?php

namespace Flow\Import;

use DOMDocument;
use DOMElement;
use DOMXPath;
use Flow\Conversion\Utils;
use Title;

class TemplateHelper {

	/**
	 * @param string $htmlContent
	 * @param string $templateName
	 * @return string
	 * @throws \Flow\Exception\WikitextException
	 */
	public static function removeFromHtml( $htmlContent, $templateName ) {
		$dom = Utils::createDOM( $htmlContent );
		$xpath = new DOMXPath( $dom );
		$templateNodes = $xpath->query( '//*[@typeof="mw:Transclusion"]' );

		foreach ( $templateNodes as $templateNode ) {
			/** @var DOMElement $templateNode */
			if ( $templateNode->hasAttribute( 'data-mw' ) ) {
				$name = self::getTemplateName( $templateNode->getAttribute( 'data-mw' ) );
				if ( $name === $templateName ) {
					$templateNode->parentNode->removeChild( $templateNode );
					if ( $templateNode->hasAttribute( 'about' ) ) {
						$about = $templateNode->getAttribute( 'about' );
						self::removeAboutNodes( $dom, $about );
					}
				}
			}
		}

		$body = $xpath->query( '/html/body' )->item(0);
		return $dom->saveHTML( $body );
	}

	/**
	 * @param string $dataMW
	 * @return string|null
	 */
	private static function getTemplateName( $dataMW ) {
		try {
			$mwAttr = json_decode( $dataMW );
			return $mwAttr->parts[0]->template->target->wt;
		} catch ( \Exception $e ) {
			return null;
		}
	}

	/**
	 * @param DOMDocument $dom
	 * @param string $about
	 */
	private static function removeAboutNodes( DOMDocument $dom, $about ) {
		$xpath = new DOMXPath( $dom );
		$aboutNodes = $xpath->query( '//*[@about="' . $about . '"]' );
		foreach ( $aboutNodes as $aboutNode ) {
			$aboutNode->parentNode->removeChild( $aboutNode );
		}
	}

	/**
	 * Extract templates from wikitext content.
	 * Requires Parsoid to reliably extract templates.
	 *
	 * @param string $content
	 * @param Title $title
	 * @return string
	 */
	public static function extractTemplates( $content, Title $title ) {
		$content = Utils::convert( 'wikitext', 'html', $content, $title );
		$dom = Utils::createDOM( $content );
		$xpath = new \DOMXPath( $dom );
		$templates = $xpath->query( '//*[@typeof="mw:Transclusion"]' );

		$content = '';
		foreach ( $templates as $template ) {
			$content .= $dom->saveHTML( $template ) . "\n";
		}

		return Utils::convert( 'html', 'wikitext', $content, $title );
	}

}