utils.js 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. const fs = require('fs-extra');
  2. const path = require('path');
  3. const crypto = require('crypto');
  4. function sleep(ms) {
  5. return new Promise(resolve => setTimeout(resolve, ms));
  6. }
  7. function versionText(config) {
  8. return `${config.name} v${config.version}, Node.js ${process.version}`;
  9. }
  10. async function findFiles(callback, dir, recursive = true) {
  11. if (!(callback && dir))
  12. return;
  13. const files = await fs.readdir(dir, { withFileTypes: true });
  14. for (const file of files) {
  15. const found = path.resolve(dir, file.name);
  16. if (file.isDirectory()) {
  17. if (recursive)
  18. await findFiles(callback, found);
  19. } else {
  20. await callback(found);
  21. }
  22. }
  23. }
  24. async function touchFile(filename) {
  25. await fs.utimes(filename, Date.now()/1000, Date.now()/1000);
  26. }
  27. function hasProp(obj, prop) {
  28. return Object.prototype.hasOwnProperty.call(obj, prop);
  29. }
  30. function freeMemory() {
  31. if (global.gc) {
  32. global.gc();
  33. global.gc();
  34. global.gc();
  35. }
  36. }
  37. function getFileHash(filename, hashName, enc) {
  38. return new Promise((resolve, reject) => {
  39. const hash = crypto.createHash(hashName);
  40. const rs = fs.createReadStream(filename);
  41. rs.on('error', reject);
  42. rs.on('data', chunk => hash.update(chunk));
  43. rs.on('end', () => resolve(hash.digest(enc)));
  44. });
  45. }
  46. function getBufHash(buf, hashName, enc) {
  47. const hash = crypto.createHash(hashName);
  48. hash.update(buf);
  49. return hash.digest(enc);
  50. }
  51. module.exports = {
  52. sleep,
  53. versionText,
  54. findFiles,
  55. touchFile,
  56. hasProp,
  57. freeMemory,
  58. getFileHash,
  59. getBufHash,
  60. };