src: fail workflow if cleanup fails

src/main.ts

@@ -81,7 +81,7 @@ export async function startBlacksmithBuilder(inputs: context.Inputs): Promise<{a
   } catch (error) {
     // If the builder setup fails for any reason, we check if we should fallback to a local build.
     // If we should not fallback, we rethrow the error and fail the build.
-    await reporter.reportBuildPushActionFailure(error);
+    await reporter.reportBuildPushActionFailure(error, "starting blacksmith builder");
 
     let errorMessage = `Error during Blacksmith builder setup: ${error.message}`;
     if (error.message.includes('buildkitd')) {
@@ -137,8 +137,14 @@ actionsToolkit.run(
       buildId: null as string | null,
       exposeId: '' as string
     };
+
+    let setupBlacksmithBuilderError: Error | undefined;
     await core.group(`Starting Blacksmith builder`, async () => {
-      builderInfo = await startBlacksmithBuilder(inputs);
+      try {
+        builderInfo = await startBlacksmithBuilder(inputs);
+      } catch (error) {
+        setupBlacksmithBuilderError = error as Error;
+      }
     });
 
     let buildError: Error | undefined;
@@ -315,22 +321,43 @@ actionsToolkit.run(
             refs: ref ? [ref] : []
           });
         }
-        await shutdownBuildkitd();
-        core.info('Shutdown buildkitd');
-        for (let attempt = 1; attempt <= 10; attempt++) {
-          try {
-            await execAsync(`sudo umount ${mountPoint}`);
-            core.debug(`${mountPoint} has been unmounted`);
-            break;
-          } catch (error) {
-            if (attempt === 10) {
-              throw error;
+        try {
+          const {stdout} = await execAsync('pgrep -f buildkitd');
+          if (stdout.trim()) {
+            await shutdownBuildkitd();
+            core.info('Shutdown buildkitd');
+          }
+        } catch (error) {
+          // No buildkitd process found, nothing to shutdown
+          core.debug('No buildkitd process found running');
+        }
+        try {
+          const {stdout: mountOutput} = await execAsync(`mount | grep ${mountPoint}`);
+          if (mountOutput) {
+            for (let attempt = 1; attempt <= 3; attempt++) {
+              try {
+                await execAsync(`sudo umount ${mountPoint}`);
+                core.debug(`${mountPoint} has been unmounted`);
+                break;
+              } catch (error) {
+                if (attempt === 3) {
+                  throw error;
+                }
+                core.warning(`Unmount failed, retrying (${attempt}/3)...`);
+                await new Promise(resolve => setTimeout(resolve, 100));
+              }
             }
-            core.warning(`Unmount failed, retrying (${attempt}/10)...`);
-            await new Promise(resolve => setTimeout(resolve, 300));
+            core.info('Unmounted device');
+          }
+        } catch (error) {
+          // grep returns exit code 1 when no matches are found.
+          if (error.code === 1) {
+            core.debug('No dangling mounts found to clean up');
+          } else {
+            // Only warn for actual errors, not for the expected case where grep finds nothing.
+            core.warning(`Error during cleanup: ${error.message}`);
           }
         }
-        core.info('Unmounted device');
 
         if (builderInfo.addr) {
           if (!buildError) {
@@ -341,9 +368,10 @@ actionsToolkit.run(
         }
       } catch (error) {
         core.warning(`Error during Blacksmith builder shutdown: ${error.message}`);
-        await reporter.reportBuildPushActionFailure(error);
+        await reporter.reportBuildPushActionFailure(error, "shutting down blacksmith builder");
+        core.setFailed(`Error during Blacksmith builder shutdown: ${error.message}`);
       } finally {
-        if (buildError) {
+        if (buildError || setupBlacksmithBuilderError) {
           try {
             const buildkitdLog = fs.readFileSync('buildkitd.log', 'utf8');
             core.info('buildkitd.log contents:');

src/reporter.ts

@@ -43,15 +43,15 @@ export function createBlacksmithAgentClient() {
   return createClient(StickyDiskService, transport);
 }
 
-export async function reportBuildPushActionFailure(error?: Error) {
+export async function reportBuildPushActionFailure(error?: Error, event?: string) {
   const requestOptions = {
     stickydisk_key: process.env.GITHUB_REPO_NAME || '',
     repo_name: process.env.GITHUB_REPO_NAME || '',
     region: process.env.BLACKSMITH_REGION || 'eu-central',
     arch: process.env.BLACKSMITH_ENV?.includes('arm') ? 'arm64' : 'amd64',
     vm_id: process.env.VM_ID || '',
     petname: process.env.PETNAME || '',
-    message: error?.message || ''
+    message: event ? `${event}: ${error?.message || ''}` : error?.message || ''
   };
 
   const client = createBlacksmithAPIClient();
@@ -66,7 +66,7 @@ export async function reportBuildCompleted(exportRes?: ExportRecordResponse, bla
   }
 
   try {
-    const agentClient = await createBlacksmithAgentClient();
+    const agentClient = createBlacksmithAgentClient();
 
     await agentClient.commitStickyDisk({
       exposeId: exposeId || '',
@@ -117,7 +117,7 @@ export async function reportBuildFailed(dockerBuildId: string | null, dockerBuil
   }
 
   try {
-    const blacksmithAgentClient = await createBlacksmithAgentClient();
+    const blacksmithAgentClient = createBlacksmithAgentClient();
     await blacksmithAgentClient.commitStickyDisk({
       exposeId: exposeId || '',
       stickyDiskKey: process.env.GITHUB_REPO_NAME || '',
@@ -166,17 +166,6 @@ export async function reportBuild(dockerfilePath: string) {
   }
 }
 
-export async function get(client: AxiosInstance, url: string, formData: FormData | null, options?: {signal?: AbortSignal}): Promise<AxiosResponse> {
-  return await client.get(url, {
-    ...(formData && {data: formData}),
-    headers: {
-      ...client.defaults.headers.common,
-      ...(formData && {'Content-Type': 'multipart/form-data'})
-    },
-    signal: options?.signal
-  });
-}
-
 export async function post(client: AxiosInstance, url: string, formData: FormData | null, options?: {signal?: AbortSignal}): Promise<AxiosResponse> {
   return await client.post(url, formData, {
     headers: {

src/setup_builder.ts

@@ -199,7 +199,7 @@ export async function startAndConfigureBuildkitd(parallelism: number, device: st
 export async function setupStickyDisk(dockerfilePath: string): Promise<{device: string; buildId?: string | null; exposeId: string}> {
   try {
     const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 10000);
+    const timeoutId = setTimeout(() => controller.abort(), 15000);
 
     let buildResponse: {docker_build_id: string} | null = null;
     let exposeId: string = '';