From 584a4dd9db49d97a07f85af371ab534bff74c7ef Mon Sep 17 00:00:00 2001
From: John McKenna <63067728+johnmckenna-snd@users.noreply.github.com>
Date: Mon, 12 Aug 2024 12:35:34 -0400
Subject: [PATCH] [FEAT] add timeout to cook command and adjust command
 variable (#110)

* add timeout to cook command and adjust command variable

* a few updates to fix some bugs with the new cookTimeout
---
 cmd/grlx/cmd/cook.go | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/cmd/grlx/cmd/cook.go b/cmd/grlx/cmd/cook.go
index bcd1d94..509e0b0 100644
--- a/cmd/grlx/cmd/cook.go
+++ b/cmd/grlx/cmd/cook.go
@@ -15,7 +15,10 @@ import (
 	"github.com/gogrlx/grlx/types"
 )
 
-var async bool
+var (
+	async   bool
+	cookTimeout int
+)
 
 // cmdCmd represents the cmd command
 var cookCmd = &cobra.Command{
@@ -114,7 +117,7 @@ var cmdCook = &cobra.Command{
 		}
 		// TODO convert this to a request and get back the list of targeted sprouts
 		ec.Publish(fmt.Sprintf("grlx.farmer.cook.trigger.%s", jid), types.TriggerMsg{JID: jid})
-		timeout := time.After(30 * time.Second)
+		localTimeout := time.After(time.Duration(cookTimeout) * time.Second)
 		dripTimeout := time.After(120 * time.Second)
 		concurrent := 0
 		defer sub.Unsubscribe()
@@ -135,14 +138,14 @@ var cmdCook = &cobra.Command{
 				}
 
 				completionSteps[completion.SproutID] = append(completionSteps[completion.SproutID], completion.CompletedStep)
-				timeout = time.After(30 * time.Second)
+				localTimeout = time.After(time.Duration(cookTimeout) * time.Second)
 			case <-finished:
 				break waitLoop
 			case <-dripTimeout:
 				finished <- struct{}{}
 				break waitLoop
-			case <-timeout:
-				color.Red("Cooking timed out after 30 seconds.")
+			case <-localTimeout:
+				color.Red(fmt.Sprintf("Cooking timed out after %d seconds.", cookTimeout))
 				finished <- struct{}{}
 				break waitLoop
 			}
@@ -190,6 +193,7 @@ func init() {
 	cmdCook.Flags().StringVarP(&environment, "environment", "E", "", "")
 	cmdCook.Flags().BoolVar(&async, "async", false, "Don't print any output, just return the JID to look up results later")
 	cmdCook.PersistentFlags().StringVarP(&sproutTarget, "target", "T", "", "list of sprouts to target")
+	cmdCook.Flags().IntVar(&cookTimeout, "cook-timeout", 30, "Cancel cook execution and return after X seconds")
 	cmdCook.MarkPersistentFlagRequired("target")
 	rootCmd.AddCommand(cmdCook)
 }