<?php
namespace Guzzle\Http\Message;
use Guzzle\Http\EntityBody;
use Guzzle\Http\EntityBodyInterface;
use Guzzle\Http\QueryString;
use Guzzle\Http\RedirectPlugin;
use Guzzle\Http\Exception\RequestException;
class EntityEnclosingRequest extends Request implements EntityEnclosingRequestInterface {
protected $expectCutoff = 1048576;
protected $body;
protected $postFields;
protected $postFiles = array();
public function __construct($method, $url, $headers = array()) {
$this->postFields = new QueryString();
$this->postFields
->setPrefix('');
parent::__construct($method, $url, $headers);
}
public function __toString() {
if (count($this->postFields) && empty($this->postFiles)) {
return parent::__toString() . (string) $this->postFields;
}
return parent::__toString() . $this->body;
}
public function setState($state) {
parent::setState($state);
if ($state == self::STATE_TRANSFER && !$this->body && !count($this->postFields) && !count($this->postFiles)) {
$this
->setHeader('Content-Length', 0)
->removeHeader('Transfer-Encoding');
}
return $this;
}
public function setBody($body, $contentType = null, $tryChunkedTransfer = false) {
$this->body = EntityBody::factory($body);
$this
->removeHeader('Content-Length');
if ($contentType) {
$this
->setHeader('Content-Type', (string) $contentType);
}
if (!$this->body
->isSeekable() && $this->expectCutoff !== false) {
$this
->setHeader('Expect', '100-Continue');
}
if ($tryChunkedTransfer) {
$this
->setHeader('Transfer-Encoding', 'chunked');
}
else {
$this
->removeHeader('Transfer-Encoding');
$size = $this->body
->getContentLength();
if ($size !== null && $size !== false) {
$this
->setHeader('Content-Length', $size);
if ($size > $this->expectCutoff) {
$this
->setHeader('Expect', '100-Continue');
}
}
elseif ('1.1' == $this->protocolVersion) {
$this
->setHeader('Transfer-Encoding', 'chunked');
}
else {
throw new RequestException('Cannot determine Content-Length and cannot use chunked Transfer-Encoding when using HTTP/1.0');
}
}
return $this;
}
public function getBody() {
return $this->body;
}
public function setExpectHeaderCutoff($size) {
$this->expectCutoff = $size;
if ($size === false || !$this->body) {
$this
->removeHeader('Expect');
}
elseif ($this->body && $this->body
->getSize() && $this->body
->getSize() > $size) {
$this
->setHeader('Expect', '100-Continue');
}
return $this;
}
public function configureRedirects($strict = false, $maxRedirects = 5) {
$this
->getParams()
->set(RedirectPlugin::STRICT_REDIRECTS, $strict);
if ($maxRedirects == 0) {
$this
->getParams()
->set(RedirectPlugin::DISABLE, true);
}
else {
$this
->getParams()
->set(RedirectPlugin::MAX_REDIRECTS, $maxRedirects);
}
return $this;
}
public function getPostField($field) {
return $this->postFields
->get($field);
}
public function getPostFields() {
return $this->postFields;
}
public function setPostField($key, $value) {
$this->postFields
->set($key, $value);
$this
->processPostFields();
return $this;
}
public function addPostFields($fields) {
$this->postFields
->merge($fields);
$this
->processPostFields();
return $this;
}
public function removePostField($field) {
$this->postFields
->remove($field);
$this
->processPostFields();
return $this;
}
public function getPostFiles() {
return $this->postFiles;
}
public function getPostFile($fieldName) {
return isset($this->postFiles[$fieldName]) ? $this->postFiles[$fieldName] : null;
}
public function removePostFile($fieldName) {
unset($this->postFiles[$fieldName]);
$this
->processPostFields();
return $this;
}
public function addPostFile($field, $filename = null, $contentType = null) {
$data = null;
if ($field instanceof PostFileInterface) {
$data = $field;
}
elseif (!is_string($filename)) {
throw new RequestException('The path to a file must be a string');
}
elseif (!empty($filename)) {
$data = new PostFile($field, $filename, $contentType);
}
if ($data) {
if (!isset($this->postFiles[$data
->getFieldName()])) {
$this->postFiles[$data
->getFieldName()] = array(
$data,
);
}
else {
$this->postFiles[$data
->getFieldName()][] = $data;
}
$this
->processPostFields();
}
return $this;
}
public function addPostFiles(array $files) {
foreach ($files as $key => $file) {
if ($file instanceof PostFileInterface) {
$this
->addPostFile($file, null, null, false);
}
elseif (is_string($file)) {
if (is_numeric($key)) {
$key = 'file';
}
$this
->addPostFile($key, $file, null, false);
}
else {
throw new RequestException('File must be a string or instance of PostFileInterface');
}
}
return $this;
}
protected function processPostFields() {
if (empty($this->postFiles)) {
$this
->removeHeader('Expect')
->setHeader('Content-Type', self::URL_ENCODED);
}
else {
$this
->setHeader('Content-Type', self::MULTIPART);
if ($this->expectCutoff !== false) {
$this
->setHeader('Expect', '100-Continue');
}
}
}
}