From 6c50b15a273d29dc3820b5e4d50d78eeb113d335 Mon Sep 17 00:00:00 2001
From: HIROSE Masaaki <hirose31@gmail.com>
Date: Fri, 11 Nov 2016 13:11:50 +0900
Subject: [PATCH] Exit with the code of ExitCoder if exists

---
 errors.go | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/errors.go b/errors.go
index fd67b96..583f89f 100644
--- a/errors.go
+++ b/errors.go
@@ -85,10 +85,8 @@ func HandleExitCoder(err error) {
 	}
 
 	if multiErr, ok := err.(MultiError); ok {
-		for _, merr := range multiErr.Errors {
-			fmt.Fprintln(ErrWriter, merr)
-		}
-		OsExiter(1)
+		code := handleMultiError(multiErr)
+		OsExiter(code)
 		return
 	}
 
@@ -97,3 +95,18 @@ func HandleExitCoder(err error) {
 	}
 	OsExiter(1)
 }
+
+func handleMultiError(multiErr MultiError) int {
+	code := 1
+	for _, merr := range multiErr.Errors {
+		if multiErr2, ok := merr.(MultiError); ok {
+			code = handleMultiError(multiErr2)
+		} else {
+			fmt.Fprintln(ErrWriter, merr)
+			if exitErr, ok := merr.(ExitCoder); ok {
+				code = exitErr.ExitCode()
+			}
+		}
+	}
+	return code
+}