go/lazygit
1
0
Fork 0
mirror of https://github.com/jesseduffield/lazygit.git synced 2025-08-06 22:33:07 +02:00
Code Issues Releases Activity

Close the pty instead of killing the process for runAndDetectCredentialRequest

Browse Source 
As with the previous commit, this is not strictly necessary for anything, just
cleaner.
This commit is contained in:
Stefan Haller
2025-07-29 10:43:42 +02:00
parent 83046a05d4
commit 7d9eaead54
2 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
pkg/commands/oscommands/cmd_obj_runner.go
View File

@ -336,7 +336,7 @@ func (self *cmdObjRunner) runAndDetectCredentialRequest(
tr := io.TeeReader(handler.stdoutPipe, cmdWriter)
go utils.Safe(func() {
self.processOutput(tr, handler.stdinPipe, promptUserForCredential, cmdObj)
self.processOutput(tr, handler.stdinPipe, promptUserForCredential, handler.close, cmdObj)
})
})
}
@ -345,6 +345,7 @@ func (self *cmdObjRunner) processOutput(
reader io.Reader,
writer io.Writer,
promptUserForCredential func(CredentialType) <-chan string,
closeFunc func() error,
cmdObj *CmdObj,
) {
checkForCredentialRequest := self.getCheckForCredentialRequestFunc()
@ -358,14 +359,16 @@ func (self *cmdObjRunner) processOutput(
if ok {
responseChan := promptUserForCredential(askFor)
if responseChan == nil {
// Returning a nil channel means we should kill the process.
// Note that we don't break the loop after this, because we
// still need to drain the output, otherwise the Wait() call
// later might block.
if err := Kill(cmdObj.GetCmd()); err != nil {
// Returning a nil channel means we should terminate the process.
// We achieve this by closing the pty that it's running in. Note that this won't
// work for the case where we're not running in a pty (i.e. on Windows), but
// in that case we'll never be prompted for credentials, so it's not a concern.
if err := closeFunc(); err != nil {
self.log.Error(err)
}
} else {
break
}
if task != nil {
task.Pause()
}
@ -379,7 +382,6 @@ func (self *cmdObjRunner) processOutput(
}
}
}
}
}
// having a function that returns a function because we need to maintain some state inbetween calls hence the closure

2
pkg/commands/oscommands/cmd_obj_runner_test.go
View File

@ -127,7 +127,7 @@ func TestProcessOutput(t *testing.T) {
writer := &strings.Builder{}
cmdObj := &CmdObj{task: gocui.NewFakeTask()}
runner.processOutput(reader, writer, toChanFn(scenario.promptUserForCredential), cmdObj)
runner.processOutput(reader, writer, toChanFn(scenario.promptUserForCredential), func() error { return nil }, cmdObj)
if writer.String() != scenario.expectedToWrite {
t.Errorf("expected to write '%s' but got '%s'", scenario.expectedToWrite, writer.String())