2024-03-02 19:16:05 +08:00
|
|
|
<?php
|
|
|
|
|
|
|
|
namespace app\utils;
|
|
|
|
|
2024-03-10 13:20:32 +08:00
|
|
|
use app\models\User;
|
2024-03-02 19:16:05 +08:00
|
|
|
use RecursiveDirectoryIterator;
|
|
|
|
use RecursiveIteratorIterator;
|
2024-03-10 13:20:32 +08:00
|
|
|
use Yii;
|
2024-03-02 19:16:05 +08:00
|
|
|
|
|
|
|
class FileSizeHelper
|
|
|
|
{
|
|
|
|
/**
|
|
|
|
* 计算指定目录的大小
|
|
|
|
*
|
|
|
|
* @param string $directory 目录路径
|
|
|
|
* @return int 目录的大小(字节)
|
|
|
|
*/
|
|
|
|
public static function getDirectorySize(string $directory): int
|
|
|
|
{
|
2024-03-19 15:03:33 +08:00
|
|
|
if (!is_dir($directory)) {
|
|
|
|
return 0;
|
|
|
|
}
|
2024-03-02 19:16:05 +08:00
|
|
|
$size = 0;
|
|
|
|
$files = new RecursiveIteratorIterator(new RecursiveDirectoryIterator($directory));
|
|
|
|
foreach ($files as $file) {
|
|
|
|
if ($file->isFile()) {
|
|
|
|
$size += $file->getSize();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return $size;
|
|
|
|
}
|
|
|
|
|
2024-03-10 13:20:32 +08:00
|
|
|
/**
|
|
|
|
* 判断用户home是否有足够的容量存放文件
|
|
|
|
* @param int $file_size 文件大小B(可选,如果文件已经添加到网盘时,不需要这个参数)
|
|
|
|
* @return bool
|
|
|
|
*/
|
|
|
|
public static function hasEnoughSpace(int $file_size = 0, int $user_id = null): bool
|
|
|
|
{
|
|
|
|
if ($user_id === null) {
|
|
|
|
$user_id = Yii::$app->user->id;
|
|
|
|
}
|
2024-03-10 16:35:05 +08:00
|
|
|
$userHomeDirSize = self::getUserHomeDirSize($user_id);
|
|
|
|
$userHomeDirSize_MB = $userHomeDirSize / 1024 / 1024;
|
|
|
|
$userVaultDirSize = self::getUserVaultDirSize($user_id);
|
|
|
|
$userVaultDirSize_MB = $userVaultDirSize / 1024 / 1024;
|
|
|
|
|
|
|
|
$file_size_MB = $file_size / 1024 / 1024; // 即将新增的文件
|
|
|
|
|
2024-03-10 13:20:32 +08:00
|
|
|
$user = User::findOne($user_id);
|
|
|
|
$limit = $user->storage_limit;
|
|
|
|
if ($limit == -1) {
|
|
|
|
return true;
|
|
|
|
}
|
2024-03-10 16:35:05 +08:00
|
|
|
|
|
|
|
return $userHomeDirSize_MB + $userVaultDirSize_MB + $file_size_MB <= $limit;
|
|
|
|
}
|
|
|
|
|
|
|
|
public static function getUserHomeDirSize(int $user_id = null): int
|
|
|
|
{
|
|
|
|
if ($user_id === null) {
|
|
|
|
$user_id = Yii::$app->user->id;
|
|
|
|
}
|
|
|
|
$userHomeDir = Yii::getAlias(Yii::$app->params['dataDirectory']) . '/' . $user_id;
|
|
|
|
return self::getDirectorySize($userHomeDir);
|
|
|
|
}
|
|
|
|
|
|
|
|
public static function getUserVaultDirSize(int $user_id = null): int
|
|
|
|
{
|
|
|
|
if ($user_id === null) {
|
|
|
|
$user_id = Yii::$app->user->id;
|
|
|
|
}
|
|
|
|
$userHomeDir = Yii::getAlias(Yii::$app->params['dataDirectory']) . '/' . $user_id . '.secret';
|
|
|
|
return self::getDirectorySize($userHomeDir);
|
2024-03-10 13:20:32 +08:00
|
|
|
}
|
2024-03-22 14:48:17 +08:00
|
|
|
public static function getUserAllDirSize(int $user_id = null): int
|
|
|
|
{
|
|
|
|
if ($user_id === null) {
|
|
|
|
$user_id = Yii::$app->user->id;
|
|
|
|
}
|
|
|
|
return self::getUserHomeDirSize($user_id) + self::getUserVaultDirSize($user_id);
|
|
|
|
}
|
|
|
|
public static function getFormatUserAllDirSize(int $user_id = null): string
|
|
|
|
{
|
|
|
|
if ($user_id === null) {
|
|
|
|
$user_id = Yii::$app->user->id;
|
|
|
|
}
|
|
|
|
return self::formatBytes(self::getUserAllDirSize($user_id));
|
|
|
|
}
|
2024-03-02 19:16:05 +08:00
|
|
|
/**
|
|
|
|
* @param $bytes
|
|
|
|
* @param $precision
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
public static function formatBytes($bytes, $precision = 2): string
|
|
|
|
{
|
|
|
|
$units = array('B', 'KB', 'MB', 'GB', 'TB');
|
|
|
|
|
|
|
|
$bytes = max($bytes, 0);
|
|
|
|
$pow = floor(($bytes ? log($bytes) : 0) / log(1024));
|
|
|
|
$pow = min($pow, count($units) - 1);
|
|
|
|
|
|
|
|
$bytes /= pow(1024, $pow);
|
|
|
|
|
|
|
|
return round($bytes, $precision) . ' ' . $units[$pow];
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param $megabytes
|
|
|
|
* @param $precision
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
public static function formatMegaBytes($megabytes, $precision = 2): string
|
|
|
|
{
|
2024-03-10 13:20:32 +08:00
|
|
|
if ($megabytes === -1) {
|
2024-03-02 19:16:05 +08:00
|
|
|
return '∞';
|
|
|
|
}
|
|
|
|
$bytes = $megabytes * pow(1024, 2); // Convert megabytes to bytes
|
|
|
|
return self::formatBytes($bytes, $precision);
|
|
|
|
}
|
|
|
|
}
|