Refactor: full span tracer

Build/lib/create-file.ts

@@ -3,8 +3,10 @@ import { readFileByLine } from './fetch-text-by-line';
 import { surgeDomainsetToClashDomainset, surgeRulesetToClashClassicalTextRuleset } from './clash';
 import { traceAsync } from './trace-runner';
 import picocolors from 'picocolors';
+import type { Span } from '../trace';
+import path from 'path';
 
-export async function compareAndWriteFile(linesA: string[], filePath: string) {
+export async function compareAndWriteFile(span: Span, linesA: string[], filePath: string) {
   let isEqual = true;
   const file = Bun.file(filePath);
 
@@ -17,48 +19,44 @@ export async function compareAndWriteFile(linesA: string[], filePath: string) {
     console.log(`Nothing to write to ${filePath}...`);
     isEqual = false;
   } else {
-    isEqual = await traceAsync(
-      picocolors.gray(`comparing ${filePath}`),
-      async () => {
-        let index = 0;
+    isEqual = await span.traceChild(`comparing ${filePath}`).traceAsyncFn(async () => {
+      let index = 0;
 
-        for await (const lineB of readFileByLine(file)) {
-          const lineA = linesA[index];
-          index++;
+      for await (const lineB of readFileByLine(file)) {
+        const lineA = linesA[index];
+        index++;
 
-          if (lineA == null) {
-            // The file becomes smaller
-            return false;
-          }
+        if (lineA == null) {
+          // The file becomes smaller
+          return false;
+        }
 
-          if (lineA[0] === '#' && lineB[0] === '#') {
-            continue;
-          }
-          if (
-            lineA[0] === '/'
+        if (lineA[0] === '#' && lineB[0] === '#') {
+          continue;
+        }
+        if (
+          lineA[0] === '/'
             && lineA[1] === '/'
             && lineA[3] === '#'
             && lineB[0] === '/'
             && lineB[1] === '/'
             && lineB[3] === '#'
-          ) {
-            continue;
-          }
-
-          if (lineA !== lineB) {
-            return false;
-          }
+        ) {
+          continue;
         }
 
-        if (index !== linesALen) {
-          // The file becomes larger
+        if (lineA !== lineB) {
           return false;
         }
+      }
 
-        return true;
-      },
-      picocolors.gray
-    );
+      if (index !== linesALen) {
+        // The file becomes larger
+        return false;
+      }
+
+      return true;
+    });
   }
 
   if (isEqual) {
@@ -66,7 +64,7 @@ export async function compareAndWriteFile(linesA: string[], filePath: string) {
     return;
   }
 
-  await traceAsync(picocolors.gray(`writing ${filePath}`), async () => {
+  await span.traceChild(`writing ${filePath}`).traceAsyncFn(async () => {
     if (linesALen < 10000) {
       return Bun.write(file, `${linesA.join('\n')}\n`);
     }
@@ -79,7 +77,7 @@ export async function compareAndWriteFile(linesA: string[], filePath: string) {
     }
 
     return writer.end();
-  }, picocolors.gray);
+  });
 }
 
 export const withBannerArray = (title: string, description: string[] | readonly string[], date: Date, content: string[]) => {
@@ -96,27 +94,28 @@ export const withBannerArray = (title: string, description: string[] | readonly
 };
 
 export const createRuleset = (
+  parentSpan: Span,
   title: string, description: string[] | readonly string[], date: Date, content: string[],
   type: 'ruleset' | 'domainset', surgePath: string, clashPath: string
-) => {
+) => parentSpan.traceChild(`create ruleset: ${path.basename(surgePath, path.extname(surgePath))}`).traceAsyncFn((childSpan) => {
   const surgeContent = withBannerArray(title, description, date, content);
+  const clashContent = childSpan.traceChild('convert incoming ruleset to clash').traceSyncFn(() => {
+    let _clashContent;
+    switch (type) {
+      case 'domainset':
+        _clashContent = surgeDomainsetToClashDomainset(content);
+        break;
+      case 'ruleset':
+        _clashContent = surgeRulesetToClashClassicalTextRuleset(content);
+        break;
+      default:
+        throw new TypeError(`Unknown type: ${type as any}`);
+    }
+    return withBannerArray(title, description, date, _clashContent);
+  });
 
-  let _clashContent;
-  switch (type) {
-    case 'domainset':
-      _clashContent = surgeDomainsetToClashDomainset(content);
-      break;
-    case 'ruleset':
-      _clashContent = surgeRulesetToClashClassicalTextRuleset(content);
-      break;
-    default:
-      throw new TypeError(`Unknown type: ${type as any}`);
-  }
-
-  const clashContent = withBannerArray(title, description, date, _clashContent);
-
-  return [
-    compareAndWriteFile(surgeContent, surgePath),
-    compareAndWriteFile(clashContent, clashPath)
-  ];
-};
+  return Promise.all([
+    compareAndWriteFile(childSpan, surgeContent, surgePath),
+    compareAndWriteFile(childSpan, clashContent, clashPath)
+  ]);
+});