139 lines
3.3 KiB
PHP
139 lines
3.3 KiB
PHP
<?php
|
|
namespace Szurubooru\Services;
|
|
|
|
class FileService
|
|
{
|
|
private $dataDirectory;
|
|
private $httpHelper;
|
|
|
|
public function __construct(\Szurubooru\Config $config, \Szurubooru\Helpers\HttpHelper $httpHelper)
|
|
{
|
|
$this->dataDirectory = $config->getPublicDataDirectory();
|
|
$this->httpHelper = $httpHelper;
|
|
}
|
|
|
|
public function serve($target, $options = [])
|
|
{
|
|
$fullPath = $this->getFullPath($target);
|
|
|
|
$daysToLive = isset($options->daysToLive)
|
|
? $options->daysToLive
|
|
: 7;
|
|
|
|
$secondsToLive = $daysToLive * 24 * 60 * 60;
|
|
$lastModified = filemtime($fullPath);
|
|
$eTag = md5(file_get_contents($fullPath)); //todo: faster
|
|
|
|
$ifModifiedSince = isset($_SERVER['HTTP_IF_MODIFIED_SINCE'])
|
|
? $_SERVER['HTTP_IF_MODIFIED_SINCE']
|
|
: false;
|
|
|
|
$eTagHeader = isset($_SERVER['HTTP_IF_NONE_MATCH'])
|
|
? trim($_SERVER['HTTP_IF_NONE_MATCH'], "\" \t\r\n")
|
|
: null;
|
|
|
|
$this->httpHelper->setHeader('ETag', '"' . $eTag . '"');
|
|
$this->httpHelper->setHeader('Last-Modified', gmdate('D, d M Y H:i:s \G\M\T', $lastModified));
|
|
$this->httpHelper->setHeader('Pragma', 'public');
|
|
$this->httpHelper->setHeader('Cache-Control', 'public, max-age=' . $secondsToLive);
|
|
$this->httpHelper->setHeader('Expires', gmdate('D, d M Y H:i:s \G\M\T', time() + $secondsToLive));
|
|
|
|
if (isset($options->customFileName))
|
|
$this->httpHelper->setHeader('Content-Disposition', 'inline; filename="' . $options->customFileName . '"');
|
|
|
|
$this->httpHelper->setHeader(
|
|
'Content-Type',
|
|
isset($options->mimeType)
|
|
? $options->mimeType
|
|
: mime_content_type($fullPath));
|
|
|
|
if (strtotime($ifModifiedSince) === $lastModified or $eTagHeader === $eTag)
|
|
{
|
|
$this->httpHelper->setResponseCode(304);
|
|
}
|
|
else
|
|
{
|
|
$this->httpHelper->setResponseCode(200);
|
|
readfile($fullPath);
|
|
}
|
|
exit;
|
|
}
|
|
|
|
public function createFolders($target)
|
|
{
|
|
$fullPath = $this->getFullPath(dirname($target));
|
|
if (!file_exists($fullPath))
|
|
mkdir($fullPath, 0777, true);
|
|
}
|
|
|
|
public function exists($target)
|
|
{
|
|
$fullPath = $this->getFullPath($target);
|
|
return $target and file_exists($fullPath);
|
|
}
|
|
|
|
public function delete($target)
|
|
{
|
|
$fullPath = $this->getFullPath($target);
|
|
if (file_exists($fullPath))
|
|
unlink($fullPath);
|
|
}
|
|
|
|
public function load($source)
|
|
{
|
|
if (!$this->exists($source))
|
|
return null;
|
|
$fullPath = $this->getFullPath($source);
|
|
return file_get_contents($fullPath);
|
|
}
|
|
|
|
public function save($destination, $data)
|
|
{
|
|
$this->createFolders($destination);
|
|
$finalDestination = $this->getFullPath($destination);
|
|
file_put_contents($finalDestination, $data);
|
|
}
|
|
|
|
public function getFullPath($destination)
|
|
{
|
|
return $this->dataDirectory . DIRECTORY_SEPARATOR . $destination;
|
|
}
|
|
|
|
public function download($url, $maxBytes = null)
|
|
{
|
|
set_time_limit(60);
|
|
try
|
|
{
|
|
$srcHandle = fopen($url, 'rb');
|
|
}
|
|
catch (Exception $e)
|
|
{
|
|
throw new \Exception('Cannot open URL for reading: ' . $e->getMessage());
|
|
}
|
|
|
|
if (!$srcHandle)
|
|
throw new \Exception('Cannot open URL for reading');
|
|
|
|
$result = '';
|
|
try
|
|
{
|
|
while (!feof($srcHandle))
|
|
{
|
|
$buffer = fread($srcHandle, 4 * 1024);
|
|
if ($maxBytes !== null and strlen($result) > $maxBytes)
|
|
{
|
|
throw new \Exception(
|
|
'File is too big (maximum size: %s)',
|
|
TextHelper::useBytesUnits($maxBytes));
|
|
}
|
|
$result .= $buffer;
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
fclose($srcHandle);
|
|
}
|
|
|
|
return $result;
|
|
}
|
|
}
|