198 lines
5.1 KiB
PHP
198 lines
5.1 KiB
PHP
|
<?php
|
||
|
|
||
|
/*
|
||
|
* This file is part of SwiftMailer.
|
||
|
* (c) 2004-2009 Chris Corbyn
|
||
|
*
|
||
|
* For the full copyright and license information, please view the LICENSE
|
||
|
* file that was distributed with this source code.
|
||
|
*/
|
||
|
|
||
|
/**
|
||
|
* A basic KeyCache backed by an array.
|
||
|
*
|
||
|
* @author Chris Corbyn
|
||
|
*/
|
||
|
class Swift_KeyCache_ArrayKeyCache implements Swift_KeyCache
|
||
|
{
|
||
|
/**
|
||
|
* Cache contents.
|
||
|
*
|
||
|
* @var array
|
||
|
*/
|
||
|
private $contents = [];
|
||
|
|
||
|
/**
|
||
|
* An InputStream for cloning.
|
||
|
*
|
||
|
* @var Swift_KeyCache_KeyCacheInputStream
|
||
|
*/
|
||
|
private $stream;
|
||
|
|
||
|
/**
|
||
|
* Create a new ArrayKeyCache with the given $stream for cloning to make
|
||
|
* InputByteStreams.
|
||
|
*/
|
||
|
public function __construct(Swift_KeyCache_KeyCacheInputStream $stream)
|
||
|
{
|
||
|
$this->stream = $stream;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Set a string into the cache under $itemKey for the namespace $nsKey.
|
||
|
*
|
||
|
* @see MODE_WRITE, MODE_APPEND
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
* @param string $string
|
||
|
* @param int $mode
|
||
|
*/
|
||
|
public function setString($nsKey, $itemKey, $string, $mode)
|
||
|
{
|
||
|
$this->prepareCache($nsKey);
|
||
|
switch ($mode) {
|
||
|
case self::MODE_WRITE:
|
||
|
$this->contents[$nsKey][$itemKey] = $string;
|
||
|
break;
|
||
|
case self::MODE_APPEND:
|
||
|
if (!$this->hasKey($nsKey, $itemKey)) {
|
||
|
$this->contents[$nsKey][$itemKey] = '';
|
||
|
}
|
||
|
$this->contents[$nsKey][$itemKey] .= $string;
|
||
|
break;
|
||
|
default:
|
||
|
throw new Swift_SwiftException('Invalid mode ['.$mode.'] used to set nsKey='.$nsKey.', itemKey='.$itemKey);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Set a ByteStream into the cache under $itemKey for the namespace $nsKey.
|
||
|
*
|
||
|
* @see MODE_WRITE, MODE_APPEND
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
* @param int $mode
|
||
|
*/
|
||
|
public function importFromByteStream($nsKey, $itemKey, Swift_OutputByteStream $os, $mode)
|
||
|
{
|
||
|
$this->prepareCache($nsKey);
|
||
|
switch ($mode) {
|
||
|
case self::MODE_WRITE:
|
||
|
$this->clearKey($nsKey, $itemKey);
|
||
|
// no break
|
||
|
case self::MODE_APPEND:
|
||
|
if (!$this->hasKey($nsKey, $itemKey)) {
|
||
|
$this->contents[$nsKey][$itemKey] = '';
|
||
|
}
|
||
|
while (false !== $bytes = $os->read(8192)) {
|
||
|
$this->contents[$nsKey][$itemKey] .= $bytes;
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new Swift_SwiftException('Invalid mode ['.$mode.'] used to set nsKey='.$nsKey.', itemKey='.$itemKey);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Provides a ByteStream which when written to, writes data to $itemKey.
|
||
|
*
|
||
|
* NOTE: The stream will always write in append mode.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
*
|
||
|
* @return Swift_InputByteStream
|
||
|
*/
|
||
|
public function getInputByteStream($nsKey, $itemKey, Swift_InputByteStream $writeThrough = null)
|
||
|
{
|
||
|
$is = clone $this->stream;
|
||
|
$is->setKeyCache($this);
|
||
|
$is->setNsKey($nsKey);
|
||
|
$is->setItemKey($itemKey);
|
||
|
if (isset($writeThrough)) {
|
||
|
$is->setWriteThroughStream($writeThrough);
|
||
|
}
|
||
|
|
||
|
return $is;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get data back out of the cache as a string.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
*
|
||
|
* @return string
|
||
|
*/
|
||
|
public function getString($nsKey, $itemKey)
|
||
|
{
|
||
|
$this->prepareCache($nsKey);
|
||
|
if ($this->hasKey($nsKey, $itemKey)) {
|
||
|
return $this->contents[$nsKey][$itemKey];
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get data back out of the cache as a ByteStream.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
* @param Swift_InputByteStream $is to write the data to
|
||
|
*/
|
||
|
public function exportToByteStream($nsKey, $itemKey, Swift_InputByteStream $is)
|
||
|
{
|
||
|
$this->prepareCache($nsKey);
|
||
|
$is->write($this->getString($nsKey, $itemKey));
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Check if the given $itemKey exists in the namespace $nsKey.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
*
|
||
|
* @return bool
|
||
|
*/
|
||
|
public function hasKey($nsKey, $itemKey)
|
||
|
{
|
||
|
$this->prepareCache($nsKey);
|
||
|
|
||
|
return \array_key_exists($itemKey, $this->contents[$nsKey]);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Clear data for $itemKey in the namespace $nsKey if it exists.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
* @param string $itemKey
|
||
|
*/
|
||
|
public function clearKey($nsKey, $itemKey)
|
||
|
{
|
||
|
unset($this->contents[$nsKey][$itemKey]);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Clear all data in the namespace $nsKey if it exists.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
*/
|
||
|
public function clearAll($nsKey)
|
||
|
{
|
||
|
unset($this->contents[$nsKey]);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Initialize the namespace of $nsKey if needed.
|
||
|
*
|
||
|
* @param string $nsKey
|
||
|
*/
|
||
|
private function prepareCache($nsKey)
|
||
|
{
|
||
|
if (!\array_key_exists($nsKey, $this->contents)) {
|
||
|
$this->contents[$nsKey] = [];
|
||
|
}
|
||
|
}
|
||
|
}
|