[AP 3560] Build UI for build status

package.json

@@ -62,6 +62,7 @@
     "@storybook/design-system": "^7.15.15",
     "chromatic": "^7.1.0",
     "date-fns": "^2.30.0",
+    "filesize": "^10.0.12",
     "pluralize": "^8.0.0",
     "ts-dedent": "^2.2.0",
     "urql": "^4.0.3",

src/Panel.tsx

@@ -4,22 +4,13 @@ import { useChannel, useStorybookState } from "@storybook/manager-api";
 import React, { useCallback } from "react";
 
 import { Sections } from "./components/layout";
-import {
-  ADDON_ID,
-  GIT_INFO,
-  GitInfoPayload,
-  IS_OUTDATED,
-  PANEL_ID,
-  RUNNING_BUILD,
-  RunningBuildPayload,
-  START_BUILD,
-} from "./constants";
+import { ADDON_ID, GIT_INFO, IS_OUTDATED, PANEL_ID, RUNNING_BUILD, START_BUILD } from "./constants";
 import { Authentication } from "./screens/Authentication/Authentication";
 import { LinkedProject } from "./screens/LinkProject/LinkedProject";
 import { LinkingProjectFailed } from "./screens/LinkProject/LinkingProjectFailed";
 import { LinkProject } from "./screens/LinkProject/LinkProject";
 import { VisualTests } from "./screens/VisualTests/VisualTests";
-import { UpdateStatusFunction } from "./types";
+import { GitInfoPayload, RunningBuildPayload, UpdateStatusFunction } from "./types";
 import { useAddonState } from "./useAddonState/manager";
 import { client, Provider, useAccessToken } from "./utils/graphQLClient";
 import { useProjectId } from "./utils/useProjectId";

src/SidebarTop.tsx

@@ -4,13 +4,8 @@ import pluralize from "pluralize";
 import React, { useEffect, useRef } from "react";
 
 import { SidebarTopButton } from "./components/SidebarTopButton";
-import {
-  ADDON_ID,
-  IS_OUTDATED,
-  RUNNING_BUILD,
-  RunningBuildPayload,
-  START_BUILD,
-} from "./constants";
+import { ADDON_ID, IS_OUTDATED, RUNNING_BUILD, START_BUILD } from "./constants";
+import { RunningBuildPayload } from "./types";
 import { useAddonState } from "./useAddonState/manager";
 import { useAccessToken } from "./utils/graphQLClient";
 import { useProjectId } from "./utils/useProjectId";
@@ -28,14 +23,14 @@ export const SidebarTop = ({ api }: SidebarTopProps) => {
 
   const [isOutdated] = useAddonState<boolean>(IS_OUTDATED);
   const [runningBuild] = useAddonState<RunningBuildPayload>(RUNNING_BUILD);
-  const isRunning = !!runningBuild && runningBuild.step !== "complete";
+  const isRunning = !!runningBuild && runningBuild.currentStep !== "complete";
 
-  const lastStep = useRef(runningBuild?.step);
+  const lastStep = useRef(runningBuild?.currentStep);
   useEffect(() => {
-    if (runningBuild?.step === lastStep.current) return;
-    lastStep.current = runningBuild?.step;
+    if (runningBuild?.currentStep === lastStep.current) return;
+    lastStep.current = runningBuild?.currentStep;
 
-    if (runningBuild?.step === "initialize") {
+    if (runningBuild?.currentStep === "initialize") {
       addNotification({
         id: `${ADDON_ID}/build-initialize`,
         content: {
@@ -51,7 +46,7 @@ export const SidebarTop = ({ api }: SidebarTopProps) => {
       setTimeout(() => clearNotification(`${ADDON_ID}/build-initialize`), 10_000);
     }
 
-    if (runningBuild?.step === "complete") {
+    if (runningBuild?.currentStep === "complete") {
       addNotification({
         id: `${ADDON_ID}/build-complete`,
         content: {
@@ -72,7 +67,7 @@ export const SidebarTop = ({ api }: SidebarTopProps) => {
       setTimeout(() => clearNotification(`${ADDON_ID}/build-complete`), 10_000);
     }
 
-    if (runningBuild?.step === "error") {
+    if (runningBuild?.currentStep === "error") {
       addNotification({
         id: `${ADDON_ID}/build-error`,
         content: {
@@ -89,7 +84,7 @@ export const SidebarTop = ({ api }: SidebarTopProps) => {
   }, [
     addNotification,
     clearNotification,
-    runningBuild?.step,
+    runningBuild?.currentStep,
     runningBuild?.errorCount,
     runningBuild?.changeCount,
   ]);

src/buildSteps.ts

@@ -0,0 +1,115 @@
+// eslint-disable-next-line import/no-unresolved
+import { TaskName } from "chromatic/node";
+import { filesize } from "filesize";
+
+import { KnownStep, RunningBuildPayload, StepProgressPayload } from "./types";
+
+export const isKnownStep = (task: TaskName): task is KnownStep =>
+  BUILD_STEP_ORDER.includes(task as KnownStep);
+
+// Note this does not include the "complete" and "error" steps
+export const BUILD_STEP_ORDER: KnownStep[] = [
+  "initialize",
+  "build",
+  "upload",
+  "verify",
+  "snapshot",
+];
+
+export const BUILD_STEP_CONFIG: Record<
+  RunningBuildPayload["currentStep"],
+  {
+    key: RunningBuildPayload["currentStep"];
+    emoji: string;
+    renderName: () => string;
+    renderProgress: (payload: RunningBuildPayload) => string;
+    renderComplete: () => string;
+    estimateDuration: number;
+  }
+> = {
+  initialize: {
+    key: "initialize",
+    emoji: "🚀",
+    renderName: () => `Initialize build`,
+    renderProgress: () => `Initializing build`,
+    renderComplete: () => `Initialized`,
+    estimateDuration: 2000,
+  },
+  build: {
+    key: "build",
+    emoji: "🏗",
+    renderName: () => `Build Storybook`,
+    renderProgress: () => `Building your Storybook...`,
+    renderComplete: () => `Storybook built`,
+    estimateDuration: 30_000,
+  },
+  upload: {
+    key: "upload",
+    emoji: "📡",
+    renderName: () => `Publish your Storybook`,
+    renderProgress: ({ stepProgress }) => {
+      const { numerator, denominator } = stepProgress.upload;
+      if (!denominator) return `Uploading files`;
+      const { value: total, exponent } = filesize(denominator, {
+        output: "object",
+        round: 1,
+      });
+      const { value: progress, symbol } = filesize(numerator, {
+        exponent,
+        output: "object",
+        round: 1,
+      });
+      return `Uploading files (${progress}/${total} ${symbol})`;
+    },
+    renderComplete: () => `Publish complete`,
+    estimateDuration: 30_000,
+  },
+  verify: {
+    key: "verify",
+    emoji: "🔍",
+    renderName: () => `Verify your Storybook`,
+    renderProgress: () => `Verifying contents...`,
+    renderComplete: () => `Storybook verified`,
+    estimateDuration: 10_000,
+  },
+  snapshot: {
+    key: "snapshot",
+    emoji: "📸",
+    renderName: () => `Run visual tests`,
+    renderProgress: ({ stepProgress }) => {
+      const { numerator, denominator } = stepProgress.snapshot;
+      return denominator
+        ? `Running visual tests (${numerator}/${denominator})`
+        : `Running visual tests`;
+    },
+    renderComplete: () => `Tested your stories`,
+    estimateDuration: 60_000,
+  },
+
+  // These are special steps that are not part of the build process
+  complete: {
+    key: "complete",
+    emoji: "🎉",
+    renderName: () => `Visual tests completed!`,
+    renderProgress: () => `Visual tests completed!`,
+    renderComplete: () => `Visual tests completed!`,
+    estimateDuration: 0,
+  },
+  error: {
+    key: "error",
+    emoji: "🚨",
+    renderName: () => `Build failed`,
+    renderProgress: () => `Build failed`,
+    renderComplete: () => `Build failed`,
+    estimateDuration: 0,
+  },
+};
+
+export const INITIAL_BUILD_PAYLOAD = {
+  buildProgressPercentage: 0,
+  currentStep: BUILD_STEP_ORDER[0],
+  stepProgress: Object.fromEntries(BUILD_STEP_ORDER.map((step) => [step, {}])) as Record<
+    KnownStep,
+    StepProgressPayload
+  >,
+};

src/components/BuildProgressLabel.tsx

@@ -1,21 +1,17 @@
 import React from "react";
 
-import { RunningBuildPayload } from "../constants";
-
-const messageMap: Record<RunningBuildPayload["step"], (payload: RunningBuildPayload) => string> = {
-  initialize: () => `📦 Validating Storybook files...`,
-  build: () => `📦 Validating Storybook files...`,
-  upload: () => `📡 Uploading to Chromatic...`, // TODO represent progress in bytes
-  verify: () => `🛠️ Initiating build...`, // TODO build number
-  snapshot: () => `👀 Running visual tests...`, // TODO count
-  complete: () => `🎉 Visual tests completed!`,
-  error: () => `❌ Build failed`, // TODO error
-};
+import { BUILD_STEP_CONFIG } from "../buildSteps";
+import { RunningBuildPayload } from "../types";
 
 interface BuildProgressLabelProps {
   runningBuild: RunningBuildPayload;
 }
 
-export const BuildProgressLabel = ({ runningBuild }: BuildProgressLabelProps) => (
-  <>{messageMap[runningBuild.step](runningBuild)}</>
-);
+export const BuildProgressLabel = ({ runningBuild }: BuildProgressLabelProps) => {
+  const { emoji, renderProgress } = BUILD_STEP_CONFIG[runningBuild.currentStep];
+  return (
+    <>
+      {emoji} {renderProgress(runningBuild)}
+    </>
+  );
+};

src/components/SidebarTopButton.stories.ts

@@ -2,6 +2,7 @@ import { action } from "@storybook/addon-actions";
 import type { Meta, StoryObj } from "@storybook/react";
 import { userEvent, within } from "@storybook/testing-library";
 
+import { INITIAL_BUILD_PAYLOAD } from "../buildSteps";
 import { playAll } from "../utils/playAll";
 import { SidebarTopButton } from "./SidebarTopButton";
 
@@ -33,8 +34,9 @@ export const IsRunning: Story = {
   args: {
     isRunning: true,
     runningBuild: {
-      step: "build",
+      ...INITIAL_BUILD_PAYLOAD,
       buildProgressPercentage: 40,
+      currentStep: "build",
     },
   },
   play: playAll(async ({ canvasElement }) => {

src/components/SidebarTopButton.tsx

@@ -2,13 +2,13 @@ import { Icons, WithTooltip } from "@storybook/components";
 import { styled } from "@storybook/theming";
 import React, { ComponentProps } from "react";
 
-import { RunningBuildPayload } from "../constants";
+import { RunningBuildPayload } from "../types";
 import { BuildProgressLabel } from "./BuildProgressLabel";
 import { IconButton } from "./IconButton";
 import { StatusDotWrapper } from "./StatusDot";
 
 export const TooltipContent = styled.div(({ theme }) => ({
-  width: 200,
+  width: 220,
   padding: 3,
   color: theme.color.defaultText,
 
@@ -25,6 +25,7 @@ export const ProgressTrack = styled.div(({ theme }) => ({
 
 export const ProgressBar = styled(ProgressTrack)(({ theme }) => ({
   background: theme.color.secondary,
+  transition: "width 3s ease-out",
 }));
 
 export const ProgressCircle = styled.svg(({ theme }) => ({