_artifactcache.py: Added client progress callback to ArtifactCache.clean()

buildstream/_artifactcache.py

@@ -250,10 +250,13 @@ class ArtifactCache():
     #
     # Clean the artifact cache as much as possible.
     #
+    # Args:
+    #    progress (callable): A callback to call when a ref is removed
+    #
     # Returns:
     #    (int): The size of the cache after having cleaned up
     #
-    def clean(self):
+    def clean(self, progress=None):
         artifacts = self.list_artifacts()
         context = self.context
 
@@ -327,6 +330,10 @@ class ArtifactCache():
                 # Remove the size from the removed size
                 self.set_cache_size(self._cache_size - size)
 
+                # User callback
+                if progress:
+                    progress()
+
         # Informational message about the side effects of the cleanup
         self._message(MessageType.INFO, "Cleanup completed",
                       detail=("Removed {} refs and saving {} disk space.\n" +