feat: binfmt image local cached

src/context.ts

@@ -5,12 +5,14 @@ export interface Inputs {
   image: string;
   platforms: string;
   cacheImage: boolean;
+  localCachePath: string;
 }
 
 export function getInputs(): Inputs {
   return {
     image: core.getInput('image') || 'docker.io/tonistiigi/binfmt:latest',
     platforms: Util.getInputList('platforms').join(',') || 'all',
-    cacheImage: core.getBooleanInput('cache-image')
+    cacheImage: core.getBooleanInput('cache-image'),
+    localCachePath: core.getInput('local-cache-path') || ''
   };
 }

src/local-cache.ts

@@ -0,0 +1,57 @@
+import * as fs from 'fs';
+import * as path from 'path';
+import * as core from '@actions/core';
+import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
+
+function getImageCacheFileName(imageName: string): string {
+  return `${imageName.replace(/[\/\:]/g, '-')}.tar`;
+}
+
+export async function loadDockerImageFromCache(localCachePath: string, imageName: string): Promise<void> {
+  const cacheFilePath = path.join(localCachePath, getImageCacheFileName(imageName));
+
+  try {
+    if (fs.existsSync(cacheFilePath)) {
+      await Docker.getExecOutput(['load', '-i', cacheFilePath], {
+        ignoreReturnCode: true
+      }).then(res => {
+        if (res.stderr.length > 0 && res.exitCode != 0) {
+          throw new Error(res.stderr.match(/(.*)\s*$/)?.[0]?.trim() ?? 'unknown error');
+        }
+        core.info(`Loaded image from ${cacheFilePath}`);
+      });
+    } else {
+      core.info(`Cache file not found at ${cacheFilePath}, pulling image instead`);
+      await Docker.pull(imageName);
+    }
+  } catch (error) {
+    core.warning(`Failed to check/load cache file: ${error}`);
+    await Docker.pull(imageName);
+  }
+}
+
+export async function saveDockerImageToCache(localCachePath: string, imageName: string): Promise<void> {
+  const cacheFilePath = path.join(localCachePath, getImageCacheFileName(imageName));
+
+  try {
+    if (!fs.existsSync(localCachePath)) {
+      fs.mkdirSync(localCachePath, {recursive: true});
+    }
+
+    if (fs.existsSync(cacheFilePath)) {
+      core.info(`Cache file already exists at ${cacheFilePath}, skipping save`);
+      return;
+    }
+
+    await Docker.getExecOutput(['save', '-o', cacheFilePath, imageName], {
+      ignoreReturnCode: true
+    }).then(res => {
+      if (res.stderr.length > 0 && res.exitCode != 0) {
+        throw new Error(res.stderr.match(/(.*)\s*$/)?.[0]?.trim() ?? 'unknown error');
+      }
+      core.info(`Saved image to ${cacheFilePath}`);
+    });
+  } catch (error) {
+    core.warning(`Failed to save image to cache file: ${error}`);
+  }
+}

src/main.ts

@@ -3,6 +3,7 @@ import * as core from '@actions/core';
 import * as actionsToolkit from '@docker/actions-toolkit';
 
 import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
+import {loadDockerImageFromCache, saveDockerImageToCache} from './local-cache';
 
 interface Platforms {
   supported: string[];
@@ -19,9 +20,15 @@ actionsToolkit.run(
       await Docker.printInfo();
     });
 
-    await core.group(`Pulling binfmt Docker image`, async () => {
-      await Docker.pull(input.image, input.cacheImage);
-    });
+    if (input.localCachePath !== '') {
+      await core.group(`Pulling binfmt Docker image`, async () => {
+        await loadDockerImageFromCache(input.localCachePath, input.image);
+      });
+    } else {
+      await core.group(`Pulling binfmt Docker image`, async () => {
+        await Docker.pull(input.image, input.cacheImage);
+      });
+    }
 
     await core.group(`Image info`, async () => {
       await Docker.getExecOutput(['image', 'inspect', input.image], {
@@ -56,5 +63,15 @@ actionsToolkit.run(
         core.setOutput('platforms', platforms.supported.join(','));
       });
     });
+  },
+
+  // post
+  async () => {
+    const input: context.Inputs = context.getInputs();
+    if (input.localCachePath !== '') {
+      await core.group(`Saving binfmt Docker image`, async () => {
+        await saveDockerImageToCache(input.localCachePath, input.image);
+      });
+    }
   }
 );