From 62e7f9eee82d86a673c9a15a8708aabe6f6c99f6 Mon Sep 17 00:00:00 2001
From: Kamil Samigullin <kamil@samigullin.info>
Date: Fri, 23 Apr 2021 16:21:18 +0300
Subject: [PATCH] issue #5: implement panic command

---
 internal/cmd/demo/panic.go | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/internal/cmd/demo/panic.go b/internal/cmd/demo/panic.go
index 61c5ff0..9916164 100644
--- a/internal/cmd/demo/panic.go
+++ b/internal/cmd/demo/panic.go
@@ -1,17 +1,23 @@
 package demo
 
-import "github.com/spf13/cobra"
+import (
+	"strings"
+
+	"github.com/spf13/cobra"
+)
 
 // Panic returns a demo cobra.Command to raise a panic.
 //
-// 	$ go run main.go panic
+// 	$ go run main.go panic [message]
 //
 func Panic() *cobra.Command {
 	command := cobra.Command{
-		Use:  "panic",
-		Args: cobra.NoArgs,
+		Use: "panic",
 		Run: func(cmd *cobra.Command, args []string) {
-			panic("unexpected panic")
+			if len(args) == 0 {
+				args = []string{"unexpected panic"}
+			}
+			panic(strings.Join(args, " "))
 		},
 	}