summaryrefslogtreecommitdiffstats
path: root/stash_panel.go
blob: 9fe386ac72d4a9efad3e207bd2e1ed60ced6f817 (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
86
87
88
89
90
91
92
93
package main

import (
	"fmt"

	"github.com/jesseduffield/gocui"
)

func refreshStashEntries(g *gocui.Gui) error {
	g.Update(func(g *gocui.Gui) error {
		v, err := g.View("stash")
		if err != nil {
			panic(err)
		}
		state.StashEntries = getGitStashEntries()
		v.Clear()
		for _, stashEntry := range state.StashEntries {
			fmt.Fprintln(v, stashEntry.DisplayString)
		}
		return resetOrigin(v)
	})
	return nil
}

func getSelectedStashEntry(v *gocui.View) *StashEntry {
	if len(state.StashEntries) == 0 {
		return nil
	}
	lineNumber := getItemPosition(v)
	return &state.StashEntries[lineNumber]
}

func renderStashOptions(g *gocui.Gui) error {
	return renderOptionsMap(g, map[string]string{
		"space":   "apply",
		"k":       "pop",
		"d":       "drop",
		"← → ↑ ↓": "navigate",
	})
}

func handleStashEntrySelect(g *gocui.Gui, v *gocui.View) error {
	if err := renderStashOptions(g); err != nil {
		return err
	}
	go func() {
		stashEntry := getSelectedStashEntry(v)
		if stashEntry == nil {
			renderString(g, "main", "No stash entries")
			return
		}
		diff, _ := getStashEntryDiff(stashEntry.Index)
		renderString(g, "main", diff)
	}()
	return nil
}

func handleStashApply(g *gocui.Gui, v *gocui.View) error {
	return stashDo(g, v, "apply")
}

func handleStashPop(g *gocui.Gui, v *gocui.View) error {
	return stashDo(g, v, "pop")
}

func handleStashDrop(g *gocui.Gui, v *gocui.View) error {
	return createConfirmationPanel(g, v, "Stash drop", "Are you sure you want to drop this stash entry?", func(g *gocui.Gui, v *gocui.View) error {
		return stashDo(g, v, "drop")
	}, nil)
}

func stashDo(g *gocui.Gui, v *gocui.View, method string) error {
	stashEntry := getSelectedStashEntry(v)
	if stashEntry == nil {
		return createErrorPanel(g, "No stash to "+method)
	}
	if output, err := gitStashDo(stashEntry.Index, method); err != nil {
		createErrorPanel(g, output)
	}
	refreshStashEntries(g)
	return refreshFiles(g)
}

func handleStashSave(g *gocui.Gui, filesView *gocui.View) error {
	createPromptPanel(g, filesView, "Stash changes", func(g *gocui.Gui, v *gocui.View) error {
		if output, err := gitStashSave(trimmedContent(v)); err != nil {
			createErrorPanel(g, output)
		}
		refreshStashEntries(g)
		return refreshFiles(g)
	})
	return nil
}