package helpers import ( "fmt" "github.com/jesseduffield/lazygit/pkg/commands" "github.com/jesseduffield/lazygit/pkg/commands/oscommands" "github.com/jesseduffield/lazygit/pkg/gui/types" ) type GpgHelper struct { c *types.HelperCommon os *oscommands.OSCommand git *commands.GitCommand } func NewGpgHelper( c *types.HelperCommon, os *oscommands.OSCommand, git *commands.GitCommand, ) *GpgHelper { return &GpgHelper{ c: c, os: os, git: git, } } // 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: we shouldn't need to use a shell here, but looks like that NewShell function contains some windows specific quoting stuff. We should centralise that. func (self *GpgHelper) WithGpgHandling(cmdObj oscommands.ICmdObj, waitingStatus string, onSuccess func() error) error { useSubprocess := self.git.Config.UsingGpg() if useSubprocess { success, err := self.c.RunSubprocess(self.os.Cmd.NewShell(cmdObj.ToString())) if success && onSuccess != nil { if err := onSuccess(); err != nil { return err } } if err := self.c.Refresh(types.RefreshOptions{Mode: types.ASYNC}); err != nil { return err } return err } else { return self.runAndStream(cmdObj, waitingStatus, onSuccess) } } func (self *GpgHelper) runAndStream(cmdObj oscommands.ICmdObj, waitingStatus string, onSuccess func() error) error { cmdObj = self.os.Cmd.NewShell(cmdObj.ToString()) return self.c.WithWaitingStatus(waitingStatus, func() error { if err := cmdObj.StreamOutput().Run(); err != nil { _ = self.c.Refresh(types.RefreshOptions{Mode: types.ASYNC}) return self.c.Error( fmt.Errorf( self.c.Tr.GitCommandFailed, self.c.UserConfig.Keybinding.Universal.ExtrasMenu, ), ) } if onSuccess != nil { if err := onSuccess(); err != nil { return err } } return self.c.Refresh(types.RefreshOptions{Mode: types.ASYNC}) }) }