From 5c8e47a26e6b609ef09bf74415e8025ef91aa7cc Mon Sep 17 00:00:00 2001
From: Marko Kungla <marko@mkungla.dev>
Date: Fri, 29 Mar 2024 19:28:01 +0200
Subject: [PATCH] feat: recover do func panic

Signed-off-by: Marko Kungla <marko@mkungla.dev>
---
 command.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/command.go b/command.go
index 7a5ff81..2f1879a 100644
--- a/command.go
+++ b/command.go
@@ -501,7 +501,7 @@ func (c *Command) callBeforeAction(sess *Session) error {
 	return nil
 }
 
-func (c *Command) callDoAction(session *Session) error {
+func (c *Command) callDoAction(session *Session) (err error) {
 	c.mu.Lock()
 	defer c.mu.Unlock()
 
@@ -509,12 +509,18 @@ func (c *Command) callDoAction(session *Session) error {
 		return nil
 	}
 
+	defer func() {
+		if r := recover(); r != nil {
+			err = fmt.Errorf("%w: %s: %v (recovered)", ErrCommand, c.name, r)
+		}
+	}()
+
 	args := sdk.NewArgs(c.flags)
 
 	if err := c.doAction(session, args); err != nil {
 		return fmt.Errorf("%w: %s: %w", ErrCommand, c.name, err)
 	}
-	return nil
+	return err
 }
 
 func (c *Command) callAfterFailureAction(session *Session, err error) error {