utils.js 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  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. function intersectSet(arrSet) {
  52. if (!arrSet.length)
  53. return new Set();
  54. let min = 0;
  55. let size = arrSet[0].size;
  56. for (let i = 1; i < arrSet.length; i++) {
  57. if (arrSet[i].size < size) {
  58. min = i;
  59. size = arrSet[i].size;
  60. }
  61. }
  62. const result = new Set();
  63. for (const elem of arrSet[min]) {
  64. let inAll = true;
  65. for (let i = 0; i < arrSet.length; i++) {
  66. if (i === min)
  67. continue;
  68. if (!arrSet[i].has(elem)) {
  69. inAll = false;
  70. break;
  71. }
  72. }
  73. if (inAll)
  74. result.add(elem);
  75. }
  76. return result;
  77. }
  78. module.exports = {
  79. sleep,
  80. versionText,
  81. findFiles,
  82. touchFile,
  83. hasProp,
  84. freeMemory,
  85. getFileHash,
  86. getBufHash,
  87. intersectSet,
  88. };