update

Author: satackey

main.ts

@@ -8,14 +8,18 @@ const main = async () => {
   const primaryKey = core.getInput(`key`, { required: true })
   const restoreKeys = core.getInput(`restore-keys`, { required: false }).split(`\n`).filter(key => key !== ``)
 
-  core.saveState(`already-existing-images`, JSON.stringify(await new ImageDetector().getExistingImages()))
+  const imageDetector = new ImageDetector()
+
+  const alreadyExistingImages = await imageDetector.getExistingImages()
 
   const layerCache = new LayerCache([])
   layerCache.concurrency = parseInt(core.getInput(`concurrency`, { required: true }), 10)
   const restoredKey = await layerCache.restore(primaryKey, restoreKeys)
   await layerCache.cleanUp()
 
   core.saveState(`restored-key`, JSON.stringify(restoredKey !== undefined ? restoredKey : ''))
+  core.saveState(`already-existing-images`, JSON.stringify(alreadyExistingImages))
+  core.saveState(`restored-images`, JSON.stringify(imageDetector.getImagesShouldSave(alreadyExistingImages)))
 }
 
 main().catch(e => {

post.ts

@@ -12,26 +12,24 @@ const main = async () => {
   }
 
   const primaryKey = core.getInput('key', { required: true })
-  const restoredKey = JSON.parse(core.getState(`restored-key`)) as string
 
-  const rawAlreadyExistingImages = core.getState(`already-existing-images`)
-  assertType<string>(rawAlreadyExistingImages)
-  const alreadyExistingImages = JSON.parse(rawAlreadyExistingImages)
+  const restoredKey = JSON.parse(core.getState(`restored-key`))
+  const alreadyExistingImages = JSON.parse(core.getState(`already-existing-images`))
+  const restoredImages = JSON.parse(core.getState(`restored-images`))
+
+  assertType<string>(restoredKey)
   assertType<string[]>(alreadyExistingImages)
+  assertType<string[]>(restoredImages)
 
   const imageDetector = new ImageDetector()
-  imageDetector.registerAlreadyExistedImages(alreadyExistingImages)
-  await imageDetector.getExistingImages()
-  core.debug(JSON.stringify({ imageIdsToSave: imageDetector.getImagesShouldSave() }))
-  const layerCache = new LayerCache(imageDetector.getImagesShouldSave())
-  layerCache.concurrency = parseInt(core.getInput(`concurrency`, { required: true }), 10)
-
-  layerCache.unformattedOrigianlKey = primaryKey
-  core.debug(JSON.stringify({ restoredKey, formattedOriginalCacheKey: layerCache.getFormattedOriginalCacheKey()}))
-  if (restoredKey !== `` && restoredKey === layerCache.getFormattedOriginalCacheKey()) {
-    core.info(`Key ${restoredKey} already exists, skip storing.`)
+  if (await imageDetector.checkIfImageHasAdded(restoredImages)) {
+    core.info(`Key ${restoredKey} already exists, not saving cache.`)
     return
   }
+
+  const layerCache = new LayerCache(await imageDetector.getImagesShouldSave(alreadyExistingImages))
+  layerCache.concurrency = parseInt(core.getInput(`concurrency`, { required: true }), 10)
+
   await layerCache.store(primaryKey)
   await layerCache.cleanUp()
 }

src/ImageDetector.ts

@@ -2,24 +2,24 @@ import exec from 'actions-exec-listener'
 import * as core from '@actions/core'
 
 export class ImageDetector {
-  alreadyExistedImages: Set<string> = new Set([])
-  existingImages: Set<string> = new Set([])
-  registerAlreadyExistedImages(images: string[]) {
-    images.forEach(image => this.alreadyExistedImages.add(image))
-  }
-
   async getExistingImages(): Promise<string[]> {
+    const existingSet = new Set<string>([])
     const ids = (await exec.exec(`docker image ls -q`, [], { silent: true })).stdoutStr.split(`\n`).filter(id => id !== ``)
     const repotags = (await exec.exec(`sh -c "docker image ls --format '{{ .Repository }}:{{ .Tag }}' --filter 'dangling=false'"`, [], { silent: true })).stdoutStr.split(`\n`).filter(id => id !== ``);
     core.debug(JSON.stringify({ log: "getExistingImages", ids, repotags }));
-    ([...ids, ...repotags]).forEach(image => this.existingImages.add(image))
-    core.debug(JSON.stringify({ existingImages: this.existingImages }))
-    return Array.from(this.existingImages)
+    ([...ids, ...repotags]).forEach(image => existingSet.add(image))
+    core.debug(JSON.stringify({ existingSet }))
+    return Array.from(existingSet)
   }
 
-  getImagesShouldSave(): string[] {
-    const resultSet = new Set(this.existingImages.values())
-    this.alreadyExistedImages.forEach(image => resultSet.delete(image))
+  async getImagesShouldSave(alreadRegisteredImages: string[]): Promise<string[]> {
+    const resultSet = new Set(await this.getExistingImages())
+    alreadRegisteredImages.forEach(image => resultSet.delete(image))
     return Array.from(resultSet)
   }
+
+  async checkIfImageHasAdded(restoredImages: string[]): Promise<boolean> {
+    const existing = await this.getExistingImages()
+    return JSON.stringify(restoredImages) === JSON.stringify(existing)
+  }
 }