refactor create reset menu logic

1 files changed, 43 insertions, 0 deletions

diff --git a/pkg/gui/reset_menu_panel.go b/pkg/gui/reset_menu_panel.go
new file mode 100644
index 000000000..10f2705dc
--- /dev/null
+++ b/pkg/gui/reset_menu_panel.go
@@ -0,0 +1,43 @@
+package gui
+
+import (
+	"fmt"
+
+	"github.com/fatih/color"
+)
+
+func (gui *Gui) createResetMenu(ref string) error {
+	strengths := []string{"soft", "mixed", "hard"}
+	menuItems := make([]*menuItem, len(strengths))
+	for i, strength := range strengths {
+		innerStrength := strength
+		menuItems[i] = &menuItem{
+			displayStrings: []string{
+				fmt.Sprintf("%s reset", strength),
+				color.New(color.FgRed).Sprint(
+					fmt.Sprintf("reset --%s %s", strength, ref),
+				),
+			},
+			onPress: func() error {
+				if err := gui.GitCommand.ResetToCommit(ref, innerStrength); err != nil {
+					return err
+				}
+
+				if err := gui.refreshCommits(gui.g); err != nil {
+					return err
+				}
+				if err := gui.refreshFiles(); err != nil {
+					return err
+				}
+				if err := gui.resetOrigin(gui.getCommitsView()); err != nil {
+					return err
+				}
+
+				gui.State.Panels.Commits.SelectedLine = 0
+				return gui.handleCommitSelect(gui.g, gui.getCommitsView())
+			},
+		}
+	}
+
+	return gui.createMenu(fmt.Sprintf("%s %s", gui.Tr.SLocalize("resetTo"), ref), menuItems, createMenuOptions{showCancel: true})
+}