refactor: shared fn for wrapping async errors

src/org/scratchOrgCreate.ts

@@ -12,6 +12,7 @@ import { ConfigAggregator } from '../config/configAggregator';
 import { OrgConfigProperties } from '../org/orgConfigProperties';
 import { SfProject } from '../sfProject';
 import { StateAggregator } from '../stateAggregator/stateAggregator';
+import { SfError } from '../sfError';
 import { Org } from './org';
 import {
   authorizeScratchOrg,
@@ -148,6 +149,15 @@ export const scratchOrgResume = async (jobId: string): Promise<ScratchOrgCreateR
         retry: 0,
       });
 
+  await setExitCodeIfError(68)(
+    scratchOrgAuthInfo.handleAliasAndDefaultSettings({
+      alias,
+      setDefault: setDefault ?? false,
+      setDefaultDevHub: false,
+      setTracksSource: tracksSource ?? true,
+    })
+  );
+
   const scratchOrg = await Org.create({ aliasOrUsername: username });
 
   const configAggregator = await ConfigAggregator.create();
@@ -160,23 +170,19 @@ export const scratchOrgResume = async (jobId: string): Promise<ScratchOrgCreateR
     capitalizeRecordTypes,
   });
   await settingsGenerator.extract({ ...soi, ...definitionjson });
-  const [authInfo] = await Promise.all([
-    resolveUrl(scratchOrgAuthInfo),
-    deploySettings(
-      scratchOrg,
-      settingsGenerator,
-      apiVersion ??
-        configAggregator.getPropertyValue(OrgConfigProperties.ORG_API_VERSION) ??
-        (await scratchOrg.retrieveMaxApiVersion())
-    ),
-  ]);
+  const [authInfo] = await setExitCodeIfError(68)(
+    Promise.all([
+      resolveUrl(scratchOrgAuthInfo),
+      deploySettings(
+        scratchOrg,
+        settingsGenerator,
+        apiVersion ??
+          configAggregator.getPropertyValue(OrgConfigProperties.ORG_API_VERSION) ??
+          (await scratchOrg.retrieveMaxApiVersion())
+      ),
+    ])
+  );
 
-  await scratchOrgAuthInfo.handleAliasAndDefaultSettings({
-    alias,
-    setDefault: setDefault ?? false,
-    setDefaultDevHub: false,
-    setTracksSource: tracksSource ?? true,
-  });
   cache.unset(soi.Id ?? jobId);
   const authFields = authInfo.getFields();
 
@@ -288,14 +294,16 @@ export const scratchOrgCreate = async (options: ScratchOrgCreateOptions): Promis
   });
 
   // anything after this point (org is created and auth'd) is potentially recoverable with the resume scratch command.
-  await scratchOrgAuthInfo.handleAliasAndDefaultSettings({
-    ...{
-      alias,
-      setDefault,
-      setDefaultDevHub: false,
-      setTracksSource: tracksSource === false ? false : true,
-    },
-  });
+  await setExitCodeIfError(68)(
+    scratchOrgAuthInfo.handleAliasAndDefaultSettings({
+      ...{
+        alias,
+        setDefault,
+        setDefaultDevHub: false,
+        setTracksSource: tracksSource === false ? false : true,
+      },
+    })
+  );
 
   // we'll need this scratch org connection later;
   const scratchOrg = await Org.create({ aliasOrUsername: soi.Username ?? soi.SignupUsername });
@@ -304,18 +312,20 @@ export const scratchOrgCreate = async (options: ScratchOrgCreateOptions): Promis
 
   await emit({ stage: 'deploy settings', scratchOrgInfo: soi });
   const configAggregator = await ConfigAggregator.create();
-  const [authInfo] = await Promise.all([
-    resolveUrl(scratchOrgAuthInfo),
-    deploySettings(
-      scratchOrg,
-      settingsGenerator,
-      apiversion ??
-        configAggregator.getPropertyValue(OrgConfigProperties.ORG_API_VERSION) ??
-        (await scratchOrg.retrieveMaxApiVersion()),
-      // some of our "wait" time has already been used.  Calculate how much remains that we can spend on the deployment.
-      Duration.milliseconds(wait.milliseconds - (Date.now() - startTimestamp))
-    ),
-  ]);
+  const [authInfo] = await setExitCodeIfError(68)(
+    Promise.all([
+      resolveUrl(scratchOrgAuthInfo),
+      deploySettings(
+        scratchOrg,
+        settingsGenerator,
+        apiversion ??
+          configAggregator.getPropertyValue(OrgConfigProperties.ORG_API_VERSION) ??
+          (await scratchOrg.retrieveMaxApiVersion()),
+        // some of our "wait" time has already been used.  Calculate how much remains that we can spend on the deployment.
+        Duration.milliseconds(wait.milliseconds - (Date.now() - startTimestamp))
+      ),
+    ])
+  );
 
   cache.unset(scratchOrgInfoId);
   const authFields = authInfo.getFields();
@@ -342,9 +352,18 @@ const getSignupTargetLoginUrl = async (): Promise<string | undefined> => {
 async function getCapitalizeRecordTypesConfig(): Promise<boolean | undefined> {
   const configAgg = await ConfigAggregator.create();
   const value = configAgg.getInfo('org-capitalize-record-types').value as string | undefined;
-
-  if (value !== undefined) return toBoolean(value);
-
-  // eslint-disable-next-line @typescript-eslint/no-unnecessary-type-assertion
-  return value as undefined;
+  return value !== undefined ? toBoolean(value) : undefined;
 }
+
+/** wrap an async function, intercept error and set the given exit code   */
+const setExitCodeIfError =
+  (exitCode: number) =>
+  async <P>(p: Promise<P>): Promise<P> => {
+    try {
+      return await p;
+    } catch (e) {
+      const sfError = e instanceof SfError ? e : SfError.wrap(e);
+      sfError.exitCode = exitCode;
+      throw sfError;
+    }
+  };

src/org/scratchOrgSettingsGenerator.ts

@@ -333,7 +333,6 @@ export default class SettingsGenerator {
         message: `A scratch org was created with username ${username}, but the settings failed to deploy due to: \n${failures}`,
         name: 'ProblemDeployingSettings',
         data: { ...result, username },
-        exitCode: 68,
       });
     }
   }

test/unit/org/scratchOrgSettingsGeneratorTest.ts

@@ -111,7 +111,7 @@ function createStubs() {
       id: '1',
     })
   );
-  sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'mybuffer');
+  sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'myBuffer');
   getUsernameStub = sandbox.stub(scratchOrg, 'getUsername').returns(adminTestData.username);
 }
 
@@ -266,7 +266,7 @@ describe('scratchOrgSettingsGenerator', () => {
           id: '1',
         })
       );
-      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'mybuffer');
+      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'myBuffer');
       getUsernameStub = sandbox.stub(scratchOrg, 'getUsername').returns(adminTestData.username);
       getConnectionStub = fakeConnection(sandbox, scratchOrg, deployId, 'SucceededPartial');
     });
@@ -329,7 +329,7 @@ describe('scratchOrgSettingsGenerator', () => {
           id: '1',
         })
       );
-      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'mybuffer');
+      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'myBuffer');
       getUsernameStub = sandbox.stub(scratchOrg, 'getUsername').returns(adminTestData.username);
       getConnectionStub = fakeConnection(sandbox, scratchOrg, deployId, 'Failed');
     });
@@ -392,7 +392,7 @@ describe('scratchOrgSettingsGenerator', () => {
           id: '1',
         })
       );
-      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'mybuffer');
+      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'myBuffer');
       getUsernameStub = sandbox.stub(scratchOrg, 'getUsername').returns(adminTestData.username);
       getConnectionStub = fakeConnection(sandbox, scratchOrg, deployId, ['InProgress', 'Succeeded']);
     });
@@ -401,7 +401,7 @@ describe('scratchOrgSettingsGenerator', () => {
       sandbox.restore();
     });
 
-    it('tries to deploy the settings to the org pools untill succeded', async () => {
+    it('tries to deploy the settings to the org pools until succeeded', async () => {
       const scratchDef = {
         ...TEMPLATE_SCRATCH_ORG_INFO,
         settings: {
@@ -497,7 +497,7 @@ describe('scratchOrgSettingsGenerator', () => {
           id: '1',
         })
       );
-      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'mybuffer');
+      sandbox.stub(ZipWriter.prototype, 'buffer').get(() => 'myBuffer');
       getUsernameStub = sandbox.stub(scratchOrg, 'getUsername').returns(adminTestData.username);
       getConnectionStub = fakeConnection(sandbox, scratchOrg, deployId, 'InProgress');
     });
@@ -507,7 +507,7 @@ describe('scratchOrgSettingsGenerator', () => {
       sandbox.restore();
     });
 
-    it('tries to deploy the settings to the org pools untill timeouts', async () => {
+    it('tries to deploy the settings to the org pools until timeouts', async () => {
       const timeout = 10 * 60 * 1000; // 10 minutes
       const frequency = 1000;
       const settings = new SettingsGenerator();