HtmlElement.php

<?php
namespace Omeka\View\Helper;

use Laminas\View\Helper\AbstractHtmlElement;

/**
 * View helper for rendering a HTML element.
 */
class HtmlElement extends AbstractHtmlElement
{
    /**
     * @var array Cached elements and their attributes
     */
    protected $elements = [];

    /**
     * @var string The current element
     */
    protected $element;

    /**
     * HTML element helper
     *
     * @param string $element The current element
     * @return self
     */
    public function __invoke($element)
    {
        $this->element = $element;
        if (!isset($this->elements[$element])) {
            $this->elements[$element] = [];
        }
        return $this;
    }

    /**
     * Render the element and its attributes
     *
     * @return string
     */
    public function __toString()
    {
        $attributes = $this->elements[$this->element];
        return '<' . $this->element . $this->htmlAttribs($attributes) . '>';
    }

    /**
     * Set an attribute to the current element
     *
     * @param string $key
     * @param string $value
     * @return self
     */
    public function setAttribute($key, $value)
    {
        $this->elements[$this->element][$key] = $value;
        return $this;
    }

    /**
     * Append an attribute to an element
     *
     * @param string $key
     * @param string $value
     * @return self
     */
    public function appendAttribute($key, $value)
    {
        if (isset($this->elements[$this->element][$key])) {
            $current = $this->elements[$this->element][$key];
            if (is_array($current)) {
                $this->elements[$this->element][$key][] = $value;
            } else {
                $this->elements[$this->element][$key] = [$current, $value];
            }
        } else {
            $this->elements[$this->element][$key] = $value;
        }
        return $this;
    }

    /**
     * Set attributes to the current element
     *
     * @param array $attributes
     * @return self
     */
    public function setAttributes(array $attributes)
    {
        foreach ($attributes as $key => $value) {
            $this->setAttribute($key, $value);
        }
        return $this;
    }

    /**
     * Remove an attribute from the current element
     *
     * @param string $key
     * @return self
     */
    public function removeAttribute($key)
    {
        unset($this->elements[$this->element][$key]);
        return $this;
    }

    /**
     * Remove all attributes from the current element
     *
     * @return self
     */
    public function removeAttributes()
    {
        if (isset($this->elements[$this->element])) {
            $this->elements[$this->element] = [];
        }
        return $this;
    }
}