package cmdx

import (
	"bufio"
	"bytes"
	"fmt"
	"gitee.com/fmpt/fmgo/logx"
	"gitee.com/fmpt/fmgo/stringx"
	"os/exec"
	"path/filepath"
	"time"
)

type Cmd struct {
	cmd *exec.Cmd
}

func Command(exe string, args ...string) *Cmd {
	cmd := exec.Command(exe, args...)
	//if runtime.GOOS == "windows" {
	//	cmd.SysProcAttr = &syscall.SysProcAttr{HideWindow: true}
	//}
	cmd.Dir = filepath.Dir(exe)
	return &Cmd{cmd}
}

func (c *Cmd) RunCommand() error {
	err := c.cmd.Run()
	if err != nil {
		logx.Error(err)
		return err
	}
	return nil
}

func (c *Cmd) ReturnCommand() string {
	var out bytes.Buffer
	c.cmd.Stdout = &out
	err := c.cmd.Run()
	if err != nil {
		logx.Error(err)
		return ""
	}
	return stringx.ConvertByte2String(out.Bytes(), stringx.GB18030)
}

func (c *Cmd) WaitCommand(result func(status int), run *bool) {
	err := c.cmd.Start()
	if err != nil {
		result(1) // 出错了
		return
	}
	result(0) // 启动成功
	err = c.cmd.Wait()
	if err != nil {
		if !*run {
			return
		}
		result(2) // 出错了
		time.Sleep(3 * time.Second)
		// 3秒后重启
		go c.WaitCommand(result, run)
	}

	return
}

func (c *Cmd) StartCommand() error {
	err := c.cmd.Start()
	if err != nil {
		logx.Error(err)
		return err
	}
	return nil
}

func (c *Cmd) ReadLineCommand(read chan string) bool {
	stdout, err := c.cmd.StdoutPipe()
	if err != nil {
		fmt.Println(err)
		return false
	}
	err = c.cmd.Start()
	if err != nil {
		return false
	}
	defer c.cmd.Wait()
	in := bufio.NewScanner(stdout)
	for in.Scan() {
		read <- stringx.ConvertByte2String(in.Bytes(), stringx.GB18030)
	}
	return true
}