create-file.ts 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. import { asyncWriteToStream } from 'foxts/async-write-to-stream';
  2. import { fastStringArrayJoin } from 'foxts/fast-string-array-join';
  3. import fs from 'node:fs';
  4. import picocolors from 'picocolors';
  5. import type { Span } from '../trace';
  6. import { readFileByLine } from './fetch-text-by-line';
  7. import { writeFile } from './misc';
  8. import { createCompareSource, fileEqualWithCommentComparator } from 'foxts/compare-source';
  9. import { promisify } from 'node:util';
  10. export const fileEqual = createCompareSource(fileEqualWithCommentComparator);
  11. export async function compareAndWriteFile(span: Span, linesA: string[], filePath: string) {
  12. if (linesA[linesA.length - 1] !== '') {
  13. linesA.push('');
  14. }
  15. const isEqual = await span.traceChildAsync(`compare ${filePath}`, async () => {
  16. if (fs.existsSync(filePath)) {
  17. return fileEqual(linesA, readFileByLine(filePath));
  18. }
  19. console.log(`${filePath} does not exists, writing...`);
  20. return false;
  21. });
  22. if (isEqual) {
  23. console.log(picocolors.gray(picocolors.dim(`same content, bail out writing: ${filePath}`)));
  24. return;
  25. }
  26. return span.traceChildAsync<void>(`writing ${filePath}`, async () => {
  27. const linesALen = linesA.length;
  28. // The default highwater mark is normally 16384,
  29. // So we make sure direct write to file if the content is
  30. // most likely less than 250 lines
  31. if (linesALen < 250) {
  32. return writeFile(filePath, fastStringArrayJoin(linesA, '\n'));
  33. }
  34. const writeStream = fs.createWriteStream(filePath);
  35. for (let i = 0; i < linesALen; i++) {
  36. const p = asyncWriteToStream(writeStream, linesA[i] + '\n');
  37. // eslint-disable-next-line no-await-in-loop -- stream high water mark
  38. if (p) await p;
  39. }
  40. await new Promise<void>(resolve => {
  41. writeStream.end(resolve);
  42. });
  43. await promisify(writeStream.close.bind(writeStream))();
  44. });
  45. }