Made specifying 'extension' optional. Fixed bug where downloaded file did not have extension when 'name' was specified. Fixes #2554

This commit is contained in:
ADmad 2012-02-08 01:08:04 +05:30
parent 4949a87b7e
commit bf700c826a
2 changed files with 37 additions and 5 deletions

View file

@ -244,6 +244,7 @@ class MediaViewTest extends CakeTestCase {
'path' => CAKE . 'Test' . DS . 'test_app' . DS . 'Config' . DS, 'path' => CAKE . 'Test' . DS . 'test_app' . DS . 'Config' . DS,
'id' => 'no_section.ini', 'id' => 'no_section.ini',
'extension' => 'ini', 'extension' => 'ini',
'name' => 'config'
); );
$this->MediaView->expects($this->exactly(2)) $this->MediaView->expects($this->exactly(2))
->method('_isActive') ->method('_isActive')
@ -270,7 +271,7 @@ class MediaViewTest extends CakeTestCase {
$this->MediaView->response->expects($this->once()) $this->MediaView->response->expects($this->once())
->method('download') ->method('download')
->with('no_section.ini'); ->with('config.ini');
$this->MediaView->response->expects($this->at(4)) $this->MediaView->response->expects($this->at(4))
->method('header') ->method('header')
@ -357,7 +358,7 @@ class MediaViewTest extends CakeTestCase {
* *
* @return void * @return void
*/ */
function testRenderUpperExtesnion() { public function testRenderUpperExtension() {
$this->MediaView->viewVars = array( $this->MediaView->viewVars = array(
'path' => CAKE . 'Test' . DS . 'test_app' . DS . 'Vendor' . DS .'img' . DS, 'path' => CAKE . 'Test' . DS . 'test_app' . DS . 'Vendor' . DS .'img' . DS,
'id' => 'test_2.JPG', 'id' => 'test_2.JPG',
@ -376,4 +377,27 @@ class MediaViewTest extends CakeTestCase {
$this->MediaView->render(); $this->MediaView->render();
} }
/**
* Test downloading files with extension not explicitly set.
*
* @return void
*/
public function testRenderExtensionNotSet() {
$this->MediaView->viewVars = array(
'path' => CAKE . 'Test' . DS . 'test_app' . DS . 'Vendor' . DS .'img' . DS,
'id' => 'test_2.JPG',
);
$this->MediaView->response->expects($this->any())
->method('type')
->with('jpg')
->will($this->returnArgument(0));
$this->MediaView->expects($this->at(0))
->method('_isActive')
->will($this->returnValue(true));
$this->MediaView->render();
}
} }

View file

@ -31,9 +31,11 @@ App::uses('CakeRequest', 'Network');
* - `id` The filename on the server's filesystem, including extension. * - `id` The filename on the server's filesystem, including extension.
* - `name` The filename that will be sent to the user, specified without the extension. * - `name` The filename that will be sent to the user, specified without the extension.
* - `download` Set to true to set a `Content-Disposition` header. This is ideal for file downloads. * - `download` Set to true to set a `Content-Disposition` header. This is ideal for file downloads.
* - `extension` The extension of the file being served. This is used to set the mimetype * - `extension` The extension of the file being served. This is used to set the mimetype.
* If not provided its extracted from filename provided as `id`.
* - `path` The absolute path, including the trailing / on the server's filesystem to `id`. * - `path` The absolute path, including the trailing / on the server's filesystem to `id`.
* - `mimeType` The mime type of the file if CakeResponse doesn't know about it. * - `mimeType` The mime type of the file if CakeResponse doesn't know about it.
* Must be an associative array with extension as key and mime type as value eg. array('ini' => 'text/plain')
* *
* ### Usage * ### Usage
* *
@ -113,7 +115,11 @@ class MediaView extends View {
$this->response->type($mimeType); $this->response->type($mimeType);
} }
if (isset($extension) && $this->_isActive()) { if (!isset($extension)) {
$extension = pathinfo($id, PATHINFO_EXTENSION);
}
if ($this->_isActive()) {
$extension = strtolower($extension); $extension = strtolower($extension);
$chunkSize = 8192; $chunkSize = 8192;
$buffer = ''; $buffer = '';
@ -128,7 +134,7 @@ class MediaView extends View {
} else { } else {
$modified = time(); $modified = time();
} }
if ($this->response->type($extension) === false) { if (!$extension || $this->response->type($extension) === false) {
$download = true; $download = true;
} }
@ -157,6 +163,8 @@ class MediaView extends View {
} }
if (is_null($name)) { if (is_null($name)) {
$name = $id; $name = $id;
} elseif ($extension) {
$name .= '.' . $extension;
} }
$this->response->download($name); $this->response->download($name);
$this->response->header(array('Accept-Ranges' => 'bytes')); $this->response->header(array('Accept-Ranges' => 'bytes'));