<?php

/**

 * Send XML response back to AJAX request.

 *

 * @package WordPress

 * @since 2.1.0

 */

class WP_Ajax_Response {

	/**

	 * Store XML responses to send.

	 *

	 * @since 2.1.0

	 * @var array

	 * @access private

	 */

	var $responses = array();

	/**

	 * Constructor - Passes args to {@link WP_Ajax_Response::add()}.

	 *

	 * @since 2.1.0

	 * @see WP_Ajax_Response::add()

	 *

	 * @param string|array $args Optional. Will be passed to add() method.

	 * @return WP_Ajax_Response

	 */

	function __construct( $args = '' ) {

		if ( !empty($args) )

			$this->add($args);

	}

	/**

	 * Append to XML response based on given arguments.

	 *

	 * The arguments that can be passed in the $args parameter are below. It is

	 * also possible to pass a WP_Error object in either the 'id' or 'data'

	 * argument. The parameter isn't actually optional, content should be given

	 * in order to send the correct response.

	 *

	 * 'what' argument is a string that is the XMLRPC response type.

	 * 'action' argument is a boolean or string that acts like a nonce.

	 * 'id' argument can be WP_Error or an integer.

	 * 'old_id' argument is false by default or an integer of the previous ID.

	 * 'position' argument is an integer or a string with -1 = top, 1 = bottom,

	 * html ID = after, -html ID = before.

	 * 'data' argument is a string with the content or message.

	 * 'supplemental' argument is an array of strings that will be children of

	 * the supplemental element.

	 *

	 * @since 2.1.0

	 *

	 * @param string|array $args Override defaults.

	 * @return string XML response.

	 */

	function add( $args = '' ) {

		$defaults = array(

			'what' => 'object', 'action' => false,

			'id' => '0', 'old_id' => false,

			'position' => 1,

			'data' => '', 'supplemental' => array()

		);

		$r = wp_parse_args( $args, $defaults );

		extract( $r, EXTR_SKIP );

		$position = preg_replace( '/[^a-z0-9:_-]/i', '', $position );

		if ( is_wp_error($id) ) {

			$data = $id;

			$id = 0;

		}

		$response = '';

		if ( is_wp_error($data) ) {

			foreach ( (array) $data->get_error_codes() as $code ) {

				$response .= "<wp_error code='$code'><![CDATA[" . $data->get_error_message($code) . "]]></wp_error>";

				if ( !$error_data = $data->get_error_data($code) )

					continue;

				$class = '';

				if ( is_object($error_data) ) {

					$class = ' class="' . get_class($error_data) . '"';

					$error_data = get_object_vars($error_data);

				}

				$response .= "<wp_error_data code='$code'$class>";

				if ( is_scalar($error_data) ) {

					$response .= "<![CDATA[$error_data]]>";

				} elseif ( is_array($error_data) ) {

					foreach ( $error_data as $k => $v )

						$response .= "<$k><![CDATA[$v]]></$k>";

				}

				$response .= "</wp_error_data>";

			}

		} else {

			$response = "<response_data><![CDATA[$data]]></response_data>";

		}

		$s = '';

		if ( is_array($supplemental) ) {

			foreach ( $supplemental as $k => $v )

				$s .= "<$k><![CDATA[$v]]></$k>";

			$s = "<supplemental>$s</supplemental>";

		}

		if ( false === $action )

			$action = $_POST['action'];

		$x = '';

		$x .= "<response action='{$action}_$id'>"; // The action attribute in the xml output is formatted like a nonce action

		$x .=	"<$what id='$id' " . ( false === $old_id ? '' : "old_id='$old_id' " ) . "position='$position'>";

		$x .=		$response;

		$x .=		$s;

		$x .=	"</$what>";

		$x .= "</response>";

		$this->responses[] = $x;

		return $x;

	}

	/**

	 * Display XML formatted responses.

	 *

	 * Sets the content type header to text/xml.

	 *

	 * @since 2.1.0

	 */

	function send() {

		header('Content-Type: text/xml');

		echo "<?xml version='1.0' standalone='yes'?><wp_ajax>";

		foreach ( (array) $this->responses as $response )

			echo $response;

		echo '</wp_ajax>';

		if ( defined( 'DOING_AJAX' ) && DOING_AJAX )

			wp_die();

		else

			die();

	}

}