Differential D20568 Diff 49067 src/parser/html/PhutilDOMNode.php

Changeset View

Standalone View

src/parser/html/PhutilDOMNode.php

This file was added.

				<?php

				final class PhutilDOMNode extends Phobject {

				private $content;
				private $tagName;
				private $children = array();
				private $attributes = array();
				private $parentNode;
				private $rawString;

				public function setContent($content) {
				$this->content = $content;
				return $this;
				}

				public function getContent() {
				return $this->content;
				}

				public function isContentNode() {
				return ($this->content !== null);
				}

				public function setTagName($tag_name) {
				$this->tagName = $tag_name;
				return $this;
				}

				public function getTagName() {
				return $this->tagName;
				}

				public function appendChild(PhutilDOMNode $node) {
				$node->parentNode = $this;
				$this->children[] = $node;
				return $this;
				}

				public function getChildren() {
				return $this->children;
				}

				public function getParentNode() {
				return $this->parentNode;
				}

				public function setAttributes(array $attributes) {
				$this->attributes = $attributes;
				return $this;
				}

				public function getAttributes() {
				return $this->attributes;
				}

				public function setRawString($raw_string) {
				$this->rawString = $raw_string;
				return $this;
				}

				public function getRawString() {
				return $this->rawString;
				}

				public function toDictionary() {
				if ($this->isContentNode()) {
				return array(
				'content' => $this->content,
				);
				} else {
				$children = array();

				foreach ($this->getChildren() as $child) {
				$children[] = $child->toDictionary();
				}

				return array(
				'tag' => $this->getTagName(),
				'attributes' => $this->getAttributes(),
				'children' => $children,
				);
				}
				}

				/**
				* Get a list of the children of a given DOM node, treating unexpected
				* tags as if they were raw content.
				*/
				public function selectChildrenWithTags(array $tag_list) {
				$tag_map = array_fuse($tag_list);

				$nodes = array();
				foreach ($this->getChildren() as $child) {
				// If this is already a content node, just keep it as-is.
				if ($child->isContentNode()) {
				$nodes[] = $child;
				continue;
				}

				$tag_name = $child->getTagName();

				// If this is a tag that we're allowing, keep it as-is.
				if (isset($tag_map[$tag_name])) {
				$nodes[] = $child;
				continue;
				}

				// Otherwise, this is some other tag. Convert it into a content
				// node.

				$raw_content = $child->getRawString();

				$nodes[] = id(new self())
				->setContent($raw_content)
				->setRawContent($raw_content);
				}

				return $this->mergeContentNodes($nodes);
				}

				public function getRawContentString() {
				$content_node = $this->selectChildrenWithTags(array());

				if (!$content_node) {
				return '';
				}

				return head($content_node)->getRawString();
				}

				public function mergeContent() {
				$this->children = $this->mergeContentNodes($this->children);

				foreach ($this->getChildren() as $child) {
				$child->parentNode = $this;
				$child->mergeContent();
				}

				return $this;
				}

				/**
				* Given a list of nodes, combine sequences of multiple adjacent content
				* nodes into single nodes.
				*/
				private function mergeContentNodes(array $nodes) {
				$list = array();
				$content_block = array();
				foreach ($this->getChildren() as $child) {
				if ($child->isContentNode()) {
				$content_block[] = $child;
				continue;
				}

				$list[] = $content_block;
				$content_block = array();

				$list[] = $child;
				}

				$list[] = $content_block;

				$results = array();
				foreach ($list as $item) {
				if (!is_array($item)) {
				$results[] = $item;
				continue;
				}

				if (!$item) {
				continue;
				}

				$parts = array();
				foreach ($item as $content_node) {
				$parts[] = $content_node->getRawString();
				}
				$parts = implode('', $parts);

				if (!strlen($parts)) {
				continue;
				}

				$results[] = id(new self())
				->setContent($parts)
				->setRawString($parts);
				}

				return $results;
				}

				}