Viewing File: /home/ubuntu/code_review/arcanist/src/future/aws/PhutilAWSS3Future.php

<?php

final class PhutilAWSS3Future extends PhutilAWSFuture {

  private $bucket;

  public function getServiceName() {
    return 's3';
  }

  public function setBucket($bucket) {
    $this->bucket = $bucket;
    return $this;
  }

  public function getBucket() {
    return $this->bucket;
  }

  public function setParametersForGetObject($key) {
    $bucket = $this->getBucket();

    $this->setHTTPMethod('GET');
    $this->setPath($bucket.'/'.$key);

    return $this;
  }

  public function setParametersForPutObject($key, $value) {
    $bucket = $this->getBucket();

    $this->setHTTPMethod('PUT');
    $this->setPath($bucket.'/'.$key);

    $this->addHeader('X-Amz-ACL', 'private');
    $this->addHeader('Content-Type', 'application/octet-stream');

    $this->setData($value);

    return $this;
  }

  public function setParametersForDeleteObject($key) {
    $bucket = $this->getBucket();

    $this->setHTTPMethod('DELETE');
    $this->setPath($bucket.'/'.$key);

    return $this;
  }

  protected function didReceiveResult($result) {
    list($status, $body, $headers) = $result;

    if (!$status->isError()) {
      return $body;
    }

    if ($status->getStatusCode() === 404) {
      return null;
    }

    return parent::didReceiveResult($result);
  }

  protected function shouldSignContent() {
    return true;
  }

}