remove image & fix nil cursor on remove for image ciew

docker.go

@@ -6,10 +6,32 @@ import (
 	docker "github.com/fsouza/go-dockerclient"
 )
 
-//
-// TODO: handle errors return from docker client instead of assigning to _
-//
+////////////////////////////////////////////
+// TODO: handle errors return from docker
+// client instead of assigning to _
+////////////////////////////////////////////
 
+// ---------------- Image ----------------
+func removeImage(c *docker.Client, id string) {
+	opts := docker.RemoveImageOptions{
+		Force: true,
+	}
+	// just tell em to remove the container that use this image first
+	_ = c.RemoveImageExtended(id, opts)
+}
+
+func listImages(c *docker.Client, showAll bool) []docker.APIImages {
+	opts := docker.ListImagesOptions{
+		All: showAll,
+	}
+	images, err := c.ListImages(opts)
+	if err != nil {
+		log.Fatal(err)
+	}
+	return images
+}
+
+// ---------------- Container ----------------
 func removeContainer(c *docker.Client, id string) {
 	opts := docker.RemoveContainerOptions{
 		ID:    id,
@@ -52,14 +74,3 @@ func listContainers(c *docker.Client, showAll bool) []docker.APIContainers {
 	}
 	return containers
 }
-
-func listImages(c *docker.Client, showAll bool) []docker.APIImages {
-	opts := docker.ListImagesOptions{
-		All: showAll,
-	}
-	images, err := c.ListImages(opts)
-	if err != nil {
-		log.Fatal(err)
-	}
-	return images
-}

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/charmbracelet/lipgloss v0.9.1
 	github.com/fsouza/go-dockerclient v1.10.0
 	github.com/mattn/go-runewidth v0.0.15
+	github.com/muesli/reflow v0.3.0
 )
 
 require (
@@ -30,7 +31,6 @@ require (
 	github.com/morikuni/aec v1.0.0 // indirect
 	github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
 	github.com/muesli/cancelreader v0.2.2 // indirect
-	github.com/muesli/reflow v0.3.0 // indirect
 	github.com/muesli/termenv v0.15.2 // indirect
 	github.com/opencontainers/go-digest v1.0.0 // indirect
 	github.com/opencontainers/image-spec v1.0.3-0.20211202183452-c5a74bcca799 // indirect

helper.go

@@ -8,16 +8,24 @@ import (
 	docker "github.com/fsouza/go-dockerclient"
 )
 
-type actionResult struct {
+type actionResultContainers struct {
 	success []Container
 	failed  []Container
 }
 
+type actionResultImages struct {
+	success              []Image
+	failed               []Image
+	associatedContainers []Container
+}
+
 const (
 	on int = iota
 	off
 )
 
+// checkProcess check if the container is in m.processes
+// (a.k.a process is in progress)
 func checkProcess(id string, processes map[string]string) bool {
 	if _, ok := processes[id]; ok {
 		return true
@@ -66,7 +74,7 @@ func unpauseAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		if c.state == "paused" {
 			go unpauseContainer(client, c.id)
@@ -113,7 +121,7 @@ func pauseAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		if c.state == "running" {
 			go pauseContainer(client, c.id)
@@ -160,7 +168,7 @@ func stopAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		if c.state == "running" || c.state == "restarting" {
 			go stopContainer(client, c.id)
@@ -207,7 +215,7 @@ func startAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		if c.state == "exited" || c.state == "created" {
 			go startContainer(client, c.id)
@@ -255,7 +263,7 @@ func removeAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		removeContainer(client, c.id)
 		desiredState := "x"
@@ -294,7 +302,7 @@ func restartAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		if c.state == "running" {
 			go restartContainer(client, c.id)
@@ -341,7 +349,7 @@ func killAndWriteLog(m model) (tea.Model, tea.Cmd) {
 		}
 	}
 
-	res := actionResult{}
+	res := actionResultContainers{}
 	for _, c := range targets {
 		if c.state == "running" {
 			go killContainer(client, c.id)