PHPWord/src/PhpWord/Writer/RTF/Element/Text.php

<?php
/**
 * This file is part of PHPWord - A pure PHP library for reading and writing
 * word processing documents.
 *
 * PHPWord is free software distributed under the terms of the GNU Lesser
 * General Public License version 3 as published by the Free Software Foundation.
 *
 * For the full copyright and license information, please read the LICENSE
 * file that was distributed with this source code. For the full list of
 * contributors, visit https://github.com/PHPOffice/PHPWord/contributors.
 *
 * @link        https://github.com/PHPOffice/PHPWord
 * @copyright   2010-2014 PHPWord contributors
 * @license     http://www.gnu.org/licenses/lgpl.txt LGPL version 3
 */

namespace PhpOffice\PhpWord\Writer\RTF\Element;

use PhpOffice\PhpWord\PhpWord;
use PhpOffice\PhpWord\Style;
use PhpOffice\PhpWord\Style\Font as FontStyle;
use PhpOffice\PhpWord\Style\Paragraph as ParagraphStyle;

/**
 * Text element RTF writer
 *
 * @since 0.10.0
 */
class Text extends Element
{
    /**
     * Write element
     */
    public function write()
    {
        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Text) {
            return;
        }

        $rtfText = '';

        $fontStyle = $this->element->getFontStyle();
        if (is_string($fontStyle)) {
            $fontStyle = Style::getStyle($fontStyle);
        }

        $paragraphStyle = $this->element->getParagraphStyle();
        if (is_string($paragraphStyle)) {
            $paragraphStyle = Style::getStyle($paragraphStyle);
        }

        if ($paragraphStyle && !$this->withoutP) {
            if ($this->parentWriter->getLastParagraphStyle() != $this->element->getParagraphStyle()) {
                $rtfText .= $this->writeParagraphStyle($paragraphStyle);
                $this->parentWriter->setLastParagraphStyle($this->element->getParagraphStyle());
            } else {
                $this->parentWriter->setLastParagraphStyle();
            }
        } else {
            $this->parentWriter->setLastParagraphStyle();
        }

        if ($fontStyle instanceof FontStyle) {
            $rtfText .= $this->writeFontStyleBegin($fontStyle);
        }
        if ($this->parentWriter->getLastParagraphStyle() != '' || $fontStyle) {
            $rtfText .= ' ';
        }
        $rtfText .= $this->element->getText();
        if ($fontStyle instanceof FontStyle) {
            $rtfText .= $this->writeFontStyleEnd($fontStyle);
        }
        if (!$this->withoutP) {
            $rtfText .= '\par' . PHP_EOL;
        }

        return $rtfText;
    }

    /**
     * Write paragraph style
     *
     * @return string
     */
    private function writeParagraphStyle(ParagraphStyle $paragraphStyle)
    {
        $rtfText = '\pard\nowidctlpar';
        if ($paragraphStyle->getSpaceAfter() != null) {
            $rtfText .= '\sa' . $paragraphStyle->getSpaceAfter();
        }
        if ($paragraphStyle->getAlign() != null) {
            if ($paragraphStyle->getAlign() == 'center') {
                $rtfText .= '\qc';
            }
        }

        return $rtfText;
    }

    /**
     * Write font style beginning
     *
     * @return string
     */
    private function writeFontStyleBegin(FontStyle $style)
    {
        $rtfText = '';
        if ($style->getColor() != null) {
            $idxColor = array_search($style->getColor(), $this->parentWriter->getColorTable());
            if ($idxColor !== false) {
                $rtfText .= '\cf' . ($idxColor + 1);
            }
        } else {
            $rtfText .= '\cf0';
        }
        if ($style->getName() != null) {
            $idxFont = array_search($style->getName(), $this->parentWriter->getFontTable());
            if ($idxFont !== false) {
                $rtfText .= '\f' . $idxFont;
            }
        } else {
            $rtfText .= '\f0';
        }
        if ($style->isBold()) {
            $rtfText .= '\b';
        }
        if ($style->isItalic()) {
            $rtfText .= '\i';
        }
        if ($style->getSize()) {
            $rtfText .= '\fs' . ($style->getSize() * 2);
        }

        return $rtfText;
    }

    /**
     * Write font style ending
     *
     * @return string
     */
    private function writeFontStyleEnd(FontStyle $style)
    {
        $rtfText = '';
        $rtfText .= '\cf0';
        $rtfText .= '\f0';

        if ($style->isBold()) {
            $rtfText .= '\b0';
        }
        if ($style->isItalic()) {
            $rtfText .= '\i0';
        }
        if ($style->getSize()) {
            $rtfText .= '\fs' . (PhpWord::DEFAULT_FONT_SIZE * 2);
        }

        return $rtfText;
    }
}