Image prunning (#13)

Author: bsideup

README.md

@@ -1,6 +1,6 @@
 # Moby Ryuk
 
-This project helps you to remove containers/networks/volumes by given filter after specified delay.
+This project helps you to remove containers/networks/volumes/images by given filter after specified delay.
 
 # Usage
 
@@ -33,4 +33,4 @@ This project helps you to remove containers/networks/volumes by given filter aft
         2018/01/15 18:38:52 Deleting {"label":{"something":true}}
         2018/01/15 18:38:52 Deleting {"label":{"something_else":true}}
         2018/01/15 18:38:52 Deleting {"health":{"unhealthy":true},"label":{"testing=true":true}}
-        2018/01/15 18:38:52 Removed 1 container(s), 0 network(s), 0 volume(s)
+        2018/01/15 18:38:52 Removed 1 container(s), 0 network(s), 0 volume(s), 0 image(s)

main.go

@@ -122,6 +122,7 @@ TimeoutLoop:
 	deletedContainers := make(map[string]bool)
 	deletedNetworks := make(map[string]bool)
 	deletedVolumes := make(map[string]bool)
+	deletedImages := make(map[string]bool)
 
 	for param := range deathNote {
 		log.Printf("Deleting %s\n", param)
@@ -166,7 +167,20 @@ TimeoutLoop:
 			}
 			return shouldRetry, err
 		})
+
+		try.Do(func(attempt int) (bool, error) {
+			imagesPruneReport, err := cli.ImagesPrune(context.Background(), args)
+			for _, image := range imagesPruneReport.ImagesDeleted {
+				deletedImages[image.Deleted] = true
+			}
+			shouldRetry := attempt < 10
+			if err != nil && shouldRetry {
+				log.Printf("Images pruning has failed, retrying(%d/%d). The error was: %v", attempt, 10, err)
+				time.Sleep(1 * time.Second)
+			}
+			return shouldRetry, err
+		})
 	}
 
-	log.Printf("Removed %d container(s), %d network(s), %d volume(s)", len(deletedContainers), len(deletedNetworks), len(deletedVolumes))
+	log.Printf("Removed %d container(s), %d network(s), %d volume(s) %d image(s)", len(deletedContainers), len(deletedNetworks), len(deletedVolumes), len(deletedImages))
 }