Encoder.php

<?php
/*
 * This file is part of XRL, a simple XML-RPC Library for PHP.
 *
 * Copyright (c) 2012, XRL Team. All rights reserved.
 * XRL is licensed under the 3-clause BSD License.
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace fpoirotte\XRL;

class Encoder implements \fpoirotte\XRL\EncoderInterface
{
    protected $indent;

    protected $stringTag;

    protected $timezone;

    public function __construct(
        \DateTimeZone $timezone = null,
        $indent = false,
        $stringTag = false
    ) {
        if (!is_bool($indent)) {
            throw new \InvalidArgumentException('$indent must be a boolean');
        }
        if (!is_bool($stringTag)) {
            throw new \InvalidArgumentException('$stringTag must be a boolean');
        }

        if ($timezone === null) {
            $timezone = new \DateTimeZone(@date_default_timezone_get());
        }

        $this->indent       = $indent;
        $this->stringTag    = $stringTag;
        $this->timezone     = $timezone;
    }

    protected function getWriter()
    {
        $writer = new \XMLWriter();
        $writer->openMemory();
        if ($this->indent) {
            $writer->setIndent(true);
            $writer->startDocument('1.0', 'UTF-8');
        } else {
            $writer->setIndent(false);
            $writer->startDocument();
        }
        return $writer;
    }

    protected function finalizeWrite(\XMLWriter $writer)
    {
        $writer->endDocument();
        $result = $writer->outputMemory(true);

        if (!$this->indent) {
            // Remove the XML declaration for an even
            // more compact result.
            if (!strncmp($result, '<'.'?xml', 5)) {
                $pos    = strpos($result, '?'.'>');
                if ($pos !== false) {
                    $result = (string) substr($result, $pos + 2);
                }
            }
            // Remove leading & trailing whitespace.
            $result = trim($result);
        }

        return $result;
    }

    public function encodeRequest(\fpoirotte\XRL\Request $request)
    {
        $writer = $this->getWriter();
        $writer->startElement('methodCall');
        $writer->writeElement('methodName', $request->getProcedure());
        if (count($request->getParams())) {
            $writer->startElement('params');
            foreach ($request->getParams() as $param) {
                $writer->startElement('param');
                $writer->startElement('value');
                $param->write($writer, $this->timezone, $this->stringTag);
                $writer->endElement();
                $writer->endElement();
            }
            $writer->endElement();
        }
        $writer->endElement();
        $result = $this->finalizeWrite($writer);
        return $result;
    }

    public function encodeError(\Exception $error)
    {
        $writer = $this->getWriter();
        $writer->startElement('methodResponse');
        $writer->startElement('fault');
        $writer->startElement('value');
        $exc = \fpoirotte\XRL\NativeEncoder::convert($error);
        $exc->write($writer, $this->timezone, $this->stringTag);
        $writer->endElement();
        $writer->endElement();
        $writer->endElement();
        $result = $this->finalizeWrite($writer);
        return $result;
    }

    public function encodeResponse($response)
    {
        if (!($response instanceof \fpoirotte\XRL\Types\AbstractType)) {
            throw new \InvalidArgumentException('Invalid response');
        }

        $writer = $this->getWriter();
        $writer->startElement('methodResponse');
        $writer->startElement('params');
        $writer->startElement('param');
        $writer->startElement('value');
        $response->write($writer, $this->timezone, $this->stringTag);
        $writer->endElement();
        $writer->endElement();
        $writer->endElement();
        $writer->endElement();
        $result = $this->finalizeWrite($writer);
        return $result;
    }
}