<?php
namespace Combodo\iTop\Test\UnitTest\Core\Sanitizer;

use HTMLDOMSanitizer;
use InlineImageMock;


require_once __DIR__.'/AbstractDOMSanitizerTest.php';


class HTMLDOMSanitizerTest extends AbstractDOMSanitizerTest
{
	/**
	 * @dataProvider DoSanitizeProvider
	 *
	 * @param string $sFileToTest filename
	 */
	public function testDoSanitize($sFileToTest)
	{
		$sInputHtml = $this->ReadTestFile($sFileToTest, self::INPUT_DIRECTORY);
		$sOutputHtml = $this->ReadTestFile($sFileToTest, self::OUTPUT_DIRECTORY);
		$sOutputHtml = $this->RemoveNewLines($sOutputHtml);

		$oSanitizer = new HTMLDOMSanitizer();
		$sRes = $oSanitizer->DoSanitize($sInputHtml);

		// Removing newlines as the parser gives different results depending on the PHP version
		// Didn't manage to get it right :
		// - no php.ini difference
		// - playing with the parser preserveWhitespace/formatOutput parser options didn't help
		// So we're removing new lines on both sides :/
		$sOutputHtml = $this->RemoveNewLines($sOutputHtml);
		$sRes = $this->RemoveNewLines($sRes);

		$this->debug($sRes);
		$this->assertEquals($sOutputHtml, $sRes);
	}

	public function DoSanitizeProvider()
	{
		return array(
			array(
				'scripts.html',
			),
		);
	}

	/**
	 * @dataProvider WhiteListProvider
	 *
	 * @param string $sHtmlToTest HTML content
	 */
	public function testDoSanitizeWhiteList($sHtmlToTest)
	{
		$oSanitizer = new HTMLDOMSanitizer();
		$sRes = $oSanitizer->DoSanitize($sHtmlToTest);

		// Removing newlines as the parser gives different results depending on the PHP version
		// Didn't manage to get it right :
		// - no php.ini difference
		// - playing with the parser preserveWhitespace/formatOutput parser options didn't help
		// So we're removing new lines on both sides :/
		$sHtmlToTest = $this->RemoveNewLines($sHtmlToTest);
		$sRes = $this->RemoveNewLines($sRes);

		$this->debug($sRes);
		$this->assertEquals($sHtmlToTest, $sRes);
	}

	public function WhiteListProvider()
	{
		// This is a copy of \HTMLDOMSanitizer::$aTagsWhiteList
		// should stay a copy as we want to check we're not removing something by mistake as it was done with the CENTER tag (N°2558)
		$aTagsWhiteList = array(
			// we don't test HTML and BODY as the parser removes them if context isn't appropriate
			'a' => array('href', 'name', 'style', 'target', 'title'),
			'p' => array('style'),
			'blockquote' => array('style'),
			'br' => array(),
			'span' => array('style'),
			'div' => array('style'),
			'b' => array(),
			'i' => array(),
			'u' => array(),
			'em' => array(),
			'strong' => array(),
			'img' => array('src', 'style', 'alt', 'title'),
			'ul' => array('style'),
			'ol' => array('style'),
			'li' => array('style'),
			'h1' => array('style'),
			'h2' => array('style'),
			'h3' => array('style'),
			'h4' => array('style'),
			'nav' => array('style'),
			'section' => array('style'),
			'code' => array('style'),
			'table' => array('style', 'width', 'summary', 'align', 'border', 'cellpadding', 'cellspacing'),
			'thead' => array('style'),
			'tbody' => array('style'),
			'tr' => array('style', 'colspan', 'rowspan'),
			'td' => array('style', 'colspan', 'rowspan'),
			'th' => array('style', 'colspan', 'rowspan'),
			'fieldset' => array('style'),
			'legend' => array('style'),
			'font' => array('face', 'color', 'style', 'size'),
			'big' => array(),
			'small' => array(),
			'tt' => array(),
			'kbd' => array(),
			'samp' => array(),
			'var' => array(),
			'del' => array(),
			's' => array(), // strikethrough
			'ins' => array(),
			'cite' => array(),
			'q' => array(),
			'hr' => array('style'),
			'pre' => array(),
			'center' => array(),
		);
		$aTestCaseArray = array();

		$sInputText = $this->ReadTestFile('whitelist_test.html', self::INPUT_DIRECTORY);
		foreach ($aTagsWhiteList as $sTag => $aTagAttributes) {
			$sTestCaseText = $sInputText;
			$sStartTag = "<$sTag";
			$iAttrCounter = 0;
			foreach ($aTagAttributes as $sTagAttribute) {
				$sStartTag .= $this->GetTagAttributeValue($sTagAttribute, $iAttrCounter);
				$iAttrCounter++;
			}
			$sStartTag .= '>';
			$sTestCaseText = str_replace('##START_TAG##', $sStartTag, $sTestCaseText);

			$sClosingTag = $this->IsClosingTag($sTag) ? "</$sTag>" : '';
			$sTestCaseText = str_replace('##END_TAG##', $sClosingTag, $sTestCaseText);

			$aTestCaseArray[$sTag] = array($sTestCaseText);
		}

		return $aTestCaseArray;
	}

	/**
	 * @dataProvider RemoveBlackListedTagContentProvider
	 */
	public function testDoSanitizeRemoveBlackListedTagContent($html, $expected)
	{
		$oSanitizer = new HTMLDOMSanitizer();
		$sSanitizedHtml = $oSanitizer->DoSanitize($html);

		$this->assertEquals($expected, str_replace("\n", '', $sSanitizedHtml));
	}

	public function RemoveBlackListedTagContentProvider()
	{
		return array(
			'basic' => array(
				'html' => 'foo<iframe>bar</iframe>baz',
				'expected' => '<p>foobaz</p>',
			),
			'basic with body' => array(
				'html' => '<body>foo<iframe>bar</iframe>baz</body>',
				'expected' => 'foobaz',
			),
			'basic with html and body tags' => array(
				'html' => '<html><body lang="EN-GB" link="#0563C1" vlink="#954F72">foo<iframe>bar</iframe>baz</body></html>',
				'expected' => 'foobaz',
			),
			'basic with attributes' => array(
				'html' => 'foo<iframe baz="1">bar</iframe>baz',
				'expected' => '<p>foobaz</p>',
			),
			'basic with comment' => array(
				'html' => 'foo<iframe baz="1">bar<!-- foo --></iframe>baz',
				'expected' => '<p>foobaz</p>',
			),
			'basic with contentRemovable tag' => array(
				'html' => 'foo<iframe baz="1">bar<style>foo</style><script>boo</script></iframe>baz',
				'expected' => '<p>foobaz</p>',
			),
			'nested' => array(
				'html' => 'before<iframe>foo<article>baz</article>oof<article><iframe>bar</iframe>oof</article></iframe>after',
				'expected' => '<p>beforeafter</p>',
			),
			'nested with not closed br' => array(
				'html' => 'before<iframe>foo<article>baz</article>oof<br><article><iframe>bar</iframe>oof</article></iframe>after',
				'expected' => '<p>beforeafter</p>',
			),
			'nested with allowed' => array(
				'html' => 'before<iframe><div><article><p>baz</p>zab</article></div>oof</iframe>after',
				'expected' => '<p>beforeafter</p>',
			),
			'nested with spaces' => array(
				'html' => 'before<iframe><article>baz</article> oof</iframe>after',
				'expected' => '<p>beforeafter</p>',
			),
			'nested with attributes' => array(
				'html' => 'before<iframe baz="1"><article baz="1" biz="2">baz</article>oof</iframe>after',
				'expected' => '<p>beforeafter</p>',
			),
			'nested with allowed and attributes and spaces ' => array(
				'html' => '<html><body>before<iframe baz="1"><div baz="baz"><article baz="1" biz="2">baz</article>rab</div> oof</iframe>after</body></html>',
				'expected' => 'beforeafter',
			),
			'nested with allowed and contentRemovable tags' => array(
				'html' => '<html><body>before<iframe baz="1"><div ><article>baz</article>rab</div> oof<embed>embedTExt</embed></iframe>middle<style>foo</style>after<script>boo</script></body></html>',
				'expected' => 'beforemiddleafter',
			),

			'regression: if head present => body is not trimmed' => array(
				'html' => '<html><head></head><body lang="EN-GB" link="#0563C1" vlink="#954F72">bar</body></html>',
				'expected' => 'bar',
			),
		);
	}

	/**
	 * @dataProvider CallInlineImageProcessImageTagProvider
	 * @uses         \InlineImageMock
	 */
	public function testDoSanitizeCallInlineImageProcessImageTag($sHtml, $iExpectedCount)
	{
		require_once APPROOT.'test/core/sanitizer/InlineImageMock.php';
		InlineImageMock::ResetCallCounter();

		$oSanitizer = new HTMLDOMSanitizer(InlineImageMock::class);
		$oSanitizer->DoSanitize($sHtml);

		$iCalledCount = \InlineImageMock::GetCallCounter();
		$this->assertEquals($iExpectedCount, $iCalledCount);
	}

	public function CallInlineImageProcessImageTagProvider()
	{
		return array(
			'no image' => array(
				'html' => '<p>bar</p>',
				'expected' => 0,
			),
			'basic image' => array(
				'html' => '<img />',
				'expected' => 1,
			),
			'nested images within forbidden tags' => array(
				'html' => '<html><body><img /><iframe baz="1"><div baz="baz"><article baz="1" biz="2">baz<img /><img /></article>rab</div> oof<img /></iframe><img /></body></html>',
				'expected' => 2,
			),
//          This test will be restored with the ticket n°2556
//			'nested images within forbidden and removed tags' => array(
//				'html' => '<html><body><img /><iframe baz="1"><div baz="baz"><object baz="1" biz="2">baz<img /><img /></object>rab</div> oof<img /></iframe><img /></body></html>',
//				'expected' => 2,
//			),
		);
	}

}