better-testing

AphrontHTTPProxyResponse.php
View Options

			<?php

			/**
			* Responds to a request by proxying an HTTP future.
			*
			* NOTE: This is currently very inefficient for large responses, and buffers
			* the entire response into memory before returning it. It should be updated
			* to stream the response instead, but we need to complete additional
			* infrastructure work first.
			*/
			final class AphrontHTTPProxyResponse extends AphrontResponse {

			private $future;
			private $headers;
			private $httpCode;

			public function setHTTPFuture(HTTPSFuture $future) {
			$this->future = $future;
			return $this;
			}

			public function getHTTPFuture() {
			return $this->future;
			}

			public function getCacheHeaders() {
			return array();
			}

			public function getHeaders() {
			$this->readRequestHeaders();
			return array_merge(
			parent::getHeaders(),
			$this->headers,
			array(
			array('X-Phabricator-Proxy', 'true'),
			));
			}

			public function buildResponseString() {
			// TODO: AphrontResponse needs to support streaming responses.
			return $this->readRequest();
			}

			public function getHTTPResponseCode() {
			$this->readRequestHeaders();
			return $this->httpCode;
			}

			private function readRequestHeaders() {
			// TODO: This should read only the headers.
			$this->readRequest();
			}

			private function readRequest() {
			// TODO: This is grossly inefficient for large requests.

			list($status, $body, $headers) = $this->future->resolve();
			$this->httpCode = $status->getStatusCode();
			$this->headers = $headers;

			return $body;
			}

			}