fix: update profile reporter

Author: pi0

src/reporters/profile.js

@@ -1,16 +1,38 @@
+import chalk from 'chalk';
+
+import { colorize } from '../utils/cli';
 import Profiler from '../profiler';
 
 export default class ProfileReporter {
-  constructor() {
-    this.profiler = new Profiler();
+  progress(context) {
+    if (!context.profiler) {
+      context.profiler = new Profiler();
+    }
+
+    context.profiler.onRequest(context.state.request);
   }
 
-  allDone() {
-    const formattedStats = this.profiler.getFormattedStats();
-    process.stderr.write(`\n${formattedStats}\n`);
+  done(context) {
+    if (context.profiler) {
+      context.state.profile = context.profiler.getFormattedStats();
+      delete context.profiler;
+    }
   }
 
-  progress(context) {
-    this.profiler.onRequest(context.state.request);
+  allDone(context) {
+    let str = '';
+
+    for (const state of context.statesArray) {
+      const color = colorize(state.color);
+
+      if (state.profile) {
+        str +=
+          color(`\nProfile results for ${chalk.bold(state.name)}\n`) +
+          `\n${state.profile}\n`;
+        delete state.profile;
+      }
+    }
+
+    process.stderr.write(str);
   }
 }