Add experimental_abortSignal to Node API

node-src/lib/tasks.ts

@@ -26,6 +26,7 @@ export const createTask = ({
 
     // eslint-disable-next-line no-restricted-syntax
     for (const step of steps) {
+      ctx.options.experimental_abortSignal?.throwIfAborted();
       // eslint-disable-next-line no-await-in-loop
       await step(ctx, task);
     }

node-src/main.test.ts

@@ -415,6 +415,15 @@ it('should exit with code 0 when the current branch is skipped', async () => {
   expect(ctx.exitCode).toBe(0);
 });
 
+it('should exit with code 6 and stop the build when abortSignal is aborted', async () => {
+  const abortSignal = AbortSignal.abort(new Error('Build canceled'));
+  const ctx = getContext(['--project-token=asdf1234']);
+  ctx.extraOptions = { experimental_abortSignal: abortSignal };
+  await runBuild(ctx);
+  expect(ctx.exitCode).toBe(6);
+  expect(uploadFiles).not.toHaveBeenCalled();
+});
+
 it('calls out to npm build script passed and uploads files', async () => {
   const ctx = getContext(['--project-token=asdf1234', '--build-script-name=build-storybook']);
   await runBuild(ctx);

node-src/runBuild.ts

@@ -1,12 +1,14 @@
 import Listr from 'listr';
 
 import GraphQLClient from './io/GraphQLClient';
+import { getConfiguration } from './lib/getConfiguration';
 import getOptions from './lib/getOptions';
 import NonTTYRenderer from './lib/NonTTYRenderer';
 import { exitCodes, setExitCode } from './lib/setExitCode';
 import { rewriteErrorMessage } from './lib/utils';
 import getTasks from './tasks';
-import { Context, Options } from './types';
+import { Context } from './types';
+import buildCanceled from './ui/messages/errors/buildCanceled';
 import fatalError from './ui/messages/errors/fatalError';
 import fetchError from './ui/messages/errors/fetchError';
 import graphqlError from './ui/messages/errors/graphqlError';
@@ -15,7 +17,6 @@ import runtimeError from './ui/messages/errors/runtimeError';
 import taskError from './ui/messages/errors/taskError';
 import intro from './ui/messages/info/intro';
 import { endActivity } from './ui/components/activity';
-import { getConfiguration } from './lib/getConfiguration';
 
 export async function runBuild(ctx: Context) {
   ctx.log.info('');
@@ -65,6 +66,11 @@ export async function runBuild(ctx: Context) {
         setExitCode(ctx, exitCodes.BUILD_NO_STORIES);
         throw rewriteErrorMessage(err, missingStories(ctx));
       }
+      if (ctx.extraOptions.experimental_abortSignal?.aborted) {
+        ctx.userError = true;
+        setExitCode(ctx, exitCodes.BUILD_WAS_CANCELED);
+        throw rewriteErrorMessage(err, buildCanceled());
+      }
       throw rewriteErrorMessage(err, taskError(ctx, err));
     } finally {
       // Handle potential runtime errors from JSDOM

node-src/types.ts

@@ -108,6 +108,9 @@ export interface Options {
 
   /** A callback that is called at the completion of each task */
   experimental_onTaskComplete?: (ctx: Context) => void;
+
+  /** An AbortSignal that terminates the build if aborted */
+  experimental_abortSignal?: AbortSignal;
 }
 
 export { Configuration };

node-src/ui/messages/errors/buildCanceled.stories.ts

@@ -0,0 +1,7 @@
+import buildCanceled from './buildCanceled';
+
+export default {
+  title: 'CLI/Messages/Errors',
+};
+
+export const BuildCanceled = () => buildCanceled();

node-src/ui/messages/errors/buildCanceled.ts

@@ -0,0 +1,10 @@
+import chalk from 'chalk';
+import { dedent } from 'ts-dedent';
+
+import { error } from '../../components/icons';
+
+export default () =>
+  dedent(chalk`
+    ${error} {bold Build canceled}
+    The build was canceled before it completed.
+  `);