From 4722b2edca482806c6c072eda81f5347b6a0c462 Mon Sep 17 00:00:00 2001
From: EGOIST <0x142857@gmail.com>
Date: Thu, 18 Mar 2021 22:50:12 +0800
Subject: [PATCH] fix: handle errors in worker

---
 src/index.ts  | 4 +++-
 src/rollup.ts | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 14aa088..4f45fe5 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -401,8 +401,10 @@ export async function build(_options: Options) {
       },
     })
     worker.on('message', (data) => {
-      if (data === 'has-error') {
+      if (data === 'error') {
         process.exitCode = 1
+      } else if (data === 'success') {
+        process.exitCode = 0
       }
     })
   }
diff --git a/src/rollup.ts b/src/rollup.ts
index 431516b..d7580ee 100644
--- a/src/rollup.ts
+++ b/src/rollup.ts
@@ -114,6 +114,7 @@ async function runRollup(options: RollupConfig) {
     )
   } catch (error) {
     console.log(`${makeLabel('dts', 'error')} Build error`)
+    parentPort?.postMessage('error')
     handleError(error)
   }
 }
@@ -135,9 +136,10 @@ async function watchRollup(options: {
       console.log(
         `${makeLabel('dts', 'success')} Build success in ${event.duration}ms`
       )
+      parentPort?.postMessage('success')
     } else if (event.code === 'ERROR') {
       console.log(`${makeLabel('dts', 'error')} Build failed`)
-      console.log(event.error)
+      parentPort?.postMessage('error')
       handleError(event.error)
     }
   })