summaryrefslogtreecommitdiffstats
path: root/pkg/gui/updates.go
blob: 7f492bbe78c01ee2d2be2972093dc1dd5a472160 (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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package gui

import (
	"github.com/jesseduffield/gocui"
	"github.com/jesseduffield/lazygit/pkg/gui/types"
	"github.com/jesseduffield/lazygit/pkg/utils"
)

func (gui *Gui) showUpdatePrompt(newVersion string) error {
	message := utils.ResolvePlaceholderString(
		gui.Tr.UpdateAvailable, map[string]string{
			"newVersion": newVersion,
		},
	)

	return gui.c.Confirm(types.ConfirmOpts{
		Title:  gui.Tr.UpdateAvailableTitle,
		Prompt: message,
		HandleConfirm: func() error {
			gui.startUpdating(newVersion)
			return nil
		},
	})
}

func (gui *Gui) onUserUpdateCheckFinish(newVersion string, err error) error {
	if err != nil {
		return gui.c.Error(err)
	}
	if newVersion == "" {
		return gui.c.ErrorMsg(gui.Tr.FailedToRetrieveLatestVersionErr)
	}
	return gui.showUpdatePrompt(newVersion)
}

func (gui *Gui) onBackgroundUpdateCheckFinish(newVersion string, err error) error {
	if err != nil {
		// ignoring the error for now so that I'm not annoying users
		gui.c.Log.Error(err.Error())
		return nil
	}
	if newVersion == "" {
		return nil
	}
	if gui.c.UserConfig.Update.Method == "background" {
		gui.startUpdating(newVersion)
		return nil
	}
	return gui.showUpdatePrompt(newVersion)
}

func (gui *Gui) startUpdating(newVersion string) {
	gui.State.Updating = true
	statusId := gui.statusManager.addWaitingStatus(gui.Tr.UpdateInProgressWaitingStatus)
	gui.Updater.Update(newVersion, func(err error) error { return gui.onUpdateFinish(statusId, err) })
}

func (gui *Gui) onUpdateFinish(statusId int, err error) error {
	gui.State.Updating = false
	gui.statusManager.removeStatus(statusId)
	gui.OnUIThread(func() error {
		_ = gui.renderString(gui.Views.AppStatus, "")
		if err != nil {
			errMessage := utils.ResolvePlaceholderString(
				gui.Tr.UpdateFailedErr, map[string]string{
					"errMessage": err.Error(),
				},
			)
			return gui.c.ErrorMsg(errMessage)
		}
		return gui.c.Alert(gui.Tr.UpdateCompletedTitle, gui.Tr.UpdateCompleted)
	})

	return nil
}

func (gui *Gui) createUpdateQuitConfirmation() error {
	return gui.c.Confirm(types.ConfirmOpts{
		Title:  gui.Tr.ConfirmQuitDuringUpdateTitle,
		Prompt: gui.Tr.ConfirmQuitDuringUpdate,
		HandleConfirm: func() error {
			return gocui.ErrQuit
		},
	})
}