includes/parser/BlockLevelPass.php

   1 <?php
   2
   3 /**
   4  * This is the part of the wikitext parser which handles automatic paragraphs
   5  * and conversion of start-of-line prefixes to HTML lists.
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20  * http://www.gnu.org/copyleft/gpl.html
  21  *
  22  * @file
  23  * @ingroup Parser
  24  */
  25 class BlockLevelPass {
  26         private $DTopen = false;
  27         private $inPre = false;
  28         private $lastSection = '';
  29         private $lineStart;
  30         private $text;
  31
  32         # State constants for the definition list colon extraction
  33         const COLON_STATE_TEXT = 0;
  34         const COLON_STATE_TAG = 1;
  35         const COLON_STATE_TAGSTART = 2;
  36         const COLON_STATE_CLOSETAG = 3;
  37         const COLON_STATE_TAGSLASH = 4;
  38         const COLON_STATE_COMMENT = 5;
  39         const COLON_STATE_COMMENTDASH = 6;
  40         const COLON_STATE_COMMENTDASHDASH = 7;
  41         const COLON_STATE_LC = 8;
  42
  43         /**
  44          * Make lists from lines starting with ':', '*', '#', etc.
  45          *
  46          * @param string $text
  47          * @param bool $lineStart Whether or not this is at the start of a line.
  48          * @return string The lists rendered as HTML
  49          */
  50         public static function doBlockLevels( $text, $lineStart ) {
  51                 $pass = new self( $text, $lineStart );
  52                 return $pass->execute();
  53         }
  54
  55         /**
  56          * Private constructor
  57          */
  58         private function __construct( $text, $lineStart ) {
  59                 $this->text = $text;
  60                 $this->lineStart = $lineStart;
  61         }
  62
  63         /**
  64          * If a pre or p is open, return the corresponding close tag and update
  65          * the state. If no tag is open, return an empty string.
  66          * @return string
  67          */
  68         private function closeParagraph() {
  69                 $result = '';
  70                 if ( $this->lastSection !== '' ) {
  71                         $result = '</' . $this->lastSection . ">\n";
  72                 }
  73                 $this->inPre = false;
  74                 $this->lastSection = '';
  75                 return $result;
  76         }
  77
  78         /**
  79          * getCommon() returns the length of the longest common substring
  80          * of both arguments, starting at the beginning of both.
  81          *
  82          * @param string $st1
  83          * @param string $st2
  84          *
  85          * @return int
  86          */
  87         private function getCommon( $st1, $st2 ) {
  88                 $shorter = min( strlen( $st1 ), strlen( $st2 ) );
  89
  90                 for ( $i = 0; $i < $shorter; ++$i ) {
  91                         if ( $st1[$i] !== $st2[$i] ) {
  92                                 break;
  93                         }
  94                 }
  95                 return $i;
  96         }
  97
  98         /**
  99          * Open the list item element identified by the prefix character.
 100          *
 101          * @param string $char
 102          *
 103          * @return string
 104          */
 105         private function openList( $char ) {
 106                 $result = $this->closeParagraph();
 107
 108                 if ( '*' === $char ) {
 109                         $result .= "<ul><li>";
 110                 } elseif ( '#' === $char ) {
 111                         $result .= "<ol><li>";
 112                 } elseif ( ':' === $char ) {
 113                         $result .= "<dl><dd>";
 114                 } elseif ( ';' === $char ) {
 115                         $result .= "<dl><dt>";
 116                         $this->DTopen = true;
 117                 } else {
 118                         $result = '<!-- ERR 1 -->';
 119                 }
 120
 121                 return $result;
 122         }
 123
 124         /**
 125          * Close the current list item and open the next one.
 126          * @param string $char
 127          *
 128          * @return string
 129          */
 130         private function nextItem( $char ) {
 131                 if ( '*' === $char || '#' === $char ) {
 132                         return "</li>\n<li>";
 133                 } elseif ( ':' === $char || ';' === $char ) {
 134                         $close = "</dd>\n";
 135                         if ( $this->DTopen ) {
 136                                 $close = "</dt>\n";
 137                         }
 138                         if ( ';' === $char ) {
 139                                 $this->DTopen = true;
 140                                 return $close . '<dt>';
 141                         } else {
 142                                 $this->DTopen = false;
 143                                 return $close . '<dd>';
 144                         }
 145                 }
 146                 return '<!-- ERR 2 -->';
 147         }
 148
 149         /**
 150          * Close the current list item identified by the prefix character.
 151          * @param string $char
 152          *
 153          * @return string
 154          */
 155         private function closeList( $char ) {
 156                 if ( '*' === $char ) {
 157                         $text = "</li></ul>";
 158                 } elseif ( '#' === $char ) {
 159                         $text = "</li></ol>";
 160                 } elseif ( ':' === $char ) {
 161                         if ( $this->DTopen ) {
 162                                 $this->DTopen = false;
 163                                 $text = "</dt></dl>";
 164                         } else {
 165                                 $text = "</dd></dl>";
 166                         }
 167                 } else {
 168                         return '<!-- ERR 3 -->';
 169                 }
 170                 return $text;
 171         }
 172
 173         /**
 174          * Execute the pass.
 175          * @return string
 176          */
 177         private function execute() {
 178                 $text = $this->text;
 179                 # Parsing through the text line by line.  The main thing
 180                 # happening here is handling of block-level elements p, pre,
 181                 # and making lists from lines starting with * # : etc.
 182                 $textLines = StringUtils::explode( "\n", $text );
 183
 184                 $lastPrefix = $output = '';
 185                 $this->DTopen = $inBlockElem = false;
 186                 $prefixLength = 0;
 187                 $pendingPTag = false;
 188                 $inBlockquote = false;
 189
 190                 foreach ( $textLines as $inputLine ) {
 191                         # Fix up $lineStart
 192                         if ( !$this->lineStart ) {
 193                                 $output .= $inputLine;
 194                                 $this->lineStart = true;
 195                                 continue;
 196                         }
 197                         # * = ul
 198                         # # = ol
 199                         # ; = dt
 200                         # : = dd
 201
 202                         $lastPrefixLength = strlen( $lastPrefix );
 203                         $preCloseMatch = preg_match( '/<\\/pre/i', $inputLine );
 204                         $preOpenMatch = preg_match( '/<pre/i', $inputLine );
 205                         # If not in a <pre> element, scan for and figure out what prefixes are there.
 206                         if ( !$this->inPre ) {
 207                                 # Multiple prefixes may abut each other for nested lists.
 208                                 $prefixLength = strspn( $inputLine, '*#:;' );
 209                                 $prefix = substr( $inputLine, 0, $prefixLength );
 210
 211                                 # eh?
 212                                 # ; and : are both from definition-lists, so they're equivalent
 213                                 #  for the purposes of determining whether or not we need to open/close
 214                                 #  elements.
 215                                 $prefix2 = str_replace( ';', ':', $prefix );
 216                                 $t = substr( $inputLine, $prefixLength );
 217                                 $this->inPre = (bool)$preOpenMatch;
 218                         } else {
 219                                 # Don't interpret any other prefixes in preformatted text
 220                                 $prefixLength = 0;
 221                                 $prefix = $prefix2 = '';
 222                                 $t = $inputLine;
 223                         }
 224
 225                         # List generation
 226                         if ( $prefixLength && $lastPrefix === $prefix2 ) {
 227                                 # Same as the last item, so no need to deal with nesting or opening stuff
 228                                 $output .= $this->nextItem( substr( $prefix, -1 ) );
 229                                 $pendingPTag = false;
 230
 231                                 if ( substr( $prefix, -1 ) === ';' ) {
 232                                         # The one nasty exception: definition lists work like this:
 233                                         # ; title : definition text
 234                                         # So we check for : in the remainder text to split up the
 235                                         # title and definition, without b0rking links.
 236                                         $term = $t2 = '';
 237                                         if ( $this->findColonNoLinks( $t, $term, $t2 ) !== false ) {
 238                                                 $t = $t2;
 239                                                 $output .= $term . $this->nextItem( ':' );
 240                                         }
 241                                 }
 242                         } elseif ( $prefixLength || $lastPrefixLength ) {
 243                                 # We need to open or close prefixes, or both.
 244
 245                                 # Either open or close a level...
 246                                 $commonPrefixLength = $this->getCommon( $prefix, $lastPrefix );
 247                                 $pendingPTag = false;
 248
 249                                 # Close all the prefixes which aren't shared.
 250                                 while ( $commonPrefixLength < $lastPrefixLength ) {
 251                                         $output .= $this->closeList( $lastPrefix[$lastPrefixLength - 1] );
 252                                         --$lastPrefixLength;
 253                                 }
 254
 255                                 # Continue the current prefix if appropriate.
 256                                 if ( $prefixLength <= $commonPrefixLength && $commonPrefixLength > 0 ) {
 257                                         $output .= $this->nextItem( $prefix[$commonPrefixLength - 1] );
 258                                 }
 259
 260                                 # Close an open <dt> if we have a <dd> (":") starting on this line
 261                                 if ( $this->DTopen && $commonPrefixLength > 0 && $prefix[$commonPrefixLength - 1] === ':' ) {
 262                                         $output .= $this->nextItem( ':' );
 263                                 }
 264
 265                                 # Open prefixes where appropriate.
 266                                 if ( $lastPrefix && $prefixLength > $commonPrefixLength ) {
 267                                         $output .= "\n";
 268                                 }
 269                                 while ( $prefixLength > $commonPrefixLength ) {
 270                                         $char = $prefix[$commonPrefixLength];
 271                                         $output .= $this->openList( $char );
 272
 273                                         if ( ';' === $char ) {
 274                                                 # @todo FIXME: This is dupe of code above
 275                                                 if ( $this->findColonNoLinks( $t, $term, $t2 ) !== false ) {
 276                                                         $t = $t2;
 277                                                         $output .= $term . $this->nextItem( ':' );
 278                                                 }
 279                                         }
 280                                         ++$commonPrefixLength;
 281                                 }
 282                                 if ( !$prefixLength && $lastPrefix ) {
 283                                         $output .= "\n";
 284                                 }
 285                                 $lastPrefix = $prefix2;
 286                         }
 287
 288                         # If we have no prefixes, go to paragraph mode.
 289                         if ( 0 == $prefixLength ) {
 290                                 # No prefix (not in list)--go to paragraph mode
 291                                 # @todo consider using a stack for nestable elements like span, table and div
 292                                 $openMatch = preg_match(
 293                                         '/(?:<table|<h1|<h2|<h3|<h4|<h5|<h6|<pre|<tr|'
 294                                                 . '<p|<ul|<ol|<dl|<li|<\\/tr|<\\/td|<\\/th)\\b/iS',
 295                                         $t
 296                                 );
 297                                 $closeMatch = preg_match(
 298                                         '/(?:<\\/table|<\\/h1|<\\/h2|<\\/h3|<\\/h4|<\\/h5|<\\/h6|'
 299                                                 . '<td|<th|<\\/?blockquote|<\\/?div|<hr|<\\/pre|<\\/p|<\\/mw:|'
 300                                                 . Parser::MARKER_PREFIX
 301                                                 . '-pre|<\\/li|<\\/ul|<\\/ol|<\\/dl|<\\/?center)\\b/iS',
 302                                         $t
 303                                 );
 304
 305                                 if ( $openMatch || $closeMatch ) {
 306                                         $pendingPTag = false;
 307                                         // Only close the paragraph if we're not inside a <pre> tag, or if
 308                                         // that <pre> tag has just been opened
 309                                         if ( !$this->inPre || $preOpenMatch ) {
 310                                                 // @todo T7718: paragraph closed
 311                                                 $output .= $this->closeParagraph();
 312                                         }
 313                                         if ( $preOpenMatch && !$preCloseMatch ) {
 314                                                 $this->inPre = true;
 315                                         }
 316                                         $bqOffset = 0;
 317                                         while ( preg_match( '/<(\\/?)blockquote[\s>]/i', $t,
 318                                                 $bqMatch, PREG_OFFSET_CAPTURE, $bqOffset )
 319                                         ) {
 320                                                 $inBlockquote = !$bqMatch[1][0]; // is this a close tag?
 321                                                 $bqOffset = $bqMatch[0][1] + strlen( $bqMatch[0][0] );
 322                                         }
 323                                         $inBlockElem = !$closeMatch;
 324                                 } elseif ( !$inBlockElem && !$this->inPre ) {
 325                                         if ( ' ' == substr( $t, 0, 1 )
 326                                                 && ( $this->lastSection === 'pre' || trim( $t ) != '' )
 327                                                 && !$inBlockquote
 328                                         ) {
 329                                                 # pre
 330                                                 if ( $this->lastSection !== 'pre' ) {
 331                                                         $pendingPTag = false;
 332                                                         $output .= $this->closeParagraph() . '<pre>';
 333                                                         $this->lastSection = 'pre';
 334                                                 }
 335                                                 $t = substr( $t, 1 );
 336                                         } elseif ( preg_match( '/^(?:<style\\b[^>]*>.*?<\\/style>\s*|<link\\b[^>]*>\s*)+$/iS', $t ) ) {
 337                                                 # T186965: <style> or <link> by itself on a line shouldn't open or close paragraphs.
 338                                                 # But it should clear $pendingPTag.
 339                                                 if ( $pendingPTag ) {
 340                                                         $output .= $this->closeParagraph();
 341                                                         $pendingPTag = false;
 342                                                         $this->lastSection = '';
 343                                                 }
 344                                         } else {
 345                                                 # paragraph
 346                                                 if ( trim( $t ) === '' ) {
 347                                                         if ( $pendingPTag ) {
 348                                                                 $output .= $pendingPTag . '<br />';
 349                                                                 $pendingPTag = false;
 350                                                                 $this->lastSection = 'p';
 351                                                         } else {
 352                                                                 if ( $this->lastSection !== 'p' ) {
 353                                                                         $output .= $this->closeParagraph();
 354                                                                         $this->lastSection = '';
 355                                                                         $pendingPTag = '<p>';
 356                                                                 } else {
 357                                                                         $pendingPTag = '</p><p>';
 358                                                                 }
 359                                                         }
 360                                                 } else {
 361                                                         if ( $pendingPTag ) {
 362                                                                 $output .= $pendingPTag;
 363                                                                 $pendingPTag = false;
 364                                                                 $this->lastSection = 'p';
 365                                                         } elseif ( $this->lastSection !== 'p' ) {
 366                                                                 $output .= $this->closeParagraph() . '<p>';
 367                                                                 $this->lastSection = 'p';
 368                                                         }
 369                                                 }
 370                                         }
 371                                 }
 372                         }
 373                         # somewhere above we forget to get out of pre block (T2785)
 374                         if ( $preCloseMatch && $this->inPre ) {
 375                                 $this->inPre = false;
 376                         }
 377                         if ( $pendingPTag === false ) {
 378                                 $output .= $t;
 379                                 if ( $prefixLength === 0 ) {
 380                                         $output .= "\n";
 381                                 }
 382                         }
 383                 }
 384                 while ( $prefixLength ) {
 385                         $output .= $this->closeList( $prefix2[$prefixLength - 1] );
 386                         --$prefixLength;
 387                         if ( !$prefixLength ) {
 388                                 $output .= "\n";
 389                         }
 390                 }
 391                 if ( $this->lastSection !== '' ) {
 392                         $output .= '</' . $this->lastSection . '>';
 393                         $this->lastSection = '';
 394                 }
 395
 396                 return $output;
 397         }
 398
 399         /**
 400          * Split up a string on ':', ignoring any occurrences inside tags
 401          * to prevent illegal overlapping.
 402          *
 403          * @param string $str The string to split
 404          * @param string &$before Set to everything before the ':'
 405          * @param string &$after Set to everything after the ':'
 406          * @throws MWException
 407          * @return string The position of the ':', or false if none found
 408          */
 409         private function findColonNoLinks( $str, &$before, &$after ) {
 410                 if ( !preg_match( '/:|<|-\{/', $str, $m, PREG_OFFSET_CAPTURE ) ) {
 411                         # Nothing to find!
 412                         return false;
 413                 }
 414
 415                 if ( $m[0][0] === ':' ) {
 416                         # Easy; no tag nesting to worry about
 417                         $colonPos = $m[0][1];
 418                         $before = substr( $str, 0, $colonPos );
 419                         $after = substr( $str, $colonPos + 1 );
 420                         return $colonPos;
 421                 }
 422
 423                 # Ugly state machine to walk through avoiding tags.
 424                 $state = self::COLON_STATE_TEXT;
 425                 $ltLevel = 0;
 426                 $lcLevel = 0;
 427                 $len = strlen( $str );
 428                 for ( $i = $m[0][1]; $i < $len; $i++ ) {
 429                         $c = $str[$i];
 430
 431                         switch ( $state ) {
 432                                 case self::COLON_STATE_TEXT:
 433                                         switch ( $c ) {
 434                                                 case "<":
 435                                                         # Could be either a <start> tag or an </end> tag
 436                                                         $state = self::COLON_STATE_TAGSTART;
 437                                                         break;
 438                                                 case ":":
 439                                                         if ( $ltLevel === 0 ) {
 440                                                                 # We found it!
 441                                                                 $before = substr( $str, 0, $i );
 442                                                                 $after = substr( $str, $i + 1 );
 443                                                                 return $i;
 444                                                         }
 445                                                         # Embedded in a tag; don't break it.
 446                                                         break;
 447                                                 default:
 448                                                         # Skip ahead looking for something interesting
 449                                                         if ( !preg_match( '/:|<|-\{/', $str, $m, PREG_OFFSET_CAPTURE, $i ) ) {
 450                                                                 # Nothing else interesting
 451                                                                 return false;
 452                                                         }
 453                                                         if ( $m[0][0] === '-{' ) {
 454                                                                 $state = self::COLON_STATE_LC;
 455                                                                 $lcLevel++;
 456                                                                 $i = $m[0][1] + 1;
 457                                                         } else {
 458                                                                 # Skip ahead to next interesting character.
 459                                                                 $i = $m[0][1] - 1;
 460                                                         }
 461                                                         break;
 462                                         }
 463                                         break;
 464                                 case self::COLON_STATE_LC:
 465                                         # In language converter markup -{ ... }-
 466                                         if ( !preg_match( '/-\{|\}-/', $str, $m, PREG_OFFSET_CAPTURE, $i ) ) {
 467                                                 # Nothing else interesting to find; abort!
 468                                                 # We're nested in language converter markup, but there
 469                                                 # are no close tags left.  Abort!
 470                                                 break 2;
 471                                         } elseif ( $m[0][0] === '-{' ) {
 472                                                 $i = $m[0][1] + 1;
 473                                                 $lcLevel++;
 474                                         } elseif ( $m[0][0] === '}-' ) {
 475                                                 $i = $m[0][1] + 1;
 476                                                 $lcLevel--;
 477                                                 if ( $lcLevel === 0 ) {
 478                                                         $state = self::COLON_STATE_TEXT;
 479                                                 }
 480                                         }
 481                                         break;
 482                                 case self::COLON_STATE_TAG:
 483                                         # In a <tag>
 484                                         switch ( $c ) {
 485                                                 case ">":
 486                                                         $ltLevel++;
 487                                                         $state = self::COLON_STATE_TEXT;
 488                                                         break;
 489                                                 case "/":
 490                                                         # Slash may be followed by >?
 491                                                         $state = self::COLON_STATE_TAGSLASH;
 492                                                         break;
 493                                                 default:
 494                                                         # ignore
 495                                         }
 496                                         break;
 497                                 case self::COLON_STATE_TAGSTART:
 498                                         switch ( $c ) {
 499                                                 case "/":
 500                                                         $state = self::COLON_STATE_CLOSETAG;
 501                                                         break;
 502                                                 case "!":
 503                                                         $state = self::COLON_STATE_COMMENT;
 504                                                         break;
 505                                                 case ">":
 506                                                         # Illegal early close? This shouldn't happen D:
 507                                                         $state = self::COLON_STATE_TEXT;
 508                                                         break;
 509                                                 default:
 510                                                         $state = self::COLON_STATE_TAG;
 511                                         }
 512                                         break;
 513                                 case self::COLON_STATE_CLOSETAG:
 514                                         # In a </tag>
 515                                         if ( $c === ">" ) {
 516                                                 if ( $ltLevel > 0 ) {
 517                                                         $ltLevel--;
 518                                                 } else {
 519                                                         # ignore the excess close tag, but keep looking for
 520                                                         # colons. (This matches Parsoid behavior.)
 521                                                         wfDebug( __METHOD__ . ": Invalid input; too many close tags\n" );
 522                                                 }
 523                                                 $state = self::COLON_STATE_TEXT;
 524                                         }
 525                                         break;
 526                                 case self::COLON_STATE_TAGSLASH:
 527                                         if ( $c === ">" ) {
 528                                                 # Yes, a self-closed tag <blah/>
 529                                                 $state = self::COLON_STATE_TEXT;
 530                                         } else {
 531                                                 # Probably we're jumping the gun, and this is an attribute
 532                                                 $state = self::COLON_STATE_TAG;
 533                                         }
 534                                         break;
 535                                 case self::COLON_STATE_COMMENT:
 536                                         if ( $c === "-" ) {
 537                                                 $state = self::COLON_STATE_COMMENTDASH;
 538                                         }
 539                                         break;
 540                                 case self::COLON_STATE_COMMENTDASH:
 541                                         if ( $c === "-" ) {
 542                                                 $state = self::COLON_STATE_COMMENTDASHDASH;
 543                                         } else {
 544                                                 $state = self::COLON_STATE_COMMENT;
 545                                         }
 546                                         break;
 547                                 case self::COLON_STATE_COMMENTDASHDASH:
 548                                         if ( $c === ">" ) {
 549                                                 $state = self::COLON_STATE_TEXT;
 550                                         } else {
 551                                                 $state = self::COLON_STATE_COMMENT;
 552                                         }
 553                                         break;
 554                                 default:
 555                                         throw new MWException( "State machine error in " . __METHOD__ );
 556                         }
 557                 }
 558                 if ( $ltLevel > 0 || $lcLevel > 0 ) {
 559                         wfDebug(
 560                                 __METHOD__ . ": Invalid input; not enough close tags " .
 561                                 "(level $ltLevel/$lcLevel, state $state)\n"
 562                         );
 563                         return false;
 564                 }
 565                 return false;
 566         }
 567 }