      1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 171: 172: 173: 174: 175: 176: 177: 178: 179: 180: 181: 182: 183: 184: 185: 186: 187: 188: 189: 190: 191: 192: 193: 194: 195: 196: 197: 198: 199: 200: 201: 202: 203: 204: 205: 206: 207: 208: 209: 210: 211: 212: 213: 214: 215: 216: 217: 218: 219: 220: 221: 222: 223: 224: 225: 226: 227: 228: 229: 230: 231: 232: 233: 234: 235: 236: 237: 238: 239: 240: 241: 242: 243: 244: 245: 246: 247: 248: 249: 250: 251: 252:

    <?php

    namespace WPGMZA;

    require_once(plugin_dir_path(__FILE__) . 'class.dom-element.php');

    class DOMDocument extends \DOMDocument
    {
        private $src_file;

        /**
         * Constructor
         * @see http://php.net/manual/en/class.domdocument.php
         */
        public function __construct($version='1.0', $encoding='UTF-8')
        {
            \DOMDocument::__construct($version, $encoding);

            $this->registerNodeClass('DOMElement', 'WPGMZA\DOMElement');
            $this->onReady();
        }

        public static function convertUTF8ToHTMLEntities($html)
        {
            if(function_exists('mb_convert_encoding'))
                return mb_convert_encoding($html, 'HTML-ENTITIES', 'UTF-8');
            else
            {
                trigger_error('Using fallback UTF to HTML entity conversion', E_USER_NOTICE);
                return htmlspecialchars_decode(utf8_decode(htmlentities($html, ENT_COMPAT, 'utf-8', false)));
            }
        }

        /**
         * Fired after construction when the Document is initialized
         * @return void
         */
        protected function onReady()
        {

        }

        /**
         * Fired after content has been loaded
         * @return void
         */
        protected function onLoaded()
        {

        }

        /**
         * Loads the specified file and evaulates nodes
         * @see http://php.net/manual/en/domdocument.load.php
         * @param string $src The file you want to load
         * @param int $options See http://php.net/manual/en/domdocument.load.php
         * @return boolean TRUE on success, FALSE otherwise
         */
        public function load($src, $options=null)
        {
            if(!is_string($src))
                throw new \Exception('Input must be a string');

            $result = \DOMDocument::load($src, $options);
            $this->src_file = $src;

            $this->onEvaluateNodes();
            $this->onLoaded();

            return $result;
        }

        /**
         * Loads the specified file and parses any PHP
         * @param string $src The file you want to load
         * @param int $options See http://php.net/manual/en/domdocument.load.php
         * @return boolean TRUE on success, FALSE otherwise
         */
        public function loadPHPFile($src, $options=0)
        {
            if(!file_exists($src))
                throw new \Exception("File does not exist: $src");

            ob_start();
            include $src;
            $html = ob_get_clean();

            if(empty($html))
                throw new \Exception("$src is empty");

            $html = DOMDocument::convertUTF8ToHTMLEntities($html);
            $suppress_warnings = !(defined('WP_DEBUG') && WP_DEBUG);

            // From PHP 5.4.0 onwards, loadHTML takes 2 arguments
            if(version_compare(PHP_VERSION, '5.4.0', '>='))
            {
                if($suppress_warnings)
                    $result = @$this->loadHTML($html, $options);
                else
                    $result = $this->loadHTML($html, $options);
            }
            else
            {
                if($suppress_warnings)
                    $result = @$this->loadHTML($html);
                else
                    $result = $this->loadHTML($html);
            }

            $this->src_file = $src;

            $this->onLoaded();

            return $result;
        }

        /**
         * @internal Handles imports based on the content
         */
        protected function handleImport($subject, $node, $forcePHP=false)
        {
            if(preg_match('/\.html(\.php)?$/i', $subject))
            {
                if(!file_exists($subject))
                    throw new \Exception("File '$subject' not found in {$this->src_file} line " . $node->getLineNo());

                $node->import($subject, $forcePHP);

                return;
            }

            if(preg_match('/\.php$/i', $subject))
            {
                if(!file_exists($subject))
                    throw new \Exception("File '$subject' not found");

                $before = get_declared_classes();
                require_once($subject);
                $after = get_declared_classes();
                $diff = array_diff($after, $before);

                $classes_in_file = array();

                foreach($diff as $class) {
                    $reflection = new \ReflectionClass($class);
                    $filename = $reflection->getFileName();

                    if(realpath($filename) == realpath($subject))
                        array_push($classes_in_file, $class);
                }

                if(empty($classes_in_file))
                    throw new \Exception('No classes found in file');

                $class_name = $classes_in_file[ count($classes_in_file) - 1 ];
                $instance = new $class_name();

                if(!($instance instanceof DOMDocument))
                    throw new \Exception('Class must be an instance of WPGMZA\\DOMDocument');

                $node->import($instance);

                return;
            }

            throw new \Exception("Failed to import page \"$subject\" in {$this->src_file} on line " . $node->getLineNo());
        }

        /**
         * Runs a CSS selector on the element. This is equivilant to Javascripts querySelector
         * @param string $query a CSS selector
         * @return mixed The first descendant \Smart\Element that matches the selector, or NULL if there is no match
         */
        public function querySelector($query)
        {
            if(!$this->documentElement)
                throw new \Exception('Document is empty');
            return $this->documentElement->querySelector($query);
        }

        /**
         * Runs a CSS selector on the element. This is equivilant to Javascripts querySelectorAll
         * @return mixed Any descendant \Smart\Element's that match the selector, or NULL if there is no match
         */
        public function querySelectorAll($query)
        {
            if(!$this->documentElement)
                throw new \Exception('Document is empty');
            return $this->documentElement->querySelectorAll($query);
        }

        /**
         * Takes an associative array or object, and populates this element with it
         * @param mixed $src Object or associative array
         * @return void
         */
        public function populate($src, $formatters=null)
        {
            if(!$this->documentElement)
                throw new \Exception('Document is empty');
            return $this->documentElement->populate($src, $formatters);
        }

        public function serializeFormData()
        {
            if(!$this->documentElement)
                throw new \Exception('Document is empty');
            return $this->documentElement->serializeFormData();
        }

        /**
         * Utility function to create an error element
         * @param string $message The error message
         * @return \Smart\Element
         */
        public function createErrorElement($message)
        {
            $span = $this->createElement('span');
            $span->appendText($message);
            $span->addClass('notice notice-error');
            return $span;
        }

        /**
         * This function saves only the inside of the <body> element of this document. This is useful when you want to import a HTML document into another, but you don't want to end up with nested <html> elements
         * @return string The HTML string
         */
        public function saveInnerBody()
        {
            $result = '';

            $body = $this->querySelector('body');

            if(!$body)
                return null;

            for($node = $body->firstChild; $node != null; $node = $node->nextSibling)
                $result .= $this->saveHTML($node);

            return $result;
        }

        public function __get($name)
        {
            if($name == 'html')
                return $this->saveInnerBody();

            return null;
        }
    }