mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-01-06 03:53:59 +02:00
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
// +build !windows
|
|
|
|
package commands
|
|
|
|
import (
|
|
"bufio"
|
|
"os/exec"
|
|
"regexp"
|
|
|
|
"github.com/ionrock/procs"
|
|
"github.com/kr/pty"
|
|
"github.com/mgutz/str"
|
|
)
|
|
|
|
// RunCommandWithOutputLiveWrapper runs a command and return every word that gets written in stdout
|
|
// Output is a function that executes by every word that gets read by bufio
|
|
// As return of output you need to give a string that will be written to stdin
|
|
// NOTE: If the return data is empty it won't written anything to stdin
|
|
// NOTE: You don't have to include a enter in the return data this function will do that for you
|
|
func RunCommandWithOutputLiveWrapper(c *OSCommand, command string, output func(string) string) error {
|
|
splitCmd := str.ToArgv(command)
|
|
cmd := exec.Command(splitCmd[0], splitCmd[1:]...)
|
|
|
|
cmd.Env = procs.Env(map[string]string{
|
|
"LANG": "en_US.utf8",
|
|
"LC_ALL": "en_US.UTF-8",
|
|
}, true)
|
|
|
|
tty, err := pty.Start(cmd)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer func() { _ = tty.Close() }()
|
|
|
|
go func() {
|
|
// Regex to cleanup the command output
|
|
// sometimes the output words include unneeded spaces at eatch end of the string
|
|
re := regexp.MustCompile(`(^\s*)|(\s*$)`)
|
|
|
|
scanner := bufio.NewScanner(tty)
|
|
scanner.Split(bufio.ScanWords)
|
|
for scanner.Scan() {
|
|
toWrite := output(re.ReplaceAllString(scanner.Text(), ""))
|
|
if len(toWrite) > 0 {
|
|
_, _ = tty.Write([]byte(toWrite + "\n"))
|
|
}
|
|
}
|
|
}()
|
|
|
|
if err := cmd.Wait(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|