Compare commits

..

4 Commits

  1. 6
      lib/DirScanner.php
  2. 10
      plugins/Common.php
  3. 250
      themes/beauty/controller/CommandController.php

6
lib/DirScanner.php

@ -59,12 +59,12 @@ Class DirScanner {
'm3u8', //视频 'm3u8', //视频
); );
protected $exifSupportFileExtensions = array( public $exifSupportFileExtensions = array(
'jpg', //图片 'jpg', //图片
'jpeg', //图片 'jpeg', //图片
); );
protected $iptcSupportFileExtensions = array( public $iptcSupportFileExtensions = array(
'jpg', //图片 'jpg', //图片
'jpeg', //图片 'jpeg', //图片
); );
@ -277,7 +277,7 @@ Class DirScanner {
try { try {
if (function_exists('exif_read_data')) { if (function_exists('exif_read_data')) {
$exif = exif_read_data($realpath, 0, true); $exif = @exif_read_data($realpath, 0, true);
if(!empty($exif['EXIF']['DateTimeOriginal'])) { if(!empty($exif['EXIF']['DateTimeOriginal'])) {
$createTime = strtotime($exif['EXIF']['DateTimeOriginal']); $createTime = strtotime($exif['EXIF']['DateTimeOriginal']);
} }

10
plugins/Common.php

@ -3,6 +3,8 @@
* 常用的公用方法 * 常用的公用方法
*/ */
Class Common { Class Common {
public static $cache = array();
public static function cleanSpecialChars($str) { public static function cleanSpecialChars($str) {
if (empty($str)) {return $str;} if (empty($str)) {return $str;}
@ -783,4 +785,12 @@ Class Common {
return !empty($newArr) ? $newArr : $sorted; return !empty($newArr) ? $newArr : $sorted;
} }
public static function setCache($key, $val) {
self::$cache[$key] = $val;
}
public static function getCache($key) {
return !empty(self::$cache[$key]) ? self::$cache[$key] : null;
}
} }

250
themes/beauty/controller/CommandController.php

@ -2,7 +2,15 @@
/** /**
* Command Controller * Command Controller
*/ */
require_once __DIR__ . '/../../../lib/DirScanner.php';
require_once __DIR__ . '/../../../plugins/Parsedown.php';
require_once __DIR__ . '/../../../plugins/Common.php';
Class CommandController extends Controller { Class CommandController extends Controller {
protected $logPrefix = '[MainBot]';
protected $scanedDirCacheKey = 'MainBotScanedDirs';
protected $dateIndexCacheKey = 'MainBotDateIndex';
protected $noOriginalCtimeFilesCacheKey = 'MainBotNoOriginalCtimeFiles';
public function actionIndex() { public function actionIndex() {
$commands = <<<eof $commands = <<<eof
@ -93,19 +101,249 @@ eof;
} }
//服务器端机器人程序,负责图片、视频文件拍摄时间等信息扫描,并缓存结果供前端使用 //服务器端机器人程序
/**
* 扫描照片目录里所有子目录和文件
* 建立年份、月份索引数据
* 汇总每个目录下的照片、视频、MP3音乐文件数量
* 记录没有original_ctime的文件,把它们单独归类
*/
public function actionMainBot() { public function actionMainBot() {
$thisTime = date('Y-m-d H:i:s'); $thisTime = date('Y-m-d H:i:s');
echo "[MainBot] Main bot started @{$thisTime}\n"; $botLogPrefix = $this->logPrefix;
echo "{$botLogPrefix} Main bot started @{$thisTime}\n";
while (true) { $menus = array(); //菜单,一级目录
$time = date('Y-m-d H:i:s'); $htmlReadme = array(); //Readme.md 内容,底部网站详细介绍
echo "[MainBot] {$time}\n"; $htmlCateReadme = ''; //当前目录下的Readme.md 内容
$menus_sorted = array(); //Readme_sort.txt 说明文件内容,一级目录菜单从上到下的排序
//执行一次扫描任务
$this->cleanScanCaches();
$this->scanMediaFiles();
$this->saveDateIndexIntoCacheFile();
$this->saveNoOriginalCtimeFilesIntoFile();
//while (true) {
//$time = date('Y-m-d H:i:s');
//echo "{$botLogPrefix} {$time}\n";
//sleep(3);
//}
}
//清空内存中的临时缓存数据
protected function cleanScanCaches() {
Common::setCache($this->scanedDirCacheKey, array());
Common::setCache($this->dateIndexCacheKey, array());
Common::setCache($this->noOriginalCtimeFilesCacheKey, array());
}
//扫描媒体文件:图片、视频、音乐
//TODO: 把它们按年份、月份归类,并缓存到/runtime/cache/目录,方便前端展示读取
//把当前扫描进度保存到单独的缓存文件,便于用户随时获取
protected function scanMediaFiles($dirpath = '') {
$rootDir = __DIR__ . '/../../../www/' . FSC::$app['config']['content_directory'];
if (empty($dirpath)) {
$dirpath = $rootDir;
}
echo "\n\n== Scanning directory {$dirpath} ...\n";
$scanner = new DirScanner();
$scanner->setWebRoot(FSC::$app['config']['content_directory']);
$scanner->setRootDir($dirpath);
$maxScanDeep = 0; //最大扫描目录级数
$dirTree = $scanner->scan($dirpath, $maxScanDeep);
$scanResults = $scanner->getScanResults();
echo 'Total directories or files: ' . count($scanResults);
echo "\n";
$supportedImageExts = FSC::$app['config']['supportedImageExts'];
$supportedVideoExts = FSC::$app['config']['supportedVideoExts'];
$supportedAudioExts = FSC::$app['config']['supportedAudioExts'];
$cacheKey = $this->scanedDirCacheKey;
if (!empty($scanResults)) {
$scanIndex = 0;
$scanTotal = count($scanResults);
foreach ($scanResults as $id => $item) {
$hadScanedDirs = Common::getCache($cacheKey);
if (
!empty($item['filename'])
&& empty($item['original_ctime'])
&& in_array($item['extension'], $supportedImageExts)
&& !in_array($item['extension'], $scanner->exifSupportFileExtensions)
) {
echo "Image file no original_ctime: {$item['filename']}.{$item['extension']}, {$item['realpath']}\n";
}else if (
!empty($item['filename'])
&& empty($item['original_ctime'])
&& (in_array($item['extension'], $supportedVideoExts) || in_array($item['extension'], $supportedAudioExts))
) {
echo "Video or audio file no original_ctime: {$item['filename']}.{$item['extension']}, {$item['realpath']}\n";
}else if (!empty($item['directory']) && empty($hadScanedDirs[$id])) { //if it's directory
$hadScanedDirs[$id] = true;
Common::setCache($cacheKey, $hadScanedDirs);
$this->scanMediaFiles($item['realpath']);
}
$scanIndex ++;
if (!empty($item['filename'])) {
//更新年份、月份时间索引
$this->updateDateIndex($item);
//更新没有拍摄时间的文件索引
$this->updateNoOriginalCtimeFiles($item);
}
//更新扫描进度
$stats = $this->updateScanStats($dirpath, $scanTotal, $scanIndex);
}
sleep(1);
}
}
//更新扫描进度
protected function updateScanStats($dirpath, $total, $index) {
if (empty($total)) {return false;}
$stats = array(
'currentDir' => $dirpath,
'total' => $total,
'current' => $index,
'percent' => floor($index*100/$total),
);
$botLogPrefix = $this->logPrefix;
$cacheDir = __DIR__ . '/../../../runtime/cache/';
$statsFile = "{$cacheDir}stats_scan.json";
if (!is_dir($cacheDir)) {
mkdir($cacheDir, 0777, true);
}
$rootDir = __DIR__ . '/../../../www/' . FSC::$app['config']['content_directory'];
if ($dirpath == $rootDir) {
echo "{$botLogPrefix} Scan has finished {$stats['percent']}%, total {$stats['total']}, current {$stats['current']}\n";
//保存进度文件
file_put_contents($statsFile, json_encode($stats) . "\n");
}else if (file_exists($statsFile)) { //更新当前扫描目录
$json = file_get_contents($statsFile);
if (!empty($json)) {
$jsonData = json_decode(trim($json), true);
if ($jsonData['currentDir'] != $dirpath) {
$jsonData['currentDir'] = $dirpath;
$json = json_encode($jsonData) . "\n";
file_put_contents($statsFile, $json);
}
}
}
sleep(3); return $stats;
}
//建立年份、月份时间索引
/**
* 数据格式:
* {"y2024": {"m1": [id1, id2, ...], "m10": [id1, id2, ...]}}
*/
protected function updateDateIndex($file) {
$ctime = !empty($file['original_ctime']) ? $file['original_ctime'] : $file['fstat']['ctime'];
$cacheKey = $this->dateIndexCacheKey;
$cacheData = Common::getCache($cacheKey);
if (empty($cacheData)) {
$cacheData = array();
}
$year = 'y' . date('Y', $ctime);
$month = 'm' . date('m', $ctime);
if (empty($cacheData[$year])) {
$cacheData[$year] = array();
}
if (empty($cacheData[$year][$month])) {
$cacheData[$year][$month] = array();
}
if (in_array($file['id'], $cacheData[$year][$month])) {
return false;
}
array_push($cacheData[$year][$month], $file['id']);
return Common::setCache($cacheKey, $cacheData);
}
protected function saveDateIndexIntoCacheFile() {
$cacheKey = $this->dateIndexCacheKey;
$cacheData = Common::getCache($cacheKey);
if (empty($cacheData)) {
return false;
}
return Common::saveCacheToFile($cacheKey, $cacheData);
}
//汇总每个目录下图片、视频、音乐文件数量
/**
* 数据格式:
* {dirid: {image: 10, video: 20, audio: 0}, ...}
*/
protected function updateDirCounter() {
}
//归类没有original_ctime的图片、视频文件
/**
* 数据格式:
* [id1, id2, ...]
*/
protected function updateNoOriginalCtimeFiles($file) {
if (!empty($file['original_ctime'])) {
return false;
}
$cacheKey = $this->noOriginalCtimeFilesCacheKey;
$cacheData = Common::getCache($cacheKey);
if (empty($cacheData)) {
$cacheData = array();
}
if (in_array($file['id'], $cacheData)) {
return false;
}
array_push($cacheData, $file['id']);
return Common::setCache($cacheKey, $cacheData);
}
protected function saveNoOriginalCtimeFilesIntoFile() {
$cacheKey = $this->noOriginalCtimeFilesCacheKey;
$cacheData = Common::getCache($cacheKey);
if (empty($cacheData)) {
return false;
}
return Common::saveCacheToFile($cacheKey, $cacheData);
}
public function actionTest() {
$cacheKey = 'TestSTData';
$time = Common::getCache($cacheKey);
if (empty($time)) {
$time = date('Y-m-d H:i:s');
Common::setCache($cacheKey, $time);
} }
echo "Cache time {$time}\n";
} }
} }

Loading…
Cancel
Save