96 lines
2.0 KiB
PHP
96 lines
2.0 KiB
PHP
<?php
|
|
/**
|
|
* @brief File IteratorIterator
|
|
* @author <a href='http://www.invisionpower.com'>Invision Power Services, Inc.</a>
|
|
* @copyright (c) 2001 - SVN_YYYY Invision Power Services, Inc.
|
|
* @license http://www.invisionpower.com/legal/standards/
|
|
* @package IPS Social Suite
|
|
* @since 10 Oct 2013
|
|
* @version SVN_VERSION_NUMBER
|
|
*/
|
|
|
|
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;
|
|
|
|
/**
|
|
* 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
|
|
* @return void
|
|
*/
|
|
public function __construct( \Traversable $iterator, $storageExtension, $urlField=NULL, $fileUrlsOnly=FALSE )
|
|
{
|
|
$this->storageExtension = $storageExtension;
|
|
$this->urlField = $urlField;
|
|
$this->fileUrlsOnly = $fileUrlsOnly;
|
|
return parent::__construct( $iterator );
|
|
}
|
|
|
|
/**
|
|
* Get current
|
|
*
|
|
* @return \IPS\File
|
|
*/
|
|
public function current()
|
|
{
|
|
try
|
|
{
|
|
$data = $this->data();
|
|
$obj = \IPS\File::get( $this->storageExtension, $this->urlField ? $data[ $this->urlField ] : $data );
|
|
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();
|
|
}
|
|
} |