push: Await and retry supervisor API requests to a local device

Change-type: patch

lib/utils/device/api.ts

@@ -20,6 +20,7 @@ import * as os from 'os';
 import * as request from 'request';
 import type * as Stream from 'stream';
 
+import { retry } from '../helpers';
 import Logger = require('../logger');
 import * as ApiErrors from './errors';
 
@@ -260,7 +261,8 @@ export class DeviceAPI {
 			}
 		}
 
-		return new Promise((resolve, reject) => {
+		const doRequest = async () => {
+			return await new Promise((resolve, reject) => {
 				return request(opts, (err, response, body) => {
 					if (err) {
 						return reject(err);
@@ -269,7 +271,9 @@ export class DeviceAPI {
 						case 200:
 							return resolve(body);
 						case 400:
-						return reject(new ApiErrors.BadRequestDeviceAPIError(body.message));
+							return reject(
+								new ApiErrors.BadRequestDeviceAPIError(body.message),
+							);
 						case 503:
 							return reject(
 								new ApiErrors.ServiceUnavailableAPIError(body.message),
@@ -279,6 +283,14 @@ export class DeviceAPI {
 					}
 				});
 			});
+		};
+
+		return await retry({
+			func: doRequest,
+			initialDelayMs: 2000,
+			maxAttempts: 6,
+			label: `Supervisor API (${opts.method} ${(opts as any).url})`,
+		});
 	}
 }
 

lib/utils/device/logs.ts

@@ -80,16 +80,19 @@ async function displayDeviceLogs(
 			jsonStream.on('error', (e) => {
 				logger.logWarn(`Error parsing NDJSON log chunk: ${e}`);
 			});
-			logs.once('error', reject);
+			logs.once('error', handleError);
-			logs.once('end', () => {
+			logs.once('end', handleError);
+			logs.pipe(jsonStream);
+
+			function handleError(error?: Error | string) {
 				logger.logWarn(DeviceConnectionLostError.defaultMsg);
 				if (gotSignal) {
 					reject(new SIGINTError('Log streaming aborted on SIGINT signal'));
 				} else {
-					reject(new DeviceConnectionLostError());
+					const msg = typeof error === 'string' ? error : error?.message;
+					reject(new DeviceConnectionLostError(msg));
+				}
 			}
-			});
-			logs.pipe(jsonStream);
 		});
 	} finally {
 		process.removeListener('SIGINT', handleSignal);