feat: run command with context

Pass a context down to *exec.Command and make timing out optional when
o.Timeout < 0.

Using `context.Background()` _all the time_ is not practical as
sometimes we need to stop the execution when _a_ parent context stops.

Signed-off-by: Ayman Bagabas <[email protected]>

Author: aymanbagabas

command.go

@@ -20,12 +20,18 @@ type Command struct {
 	name string
 	args []string
 	envs []string
+	ctx  context.Context
 }
 
 // CommandOptions contains options for running a command.
+// If timeout is zero, DefaultTimeout will be used.
+// If timeout is less than zero, no timeout will be set.
+// If context is nil, context.Background() will be used.
 type CommandOptions struct {
-	Args []string
-	Envs []string
+	Args    []string
+	Envs    []string
+	Timeout time.Duration
+	Context context.Context
 }
 
 // String returns the string representation of the command.
@@ -38,9 +44,16 @@ func (c *Command) String() string {
 
 // NewCommand creates and returns a new Command with given arguments for "git".
 func NewCommand(args ...string) *Command {
+	return NewCommandWithContext(context.Background(), args...)
+}
+
+// NewCommandWithContext creates and returns a new Command with given arguments
+// and context for "git".
+func NewCommandWithContext(ctx context.Context, args ...string) *Command {
 	return &Command{
 		name: "git",
 		args: args,
+		ctx:  ctx,
 	}
 }
 
@@ -56,8 +69,21 @@ func (c *Command) AddEnvs(envs ...string) *Command {
 	return c
 }
 
+// WithContext sets the context for the command.
+func (c *Command) WithContext(ctx context.Context) *Command {
+	c.ctx = ctx
+	return c
+}
+
 // AddOptions adds options to the command.
 func (c *Command) AddOptions(opts ...CommandOptions) *Command {
+	if len(opts) > 0 {
+		opt := opts[0]
+		if opt.Context != nil {
+			c.ctx = opt.Context
+		}
+	}
+
 	for _, opt := range opts {
 		c.AddArgs(opt.Args...)
 		c.AddEnvs(opt.Envs...)
@@ -124,7 +150,9 @@ func (c *Command) RunInDirWithOptions(dir string, opts ...RunInDirOptions) (err
 	if len(opts) > 0 {
 		opt = opts[0]
 	}
-	if opt.Timeout < time.Nanosecond {
+	if opt.Timeout < 0 {
+		opt.Timeout = -1
+	} else if opt.Timeout == 0 {
 		opt.Timeout = DefaultTimeout
 	}
 
@@ -147,13 +175,21 @@ func (c *Command) RunInDirWithOptions(dir string, opts ...RunInDirOptions) (err
 		}
 	}()
 
-	ctx, cancel := context.WithTimeout(context.Background(), opt.Timeout)
-	defer func() {
-		cancel()
-		if err == context.DeadlineExceeded {
-			err = ErrExecTimeout
-		}
-	}()
+	ctx := context.Background()
+	if c.ctx != nil {
+		ctx = c.ctx
+	}
+
+	if opt.Timeout > 0 {
+		var cancel context.CancelFunc
+		ctx, cancel = context.WithTimeout(ctx, opt.Timeout)
+		defer func() {
+			cancel()
+			if err == context.DeadlineExceeded {
+				err = ErrExecTimeout
+			}
+		}()
+	}
 
 	cmd := exec.CommandContext(ctx, c.name, c.args...)
 	if len(c.envs) > 0 {