utils.js 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234
  1. const fs = require('fs-extra');
  2. const path = require('path');
  3. const zlib = require('zlib');
  4. const crypto = require('crypto');
  5. function sleep(ms) {
  6. return new Promise(resolve => setTimeout(resolve, ms));
  7. }
  8. function processLoop() {
  9. return new Promise(resolve => setImmediate(resolve));
  10. }
  11. function versionText(config) {
  12. return `${config.name} v${config.version}, Node.js ${process.version}, ${process.platform}`;
  13. }
  14. async function findFiles(callback, dir, recursive = true) {
  15. if (!(callback && dir))
  16. return;
  17. const files = await fs.readdir(dir, { withFileTypes: true });
  18. for (const file of files) {
  19. const found = path.resolve(dir, file.name);
  20. if (file.isDirectory()) {
  21. if (recursive)
  22. await findFiles(callback, found);
  23. } else {
  24. await callback(found);
  25. }
  26. }
  27. }
  28. async function touchFile(filename) {
  29. await fs.utimes(filename, Date.now()/1000, Date.now()/1000);
  30. }
  31. function hasProp(obj, prop) {
  32. return Object.prototype.hasOwnProperty.call(obj, prop);
  33. }
  34. function freeMemory() {
  35. if (global.gc) {
  36. global.gc();
  37. }
  38. }
  39. function getFileHash(filename, hashName, enc) {
  40. return new Promise((resolve, reject) => {
  41. const hash = crypto.createHash(hashName);
  42. const rs = fs.createReadStream(filename);
  43. rs.on('error', reject);
  44. rs.on('data', chunk => hash.update(chunk));
  45. rs.on('end', () => resolve(hash.digest(enc)));
  46. });
  47. }
  48. function getBufHash(buf, hashName, enc) {
  49. const hash = crypto.createHash(hashName);
  50. hash.update(buf);
  51. return hash.digest(enc);
  52. }
  53. function intersectSet(arrSet) {
  54. if (!arrSet.length)
  55. return new Set();
  56. let min = 0;
  57. let size = arrSet[0].size;
  58. for (let i = 1; i < arrSet.length; i++) {
  59. if (arrSet[i].size < size) {
  60. min = i;
  61. size = arrSet[i].size;
  62. }
  63. }
  64. const result = new Set();
  65. for (const elem of arrSet[min]) {
  66. let inAll = true;
  67. for (let i = 0; i < arrSet.length; i++) {
  68. if (i === min)
  69. continue;
  70. if (!arrSet[i].has(elem)) {
  71. inAll = false;
  72. break;
  73. }
  74. }
  75. if (inAll)
  76. result.add(elem);
  77. }
  78. return result;
  79. }
  80. function randomHexString(len) {
  81. return crypto.randomBytes(len).toString('hex')
  82. }
  83. //async
  84. function gzipFile(inputFile, outputFile, level = 1) {
  85. return new Promise((resolve, reject) => {
  86. const gzip = zlib.createGzip({level});
  87. const input = fs.createReadStream(inputFile);
  88. const output = fs.createWriteStream(outputFile);
  89. input.on('error', reject)
  90. .pipe(gzip).on('error', reject)
  91. .pipe(output).on('error', reject)
  92. .on('finish', (err) => {
  93. if (err) reject(err);
  94. else resolve();
  95. }
  96. );
  97. });
  98. }
  99. function gunzipFile(inputFile, outputFile) {
  100. return new Promise((resolve, reject) => {
  101. const gzip = zlib.createGunzip();
  102. const input = fs.createReadStream(inputFile);
  103. const output = fs.createWriteStream(outputFile);
  104. input.on('error', reject)
  105. .pipe(gzip).on('error', reject)
  106. .pipe(output).on('error', reject)
  107. .on('finish', (err) => {
  108. if (err) reject(err);
  109. else resolve();
  110. }
  111. );
  112. });
  113. }
  114. function gzipBuffer(buf) {
  115. return new Promise((resolve, reject) => {
  116. zlib.gzip(buf, {level: 1}, (err, result) => {
  117. if (err) reject(err);
  118. resolve(result);
  119. });
  120. });
  121. }
  122. function gunzipBuffer(buf) {
  123. return new Promise((resolve, reject) => {
  124. zlib.gunzip(buf, (err, result) => {
  125. if (err) reject(err);
  126. resolve(result);
  127. });
  128. });
  129. }
  130. function toUnixPath(dir) {
  131. return dir.replace(/\\/g, '/');
  132. }
  133. function makeValidFileName(fileName, repl = '_') {
  134. let f = fileName.replace(/[\x00\\/:*"<>|]/g, repl); // eslint-disable-line no-control-regex
  135. f = f.trim();
  136. while (f.length && (f[f.length - 1] == '.' || f[f.length - 1] == '_')) {
  137. f = f.substring(0, f.length - 1);
  138. }
  139. if (f)
  140. return f;
  141. else
  142. throw new Error('Invalid filename');
  143. }
  144. function makeValidFileNameOrEmpty(fileName) {
  145. try {
  146. return makeValidFileName(fileName);
  147. } catch(e) {
  148. return '';
  149. }
  150. }
  151. function wordEnding(num, type = 0) {
  152. const endings = [
  153. ['ов', '', 'а', 'а', 'а', 'ов', 'ов', 'ов', 'ов', 'ов'],//0
  154. ['й', 'я', 'и', 'и', 'и', 'й', 'й', 'й', 'й', 'й'],//1
  155. ['о', '', 'о', 'о', 'о', 'о', 'о', 'о', 'о', 'о'],//2
  156. ['ий', 'ие', 'ия', 'ия', 'ия', 'ий', 'ий', 'ий', 'ий', 'ий'],//3
  157. ['о', 'а', 'о', 'о', 'о', 'о', 'о', 'о', 'о', 'о'],//4
  158. ['ок', 'ка', 'ки', 'ки', 'ки', 'ок', 'ок', 'ок', 'ок', 'ок'],//5
  159. ['ых', 'ое', 'ых', 'ых', 'ых', 'ых', 'ых', 'ых', 'ых', 'ых'],//6
  160. ['о', 'о', 'о', 'о', 'о', 'о', 'о', 'о', 'о', 'о'],//7
  161. ['', 'а', 'и', 'и', 'и', '', '', '', '', ''],//8
  162. ];
  163. const deci = num % 100;
  164. if (deci > 10 && deci < 20) {
  165. return endings[type][0];
  166. } else {
  167. return endings[type][num % 10];
  168. }
  169. }
  170. function cutString(data, len = 500) {
  171. try {
  172. if (!data)
  173. return '';
  174. if (typeof(data) !== 'string')
  175. data = JSON.stringify(data);
  176. return `${data.substring(0, len)}${data.length > len ? ' ...' : ''}`;
  177. } catch (e) {
  178. return '';
  179. }
  180. }
  181. module.exports = {
  182. sleep,
  183. processLoop,
  184. versionText,
  185. findFiles,
  186. touchFile,
  187. hasProp,
  188. freeMemory,
  189. getFileHash,
  190. getBufHash,
  191. intersectSet,
  192. randomHexString,
  193. gzipFile,
  194. gunzipFile,
  195. gzipBuffer,
  196. gunzipBuffer,
  197. toUnixPath,
  198. makeValidFileName,
  199. makeValidFileNameOrEmpty,
  200. wordEnding,
  201. cutString,
  202. };