juniorb2ss/zServices

<?php namespace zServices\ReceitaFederal\Services\Portais\AN;

use zServices\Miscellany\Interfaces\CrawlerInterface;
use Symfony\Component\DomCrawler\Crawler as BaseCrawler;
use zServices\Miscellany\Exceptions\NoSelectorsConfigured;
use zServices\Miscellany\Exceptions\InvalidCaptcha;
use zServices\Miscellany\Exceptions\ErrorFoundData;

/**
* 
*/
class Crawler extends BaseCrawler implements CrawlerInterface
{

    /**
     * Selectors to crawler
     * @var array
     */
    private $selectors = [];

    /**
     * Get selectors and initialize crawler in HTML
     * @param string $html      
     * @param array  $selectors
     * @return void
     */
    public function __construct($html, $selectors)
    {
        $this->selectors = $selectors;

        parent::__construct($html);
    }

    /**
     * Verifica antes de fazer o crawler se possui erros
     * na requisição
     * @return boolean|null 
     */
    public function hasError()
    {
        if (count($this->selectors) == 0) {
            throw new NoSelectorsConfigured("NoSelectorsConfigured", 1);
        }

        // verifica se a página seguida na requisição 
        // é página de erro da receita federal
        $node = $this->filter($this->selectors['error']);

        if ($node->count()) {
             throw new ErrorFoundData($this->clearString($node->text()), 1);
        }
        
        // CNPJ informado é válido?
        if ($this->filter('#imgCaptcha')->count()) {
            throw new InvalidCaptcha('Captcha inválido', 99);
        }
    }

    /**
     * Extrai informações do HTML através do DOM
     *
     * @return array
     */
    public function scraping()
    {
        $scrapped = [];

        $this->hasError();

        foreach ($this->selectors as $name => $selector) {
            if (is_string($selector)) {
                $node = $this->scrap($selector);

                if ($node->count()) {
                    $scrapped[$name] = $this->clearString($node->text());
                }
            } elseif (is_array($selector)) {
                foreach ($selector as $selector => $repeat) {
                    $node = $this->scrap($selector);
                    if ($node->count()) {
                        foreach ($node->filter($repeat) as $loop)
                        {
                            $scrapped[$name][] = $this->clearString($loop->nodeValue);
                        }
                    }
                }
            }
        }

        return $scrapped;
    }

    /**
     * Limpa o valor repassado
     * @param  string $string
     * @return string
     */
    public function clearString($string)
    {
        return trim(preg_replace(['/[\s]+/mu'], ' ', $string));
    }

    /**
     * Filtra selector no crawler
     */
    public function scrap($selector)
    {
        $node = $this->filter($selector);
        return $node;
    }
}