summaryrefslogtreecommitdiffstats
path: root/pkg/gui/gpg.go
blob: 60d728c4274fed927ebd882f331f7666aa4a67fe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package gui

import (
	"fmt"

	"github.com/jesseduffield/lazygit/pkg/commands/oscommands"
	"github.com/jesseduffield/lazygit/pkg/gui/style"
	"github.com/jesseduffield/lazygit/pkg/gui/types"
)

// Currently there is a bug where if we switch to a subprocess from within
// WithWaitingStatus we get stuck there and can't return to lazygit. We could
// fix this bug, or just stop running subprocesses from within there, given that
// we don't need to see a loading status if we're in a subprocess.
// TODO: work out if we actually need to use a shell command here
func (gui *Gui) withGpgHandling(cmdObj oscommands.ICmdObj, waitingStatus string, onSuccess func() error) error {
	gui.LogCommand(cmdObj.ToString(), true)

	useSubprocess := gui.git.Config.UsingGpg()
	if useSubprocess {
		success, err := gui.runSubprocessWithSuspense(gui.os.Cmd.NewShell(cmdObj.ToString()))
		if success && onSuccess != nil {
			if err := onSuccess(); err != nil {
				return err
			}
		}
		if err := gui.c.Refresh(types.RefreshOptions{Mode: types.ASYNC}); err != nil {
			return err
		}

		return err
	} else {
		return gui.RunAndStream(cmdObj, waitingStatus, onSuccess)
	}
}

func (gui *Gui) RunAndStream(cmdObj oscommands.ICmdObj, waitingStatus string, onSuccess func() error) error {
	return gui.c.WithWaitingStatus(waitingStatus, func() error {
		cmdObj := gui.os.Cmd.NewShell(cmdObj.ToString())
		cmdObj.AddEnvVars("TERM=dumb")
		cmdWriter := gui.getCmdWriter()
		cmd := cmdObj.GetCmd()
		cmd.Stdout = cmdWriter
		cmd.Stderr = cmdWriter

		if err := cmd.Run(); err != nil {
			if _, err := cmd.Stdout.Write([]byte(fmt.Sprintf("%s\n", style.FgRed.Sprint(err.Error())))); err != nil {
				gui.c.Log.Error(err)
			}
			_ = gui.c.Refresh(types.RefreshOptions{Mode: types.ASYNC})
			return gui.c.Error(
				fmt.Errorf(
					gui.c.Tr.GitCommandFailed, gui.c.UserConfig.Keybinding.Universal.ExtrasMenu,
				),
			)
		}

		if onSuccess != nil {
			if err := onSuccess(); err != nil {
				return err
			}
		}

		return gui.c.Refresh(types.RefreshOptions{Mode: types.ASYNC})
	})
}