package executor

import (
	"bufio"
	"fmt"
	"io"
	"os/exec"
)

func ReadTimeCommandExecutor(singleLineCommand []string) {
	cmd := exec.Command(singleLineCommand[0], singleLineCommand[1:]...)
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		log.ErrorF("command  %v stdout error => %v", singleLineCommand, err)
	}
	stderr, err := cmd.StderrPipe()
	if err != nil {
		log.ErrorF("command  %v stderr error => %v", singleLineCommand, err)
	}

	if err := cmd.Start(); err != nil {
		log.ErrorF("command  %v runtime error => %v", singleLineCommand, err)
	}

	go realTimeOutput(stdout)
	go realTimeOutput(stderr)

	if err := cmd.Wait(); err != nil {
		log.ErrorF("command  %v result error => %v", singleLineCommand, err)
	}

}

// AllOutputCommandExecutor 收集全部执行结果、错误并且返回
func AllOutputCommandExecutor(singleLineCommand []string) ([]string, error) {

	// result
	var resultSlice []string
	var resultError error

	cmd := exec.Command(singleLineCommand[0], singleLineCommand[1:]...)
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		log.ErrorF("command  %v stdout error => %v", singleLineCommand, err)
		resultError = err
	}
	stderr, err := cmd.StderrPipe()
	if err != nil {
		log.ErrorF("command  %v stderr error => %v", singleLineCommand, err)
		resultError = err
	}

	if err := cmd.Start(); err != nil {
		log.ErrorF("command  %v runtime error => %v", singleLineCommand, err)
	}

	resultSlice = append(resultSlice, fmt.Sprintf(" ========= 命令为 ====> %v", singleLineCommand), "↓↓↓ 命令 输出 如下 ↓↓↓")
	resultSlice = collectOutput(stdout, resultSlice)
	resultSlice = append(resultSlice, "↓↓↓ 命令 错误 如下 ↓↓↓")
	resultSlice = collectOutput(stderr, resultSlice)

	if err := cmd.Wait(); err != nil {
		log.ErrorF("command  %v result error => %v", singleLineCommand, err)
		resultError = err
	}

	log.DebugF("real time exec result are %v", resultSlice)

	return resultSlice, resultError
}

func realTimeOutput(r io.Reader) {
	scanner := bufio.NewScanner(r)
	for scanner.Scan() {
		fmt.Println(scanner.Text())
	}
}

func collectOutput(r io.Reader, resultSlice []string) []string {
	scanner := bufio.NewScanner(r)
	for scanner.Scan() {
		resultLine := scanner.Text()

		resultSlice = append(resultSlice, resultLine)
		// debug usage
		//fmt.Println(resultLine)
	}

	return resultSlice
}

// PureResultSingleExecute 执行单行命令，忽略输出，只对执行成功与否负责
func PureResultSingleExecute(singleCommand []string) bool {

	cmd := exec.Command(singleCommand[0], singleCommand[1:]...)
	err := cmd.Run()
	if err != nil {
		log.DebugF("指令 %s 执行 错误", singleCommand)
		return false
	} else {
		log.DebugF("指令 %s 执行 成功", singleCommand)
		return true
	}
}

// PureResultSingleExecuteBatch 批量 执行单行命令，忽略输出，只对执行成功与否负责
func PureResultSingleExecuteBatch(singleCommandList [][]string) bool {

	for _, singleCommand := range singleCommandList {
		if !PureResultSingleExecute(singleCommand) {
			return false
		}
	}

	return true
}