wehub-resource-sync
69 lines
1.8 KiB
TypeScript
69 lines
1.8 KiB
TypeScript
import fs from 'fs';
|
|
import path from 'path';
|
|
|
|
const FILE_SIZE_LIMIT = parseInt(process.env.FILE_SIZE_LIMIT || '100', 10) * 1024 * 1024; // MB to bytes
|
|
const DIRECTORY_SIZE_LIMIT = parseInt(process.env.DIRECTORY_SIZE_LIMIT || '500', 10) * 1024 * 1024; // MB to bytes
|
|
|
|
interface FileSizeCheck {
|
|
size: number;
|
|
exceeds: boolean;
|
|
limit: number;
|
|
}
|
|
|
|
interface DirectorySizeCheck {
|
|
size: number;
|
|
limit: number;
|
|
exceeds: boolean;
|
|
}
|
|
|
|
function checkFileSizeLimit(filePath: string): FileSizeCheck {
|
|
const stats = fs.statSync(filePath);
|
|
return {
|
|
size: stats.size,
|
|
limit: FILE_SIZE_LIMIT,
|
|
exceeds: stats.size > FILE_SIZE_LIMIT
|
|
};
|
|
}
|
|
|
|
function checkDirectorySizeLimit(directoryPath: string): DirectorySizeCheck {
|
|
const totalSize = calculateDirectorySize(directoryPath);
|
|
return {
|
|
size: totalSize,
|
|
limit: DIRECTORY_SIZE_LIMIT,
|
|
exceeds: totalSize > DIRECTORY_SIZE_LIMIT
|
|
};
|
|
}
|
|
|
|
function calculateDirectorySize(directoryPath: string): number {
|
|
let totalSize = 0;
|
|
const files = fs.readdirSync(directoryPath);
|
|
|
|
for (const file of files) {
|
|
const filePath = path.join(directoryPath, file);
|
|
const stats = fs.statSync(filePath);
|
|
|
|
if (stats.isFile()) {
|
|
totalSize += stats.size;
|
|
} else if (stats.isDirectory()) {
|
|
totalSize += calculateDirectorySize(filePath);
|
|
}
|
|
}
|
|
|
|
return totalSize;
|
|
}
|
|
|
|
function formatSize(bytes: number): string {
|
|
if (bytes < 1024) return bytes + ' bytes';
|
|
else if (bytes < 1024 * 1024) return (bytes / 1024).toFixed(2) + ' KB';
|
|
else if (bytes < 1024 * 1024 * 1024) return (bytes / (1024 * 1024)).toFixed(2) + ' MB';
|
|
else return (bytes / (1024 * 1024 * 1024)).toFixed(2) + ' GB';
|
|
}
|
|
|
|
export {
|
|
checkFileSizeLimit,
|
|
checkDirectorySizeLimit,
|
|
calculateDirectorySize,
|
|
formatSize,
|
|
FILE_SIZE_LIMIT,
|
|
DIRECTORY_SIZE_LIMIT
|
|
};
|