mirror of
https://github.com/ezyang/htmlpurifier.git
synced 2024-12-23 08:51:53 +00:00
357 lines
14 KiB
PHP
357 lines
14 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Injector that auto paragraphs text in the root node based on
|
|
* double-spacing.
|
|
* @todo Ensure all states are unit tested, including variations as well.
|
|
* @todo Make a graph of the flow control for this Injector.
|
|
*/
|
|
class HTMLPurifier_Injector_AutoParagraph extends HTMLPurifier_Injector
|
|
{
|
|
/**
|
|
* @type string
|
|
*/
|
|
public $name = 'AutoParagraph';
|
|
|
|
/**
|
|
* @type array
|
|
*/
|
|
public $needed = array('p');
|
|
|
|
/**
|
|
* @return HTMLPurifier_Token_Start
|
|
*/
|
|
private function _pStart()
|
|
{
|
|
$par = new HTMLPurifier_Token_Start('p');
|
|
$par->armor['MakeWellFormed_TagClosedError'] = true;
|
|
return $par;
|
|
}
|
|
|
|
/**
|
|
* @param HTMLPurifier_Token_Text $token
|
|
*/
|
|
public function handleText(&$token)
|
|
{
|
|
$text = $token->data;
|
|
// Does the current parent allow <p> tags?
|
|
if ($this->allowsElement('p')) {
|
|
if (empty($this->currentNesting) || strpos($text, "\n\n") !== false) {
|
|
// Note that we have differing behavior when dealing with text
|
|
// in the anonymous root node, or a node inside the document.
|
|
// If the text as a double-newline, the treatment is the same;
|
|
// if it doesn't, see the next if-block if you're in the document.
|
|
|
|
$i = $nesting = null;
|
|
if (!$this->forwardUntilEndToken($i, $current, $nesting) && $token->is_whitespace) {
|
|
// State 1.1: ... ^ (whitespace, then document end)
|
|
// ----
|
|
// This is a degenerate case
|
|
} else {
|
|
if (!$token->is_whitespace || $this->_isInline($current)) {
|
|
// State 1.2: PAR1
|
|
// ----
|
|
|
|
// State 1.3: PAR1\n\nPAR2
|
|
// ------------
|
|
|
|
// State 1.4: <div>PAR1\n\nPAR2 (see State 2)
|
|
// ------------
|
|
$token = array($this->_pStart());
|
|
$this->_splitText($text, $token);
|
|
} else {
|
|
// State 1.5: \n<hr />
|
|
// --
|
|
}
|
|
}
|
|
} else {
|
|
// State 2: <div>PAR1... (similar to 1.4)
|
|
// ----
|
|
|
|
// We're in an element that allows paragraph tags, but we're not
|
|
// sure if we're going to need them.
|
|
if ($this->_pLookAhead()) {
|
|
// State 2.1: <div>PAR1<b>PAR1\n\nPAR2
|
|
// ----
|
|
// Note: This will always be the first child, since any
|
|
// previous inline element would have triggered this very
|
|
// same routine, and found the double newline. One possible
|
|
// exception would be a comment.
|
|
$token = array($this->_pStart(), $token);
|
|
} else {
|
|
// State 2.2.1: <div>PAR1<div>
|
|
// ----
|
|
|
|
// State 2.2.2: <div>PAR1<b>PAR1</b></div>
|
|
// ----
|
|
}
|
|
}
|
|
// Is the current parent a <p> tag?
|
|
} elseif (!empty($this->currentNesting) &&
|
|
$this->currentNesting[count($this->currentNesting) - 1]->name == 'p') {
|
|
// State 3.1: ...<p>PAR1
|
|
// ----
|
|
|
|
// State 3.2: ...<p>PAR1\n\nPAR2
|
|
// ------------
|
|
$token = array();
|
|
$this->_splitText($text, $token);
|
|
// Abort!
|
|
} else {
|
|
// State 4.1: ...<b>PAR1
|
|
// ----
|
|
|
|
// State 4.2: ...<b>PAR1\n\nPAR2
|
|
// ------------
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param HTMLPurifier_Token $token
|
|
*/
|
|
public function handleElement(&$token)
|
|
{
|
|
// We don't have to check if we're already in a <p> tag for block
|
|
// tokens, because the tag would have been autoclosed by MakeWellFormed.
|
|
if ($this->allowsElement('p')) {
|
|
if (!empty($this->currentNesting)) {
|
|
if ($this->_isInline($token)) {
|
|
// State 1: <div>...<b>
|
|
// ---
|
|
// Check if this token is adjacent to the parent token
|
|
// (seek backwards until token isn't whitespace)
|
|
$i = null;
|
|
$this->backward($i, $prev);
|
|
|
|
if (!$prev instanceof HTMLPurifier_Token_Start) {
|
|
// Token wasn't adjacent
|
|
if ($prev instanceof HTMLPurifier_Token_Text &&
|
|
substr($prev->data, -2) === "\n\n"
|
|
) {
|
|
// State 1.1.4: <div><p>PAR1</p>\n\n<b>
|
|
// ---
|
|
// Quite frankly, this should be handled by splitText
|
|
$token = array($this->_pStart(), $token);
|
|
} else {
|
|
// State 1.1.1: <div><p>PAR1</p><b>
|
|
// ---
|
|
// State 1.1.2: <div><br /><b>
|
|
// ---
|
|
// State 1.1.3: <div>PAR<b>
|
|
// ---
|
|
}
|
|
} else {
|
|
// State 1.2.1: <div><b>
|
|
// ---
|
|
// Lookahead to see if <p> is needed.
|
|
if ($this->_pLookAhead()) {
|
|
// State 1.3.1: <div><b>PAR1\n\nPAR2
|
|
// ---
|
|
$token = array($this->_pStart(), $token);
|
|
} else {
|
|
// State 1.3.2: <div><b>PAR1</b></div>
|
|
// ---
|
|
|
|
// State 1.3.3: <div><b>PAR1</b><div></div>\n\n</div>
|
|
// ---
|
|
}
|
|
}
|
|
} else {
|
|
// State 2.3: ...<div>
|
|
// -----
|
|
}
|
|
} else {
|
|
if ($this->_isInline($token)) {
|
|
// State 3.1: <b>
|
|
// ---
|
|
// This is where the {p} tag is inserted, not reflected in
|
|
// inputTokens yet, however.
|
|
$token = array($this->_pStart(), $token);
|
|
} else {
|
|
// State 3.2: <div>
|
|
// -----
|
|
}
|
|
|
|
$i = null;
|
|
if ($this->backward($i, $prev)) {
|
|
if (!$prev instanceof HTMLPurifier_Token_Text) {
|
|
// State 3.1.1: ...</p>{p}<b>
|
|
// ---
|
|
// State 3.2.1: ...</p><div>
|
|
// -----
|
|
if (!is_array($token)) {
|
|
$token = array($token);
|
|
}
|
|
array_unshift($token, new HTMLPurifier_Token_Text("\n\n"));
|
|
} else {
|
|
// State 3.1.2: ...</p>\n\n{p}<b>
|
|
// ---
|
|
// State 3.2.2: ...</p>\n\n<div>
|
|
// -----
|
|
// Note: PAR<ELEM> cannot occur because PAR would have been
|
|
// wrapped in <p> tags.
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
// State 2.2: <ul><li>
|
|
// ----
|
|
// State 2.4: <p><b>
|
|
// ---
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Splits up a text in paragraph tokens and appends them
|
|
* to the result stream that will replace the original
|
|
* @param string $data String text data that will be processed
|
|
* into paragraphs
|
|
* @param HTMLPurifier_Token[] $result Reference to array of tokens that the
|
|
* tags will be appended onto
|
|
*/
|
|
private function _splitText($data, &$result)
|
|
{
|
|
$raw_paragraphs = explode("\n\n", $data);
|
|
$paragraphs = array(); // without empty paragraphs
|
|
$needs_start = false;
|
|
$needs_end = false;
|
|
|
|
$c = count($raw_paragraphs);
|
|
if ($c == 1) {
|
|
// There were no double-newlines, abort quickly. In theory this
|
|
// should never happen.
|
|
$result[] = new HTMLPurifier_Token_Text($data);
|
|
return;
|
|
}
|
|
for ($i = 0; $i < $c; $i++) {
|
|
$par = $raw_paragraphs[$i];
|
|
if (trim($par) !== '') {
|
|
$paragraphs[] = $par;
|
|
} else {
|
|
if ($i == 0) {
|
|
// Double newline at the front
|
|
if (empty($result)) {
|
|
// The empty result indicates that the AutoParagraph
|
|
// injector did not add any start paragraph tokens.
|
|
// This means that we have been in a paragraph for
|
|
// a while, and the newline means we should start a new one.
|
|
$result[] = new HTMLPurifier_Token_End('p');
|
|
$result[] = new HTMLPurifier_Token_Text("\n\n");
|
|
// However, the start token should only be added if
|
|
// there is more processing to be done (i.e. there are
|
|
// real paragraphs in here). If there are none, the
|
|
// next start paragraph tag will be handled by the
|
|
// next call to the injector
|
|
$needs_start = true;
|
|
} else {
|
|
// We just started a new paragraph!
|
|
// Reinstate a double-newline for presentation's sake, since
|
|
// it was in the source code.
|
|
array_unshift($result, new HTMLPurifier_Token_Text("\n\n"));
|
|
}
|
|
} elseif ($i + 1 == $c) {
|
|
// Double newline at the end
|
|
// There should be a trailing </p> when we're finally done.
|
|
$needs_end = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Check if this was just a giant blob of whitespace. Move this earlier,
|
|
// perhaps?
|
|
if (empty($paragraphs)) {
|
|
return;
|
|
}
|
|
|
|
// Add the start tag indicated by \n\n at the beginning of $data
|
|
if ($needs_start) {
|
|
$result[] = $this->_pStart();
|
|
}
|
|
|
|
// Append the paragraphs onto the result
|
|
foreach ($paragraphs as $par) {
|
|
$result[] = new HTMLPurifier_Token_Text($par);
|
|
$result[] = new HTMLPurifier_Token_End('p');
|
|
$result[] = new HTMLPurifier_Token_Text("\n\n");
|
|
$result[] = $this->_pStart();
|
|
}
|
|
|
|
// Remove trailing start token; Injector will handle this later if
|
|
// it was indeed needed. This prevents from needing to do a lookahead,
|
|
// at the cost of a lookbehind later.
|
|
array_pop($result);
|
|
|
|
// If there is no need for an end tag, remove all of it and let
|
|
// MakeWellFormed close it later.
|
|
if (!$needs_end) {
|
|
array_pop($result); // removes \n\n
|
|
array_pop($result); // removes </p>
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns true if passed token is inline (and, ergo, allowed in
|
|
* paragraph tags)
|
|
* @param HTMLPurifier_Token $token
|
|
* @return bool
|
|
*/
|
|
private function _isInline($token)
|
|
{
|
|
return isset($this->htmlDefinition->info['p']->child->elements[$token->name]);
|
|
}
|
|
|
|
/**
|
|
* Looks ahead in the token list and determines whether or not we need
|
|
* to insert a <p> tag.
|
|
* @return bool
|
|
*/
|
|
private function _pLookAhead()
|
|
{
|
|
if ($this->currentToken instanceof HTMLPurifier_Token_Start) {
|
|
$nesting = 1;
|
|
} else {
|
|
$nesting = 0;
|
|
}
|
|
$ok = false;
|
|
$i = null;
|
|
while ($this->forwardUntilEndToken($i, $current, $nesting)) {
|
|
$result = $this->_checkNeedsP($current);
|
|
if ($result !== null) {
|
|
$ok = $result;
|
|
break;
|
|
}
|
|
}
|
|
return $ok;
|
|
}
|
|
|
|
/**
|
|
* Determines if a particular token requires an earlier inline token
|
|
* to get a paragraph. This should be used with _forwardUntilEndToken
|
|
* @param HTMLPurifier_Token $current
|
|
* @return bool
|
|
*/
|
|
private function _checkNeedsP($current)
|
|
{
|
|
if ($current instanceof HTMLPurifier_Token_Start) {
|
|
if (!$this->_isInline($current)) {
|
|
// <div>PAR1<div>
|
|
// ----
|
|
// Terminate early, since we hit a block element
|
|
return false;
|
|
}
|
|
} elseif ($current instanceof HTMLPurifier_Token_Text) {
|
|
if (strpos($current->data, "\n\n") !== false) {
|
|
// <div>PAR1<b>PAR1\n\nPAR2
|
|
// ----
|
|
return true;
|
|
} else {
|
|
// <div>PAR1<b>PAR1...
|
|
// ----
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
|
|
// vim: et sw=4 sts=4
|