gjerokrsteski/pimf-framework

<?php
/**
 * Util
 *
 * @copyright Copyright (c)  Gjero Krsteski (http://krsteski.de)
 * @license   http://opensource.org/licenses/MIT MIT License
 */

namespace Pimf\Util;

/**
 * A powerful tool for HTML or XML document manipulation and extraction of data.
 *
 * @package Util
 * @author  Gjero Krsteski <gjero@krsteski.de>
 */
class Dom extends \DOMDocument
{
    /**
     * Retries a list of attributes-value which can live inside a HTML/XML-Element
     *
     * @param string $tag       A HTML/XML tag-name representation for the HTML/XML-Element
     * @param string $attribute A attribute inside of the HTML/XML-Element
     *
     * @return array
     */
    public function fetchValues($tag, $attribute)
    {
        $values = array();

        // loop through each tag in the dom and add it to the array
        foreach ($this->getElementsByTagName($tag) as $element) {
            /* @var $element \DOMElement */
            $values[] = $element->getAttribute($attribute);
        }

        return $values;
    }

    /**
     * Grab all links in a page
     *
     * @return array
     */
    public function getURLs()
    {
        return $this->fetchValues('a', 'href');
    }

    /**
     * Grab all URLs of an image
     *
     * @return array
     */
    public function getImageURLs()
    {
        return $this->fetchValues('img', 'src');
    }

    /**
     * Grab all URLs of an external script file like JS
     */
    public function getScriptURLs()
    {
        return $this->fetchValues('script', 'src');
    }

    /**
     * Grab all URLs location of the linked document like CSS
     */
    public function getCssURLs()
    {
        return $this->fetchValues('link', 'href');
    }
}