[UTIL][HTML][TESTS] Fix test and implementation and expand HTML generation utilities

This commit is contained in:
Hugo Sales 2021-11-16 13:12:32 +00:00
parent fe755f7c42
commit 0e104a9701
Signed by: someonewithpc
GPG Key ID: 7D0C7EAFC9D835A0
2 changed files with 50 additions and 41 deletions

View File

@ -34,17 +34,25 @@ use InvalidArgumentException;
abstract class HTML abstract class HTML
{ {
public const ALLOWED_TAGS = ['p', 'br', 'a', 'span']; /**
* Tags whose content is sensitive to indentation, so we shouldn't indent them
*/
public const NO_INDENT_TAGS = ['a', 'b', 'em', 'i', 'q', 's', 'p', 'sub', 'sup', 'u'];
public const ALLOWED_TAGS = ['p', 'br', 'a', 'span', 'div'];
public const FORBIDDEN_ATTRIBUTES = [ public const FORBIDDEN_ATTRIBUTES = [
'onerror', 'form', 'onforminput', 'onbeforescriptexecute', 'formaction', 'onfocus', 'onload', 'onerror', 'form', 'onforminput', 'onbeforescriptexecute', 'formaction', 'onfocus', 'onload',
'data', 'event', 'autofocus', 'onactivate', 'onanimationstart', 'onwebkittransitionend', 'onblur', 'poster', 'data', 'event', 'autofocus', 'onactivate', 'onanimationstart', 'onwebkittransitionend', 'onblur', 'poster',
'onratechange', 'ontoggle', 'onscroll', 'actiontype', 'dirname', 'srcdoc', 'onratechange', 'ontoggle', 'onscroll', 'actiontype', 'dirname', 'srcdoc',
]; ];
public const SELF_CLOSING_TAG = ['area', 'base', 'br', 'col', 'embed', 'hr', 'img', 'link', 'meta', 'param', 'source', 'track', 'wbr'];
/** /**
* Creates an HTML tag without attributes * Creates an HTML tag without attributes
*/ */
public static function tag(string $tag, mixed $attrs = null, mixed $content = null, array $options = []): array public static function tag(string $tag, mixed $attrs = null, mixed $content = null, array $options = []): string
{ {
return self::attr_tag($tag, $attrs ?? '', $content ?? '', $options); return self::attr_tag($tag, $attrs ?? '', $content ?? '', $options);
} }
@ -52,20 +60,20 @@ abstract class HTML
/** /**
* Create tag, possibly with attributes and indentation * Create tag, possibly with attributes and indentation
*/ */
private static function attr_tag(string $tag, mixed $attrs, mixed $content = '', array $options = []): array private static function attr_tag(string $tag, mixed $attrs, mixed $content = '', array $options = []): string
{ {
$html = '<' . $tag . (\is_string($attrs) ? ($attrs ? ' ' : '') . $attrs : self::attr($attrs, $options)); $html = '<' . $tag . (\is_string($attrs) ? ($attrs ? ' ' : '') . $attrs : self::attr($attrs, $options));
if ($options['empty'] ?? false) { if (\in_array($tag, self::SELF_CLOSING_TAG)) {
$html .= '/>'; $html .= '>';
} else { } else {
if ($options['indent'] ?? true) { if (!\in_array($tag, self::NO_INDENT_TAGS) && ($options['indent'] ?? true)) {
$inner = Formatting::indent($content); $inner = Formatting::indent($content);
$html .= ">\n" . ($inner == '' ? '' : $inner . "\n") . "</{$tag}>"; $html .= ">\n" . ($inner == '' ? '' : $inner . "\n") . "</{$tag}>";
} else { } else {
$html .= ">{$content}</{$tag}>"; $html .= ">{$content}</{$tag}>";
} }
} }
return explode("\n", $html); return $html;
} }
/** /**
@ -73,32 +81,29 @@ abstract class HTML
*/ */
private static function attr(array $attrs, array $options = []): string private static function attr(array $attrs, array $options = []): string
{ {
return ' ' . implode( return ' ' . implode(' ', F\map($attrs, [self::class, '_process_attribute']));
' ', }
F\map(
$attrs, /**
/** * Convert an attr ($key), $val pair to an HTML attribute, but validate to exclude some vectors of injection
* Convert an attr ($key), $val pair to an HTML attribute, but validate to exclude some vectors of injection */
*/ public static function _process_attribute(string $val, string $key): string
function (string $val, string $key, array $_): string { {
if (\in_array($key, array_merge($options['forbidden_attributes'] ?? [], self::FORBIDDEN_ATTRIBUTES)) if (\in_array($key, array_merge($options['forbidden_attributes'] ?? [], self::FORBIDDEN_ATTRIBUTES))
|| str_starts_with($val, 'javascript:')) { || str_starts_with($val, 'javascript:')) {
throw new InvalidArgumentException("HTML::html: Attribute {$key} is not allowed"); throw new InvalidArgumentException("HTML::html: Attribute {$key} is not allowed");
} }
if (!($options['raw'] ?? false)) { if (!($options['raw'] ?? false)) {
$val = htmlspecialchars($val, flags: \ENT_QUOTES | \ENT_SUBSTITUTE, double_encode: false); $val = htmlspecialchars($val, flags: \ENT_QUOTES | \ENT_SUBSTITUTE, double_encode: false);
} }
return "{$key}=\"{$val}\""; return "{$key}=\"{$val}\"";
},
),
);
} }
/** /**
* @param array|string $html The input to convert to HTML * @param array|string $html The input to convert to HTML
* @param array $options = [] ['allowed_tags' => string[], 'forbidden_attributes' => string[], 'raw' => bool, 'indent' => bool] * @param array $options = [] ['allowed_tags' => string[], 'forbidden_attributes' => string[], 'raw' => bool, 'indent' => bool]
*/ */
public static function html(string|array $html, array $options = [], int $indent = 1): string public static function html(string|array $html, array $options = []): string
{ {
if (\is_string($html)) { if (\is_string($html)) {
if ($options['raw'] ?? false) { if ($options['raw'] ?? false) {
@ -109,18 +114,18 @@ abstract class HTML
} else { } else {
$out = ''; $out = '';
foreach ($html as $tag => $contents) { foreach ($html as $tag => $contents) {
if ($contents['empty'] ?? false) { if (\in_array($tag, self::SELF_CLOSING_TAG)) {
$out .= "<{$tag}/>"; $out .= "<{$tag}>";
} else { } else {
$attrs = isset($contents['attrs']) ? self::attr(array_shift($contents), $options) : ''; $attrs = isset($contents['attrs']) ? self::attr(array_shift($contents), $options) : '';
$is_tag = is_string($tag) && preg_match('/[A-Za-z][A-Za-z0-9]*/', $tag); $is_tag = \is_string($tag) && preg_match('/[A-Za-z][A-Za-z0-9]*/', $tag);
$inner = self::html($contents, $options, $indent + 1); $inner = self::html($contents, $options);
if ($is_tag) { if ($is_tag) {
if (!\in_array($tag, array_merge($options['allowed_tags'] ?? [], self::ALLOWED_TAGS))) { if (!\in_array($tag, array_merge($options['allowed_tags'] ?? [], self::ALLOWED_TAGS))) {
throw new InvalidArgumentException("HTML::html: Tag {$tag} is not allowed"); throw new InvalidArgumentException("HTML::html: Tag {$tag} is not allowed");
} }
if (!empty($inner)) { if (!empty($inner) && !\in_array($tag, self::NO_INDENT_TAGS) && ($options['indent'] ?? true)) {
$inner = ($options['indent'] ?? true) ? ("\n" . Formatting::indent($inner, $indent) . "\n") : $inner; $inner = "\n" . Formatting::indent($inner) . "\n";
} }
$out .= "<{$tag}{$attrs}>{$inner}</{$tag}>"; $out .= "<{$tag}{$attrs}>{$inner}</{$tag}>";
} else { } else {

View File

@ -22,9 +22,9 @@ declare(strict_types = 1);
namespace App\Tests\Util; namespace App\Tests\Util;
use App\Util\HTML; use App\Util\HTML;
use InvalidArgumentException;
use Jchook\AssertThrows\AssertThrows; use Jchook\AssertThrows\AssertThrows;
use Symfony\Bundle\FrameworkBundle\Test\WebTestCase; use Symfony\Bundle\FrameworkBundle\Test\WebTestCase;
use TypeError;
class HTMLTest extends WebTestCase class HTMLTest extends WebTestCase
{ {
@ -33,12 +33,16 @@ class HTMLTest extends WebTestCase
public function testHTML() public function testHTML()
{ {
static::assertSame('', HTML::html('')); static::assertSame('', HTML::html(''));
static::assertSame("<a>\n\n</a>\n", HTML::html(['a' => ''])); static::assertSame('<a></a>', HTML::html(['a' => '']));
static::assertSame("<a>\n <p>\n </p>\n</a>\n", HTML::html(['a' => ['p' => '']])); static::assertSame("<div>\n <p></p>\n</div>", HTML::html(['div' => ['p' => '']]));
static::assertSame("<a href=\"test\">\n <p>\n </p>\n</a>\n", HTML::html(['a' => ['attrs' => ['href' => 'test'], 'p' => '']])); static::assertSame("<div>\n <div>\n <p></p>\n </div>\n</div>", HTML::html(['div' => ['div' => ['p' => '']]]));
static::assertSame("<a>\n <p>\n foo\n </p>\n <br/>\n</a>\n", HTML::html(['a' => ['p' => 'foo', 'br' => 'empty']])); static::assertSame("<div>\n <div>\n <div>\n <p></p>\n </div>\n </div>\n</div>", HTML::html(['div' => ['div' => ['div' => ['p' => '']]]]));
static::assertThrows(InvalidArgumentException::class, fn () => HTML::html(1)); static::assertSame('<a href="test"><p></p></a>', HTML::html(['a' => ['attrs' => ['href' => 'test'], 'p' => '']]));
static::assertSame("<a href=\"test\">\n foo\n</a>", implode("\n", HTML::tag('a', ['href' => 'test'], content: 'foo', options: ['empty' => false]))); static::assertSame('<a><p>foo</p><br></a>', HTML::html(['a' => ['p' => 'foo', 'br' => 'empty']]));
static::assertSame('<br/>', implode("\n", HTML::tag('br', attrs: null, content: null, options: ['empty' => true]))); static::assertSame("<div>\n <a><p>foo</p><br></a>\n</div>", HTML::html(['div' => ['a' => ['p' => 'foo', 'br' => 'empty']]]));
static::assertSame('<div><a><p>foo</p><br></a></div>', HTML::html(['div' => ['a' => ['p' => 'foo', 'br' => 'empty']]], options: ['indent' => false]));
static::assertThrows(TypeError::class, fn () => HTML::html(1));
static::assertSame('<a href="test">foo</a>', HTML::tag('a', ['href' => 'test'], content: 'foo', options: ['empty' => false]));
static::assertSame('<br>', HTML::tag('br', attrs: null, content: null, options: ['empty' => true]));
} }
} }