Refactor pausing a bit

jesseduffield · May 12, 2022 · a03f550 · a03f550
1 parent 0ced576
commit a03f550
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 20 deletions.
diff --git a/pkg/gui/containers_panel.go b/pkg/gui/containers_panel.go
@@ -423,13 +423,8 @@ func (gui *Gui) handleContainersRemoveMenu(g *gocui.Gui, v *gocui.View) error {
 	return gui.createMenu("", options, len(options), handleMenuPress)
 }
 
-func (gui *Gui) handleContainerPause(g *gocui.Gui, v *gocui.View) error {
-	container, err := gui.getSelectedContainer()
-	if err != nil {
-		return nil
-	}
-
-	return gui.WithWaitingStatus(gui.Tr.PausingStatus, func() error {
+func (gui *Gui) PauseContainer(container *commands.Container) error {
+	return gui.WithWaitingStatus(gui.Tr.PausingStatus, func() (err error) {
 		if container.Details.State.Paused {
 			err = container.Unpause()
 		} else {
@@ -444,6 +439,15 @@ func (gui *Gui) handleContainerPause(g *gocui.Gui, v *gocui.View) error {
 	})
 }
 
+func (gui *Gui) handleContainerPause(g *gocui.Gui, v *gocui.View) error {
+	container, err := gui.getSelectedContainer()
+	if err != nil {
+		return nil
+	}
+
+	return gui.PauseContainer(container)
+}
+
 func (gui *Gui) handleContainerStop(g *gocui.Gui, v *gocui.View) error {
 	container, err := gui.getSelectedContainer()
 	if err != nil {

diff --git a/pkg/gui/services_panel.go b/pkg/gui/services_panel.go
@@ -228,19 +228,7 @@ func (gui *Gui) handleServicePause(g *gocui.Gui, v *gocui.View) error {
 		return nil
 	}
 
-	return gui.WithWaitingStatus(gui.Tr.PausingStatus, func() error {
-		if service.Container.Details.State.Paused {
-			err = service.Container.Unpause()
-		} else {
-			err = service.Container.Pause()
-		}
-
-		if err != nil {
-			return gui.createErrorPanel(gui.g, err.Error())
-		}
-
-		return gui.refreshContainersAndServices()
-	})
+	return gui.PauseContainer(service.Container)
 }
 
 func (gui *Gui) handleServiceStop(g *gocui.Gui, v *gocui.View) error {