--- /dev/null
+<?php
+
+namespace RemexHtml\TreeBuilder;
+use RemexHtml\Tokenizer\Attributes;
+use RemexHtml\Tokenizer\PlainAttributes;
+
+/**
+ * The "after head" insertion mode
+ */
+class AfterHead extends InsertionMode {
+ public function characters( $text, $start, $length, $sourceStart, $sourceLength ) {
+ $builder = $this->builder;
+ $dispatcher = $this->dispatcher;
+
+ // Insert whitespace
+ list( $part1, $part2 ) = $this->splitInitialMatch(
+ true, "\t\n\f\r ", $text, $start, $length, $sourceStart, $sourceLength );
+ list( $start, $length, $sourceStart, $sourceLength ) = $part1;
+ if ( $length ) {
+ $builder->insertCharacters( $text, $start, $length,
+ $sourceStart, $sourceLength );
+ }
+
+ // Switch mode on non-whitespace
+ list( $start, $length, $sourceStart, $sourceLength ) = $part2;
+ if ( $length ) {
+ $builder->insertElement( 'body', new PlainAttributes, false, $sourceStart, 0 );
+ $dispatcher->switchMode( Dispatcher::IN_BODY )
+ ->characters( $text, $start, $length, $sourceStart, $sourceLength );
+ }
+ }
+
+ public function startTag( $name, Attributes $attrs, $selfClose, $sourceStart, $sourceLength ) {
+ $builder = $this->builder;
+ $stack = $builder->stack;
+ $dispatcher = $this->dispatcher;
+
+ switch ( $name ) {
+ case 'html':
+ $dispatcher->inBody->startTag( $name, $attrs, $selfClose,
+ $sourceStart, $sourceLength );
+ break;
+
+ case 'frameset':
+ $builder->insertElement( $name, $attrs, false, $sourceStart, $sourceLength );
+ $dispatcher->switchMode( Dispatcher::IN_FRAMESET );
+ break;
+
+ case 'base':
+ case 'basefont':
+ case 'bgsound':
+ case 'link':
+ case 'meta':
+ case 'noframes':
+ case 'script':
+ case 'style':
+ case 'template':
+ case 'title':
+ $builder->error( "unexpected <$name> after </head>, accepting", $sourceStart );
+ $stack->push( $builder->headElement );
+ $dispatcher->inHead->startTag(
+ $name, $attrs, $selfClose, $sourceStart, $sourceLength );
+ $stack->remove( $builder->headElement );
+ break;
+
+ case 'head':
+ $builder->error( "unexpected <head> after </head>, ignoring", $sourceStart );
+ return;
+
+ default:
+ $builder->insertElement( 'body', new PlainAttributes, false, $sourceStart, 0 );
+ $dispatcher->switchMode( Dispatcher::IN_BODY )
+ ->startTag( $name, $attrs, $selfClose, $sourceStart, $sourceLength );
+ }
+ }
+
+ public function endTag( $name, $sourceStart, $sourceLength ) {
+ $builder = $this->builder;
+ $dispatcher = $this->dispatcher;
+
+ switch ( $name ) {
+ case 'template':
+ $dispatcher->inHead->endTag( $name, $sourceStart, $sourceLength );
+ break;
+
+ case 'body':
+ case 'html':
+ case 'br':
+ $builder->insertElement( 'body', new PlainAttributes, false, $sourceStart, 0 );
+ $dispatcher->switchMode( Dispatcher::IN_BODY )
+ ->endTag( $name, $sourceStart, $sourceLength );
+ break;
+
+ default:
+ $builder->error( "unexpected </$name> after head, ignoring", $sourceStart );
+ return;
+ }
+ }
+
+ public function endDocument( $pos ) {
+ $this->builder->insertElement( 'body', new PlainAttributes, false, $pos, 0 );
+ $this->dispatcher->switchMode( Dispatcher::IN_BODY )
+ ->endDocument( $pos );
+ }
+}