You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
4.1 KiB
115 lines
4.1 KiB
<?php
|
|
/**
|
|
* @license MIT
|
|
* Copyright 2022 Dustin Wilson, et al.
|
|
* See LICENSE and AUTHORS files for details
|
|
*/
|
|
|
|
declare(strict_types=1);
|
|
namespace MensBeam\Logger;
|
|
use MensBeam\{
|
|
Filesystem as Fs,
|
|
Path
|
|
};
|
|
|
|
|
|
class StreamHandler extends Handler {
|
|
protected int $chunkSize = 10 * 1024 * 1024;
|
|
protected $resource = null;
|
|
protected ?string $url = null;
|
|
protected ?string $urlScheme = null;
|
|
|
|
protected ?string $_entryFormat = '%time% %channel% %level_name% %message%';
|
|
|
|
|
|
public function __construct($stream = 'php://stdout', array $levels = [ 0, 1, 2, 3, 4, 5, 6, 7 ], array $options = []) {
|
|
// Get the memory limit to determine the chunk size.
|
|
|
|
// The memory limit is in a shorthand format
|
|
// (https://www.php.net/manual/en/faq.using.php#faq.using.shorthandbytes), so we
|
|
// need it as a integer representation in bytes.
|
|
if (preg_match('/^\s*(?<num>\d+)(?:\.\d+)?\s*(?<unit>[gkm])\s*$/i', ini_get('memory_limit'), $matches) === 1) {
|
|
$num = (int)$matches['num'];
|
|
switch (strtolower($matches['unit'] ?? '')) {
|
|
case 'g': $num *= 1024;
|
|
case 'm': $num *= 1024;
|
|
case 'k': $num *= 1024;
|
|
}
|
|
|
|
// Use 10% of allowed memory or 100K, whichever is largest
|
|
$this->chunkSize = min($this->chunkSize, max((int)($num / 10), 100 * 1024));
|
|
}
|
|
|
|
$this->setStream($stream);
|
|
|
|
// Bad dog, no biscuit!
|
|
if (($options['entryFormat'] ?? null) === '') {
|
|
$options['entryFormat'] = $this->_entryFormat;
|
|
}
|
|
|
|
parent::__construct($levels, $options);
|
|
}
|
|
|
|
|
|
|
|
|
|
public function getStream() {
|
|
return $this->resource ?? $this->url;
|
|
}
|
|
|
|
public function setStream($value): void {
|
|
if (is_resource($value)) {
|
|
$this->resource = $value;
|
|
stream_set_chunk_size($this->resource, $this->chunkSize);
|
|
} elseif(is_string($value)) {
|
|
$value = Path::canonicalize($value);
|
|
// This wouldn't be useful for validating a URI schema, but it's fine for what this needs
|
|
preg_match('/^(?:(?<scheme>[^:\s\/]+):)?(?<slashes>\/*)/i', $value, $matches);
|
|
if (in_array($matches['scheme'], [ 'file', '' ])) {
|
|
$slashCount = strlen($matches['slashes'] ?? '');
|
|
$relative = ($matches['scheme'] === 'file') ? ($slashCount === 0 || $slashCount === 2) : ($slashCount === 0);
|
|
$value = (($relative) ? getcwd() : '') . '/' . substr($value, strlen($matches[0]));
|
|
}
|
|
|
|
$this->url = $value;
|
|
$this->urlScheme = $matches['scheme'] ?: 'file';
|
|
} else {
|
|
$type = gettype($value);
|
|
$type = ($type === 'object') ? $value::class : $type;
|
|
throw new InvalidArgumentException(sprintf('Argument #1 ($value) must be of type resource|string, %s given', $type));
|
|
}
|
|
}
|
|
|
|
|
|
protected function invokeCallback(string $time, int $level, string $channel, string $message, array $context = []): void {
|
|
// Do entry formatting here.
|
|
$output = trim(preg_replace_callback('/%([a-z_]+)%/', function($m) use ($time, $level, $channel, $message) {
|
|
switch ($m[1]) {
|
|
case 'channel': return $channel;
|
|
case 'level': return (string)$level;
|
|
case 'level_name': return strtoupper(Level::from($level)->name);
|
|
case 'message': return $message;
|
|
case 'time': return $time;
|
|
default: return '';
|
|
}
|
|
}, $this->_entryFormat));
|
|
// If output contains any newlines then add an additional newline to aid readability.
|
|
if (str_contains($output, \PHP_EOL)) {
|
|
$output .= \PHP_EOL;
|
|
}
|
|
$output .= \PHP_EOL;
|
|
|
|
if ($this->resource === null) {
|
|
if ($this->urlScheme === 'file') {
|
|
Fs::mkdir(dirname($this->url));
|
|
}
|
|
|
|
$fp = fopen($this->url, 'a');
|
|
stream_set_chunk_size($fp, $this->chunkSize);
|
|
fwrite($fp, $output);
|
|
fclose($fp);
|
|
} else {
|
|
fwrite($this->resource, $output);
|
|
}
|
|
}
|
|
}
|