Wszetko\Sitemap\Drivers\XML\XMLWriter

Complexity

Total Complexity

39

Size/Duplication

Total Lines	244
Duplicated Lines	0 %

Test Coverage

Coverage

97.35%

Importance

Changes	1
Bugs	0	Features	0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	addElement()	0	11	4
A	getSitemapSize()	0	5	2
A	closeSitemapIndex()	0	8	1
A	openSitemap()	0	16	2
A	addElementArrayNonAssoc()	0	4	2
A	addSitemap()	0	11	2
A	closeSitemap()	0	8	1
B	addElementArrayAssoc()	0	31	10
A	addElementArray()	0	12	4
A	flushData()	0	3	1
A	__construct()	0	8	2
A	addUrl()	0	7	2
A	endFile()	0	20	5
A	openSitemapIndex()	0	11	1

<?php

declare(strict_types=1);

/**
 * This file is part of Wszetko Sitemap.
 *
 * (c) Paweł Kłopotek-Główczewski <pawelkg@pawelkg.com>
 *
 * This source file is subject to the MIT license that is bundled
 * with this source code in the file LICENSE.
 */

namespace Wszetko\Sitemap\Drivers\XML;

use Exception;
use Wszetko\Sitemap\Sitemap;

/**
 * Class XMLWriter.
 *
 * @package Wszetko\Sitemap\Drivers\XML
 */
class XMLWriter extends AbstractXML
{
    /**
     * XMLWriter constructor.
     *
     * @param array $config
     *
     * @throws \InvalidArgumentException
     */
    public function __construct(array $config)
    {
        if (!isset($config['domain'])) {
            throw new \InvalidArgumentException('Domain is not set.');
        }

        $this->XMLWriter = new \XMLWriter();
        $this->setDomain($config['domain']);
    }

    /**
     * @param string $sitemap
     * @param array  $extensions
     */
    public function openSitemap(string $sitemap, array $extensions = []): void
    {
        $this->setCurrentSitemap($sitemap);
        /* @var \XMLWriter $XMLWriter */
        $XMLWriter = $this->getXMLWriter();
        $XMLWriter->openMemory();
        $XMLWriter->startDocument('1.0', 'UTF-8');
        $XMLWriter->setIndent(true);
        $XMLWriter->startElement('urlset');
        $XMLWriter->writeAttribute('xmlns', Sitemap::SCHEMA);

        foreach ($extensions as $extension => $urlset) {
            $this->getXMLWriter()->writeAttribute('xmlns:' . $extension, $urlset);
        }

        $this->flushData();
    }

    /**
     * @throws \Exception
     */
    public function closeSitemap(): void
    {
        /* @var \XMLWriter $XMLWriter */
        $XMLWriter = $this->getXMLWriter();
        $XMLWriter->endElement();
        $XMLWriter->endDocument();
        $this->flushData();
        $this->endFile();
    }

    /**
     * @return int
     */
    public function getSitemapSize(): int
    {
        clearstatcache(true, $this->getSitemapFileFullPath());

        return file_exists($this->getSitemapFileFullPath()) ? (int) filesize($this->getSitemapFileFullPath()) : 0;
    }

    /**
     * @param array $element
     */
    public function addUrl(array $element): void
    {
        foreach ($element as $el => $val) {
            $this->addElement($el, $val);
        }

        $this->flushData();
    }

    /**
     * @param string $sitemap
     */
    public function openSitemapIndex(string $sitemap): void
    {
        $this->setCurrentSitemap($sitemap);
        /* @var \XMLWriter $XMLWriter */
        $XMLWriter = $this->getXMLWriter();
        $XMLWriter->openMemory();
        $XMLWriter->startDocument('1.0', 'UTF-8');
        $XMLWriter->setIndent(true);
        $XMLWriter->startElement('sitemapindex');
        $XMLWriter->writeAttribute('xmlns', Sitemap::SCHEMA);
        $this->flushData();
    }

    /**
     * @throws \Exception
     */
    public function closeSitemapIndex(): void
    {
        /* @var \XMLWriter $XMLWriter */
        $XMLWriter = $this->getXMLWriter();
        $XMLWriter->endElement();
        $XMLWriter->endDocument();
        $this->flushData();
        $this->endFile();
    }

    /**
     * @param string      $sitemap
     * @param null|string $lastmod
     */
    public function addSitemap(string $sitemap, string $lastmod = null): void
    {
        /* @var \XMLWriter $XMLWriter */
        $XMLWriter = $this->getXMLWriter();
        $XMLWriter->startElement('sitemap');
        $XMLWriter->writeElement('loc', $sitemap);
        if (isset($lastmod)) {
            $XMLWriter->writeElement('lastmod', $lastmod);
        }
        $XMLWriter->endElement();
        $this->flushData();
    }

    /**
     * Save from buffer to file.
     *
     * @return void
     */
    private function flushData(): void
    {
        file_put_contents($this->getSitemapFileFullPath(), $this->getXMLWriter()->flush(true), FILE_APPEND);
    }

    /**
     * Remove whitespace chars from end of file (Google don't like them).
     *
     * @throws Exception
     *
     * @return void
     */
    private function endFile(): void
    {
        if ($sitemapFile = fopen($this->getSitemapFileFullPath(), 'r+')) {
            fseek($sitemapFile, -1, SEEK_END);
            $truncate = 0;
            $length = $this->getSitemapSize();
            $end = false;

            do {
                $s = fread($sitemapFile, 1);
                if (is_string($s) && ctype_space($s)) {
                    ++$truncate;
                    fseek($sitemapFile, -2, SEEK_CUR);
                } else {
                    $end = true;
                }
            } while (!$end);

            ftruncate($sitemapFile, $length - $truncate);
            fclose($sitemapFile);
        }
    }

    /**
     * @param string      $element
     * @param mixed       $value
     * @param null|string $namespace
     */
    private function addElement(string $element, $value, ?string $namespace = null): void
    {
        if (!is_array($value)) {
            /* @var \XMLWriter $XMLWriter */
            $XMLWriter = $this->getXMLWriter();
            $XMLWriter->writeElement(($namespace ? $namespace . ':' : '') . $element, (string) $value);
        } else {
            if (isset($value['_namespace'])) {
                $this->addElement($value['_element'], $value[$value['_element']], $value['_namespace']);
            } else {
                $this->addElementArray($element, $value, $namespace);
            }
        }
    }

    /**
     * @param string      $element
     * @param mixed       $value
     * @param null|string $namespace
     */
    private function addElementArray(string $element, $value, ?string $namespace = null): void
    {
        if (!$this->isAssoc($value)) {
            if (!empty($value)) {
                $this->addElementArrayNonAssoc($element, $value, $namespace);
            } else {
                /* @var \XMLWriter $XMLWriter */
                $XMLWriter = $this->getXMLWriter();
                $XMLWriter->writeElement(($namespace ? $namespace . ':' : '') . $element);
            }
        } else {
            $this->addElementArrayAssoc($element, $value, $namespace);
        }
    }

    private function addElementArrayAssoc(string $element, $value, ?string $namespace = null): void
    {
        /* @var \XMLWriter $XMLWriter */
        $XMLWriter = $this->getXMLWriter();
        $XMLWriter->startElement(($namespace ? $namespace . ':' : '') . $element);

        if (isset($value['_attributes'])) {
            foreach ($value['_attributes'] as $attribute => $val) {
                $XMLWriter->writeAttribute($attribute, $val);
            }

            if (isset($value['_value'])) {
                if (is_array($value['_value'])) {
                    foreach ($value['_value'] as $el => $val) {
                        $this->addElement($el, $val);
                    }
                } else {
                    $XMLWriter->text((string) $value['_value']);
                }
            }
        } else {
            foreach ($value as $el => $val) {
                if (is_array($val)) {
                    $this->addElement($el, $val, $namespace);
                } else {
                    $XMLWriter->writeElement(($namespace ? $namespace . ':' : '') . $el, (string) $val);
                }
            }
        }

        $XMLWriter->endElement();
    }

    /**
     * @param string      $element
     * @param mixed       $value
     * @param null|string $namespace
     */
    private function addElementArrayNonAssoc(string $element, $value, ?string $namespace = null): void
    {
        foreach ($value as $val) {
            $this->addElement($element, $val, $namespace);
        }
    }
}