Spatie\Sitemap\SitemapGenerator

Complexity

Total Complexity

9

Size/Duplication

Total Lines	94
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes	1
Bugs	0	Features	0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	create()	0	4	1
A	__construct()	0	14	1
A	setUrl()	0	6	1
A	hasCrawled()	0	6	1
A	getSitemap()	0	9	1
A	writeToFile()	0	6	1
A	getObserver()	0	12	2
A	getProfile()	0	4	1

<?php

namespace Spatie\Sitemap;

use Spatie\Crawler\Crawler;
use Spatie\Crawler\Url as CrawlerUrl;
use Spatie\Sitemap\Crawler\Observer;
use Spatie\Sitemap\Crawler\Profile;
use Spatie\Sitemap\Tags\Url;

class SitemapGenerator
{
    /** @var string */
    protected $url = '';

    /** @var \Spatie\Crawler\Crawler */
    protected $crawler;

    /** @var callable */
    protected $hasCrawled;

    /** @var callable */
    protected $crawlProfile;

    /** @var \Spatie\Sitemap\Sitemap */
    protected $sitemap;

    /**
     * @param string $url
     *
     * @return static
     */
    public static function create(string $url)
    {
        return app(self::class)->setUrl($url);
    }

    public function __construct(Crawler $crawler)
    {
        $this->crawler = $crawler;

        $this->sitemap = new Sitemap();

        $this->hasCrawled = function (Url $url) {
            return $url;
        };

        $this->crawlProfile = function (CrawlerUrl $url) {
            return $url->host == CrawlerUrl::create($this->url)->host;
        };
    }

    public function setUrl(string $url)
    {
        $this->url = $url;

        return $this;
    }

    public function hasCrawled(callable $hasCrawled)
    {
        $this->hasCrawled = $hasCrawled;

        return $this;
    }

    /**
     * @return \Spatie\Sitemap\Sitemap
     */
    public function getSitemap()
    {
        $this->crawler
            ->setCrawlProfile($this->getProfile())
            ->setCrawlObserver($this->getObserver())
            ->startCrawling($this->url);

        return $this->sitemap;
    }

    public function writeToFile($path)
    {
        $this->getSitemap()->writeToFile($path);

        return $this;
    }

    protected function getObserver(): Observer
    {
        $performAfterUrlHasBeenCrawled = function (CrawlerUrl $crawlerUrl) {
            $sitemapUrl = ($this->hasCrawled)(Url::create((string) $crawlerUrl));

            if ($sitemapUrl) {
                $this->sitemap->add($sitemapUrl);
            }
        };

        return new Observer($performAfterUrlHasBeenCrawled);
    }

    protected function getProfile(): Profile
    {
        return new Profile($this->crawlProfile);
    }
}