package execUtil

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"os"
	"os/exec"
	"strings"
	"time"
)

// 执行系统命令行命令
// 参数：cmdDict  系统命令行命令及参数字典
//
//	cmdDict["cmd"]:命令行命令
//	cmdDict["cmdArgs"]:对应命令的参数
//	iTimeOut:命令执行超时时间，如果为0则表示一直等待至命令运行完毕
//	cmdDir:命令运行的工作目录
func ExecCmd(cmdDict map[string]string, iTimeOut int, cmdDir string) error {
	_, err := ExecCmdWithErrInfo(cmdDict, iTimeOut, cmdDir)
	return err
}
func ExecCmdContext(cmdDict map[string]string, iTimeOut int, cmdDir string) (cmdOutInfo *bytes.Buffer, cmdErrInfo *bytes.Buffer, err error) {
	cmd := cmdDict["cmd"]
	cmdArgs := cmdDict["cmdArgs"]
	stderr := &bytes.Buffer{}
	stdout := &bytes.Buffer{}
	ctx := context.TODO()
	if iTimeOut != 0 {
		var cancel context.CancelFunc
		ctx, cancel = context.WithTimeout(ctx, time.Duration(iTimeOut)*time.Second)
		defer cancel()
	}
	c := exec.CommandContext(ctx, cmd, cmdArgs)
	c.Stdout = stdout
	c.Stderr = stderr
	c.Dir = cmdDir
	err = c.Run()
	if ctx.Err() == context.DeadlineExceeded {
		fmt.Printf("exec %s timeout\n", cmd)
		return nil, nil, ctx.Err()
	}
	if err != nil {
		return stdout, stderr, err
	}
	return stdout, stderr, nil
}
func ExecCmdWithErrInfo(cmdDict map[string]string, iTimeOut int, cmdDir string) ([]string, error) {
	cmd := cmdDict["cmd"]
	cmdArgs := cmdDict["cmdArgs"]
	stderr := &bytes.Buffer{}
	ctx := context.TODO()
	if iTimeOut != 0 {
		var cancel context.CancelFunc
		ctx, cancel = context.WithTimeout(ctx, time.Duration(iTimeOut)*time.Second)
		defer cancel()
	}
	c := exec.CommandContext(ctx, cmd, cmdArgs)
	// c.Stdout = stdout
	c.Stdout = os.Stdout
	c.Stderr = stderr
	c.Dir = cmdDir
	err := c.Run()
	if ctx.Err() == context.DeadlineExceeded {
		fmt.Printf("exec %s timeout\n", cmd)
		return nil, ctx.Err()
	}
	if err != nil {
		// fmt.Printf("errors happen:%s %s %s\n", cmd, err.Error(), stderr.String())
		slurp, _ := io.ReadAll(stderr)
		// fmt.Printf("%s\n", slurp)
		return strings.Split(string(slurp), "\r\n"), err
	}
	return nil, nil
}