122 lines
2.8 KiB
PHP
122 lines
2.8 KiB
PHP
<?php
|
|
/**
|
|
* @brief File IteratorIterator
|
|
* @author <a href='https://www.invisioncommunity.com'>Invision Power Services, Inc.</a>
|
|
* @copyright (c) Invision Power Services, Inc.
|
|
* @license https://www.invisioncommunity.com/legal/standards/
|
|
* @package Invision Community
|
|
* @since 10 Oct 2013
|
|
*/
|
|
|
|
namespace IPS\File;
|
|
|
|
/* To prevent PHP errors (extending class does not exist) revealing path */
|
|
if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
|
|
{
|
|
header( ( isset( $_SERVER['SERVER_PROTOCOL'] ) ? $_SERVER['SERVER_PROTOCOL'] : 'HTTP/1.0' ) . ' 403 Forbidden' );
|
|
exit;
|
|
}
|
|
|
|
/**
|
|
* File IteratorIterator
|
|
*/
|
|
class _Iterator extends \IteratorIterator implements \Countable
|
|
{
|
|
/**
|
|
* @brief Stroage Extension
|
|
*/
|
|
protected $storageExtension;
|
|
|
|
/**
|
|
* @brief URL Field
|
|
*/
|
|
protected $urlField;
|
|
|
|
/**
|
|
* @brief URLs Only
|
|
*/
|
|
protected $fileUrlsOnly;
|
|
|
|
/**
|
|
* @brief Used to restore 'real' names when filenames cleaned (eg adh1029_file.php back to fi^le.php)
|
|
*/
|
|
protected $replaceNameField;
|
|
|
|
/**
|
|
* Constructor
|
|
*
|
|
* @param Traversable $iterator The iterator
|
|
* @param string $storageExtension The storage extension
|
|
* @param string|NULL $urlField If passed a string, will look for an element with that key in the data returned from the iterator
|
|
* @param string|NULL $replaceNameField If passed a string, it will replace the originalFilename with the data in the array. Used to restore 'real' names when filenames cleaned (eg adh1029_file.php back to fi^le.php)
|
|
* @return void
|
|
*/
|
|
public function __construct( \Traversable $iterator, $storageExtension, $urlField=NULL, $fileUrlsOnly=FALSE, $replaceNameField=NULL )
|
|
{
|
|
$this->storageExtension = $storageExtension;
|
|
$this->urlField = $urlField;
|
|
$this->fileUrlsOnly = $fileUrlsOnly;
|
|
$this->replaceNameField = $replaceNameField;
|
|
return parent::__construct( $iterator );
|
|
}
|
|
|
|
/**
|
|
* Get current
|
|
*
|
|
* @return \IPS\File
|
|
*/
|
|
public function current()
|
|
{
|
|
try
|
|
{
|
|
$data = $this->data();
|
|
$urlField = NULL;
|
|
|
|
if ( $this->urlField )
|
|
{
|
|
if ( !is_string( $this->urlField ) and is_callable( $this->urlField ) )
|
|
{
|
|
$urlField = call_user_func( $this->urlField, $data );
|
|
}
|
|
else
|
|
{
|
|
$urlField = $this->urlField;
|
|
}
|
|
}
|
|
|
|
$obj = \IPS\File::get( $this->storageExtension, $urlField ? $data[ $urlField ] : $data );
|
|
|
|
if ( $this->replaceNameField and ! empty( $data[ $this->replaceNameField ] ) )
|
|
{
|
|
$obj->originalFilename = $data[ $this->replaceNameField ];
|
|
}
|
|
|
|
return ( $this->fileUrlsOnly ) ? (string) $obj->url : $obj;
|
|
}
|
|
catch ( \Exception $e )
|
|
{
|
|
$this->next();
|
|
return $this->current();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get data
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function data()
|
|
{
|
|
return parent::current();
|
|
}
|
|
|
|
/**
|
|
* Get count
|
|
*
|
|
* @return int
|
|
*/
|
|
public function count()
|
|
{
|
|
return $this->getInnerIterator()->count();
|
|
}
|
|
} |