vladahejda / nettecsvresponse Goto Github PK

<?php
namespace App\Utils\Responses;

use Nette;

class CsvResponseWithLineEndingsSetting extends \OHWeb\Application\Responses\CsvResponse {

  /** line endings */
  const LF = "\n",
    CRLF = "\r\n";

  /** @var string */
  protected $line_endings = self::CRLF;


  /**
   * Line endings
   * @param string $line_endings
   * @return self
   * @throws \InvalidArgumentException
   */
   public function setLineEndings($line_endings) {
     if (empty($line_endings) || ($line_endings != self::LF && $line_endings != self::CRLF)) {
       throw new \InvalidArgumentException(__CLASS__.": line endings cannot be an empty and must be self::LF or self::CRLF.");
     }
     $this->line_endings = $line_endings;
     return $this;
   }


  /**
   * Sends response to output.
   * @param  NetteHttpIRequest  $httpRequest
   * @param  NetteHttpIResponse $httpResponse
   */
  public function send(Nette\Http\IRequest $httpRequest, Nette\Http\IResponse $httpResponse) {

    $httpResponse->setContentType($this->contentType, $this->outputCharset);

    $attachment = 'attachment';
    if (!empty($this->filename)) {
      $attachment .= '; filename="' . $this->filename . '"';
    }

    $httpResponse->setHeader('Content-Disposition', $attachment);

    $data = $this->formatCsv();

    // fix line endings
    if ($this->line_endings == self::CRLF) {
      $data = str_replace("\n", "\r\n", str_replace("\r", '', $data));
    }

    $httpResponse->setHeader('Content-Length', strlen($data));
    print $data;

  }

}