mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-14 11:23:09 +02:00
75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
|
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})
|
||
|
})
|
||
|
}
|