|
|
|
<?php
|
|
|
|
/**
|
|
|
|
* Class DirScanner
|
|
|
|
*/
|
|
|
|
|
|
|
|
Class DirScanner {
|
|
|
|
private $nginxSecureOn = false; //Nginx防盗链开启状态
|
|
|
|
private $nginxSecret = 'foo=bar'; //Nginx防盗链密钥
|
|
|
|
private $userIp = '127.0.0.1'; //用户IP地址
|
|
|
|
private $nginxSecureTimeout = 1800; //Nginx防盗链有效期,单位:秒
|
|
|
|
private $nginxSecureLinkMd5Pattern = '{secure_link_expires}{uri}{remote_addr} {secret}'; //Nginx防盗链MD5加密方式
|
|
|
|
private $allowReadContentFileExtensions = array( //允许读取文件内容的文件类型
|
|
|
|
'txt',
|
|
|
|
'md',
|
|
|
|
'url',
|
|
|
|
);
|
|
|
|
private $fields = array( //私有属性字段名和说明
|
|
|
|
'directory' => '目录名',
|
|
|
|
'filename' => '文件名',
|
|
|
|
'realpath' => '完整路径',
|
|
|
|
'path' => '相对网址',
|
|
|
|
'extension' => '文件后缀',
|
|
|
|
'fstat' => '资源状态', //同php方法fstat: https://www.php.net/manual/en/function.fstat.php
|
|
|
|
'content' => 'MD文件内容',
|
|
|
|
'shortcut' => 'URL快捷方式',
|
|
|
|
|
|
|
|
'description' => '描述',
|
|
|
|
'keywords' => '关键词',
|
|
|
|
'snapshot' => '快照图片',
|
|
|
|
);
|
|
|
|
private $rootDir; //当前扫描的根目录
|
|
|
|
private $webRoot = '/content/'; //网站静态文件相对路径的根目录
|
|
|
|
private $scanningDirLevel = 0; //当前扫描的目录深度
|
|
|
|
private $scanStartTime = 0; //扫描开始时间,单位:秒
|
|
|
|
private $scanResults = array(); //目录扫描结果
|
|
|
|
private $tree = array(); //目录扫描树形结构
|
|
|
|
|
|
|
|
|
|
|
|
private $maxFileNumber = 1000; //目录下最大文件数量支持
|
|
|
|
|
|
|
|
|
|
|
|
protected $supportFileExtensions = array( //支持的文件类型
|
|
|
|
'txt', //纯文本
|
|
|
|
'md', //纯文本
|
|
|
|
'url', //快捷方式
|
|
|
|
'jpg', //图片
|
|
|
|
'jpeg', //图片
|
|
|
|
'png', //图片
|
|
|
|
'webp', //图片
|
|
|
|
'gif', //图片
|
|
|
|
'ico', //图标
|
|
|
|
'mp3', //音乐
|
|
|
|
'mp4', //视频
|
|
|
|
'ts', //视频
|
|
|
|
'm3u8', //视频
|
|
|
|
);
|
|
|
|
protected $maxReadFilesize = array( //默认每种文件读取内容最大大小
|
|
|
|
'txt' => 102400, //纯文本
|
|
|
|
'md' => 5242880, //纯文本
|
|
|
|
'url' => 20480, //快捷方式
|
|
|
|
'jpg' => 512000, //图片
|
|
|
|
'jpeg' => 512000, //图片
|
|
|
|
'png' => 512000, //图片
|
|
|
|
'webp' => 512000, //图片
|
|
|
|
'gif' => 512000, //图片
|
|
|
|
'ico' => 51200, //图标
|
|
|
|
'mp3' => 10485760, //音乐,10M
|
|
|
|
'mp4' => 104857600, //视频,100M
|
|
|
|
'ts' => 10485760, //视频,10M
|
|
|
|
'm3u8' => 10485760, //视频,10M
|
|
|
|
);
|
|
|
|
protected $securedFileExtensions = array( //开启Nginx防盗链的文件类型
|
|
|
|
'jpg', //图片
|
|
|
|
'jpeg', //图片
|
|
|
|
'png', //图片
|
|
|
|
'webp', //图片
|
|
|
|
'gif', //图片
|
|
|
|
'ico', //图标
|
|
|
|
'mp3', //音乐
|
|
|
|
'mp4', //视频
|
|
|
|
'ts', //视频
|
|
|
|
'm3u8', //视频
|
|
|
|
);
|
|
|
|
|
|
|
|
public $scanTimeCost = 0; //上一次目录扫描耗时,单位:毫秒
|
|
|
|
public $isApi = false; //如果为API获取数据,则realpath只返回相对路径
|
|
|
|
|
|
|
|
|
|
|
|
//判断目录名或文件名是否合法
|
|
|
|
//不允许包含斜杠/,反斜杠\,单引号',双引号",空格字符
|
|
|
|
//忽略.开头的隐藏文件
|
|
|
|
private function isValid($name) {
|
|
|
|
return str_replace(['/', '\\', "'", '"'], '', $name) == $name && !preg_match('/^\..+/', $name);
|
|
|
|
}
|
|
|
|
|
|
|
|
//解析描述文件内容
|
|
|
|
//snapshot相对路径完善,支持secure_link
|
|
|
|
//bug fix: 允许空内容文件,如:刚创建的分类文件
|
|
|
|
private function parseDescriptionFiles($realpath) {
|
|
|
|
$filename = $this->getFilenameWithoutExtension($realpath);
|
|
|
|
$pathinfo = pathinfo($realpath);
|
|
|
|
$tmp = explode('_', $filename);
|
|
|
|
$field = array_pop($tmp); //['title', 'snapshot']
|
|
|
|
$content = @file_get_contents($realpath);
|
|
|
|
if (!empty($content)) {
|
|
|
|
$content = trim($content);
|
|
|
|
}
|
|
|
|
|
|
|
|
$data = array();
|
|
|
|
|
|
|
|
//如果是快照图片,获取图片相对路径
|
|
|
|
if (!empty($content) && $field == 'snapshot') {
|
|
|
|
$img_realpath = realpath("{$pathinfo['dirname']}/{$content}");
|
|
|
|
if (file_exists($img_realpath)) {
|
|
|
|
$id = $this->getId($img_realpath);
|
|
|
|
$fp = fopen($img_realpath, 'r');
|
|
|
|
$fstat = fstat($fp);
|
|
|
|
fclose($fp);
|
|
|
|
$img_filename = $this->getFilenameWithoutExtension($img_realpath);
|
|
|
|
$img_pathinfo = pathinfo($img_realpath);
|
|
|
|
$extension = strtolower($img_pathinfo['extension']);
|
|
|
|
$content = $this->getFilePath( $id, $this->getRelativeDirname($img_pathinfo['dirname']), $img_filename, $extension, $fstat['mtime'] );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//返回文件数据
|
|
|
|
$data[$field] = $content;
|
|
|
|
|
|
|
|
return $data;
|
|
|
|
}
|
|
|
|
|
|
|
|
//解析快捷方式文件内容
|
|
|
|
private function parseShortCuts($realpath, $filename) {
|
|
|
|
$content = @file_get_contents($realpath);
|
|
|
|
if (empty($content) || !preg_match('/\[InternetShortcut\]/i', $content)) {return false;}
|
|
|
|
$content = trim($content);
|
|
|
|
|
|
|
|
preg_match('/URL=(\S+)/i', $content, $matches);
|
|
|
|
if (empty($matches) || empty($matches[1])) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return [
|
|
|
|
'name' => $filename,
|
|
|
|
'url' => $matches[1],
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据文件路径生成唯一编号
|
|
|
|
public function getId($realpath) {
|
|
|
|
return !empty($realpath) ? md5($realpath) : '';
|
|
|
|
}
|
|
|
|
|
|
|
|
//判断Nginx防盗链MD5加密方式字符串是否合格
|
|
|
|
private function isNginxSecureLinkMd5PatternValid($pattern) {
|
|
|
|
$valid = true;
|
|
|
|
|
|
|
|
$fieldsNeeded = array(
|
|
|
|
'{secure_link_expires}',
|
|
|
|
'{uri}',
|
|
|
|
'{remote_addr}',
|
|
|
|
'{secret}',
|
|
|
|
);
|
|
|
|
foreach($fieldsNeeded as $needle) {
|
|
|
|
if (strstr($pattern, $needle) === false) {
|
|
|
|
$valid = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $valid;
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取路径中的最后一个目录名,支持中文
|
|
|
|
private function basename($realpath) {
|
|
|
|
$realpath = preg_replace('/\/$/', '', $realpath);
|
|
|
|
$arr = explode('/', $realpath);
|
|
|
|
if (count($arr) < 2) {return $realpath;}
|
|
|
|
|
|
|
|
return array_pop($arr);
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取文件名,不含文件后缀
|
|
|
|
private function getFilenameWithoutExtension($realpath) {
|
|
|
|
return preg_replace('/\.[^\.]+$/i', '', $this->basename($realpath));
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据路径生成目录数组
|
|
|
|
private function getDirData($realpath, $files) {
|
|
|
|
$id = $this->getId($realpath);
|
|
|
|
$data = array(
|
|
|
|
'id' => $id,
|
|
|
|
'directory' => $this->basename($realpath),
|
|
|
|
'realpath' => $this->isApi ? $this->getRelativeDirname($realpath) : $realpath,
|
|
|
|
'path' => $this->getDirPath($id),
|
|
|
|
);
|
|
|
|
|
|
|
|
$sub_dirs = array();
|
|
|
|
$sub_files = array();
|
|
|
|
|
|
|
|
//try to merge description data
|
|
|
|
if (!empty($files[$id])) {
|
|
|
|
$data = array_merge($data, $files[$id]);
|
|
|
|
unset($files[$id]);
|
|
|
|
}
|
|
|
|
|
|
|
|
//区分目录和文件
|
|
|
|
foreach ($files as $id => $item) {
|
|
|
|
if (!empty($item['directory'])) {
|
|
|
|
$sub_dirs[$id] = $item;
|
|
|
|
}else {
|
|
|
|
$sub_files[$id] = $item;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($sub_dirs)) {
|
|
|
|
$data['directories'] = $sub_dirs;
|
|
|
|
}
|
|
|
|
if (!empty($sub_files)) {
|
|
|
|
$data['files'] = $sub_files;
|
|
|
|
}
|
|
|
|
|
|
|
|
return $data;
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据路径生成文件数组,兼容URL文件
|
|
|
|
private function getFileData($realpath) {
|
|
|
|
$id = $this->getId($realpath);
|
|
|
|
$fp = fopen($realpath, 'r');
|
|
|
|
$fstat = fstat($fp);
|
|
|
|
fclose($fp);
|
|
|
|
$pathinfo = pathinfo($realpath);
|
|
|
|
$extension = strtolower($pathinfo['extension']);
|
|
|
|
$filename = $this->getFilenameWithoutExtension($realpath);
|
|
|
|
$data = array(
|
|
|
|
'id' => $id,
|
|
|
|
'filename' => $filename,
|
|
|
|
'extension' => $extension,
|
|
|
|
'fstat' => array(
|
|
|
|
'size' => $fstat['size'],
|
|
|
|
'atime' => $fstat['atime'],
|
|
|
|
'mtime' => $fstat['mtime'],
|
|
|
|
'ctime' => $fstat['ctime'],
|
|
|
|
),
|
|
|
|
'realpath' => $this->isApi ? $this->getRelativeDirname($realpath) : $realpath,
|
|
|
|
'path' => $this->getFilePath( $id, $this->getRelativeDirname($pathinfo['dirname']), $filename, $extension, $fstat['mtime'] ),
|
|
|
|
);
|
|
|
|
|
|
|
|
if ($extension == 'url') {
|
|
|
|
$data['shortcut'] = $this->parseShortCuts($realpath, $filename);
|
|
|
|
}
|
|
|
|
|
|
|
|
return $data;
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据路径和根目录获取当前扫描的目录深度
|
|
|
|
private function getScanningLevel($rootDir, $dir) {
|
|
|
|
$level = 0;
|
|
|
|
|
|
|
|
$rootDir = realpath($rootDir);
|
|
|
|
$dir = realpath($dir);
|
|
|
|
|
|
|
|
if ($dir == $rootDir) {
|
|
|
|
$level = 1;
|
|
|
|
}else {
|
|
|
|
$dirs = explode('/', str_replace($rootDir, '', $dir));
|
|
|
|
$level = count($dirs);
|
|
|
|
}
|
|
|
|
|
|
|
|
return $level;
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据路径和当前扫描深度获取目录名
|
|
|
|
private function getRelativeDirname($dirname) {
|
|
|
|
return str_replace($this->rootDir, '', $dirname);
|
|
|
|
}
|
|
|
|
|
|
|
|
//合并描述文件内容到md文件或者目录数据
|
|
|
|
//增加视频文件:mp4, m3u8描述文件支持
|
|
|
|
//增加.url文件支持
|
|
|
|
private function mergeDescriptionData($realpath) {
|
|
|
|
$data = array();
|
|
|
|
$ext = $this->parseDescriptionFiles($realpath);
|
|
|
|
|
|
|
|
//try to find the md file
|
|
|
|
$targetFile = '';
|
|
|
|
$targetFile_md = preg_replace('/_?[a-z0-9]+\.txt$/U', '.md', $realpath);
|
|
|
|
$targetFile_mp4 = preg_replace('/_?[a-z0-9]+\.txt$/U', '.mp4', $realpath);
|
|
|
|
$targetFile_m3u8 = preg_replace('/_?[a-z0-9]+\.txt$/U', '.m3u8', $realpath);
|
|
|
|
$targetFile_url = preg_replace('/_?[a-z0-9]+\.txt$/U', '.url', $realpath);
|
|
|
|
if (file_exists($targetFile_md)) {
|
|
|
|
$targetFile = $targetFile_md;
|
|
|
|
}else if (file_exists($targetFile_mp4)) {
|
|
|
|
$targetFile = $targetFile_mp4;
|
|
|
|
}else if (file_exists($targetFile_m3u8)) {
|
|
|
|
$targetFile = $targetFile_m3u8;
|
|
|
|
}else if (file_exists($targetFile_url)) {
|
|
|
|
$targetFile = $targetFile_url;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($targetFile) && $targetFile != $realpath) {
|
|
|
|
$fileId = $this->getId($targetFile);
|
|
|
|
if (empty($this->scanResults[$fileId])) {
|
|
|
|
$ext['id'] = $fileId;
|
|
|
|
$this->scanResults[$fileId] = $ext;
|
|
|
|
$data = $ext;
|
|
|
|
}else {
|
|
|
|
$data = $this->scanResults[$fileId];
|
|
|
|
$data = array_merge($data, $ext);
|
|
|
|
$this->scanResults[$fileId] = $data;
|
|
|
|
}
|
|
|
|
}else {
|
|
|
|
//try to merge to the parent directory
|
|
|
|
$targetDir = preg_replace('/\/[a-z0-9]+\.txt$/U', '', $realpath);
|
|
|
|
if (is_dir($targetDir)) {
|
|
|
|
$dirId = $this->getId($targetDir);
|
|
|
|
if (empty($this->scanResults[$dirId])) {
|
|
|
|
$ext['id'] = $dirId;
|
|
|
|
$this->scanResults[$dirId] = $ext;
|
|
|
|
$data = $ext;
|
|
|
|
}else {
|
|
|
|
$data = $this->scanResults[$dirId];
|
|
|
|
$data = array_merge($data, $ext);
|
|
|
|
$this->scanResults[$dirId] = $data;
|
|
|
|
}
|
|
|
|
}else {
|
|
|
|
//keep it in files
|
|
|
|
$fileId = $this->getId($realpath);
|
|
|
|
if (empty($this->scanResults[$fileId])) {
|
|
|
|
$ext['id'] = $fileId;
|
|
|
|
$this->scanResults[$fileId] = $ext;
|
|
|
|
$data = $ext;
|
|
|
|
}else {
|
|
|
|
$data = $this->scanResults[$fileId];
|
|
|
|
$data = array_merge($data, $ext);
|
|
|
|
$this->scanResults[$fileId] = $data;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $data;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
//根据文件生成防盗链网址
|
|
|
|
//参考:https://nginx.org/en/docs/http/ngx_http_secure_link_module.html#secure_link
|
|
|
|
//防盗链参数名:md5, expires
|
|
|
|
protected function getSecureLink($path) {
|
|
|
|
$expires = time() + $this->nginxSecureTimeout;
|
|
|
|
$originStr = str_replace([
|
|
|
|
'{secure_link_expires}',
|
|
|
|
'{uri}',
|
|
|
|
'{remote_addr}',
|
|
|
|
'{secret}',
|
|
|
|
], [
|
|
|
|
$expires,
|
|
|
|
$path,
|
|
|
|
$this->userIp,
|
|
|
|
$this->nginxSecret,
|
|
|
|
], $this->nginxSecureLinkMd5Pattern);
|
|
|
|
|
|
|
|
$md5 = base64_encode( md5($originStr, true) );
|
|
|
|
$md5 = strtr($md5, '+/', '-_');
|
|
|
|
$md5 = str_replace('=', '', $md5);
|
|
|
|
|
|
|
|
return "{$path}?md5={$md5}&expires={$expires}";
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据文件生成相对路径
|
|
|
|
//@ver: 增加静态文件的更新时间戳作为文件版本参数
|
|
|
|
protected function getFilePath($id, $directory, $filename, $extension, $mtime) {
|
|
|
|
if (empty($directory)) {
|
|
|
|
$directory = '/';
|
|
|
|
}
|
|
|
|
if (!preg_match('/\/$/', $directory)) {
|
|
|
|
$directory .= '/';
|
|
|
|
}
|
|
|
|
if (!preg_match('/^\//', $directory)) {
|
|
|
|
$directory = "/{$directory}";
|
|
|
|
}
|
|
|
|
|
|
|
|
$webRoot = preg_replace('/\/$/', '', $this->webRoot);
|
|
|
|
$extensionPathMap = array( //默认每种文件读取内容最大大小
|
|
|
|
'txt' => '',
|
|
|
|
'md' => '/view/',
|
|
|
|
'url' => '/link/',
|
|
|
|
'm3u8' => '/m3u8/',
|
|
|
|
'jpg' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'jpeg' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'png' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'webp' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'gif' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'ico' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'mp3' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'mp4' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
'ts' => "{$webRoot}{$directory}{$filename}.{$extension}",
|
|
|
|
);
|
|
|
|
|
|
|
|
$path = isset($extensionPathMap[$extension]) ? $extensionPathMap[$extension] : '';
|
|
|
|
|
|
|
|
if (!empty($path) && in_array($extension, ['md', 'url', 'm3u8'])) {
|
|
|
|
if ($this->nginxSecureOn && $extension == 'm3u8') {
|
|
|
|
$path = $this->getSecureLink($path);
|
|
|
|
$path = "{$path}&id={$id}";
|
|
|
|
}else {
|
|
|
|
$path = "{$path}?id={$id}";
|
|
|
|
}
|
|
|
|
}else if (!empty($path) && $this->nginxSecureOn) {
|
|
|
|
$path = $this->getSecureLink($path);
|
|
|
|
}
|
|
|
|
|
|
|
|
//增加版本号ver参数
|
|
|
|
if (!in_array($extension, ['md', 'url', 'txt'])) {
|
|
|
|
$path .= strpos($path, '?') !== false ? "&ver={$mtime}" : "?ver={$mtime}";
|
|
|
|
}
|
|
|
|
|
|
|
|
return $path;
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据目录生成相对路径
|
|
|
|
protected function getDirPath($id) {
|
|
|
|
return "/list/?id={$id}";
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
//设置Nginx防盗链开启或关闭,以及密钥、加密方式、超时时长
|
|
|
|
public function setNginxSecure($secureOn, $secret = '', $userIp = '', $pattern = '', $timeout = 0) {
|
|
|
|
$status = false;
|
|
|
|
if (is_string($secureOn) && strtolower($secureOn) == 'on') {
|
|
|
|
$status = true;
|
|
|
|
}else if (is_string($secureOn) && strtolower($secureOn) == 'off') {
|
|
|
|
$status = false;
|
|
|
|
}else if ((bool)$secureOn == true) {
|
|
|
|
$status = true;
|
|
|
|
}
|
|
|
|
$this->nginxSecureOn = $status;
|
|
|
|
|
|
|
|
if (!empty($secret) && is_string($secret)) {
|
|
|
|
$this->nginxSecret = $secret;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($userIp) && is_string($userIp)) {
|
|
|
|
$this->userIp = $userIp;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($pattern) && is_string($pattern)) {
|
|
|
|
if ($this->isNginxSecureLinkMd5PatternValid($pattern) == false) {
|
|
|
|
throw new Exception("Invalid Nginx secure link md5 pattern: {$pattern}", 500);
|
|
|
|
}
|
|
|
|
$this->nginxSecureLinkMd5Pattern = $pattern;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((int)$timeout > 0) {
|
|
|
|
$this->nginxSecureTimeout = (int)$timeout;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置Nginx防盗链密钥
|
|
|
|
public function setNginxSecret($secret) {
|
|
|
|
if (!empty($secret) && is_string($secret)) {
|
|
|
|
$this->nginxSecret = $secret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取Nginx防盗链密钥
|
|
|
|
public function getNginxSecret() {
|
|
|
|
return $this->nginxSecret;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置Nginx防盗链密钥
|
|
|
|
public function setUserIp($userIp) {
|
|
|
|
if (!empty($userIp) && is_string($userIp)) {
|
|
|
|
$this->userIp = $userIp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取Nginx防盗链密钥
|
|
|
|
public function getUserIp() {
|
|
|
|
return $this->userIp;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置Nginx防盗链MD5加密方式
|
|
|
|
/**
|
|
|
|
* Nginx防盗链MD5加密方式参考下面网址中的示例,
|
|
|
|
* 将Nginx的变量替换$符号为英文大括号;
|
|
|
|
*
|
|
|
|
* 示例:
|
|
|
|
* ```
|
|
|
|
* {secure_link_expires}{uri}{remote_addr} {secret}
|
|
|
|
* ```
|
|
|
|
* Nginx文档参考:http://nginx.org/en/docs/http/ngx_http_secure_link_module.html#secure_link_md5
|
|
|
|
*/
|
|
|
|
public function setNginxSecureLinkMd5Pattern($pattern) {
|
|
|
|
if (!empty($pattern) && is_string($pattern)) {
|
|
|
|
if ($this->isNginxSecureLinkMd5PatternValid($pattern) == false) {
|
|
|
|
throw new Exception("Invalid Nginx secure link md5 pattern: {$pattern}", 500);
|
|
|
|
}
|
|
|
|
$this->nginxSecureLinkMd5Pattern = $pattern;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取Nginx防盗链MD5加密方式
|
|
|
|
public function getNginxSecureLinkMd5Pattern() {
|
|
|
|
return $this->nginxSecureLinkMd5Pattern;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置Nginx防盗链超时时长,单位:秒
|
|
|
|
public function setNginxSecureTimeout($timeout) {
|
|
|
|
if ((int)$timeout > 0) {
|
|
|
|
$this->nginxSecureTimeout = (int)$timeout;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取Nginx防盗链超时时长,单位:秒
|
|
|
|
public function getNginxSecureTimeout() {
|
|
|
|
return $this->nginxSecureTimeout;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置网站静态文件相对根目录
|
|
|
|
public function setWebRoot($webRoot) {
|
|
|
|
if (!empty($webRoot) && !preg_match('/^\//', $webRoot)) {
|
|
|
|
$webRoot = "/{$webRoot}";
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($webRoot) && !preg_match('/\/$/', $webRoot)) {
|
|
|
|
$webRoot = "{$webRoot}/";
|
|
|
|
}
|
|
|
|
|
|
|
|
$this->webRoot = $webRoot;
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取网站静态文件相对根目录
|
|
|
|
public function getWebRoot() {
|
|
|
|
return $this->webRoot;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置扫描绝对根目录
|
|
|
|
public function setRootDir($dir) {
|
|
|
|
$this->rootDir = realpath($dir);
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取是否开启防盗链
|
|
|
|
public function isSecureOn() {
|
|
|
|
return $this->nginxSecureOn;
|
|
|
|
}
|
|
|
|
|
|
|
|
//扫描目录获取目录和文件列表,支持指定目录扫描深度(目录级数)
|
|
|
|
public function scan($dir, $levels = 3) {
|
|
|
|
if (empty($this->scanStartTime)) {
|
|
|
|
$this->scanStartTime = microtime(true);
|
|
|
|
}
|
|
|
|
|
|
|
|
$tree = array();
|
|
|
|
|
|
|
|
$ignore_files = array('.', '..');
|
|
|
|
if (is_dir($dir)) {
|
|
|
|
if (!preg_match('/\/$/', $dir)) {$dir .= '/';}
|
|
|
|
if (empty($this->rootDir)) {
|
|
|
|
$this->rootDir = realpath($dir);
|
|
|
|
}
|
|
|
|
$this->scanningDirLevel = $this->getScanningLevel($this->rootDir, $dir);
|
|
|
|
$nextLevels = $levels - $this->scanningDirLevel;
|
|
|
|
|
|
|
|
$files = scandir($dir);
|
|
|
|
$counter = 0;
|
|
|
|
foreach($files as $file) {
|
|
|
|
if (in_array($file, $ignore_files) || !$this->isValid($file)) {continue;}
|
|
|
|
|
|
|
|
$branch = array();
|
|
|
|
$realpath = realpath("{$dir}{$file}");
|
|
|
|
if (is_dir($realpath)) {
|
|
|
|
$files = array();
|
|
|
|
if ($nextLevels >= 0) {
|
|
|
|
$files = $this->scan($realpath, $levels);
|
|
|
|
if (!empty($files)) {
|
|
|
|
foreach($files as $file) {
|
|
|
|
$this->scanResults[$file['id']] = $file;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
$branch = $this->getDirData($realpath, $files);
|
|
|
|
|
|
|
|
//add parent directory's id
|
|
|
|
$pid = $this->getId(realpath($dir));
|
|
|
|
if (!empty($pid)) {
|
|
|
|
$branch = array_merge(['pid' => $pid], $branch);
|
|
|
|
}
|
|
|
|
}else {
|
|
|
|
if ($counter >= $this->maxFileNumber) {break;} //最大文件数量限制
|
|
|
|
|
|
|
|
$pathinfo = pathinfo($realpath);
|
|
|
|
$extension = strtolower($pathinfo['extension']);
|
|
|
|
if ( in_array($extension, $this->supportFileExtensions) ) {
|
|
|
|
if ($extension != 'txt') {
|
|
|
|
$branch = $this->getFileData($realpath);
|
|
|
|
|
|
|
|
//add parent directory's id
|
|
|
|
$pid = $this->getId(realpath($dir));
|
|
|
|
if (!empty($pid)) {
|
|
|
|
$branch = array_merge(['pid' => $pid], $branch);
|
|
|
|
}
|
|
|
|
}else {
|
|
|
|
//把描述文件内容合并到被描述的目录或md文件数据中
|
|
|
|
$branch = $this->mergeDescriptionData($realpath);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($branch)) {
|
|
|
|
$this->scanResults[$branch['id']] = $branch;
|
|
|
|
$tree[$branch['id']] = $branch;
|
|
|
|
}
|
|
|
|
|
|
|
|
$counter ++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
$this->tree = $tree;
|
|
|
|
|
|
|
|
$time = microtime(true);
|
|
|
|
$this->scanTimeCost = $this->scanStartTime > 0 ? ceil( ($time - $this->scanStartTime)*1000 ) : 0;
|
|
|
|
|
|
|
|
return $tree;
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取扫描结果
|
|
|
|
public function getScanResults() {
|
|
|
|
return $this->scanResults;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置扫描结果,以支持缓存
|
|
|
|
public function setScanResults($data) {
|
|
|
|
$this->scanResults = $data;
|
|
|
|
}
|
|
|
|
|
|
|
|
//设置tree,以支持缓存
|
|
|
|
public function setTreeData($data) {
|
|
|
|
$this->tree = $data;
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取菜单,扫描结果中的目录结构
|
|
|
|
public function getMenus($tree = array()) {
|
|
|
|
$results = empty($tree) ? $this->tree : $tree;
|
|
|
|
$menus = array();
|
|
|
|
if (empty($results)) {return $menus;}
|
|
|
|
|
|
|
|
foreach ($results as $id => $item) {
|
|
|
|
$dir = array();
|
|
|
|
if (!empty($item['directory'])) {
|
|
|
|
$dir = array(
|
|
|
|
'id' => $item['id'],
|
|
|
|
'directory' => $item['directory'],
|
|
|
|
'path' => $item['path'],
|
|
|
|
);
|
|
|
|
if (!empty($item['snapshot'])) {
|
|
|
|
$dir['snapshot'] = $item['snapshot'];
|
|
|
|
}
|
|
|
|
if (!empty($item['title'])) {
|
|
|
|
$dir['title'] = $item['title'];
|
|
|
|
}
|
|
|
|
if (!empty($item['description'])) {
|
|
|
|
$dir['description'] = $item['description'];
|
|
|
|
}
|
|
|
|
if (!empty($item['pid'])) {
|
|
|
|
$dir['pid'] = $item['pid'];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($item['directories'])) {
|
|
|
|
$dirs = $this->getMenus($item['directories']);
|
|
|
|
if (!empty($dirs)) {
|
|
|
|
$dir['directories'] = $dirs;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($dir)) {
|
|
|
|
$menus[] = $dir;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $menus;
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取.md文件中的h1-h6标题
|
|
|
|
public function getMDTitles($id) {
|
|
|
|
if (empty($this->scanResults[$id])) {return [];}
|
|
|
|
$file = $this->scanResults[$id];
|
|
|
|
|
|
|
|
$content = @file_get_contents($file['realpath']);
|
|
|
|
if (empty($content)) {return [];}
|
|
|
|
$content = trim($content);
|
|
|
|
|
|
|
|
# standardize line breaks
|
|
|
|
$content = str_replace(array("\r\n", "\r"), "\n", $content);
|
|
|
|
# remove surrounding line breaks
|
|
|
|
$content = trim($content, "\n");
|
|
|
|
# split text into lines
|
|
|
|
$lines = explode("\n", $content);
|
|
|
|
|
|
|
|
$titles = array();
|
|
|
|
if (!empty($lines)) {
|
|
|
|
foreach($lines as $line) {
|
|
|
|
preg_match_all('/^#(.+)/u', $line, $matches);
|
|
|
|
if (!empty($matches[1])) {
|
|
|
|
foreach($matches[1] as $title) {
|
|
|
|
$num = substr_count($title, '#');
|
|
|
|
$titles[] = array(
|
|
|
|
'name' => trim(str_replace('#', '', $title)),
|
|
|
|
'heading' => 'h' . ($num+1),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $titles;
|
|
|
|
}
|
|
|
|
|
|
|
|
//替换.md文件解析之后的HTML中的静态文件URL为相对路径path
|
|
|
|
public function fixMDUrls($realpath, $html) {
|
|
|
|
$pathinfo = pathinfo($realpath);
|
|
|
|
|
|
|
|
$matches = array();
|
|
|
|
|
|
|
|
//匹配图片地址
|
|
|
|
$reg_imgs = '/src="([^"]+)"/i';
|
|
|
|
preg_match_all($reg_imgs, $html, $img_matches);
|
|
|
|
if (!empty($img_matches[1])) {
|
|
|
|
$matches = $img_matches[1];
|
|
|
|
}
|
|
|
|
|
|
|
|
//匹配a链接
|
|
|
|
$reg_links = '/href="([^"]+)"/i';
|
|
|
|
preg_match_all($reg_links, $html, $link_matches);
|
|
|
|
if (!empty($link_matches[1])) {
|
|
|
|
$matches = array_merge($matches, $link_matches[1]);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!empty($matches)) {
|
|
|
|
foreach ($matches as $url) {
|
|
|
|
if (preg_match('/^http(s)?:\/\//i', $url) || preg_match('/^\//i', $url)) {continue;}
|
|
|
|
|
|
|
|
$src_realpath = realpath("{$pathinfo['dirname']}/{$url}");
|
|
|
|
if (file_exists($src_realpath)) {
|
|
|
|
$id = $this->getId($src_realpath);
|
|
|
|
$fp = fopen($src_realpath, 'r');
|
|
|
|
$fstat = fstat($fp);
|
|
|
|
fclose($fp);
|
|
|
|
$src_filename = $this->getFilenameWithoutExtension($src_realpath);
|
|
|
|
$src_pathinfo = pathinfo($src_realpath);
|
|
|
|
$extension = strtolower($src_pathinfo['extension']);
|
|
|
|
|
|
|
|
$src_path = $this->getFilePath( $id, $this->getRelativeDirname($src_pathinfo['dirname']), $src_filename, $extension, $fstat['mtime'] );
|
|
|
|
|
|
|
|
$html = str_replace("\"{$url}\"", "\"{$src_path}\"", $html);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $html;
|
|
|
|
}
|
|
|
|
|
|
|
|
//获取指定目录下名称为README.md的文件
|
|
|
|
public function getDefaultReadme($dirid = '') {
|
|
|
|
return $this->getDefaultFile('md', $dirid);
|
|
|
|
}
|
|
|
|
|
|
|
|
//根据扩展名在根目录下或者指定目录下获取一个文件
|
|
|
|
public function getDefaultFile($extension, $dirid = '') {
|
|
|
|
$readme = null;
|
|
|
|
$md = null;
|
|
|
|
|
|
|
|
if (empty($dirid) && !empty($this->tree)) {
|
|
|
|
foreach($this->tree as $id => $file) {
|
|
|
|
if (!empty($file['extension']) && $file['extension'] == $extension) {
|
|
|
|
$md = $file;
|
|
|
|
if ($extension == 'md' && strtoupper($file['filename']) == 'README') {
|
|
|
|
$readme = $file;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}else if (!empty($dirid) && !empty($this->scanResults)) {
|
|
|
|
$directory = $this->scanResults[$dirid];
|
|
|
|
if (!empty($directory) && !empty($directory['files'])) {
|
|
|
|
foreach($directory['files'] as $id => $file) {
|
|
|
|
if (!empty($file['extension']) && $file['extension'] == $extension) {
|
|
|
|
if (empty($md)) {$md = $file;} //取第一个md文件
|
|
|
|
if ($extension == 'md' && strtoupper($file['filename']) == 'README') {
|
|
|
|
$readme = $file;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (empty($readme) && !empty($md)) {
|
|
|
|
$readme = $md;
|
|
|
|
}
|
|
|
|
|
|
|
|
return $readme;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|