<?php
/**
 * PHPWord
 *
 * @link        https://github.com/PHPOffice/PHPWord
 * @copyright   2014 PHPWord
 * @license     http://www.gnu.org/licenses/old-licenses/lgpl-2.1.txt LGPL
 */

namespace PhpOffice\PhpWord\Container;

use PhpOffice\PhpWord\Exception\InvalidImageException;
use PhpOffice\PhpWord\Exception\InvalidObjectException;
use PhpOffice\PhpWord\Media;
use PhpOffice\PhpWord\Style;
use PhpOffice\PhpWord\TOC;
use PhpOffice\PhpWord\Footnote;
use PhpOffice\PhpWord\Shared\String;
use PhpOffice\PhpWord\Element\Text;
use PhpOffice\PhpWord\Element\TextRun;
use PhpOffice\PhpWord\Element\Link;
use PhpOffice\PhpWord\Element\Title;
use PhpOffice\PhpWord\Element\PreserveText;
use PhpOffice\PhpWord\Element\TextBreak;
use PhpOffice\PhpWord\Element\ListItem;
use PhpOffice\PhpWord\Element\Table;
use PhpOffice\PhpWord\Element\Image;
use PhpOffice\PhpWord\Element\Object;
use PhpOffice\PhpWord\Element\Footnote as FootnoteElement;
use PhpOffice\PhpWord\Element\CheckBox;

/**
 * Container abstract class
 *
 * @since 0.9.2
 */
abstract class Container
{
    /**
     * Container type section|header|footer
     *
     * @var string
     */
    protected $containerType;

    /**
     * Section Id
     *
     * @var int
     */
    protected $sectionId;

    /**
     * Footer Element Collection
     *
     * @var int
     */
    protected $elements = array();

    /**
     * Relation Id
     *
     * @var int
     */
    private $relationId;

    /**
     * Add text element
     *
     * @param string $text
     * @param mixed $styleFont
     * @param mixed $styleParagraph
     * @return Text
     */
    public function addText($text, $styleFont = null, $styleParagraph = null)
    {
        if (!String::isUTF8($text)) {
            $text = utf8_encode($text);
        }
        $text = new Text($text, $styleFont, $styleParagraph);
        $this->elements[] = $text;

        return $text;
    }

    /**
     * Add text break element
     *
     * @param int $count
     * @param mixed $fontStyle
     * @param mixed $paragraphStyle
     */
    public function addTextBreak($count = 1, $fontStyle = null, $paragraphStyle = null)
    {
        for ($i = 1; $i <= $count; $i++) {
            $this->elements[] = new TextBreak($fontStyle, $paragraphStyle);
        }
    }

    /**
     * Add textrun element
     *
     * @param mixed $styleParagraph
     * @return TextRun
     */
    public function addTextRun($styleParagraph = null)
    {
        $textRun = new TextRun($styleParagraph);
        $this->elements[] = $textRun;

        return $textRun;
    }

    /**
     * Add link element
     *
     * @param string $linkSrc
     * @param string $linkName
     * @param mixed $styleFont
     * @param mixed $styleParagraph
     * @return Link
     * @todo Enable link element in header and footer
     */
    public function addLink($linkSrc, $linkName = null, $styleFont = null, $styleParagraph = null)
    {
        if ($this->containerType != 'section') {
            throw new \BadMethodCallException();
        }

        if (!String::isUTF8($linkSrc)) {
            $linkSrc = utf8_encode($linkSrc);
        }
        if (!is_null($linkName)) {
            if (!String::isUTF8($linkName)) {
                $linkName = utf8_encode($linkName);
            }
        }
        $link = new Link($linkSrc, $linkName, $styleFont, $styleParagraph);
        $rID = Media::addSectionLinkElement($linkSrc);
        $link->setRelationId($rID);
        $this->elements[] = $link;
        return $link;
    }

    /**
     * Add a Title Element
     *
     * @param string $text
     * @param int $depth
     * @return Title
     * @todo Enable title element in header and footer
     */
    public function addTitle($text, $depth = 1)
    {
        if ($this->containerType != 'section') {
            throw new \BadMethodCallException();
        }

        if (!String::isUTF8($text)) {
            $text = utf8_encode($text);
        }
        $styles = Style::getStyles();
        if (array_key_exists('Heading_' . $depth, $styles)) {
            $style = 'Heading' . $depth;
        } else {
            $style = null;
        }
        $title = new Title($text, $depth, $style);
        $data = TOC::addTitle($text, $depth);
        $anchor = $data[0];
        $bookmarkId = $data[1];
        $title->setAnchor($anchor);
        $title->setBookmarkId($bookmarkId);
        $this->elements[] = $title;
        return $title;
    }

    /**
     * Add preserve text element
     *
     * @param string $text
     * @param mixed $styleFont
     * @param mixed $styleParagraph
     * @return PreserveText
     */
    public function addPreserveText($text, $styleFont = null, $styleParagraph = null)
    {
        if ($this->containerType == 'section') {
            throw new \BadMethodCallException();
        }

        if (!String::isUTF8($text)) {
            $text = utf8_encode($text);
        }
        $ptext = new PreserveText($text, $styleFont, $styleParagraph);
        $this->elements[] = $ptext;

        return $ptext;
    }

    /**
     * Add listitem element
     *
     * @param string $text
     * @param int $depth
     * @param mixed $styleFont
     * @param mixed $styleList
     * @param mixed $styleParagraph
     * @return ListItem
     * @todo Enable list item element in header and footer
     */
    public function addListItem($text, $depth = 0, $styleFont = null, $styleList = null, $styleParagraph = null)
    {
        if ($this->containerType != 'section') {
            throw new \BadMethodCallException();
        }

        if (!String::isUTF8($text)) {
            $text = utf8_encode($text);
        }
        $listItem = new ListItem($text, $depth, $styleFont, $styleList, $styleParagraph);
        $this->elements[] = $listItem;
        return $listItem;
    }

    /**
     * Add table element
     *
     * @param mixed $style
     * @return Table
     */
    public function addTable($style = null)
    {
        $table = new Table($this->containerType, $this->sectionId, $style);
        $this->elements[] = $table;

        return $table;
    }

    /**
     * Add image element
     *
     * @param string $src
     * @param mixed $style Image style
     * @param boolean $isWatermark
     * @return Image
     */
    public function addImage($src, $style = null, $isWatermark = false)
    {
        $image = new Image($src, $style, $isWatermark);
        if (!is_null($image->getSource())) {
            switch ($this->containerType) {
                case 'section':
                    $rID = Media::addSectionMediaElement($src, 'image', $image);
                    break;
                case 'header':
                    $rID = Media::addHeaderMediaElement($this->sectionId, $src, $image);
                    break;
                case 'footer':
                    $rID = Media::addFooterMediaElement($this->sectionId, $src, $image);
                    break;
            }
            $image->setRelationId($rID);
            $this->elements[] = $image;
            return $image;
        } else {
            throw new InvalidImageException;
        }
    }

    /**
     * Add OLE-object element
     *
     * All exceptions should be handled by PhpOffice\PhpWord\Element\Object
     *
     * @param string $src
     * @param mixed $style
     * @return Object
     * @todo Enable OLE object element in header and footer
     */
    public function addObject($src, $style = null)
    {
        if ($this->containerType != 'section') {
            throw new \BadMethodCallException();
        }

        $object = new Object($src, $style);
        if (!is_null($object->getSource())) {
            $inf = pathinfo($src);
            $ext = $inf['extension'];
            if (strlen($ext) == 4 && strtolower(substr($ext, -1)) == 'x') {
                $ext = substr($ext, 0, -1);
            }
            $icon = __DIR__ . "/../_staticDocParts/_{$ext}.png";
            $rIDimg = Media::addSectionMediaElement($icon, 'image', new Image($icon));
            $data = Media::addSectionMediaElement($src, 'oleObject');
            $rID = $data[0];
            $objectId = $data[1];
            $object->setRelationId($rID);
            $object->setObjectId($objectId);
            $object->setImageRelationId($rIDimg);
            $this->elements[] = $object;
            return $object;
        } else {
            throw new InvalidObjectException();
        }
    }

    /**
     * Add footnote element
     *
     * @param mixed $styleParagraph
     * @return FootnoteElement
     * @todo Enable footnote element in header and footer
     */
    public function addFootnote($styleParagraph = null)
    {
        if ($this->containerType != 'section') {
            throw new \BadMethodCallException();
        }

        $footnote = new FootnoteElement($styleParagraph);
        $refID = Footnote::addFootnoteElement($footnote);
        $footnote->setReferenceId($refID);
        $this->elements[] = $footnote;
        return $footnote;
    }

    /**
     * Add a CheckBox Element
     *
     * @param string $name
     * @param string $text
     * @param mixed $styleFont
     * @param mixed $styleParagraph
     * @return CheckBox
     * @todo Enable checkbox element in header and footer
     */
    public function addCheckBox($name, $text, $styleFont = null, $styleParagraph = null)
    {
        if ($this->containerType != 'section') {
            throw new \BadMethodCallException();
        }

        if (!String::isUTF8($name)) {
            $name = utf8_encode($name);
        }
        if (!String::isUTF8($text)) {
            $text = utf8_encode($text);
        }
        $element = new CheckBox($name, $text, $styleFont, $styleParagraph);
        $this->elements[] = $element;

        return $element;
    }

    /**
     * Get section number
     * getFooterCount
     */
    public function getSectionId()
    {
        return $this->sectionId;
    }

    /**
     * Get all elements
     *
     * @return array
     */
    public function getElements()
    {
        return $this->elements;
    }

    /**
     * Get relation Id
     *
     * @return int
     */
    public function getRelationId()
    {
        if ($this->containerType == 'section') {
            throw new \BadMethodCallException();
        }

        return $this->relationId;
    }

    /**
     * Set relation Id
     *
     * @param int $rId
     */
    public function setRelationId($rId)
    {
        if ($this->containerType == 'section') {
            throw new \BadMethodCallException();
        }

        $this->relationId = $rId;
    }

    /**
     * Add memory image element
     *
     * @param string $src
     * @param mixed $style
     * @deprecated 0.9.0
     */
    public function addMemoryImage($src, $style = null)
    {
        return $this->addImage($src, $style);
    }

    /**
     * Create textrun element
     *
     * @param mixed $styleParagraph
     * @deprecated 0.9.2
     */
    public function createTextRun($styleParagraph = null)
    {
        return $this->addTextRun($styleParagraph);
    }

    /**
     * Create footnote element
     *
     * @param mixed $styleParagraph
     * @deprecated 0.9.2
     */
    public function createFootnote($styleParagraph = null)
    {
        return $this->addFootnote($styleParagraph);
    }
}