// Copyright 2019 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package exec2

import (
	"context"
	"os/exec"
)

// Cmd is like exec.Cmd, but supports terminate and process containment.
type Cmd struct {
	*exec.Cmd

	attr attr
}

// CommandContext is like exec.CommandContext, but it uses process group by
// default and supports timeout in Wait function.
func CommandContext(ctx context.Context, name string, arg ...string) *Cmd {
	cmd := &Cmd{
		Cmd: exec.CommandContext(ctx, name, arg...),
	}

	cmd.setupCmd()

	return cmd
}

// Start starts command with appropriate setup.
func (c *Cmd) Start() error {
	return c.start()
}

// Wait waits to process to finish.
func (c *Cmd) Wait() error {
	return c.wait()
}

// Terminate sends SIGTERM on unix or CTRL+BREAK on windows.
func (c *Cmd) Terminate() error {
	return c.terminate()
}

// Kill kills process.
func (c *Cmd) Kill() error {
	return c.kill()
}