@gfxlabs/third-eye-cli 3.24.3 → 3.25.1

package/dist/bin.mjs

@@ -27,6 +27,16 @@ import { RPCLink } from "@orpc/client/fetch";
 import { execa } from "execa";
 import { XMLParser } from "fast-xml-parser";
 //#region \0rolldown/runtime.js
+var __defProp = Object.defineProperty;
+var __exportAll = (all, no_symbols) => {
+	let target = {};
+	for (var name in all) __defProp(target, name, {
+		get: all[name],
+		enumerable: true
+	});
+	if (!no_symbols) __defProp(target, Symbol.toStringTag, { value: "Module" });
+	return target;
+};
 var __require = /* @__PURE__ */ createRequire(import.meta.url);
 //#endregion
 //#region src/log.ts
@@ -184,7 +194,8 @@ z$1.object({
 	storyName: z$1.string().optional(),
 	storyId: z$1.string().optional(),
 	storyArgs: z$1.record(z$1.string(), z$1.unknown()).optional(),
-	tags: z$1.array(z$1.string()).optional()
+	tags: z$1.array(z$1.string()).optional(),
+	designUrl: z$1.string().optional()
 });
 //#endregion
 //#region src/config.ts
@@ -1035,7 +1046,8 @@ const generateStorybookShotItems = (baseUrl, stories, mask, modeBreakpoints, bro
 			componentPath: story.kind,
 			storyName: story.story,
 			storyId: story.id,
-			storyArgs: story.parameters?.thirdeye?.args
+			storyArgs: story.parameters?.thirdeye?.args,
+			designUrl: (story.parameters?.design)?.url
 		};
 		const storyLevelBreakpoints = story.parameters?.thirdeye?.breakpoints ?? [];
 		const breakpoints = selectBreakpoints(config.breakpoints, modeBreakpoints, storyLevelBreakpoints);
@@ -1229,6 +1241,22 @@ const getPagesFromExternalLoader = async () => {
 };
 //#endregion
 //#region src/shots/shots.ts
+/**
+* Wait for Storybook play functions to complete before screenshotting.
+* Checks __STORYBOOK_PREVIEW__.currentRender.phase and data-play-fn-done attribute.
+*/
+const waitForPlayFunction = async (page, shotItem, logger) => {
+	if (shotItem.shotMode !== "storybook") return;
+	try {
+		await page.waitForFunction(() => {
+			if (window.__STORYBOOK_PREVIEW__?.currentRender?.phase === "playing") return false;
+			if (document.getElementById("storybook-root")?.dataset.playFnDone === "false") return false;
+			return true;
+		}, { timeout: config.timeouts?.loadState ?? 3e4 });
+	} catch {
+		logger.process("info", "general", `Play function wait timed out for ${shotItem.id} — proceeding with screenshot`);
+	}
+};
 const takeScreenShot = async ({ browser, shotItem, logger }) => {
 	const context = await browser.newContext(shotItem.browserConfig);
 	const page = await context.newPage();
@@ -1273,6 +1301,7 @@ const takeScreenShot = async ({ browser, shotItem, logger }) => {
 	} catch (error) {
 		logger.process("error", "timeout", `Timeout while waiting for all network requests: ${shotItem.url}`, error);
 	}
+	await waitForPlayFunction(page, shotItem, logger);
 	if (config.beforeScreenshot) await config.beforeScreenshot(page, {
 		shotMode: shotItem.shotMode,
 		id: shotItem.id,
@@ -1640,6 +1669,12 @@ const createShots = async (turboSnapFilter) => {
 //#region src/git.ts
 const INITIAL_BATCH_SIZE = 20;
 /**
+* Check if a commit exists in the local git index.
+*/
+const commitExists = (commit) => {
+	return execGit(`git cat-file -t ${commit}`) === "commit";
+};
+/**
 * Execute a git command and return the trimmed output.
 */
 const execGit = (command) => {
@@ -1652,34 +1687,28 @@ const execGit = (command) => {
 		return "";
 	}
 };
-/**
-* Find the "covering" set of ancestor commits that have builds on the server.
-*
-* This mirrors Chromatic's approach:
-* 1. Walk git history with `git rev-list`
-* 2. Ask the server which commits have builds
-* 3. Use `--not` to exclude ancestors of commits with builds
-* 4. Repeat with exponentially larger batches until no more uncovered commits
-*
-* The result is the minimal set of ancestor commits with builds such that
-* every ancestor of HEAD either has a build or is an ancestor of a commit
-* with a build.
-*/
-const getParentCommits = async (hasBuildsWithCommits) => {
+const getParentCommits = async (hasBuildsWithCommits, options) => {
 	if (!execGit("git rev-parse HEAD")) {
 		log.process("info", "general", "Not a git repository, skipping ancestor detection");
-		return [];
+		return {
+			parentCommits: [],
+			visitedCommitsWithoutBuilds: []
+		};
 	}
 	if (!execGit("git --no-pager log -n 1 --skip=1 --format=\"%H\"")) {
 		log.process("info", "general", "Initial commit, no ancestors");
-		return [];
+		return {
+			parentCommits: [],
+			visitedCommitsWithoutBuilds: []
+		};
 	}
-	let commitsWithBuilds = [];
+	const sinceArg = options?.firstCommittedAtSeconds ? `--since ${options.firstCommittedAtSeconds}` : "";
+	let commitsWithBuilds = [...options?.initialCommitsWithBuilds ?? []];
 	let commitsWithoutBuilds = [];
 	let limit = INITIAL_BATCH_SIZE;
 	for (;;) {
 		const notArgs = commitsWithBuilds.map((c) => c.trim()).join(" ");
-		const output = execGit(`git rev-list HEAD -n ${limit + commitsWithoutBuilds.length}${notArgs ? ` --not ${notArgs}` : ""}`);
+		const output = execGit(`git rev-list HEAD ${sinceArg} -n ${limit + commitsWithoutBuilds.length}${notArgs ? ` --not ${notArgs}` : ""}`);
 		const candidates = (output ? output.split("\n").filter(Boolean) : []).filter((c) => !commitsWithBuilds.includes(c)).filter((c) => !commitsWithoutBuilds.includes(c)).slice(0, limit);
 		if (candidates.length === 0) break;
 		log.process("info", "general", `🔍 Checking ${candidates.length} commits for builds (batch size ${limit})`);
@@ -1695,7 +1724,10 @@ const getParentCommits = async (hasBuildsWithCommits) => {
 	}
 	if (commitsWithBuilds.length === 0) {
 		log.process("info", "general", "No ancestor builds found — this may be the first build");
-		return [];
+		return {
+			parentCommits: [],
+			visitedCommitsWithoutBuilds: commitsWithoutBuilds
+		};
 	}
 	if (commitsWithBuilds.length > 1) {
 		const parentArgs = commitsWithBuilds.map((c) => `"${c}^@"`).join(" ");
@@ -1704,7 +1736,31 @@ const getParentCommits = async (hasBuildsWithCommits) => {
 		if (maxCommits.length > 0) commitsWithBuilds = maxCommits;
 	}
 	log.process("info", "general", `📌 Found ${commitsWithBuilds.length} ancestor build(s): ${commitsWithBuilds.map((c) => c.slice(0, 7)).join(", ")}`);
-	return commitsWithBuilds;
+	return {
+		parentCommits: commitsWithBuilds,
+		visitedCommitsWithoutBuilds: commitsWithoutBuilds
+	};
+};
+/**
+* Compute the merge base commit between the current HEAD and a base branch,
+* plus a list of ancestor commits on the base branch from that point backwards.
+* This gives the server enough git history to walk back and find a build
+* even if the exact merge base commit doesn't have one.
+*/
+const getMergeBaseInfo = (baseBranch) => {
+	const mergeBaseCommit = execGit(`git merge-base HEAD ${execGit(`git rev-parse origin/${baseBranch}`) ? `origin/${baseBranch}` : baseBranch}`);
+	if (!mergeBaseCommit) {
+		log.process("info", "general", `Could not compute merge base with ${baseBranch}`);
+		return null;
+	}
+	log.process("info", "general", `📍 Merge base with ${baseBranch}: ${mergeBaseCommit.slice(0, 7)}`);
+	const ancestorsOutput = execGit(`git rev-list ${mergeBaseCommit} -n 50`);
+	const mergeBaseAncestors = ancestorsOutput ? ancestorsOutput.split("\n").filter(Boolean) : [];
+	log.process("info", "general", `📍 Got ${mergeBaseAncestors.length} merge base ancestors for build lookup`);
+	return {
+		mergeBaseCommit,
+		mergeBaseAncestors
+	};
 };
 //#endregion
 //#region ../shared/dist/client.js
@@ -1744,6 +1800,21 @@ const createTypedClient = (options) => {
 };
 //#endregion
 //#region src/api.ts
+var api_exports = /* @__PURE__ */ __exportAll({
+	getAffectedStories: () => getAffectedStories,
+	getApiToken: () => getApiToken,
+	getGitInfoFromAPI: () => getGitInfoFromAPI,
+	prepareUpload: () => prepareUpload,
+	processShots: () => processShots,
+	sendCheckCacheToAPI: () => sendCheckCacheToAPI,
+	sendFinalizeToAPI: () => sendFinalizeToAPI,
+	sendFindSquashMergeParentsToAPI: () => sendFindSquashMergeParentsToAPI,
+	sendHasBuildsWithCommitsToAPI: () => sendHasBuildsWithCommitsToAPI,
+	sendInitToAPI: () => sendInitToAPI,
+	sendRecordLogsToAPI: () => sendRecordLogsToAPI,
+	uploadShot: () => uploadShot,
+	uploadStorybookArchive: () => uploadStorybookArchive
+});
 const createClient = (platformUrl, apiKey, apiToken) => createTypedClient({
 	url: platformUrl,
 	apiKey,
@@ -1775,7 +1846,7 @@ const getApiToken = async (config) => {
 		process.exit(1);
 	}
 };
-const sendInitToAPI = async (config, apiToken, parentCommits) => {
+const sendInitToAPI = async (config, apiToken, parentCommits, mergeBaseInfo) => {
 	const client = createClient(config.thirdEyePlatform, void 0, apiToken);
 	return withRetry("init", () => client.orgs.projects.builds.init({
 		orgId: config.thirdEyeOrgId,
@@ -1785,7 +1856,9 @@ const sendInitToAPI = async (config, apiToken, parentCommits) => {
 		buildNumber: config.ciBuildNumber,
 		baseBranch: config.baseBranch || void 0,
 		prNumber: config.prNumber,
-		parentCommits
+		parentCommits,
+		mergeBaseCommit: mergeBaseInfo?.mergeBaseCommit,
+		mergeBaseAncestors: mergeBaseInfo?.mergeBaseAncestors
 	}));
 };
 const sendHasBuildsWithCommitsToAPI = async (config, apiToken, commits) => {
@@ -1796,6 +1869,22 @@ const sendHasBuildsWithCommitsToAPI = async (config, apiToken, commits) => {
 		commits
 	}))).commits;
 };
+const getGitInfoFromAPI = async (config, apiToken, branch) => {
+	const client = createClient(config.thirdEyePlatform, void 0, apiToken);
+	return withRetry("getGitInfo", () => client.orgs.projects.builds.getGitInfo({
+		orgId: config.thirdEyeOrgId,
+		projectId: config.thirdEyeProjectId,
+		branch
+	}));
+};
+const sendFindSquashMergeParentsToAPI = async (config, apiToken, commits) => {
+	const client = createClient(config.thirdEyePlatform, void 0, apiToken);
+	return (await withRetry("findSquashMergeParents", () => client.orgs.projects.builds.findSquashMergeParents({
+		orgId: config.thirdEyeOrgId,
+		projectId: config.thirdEyeProjectId,
+		commits
+	}))).parents;
+};
 const sendFinalizeToAPI = async (config, apiToken) => {
 	const client = createClient(config.thirdEyePlatform, void 0, apiToken);
 	return withRetry("finalize", () => client.orgs.projects.builds.finalize({
@@ -1918,7 +2007,8 @@ const uploadRequiredShots = async ({ config, apiToken, uploadToken, requiredFile
 					...shotItem.viewport ? { viewport: shotItem.viewport } : {},
 					...shotItem.breakpoint !== void 0 ? { breakpoint: shotItem.breakpoint } : {},
 					...domHtml ? { dom_html: domHtml } : {},
-					...dependencyMap && shotItem.importPath && dependencyMap.has(shotItem.importPath) ? { dependencies: dependencyMap.get(shotItem.importPath) } : {}
+					...dependencyMap && shotItem.importPath && dependencyMap.has(shotItem.importPath) ? { dependencies: dependencyMap.get(shotItem.importPath) } : {},
+					...shotItem.designUrl ? { designUrl: shotItem.designUrl } : {}
 				},
 				logger
 			});
@@ -2195,8 +2285,44 @@ const platformRunner = async (config, apiToken) => {
 			`commitRefName = ${config.commitRefName}`,
 			`commitHash = ${config.commitHash}`
 		].join("\n   - "));
+		log.process("info", "general", "🔍 Querying server for git info...");
+		const gitInfo = await getGitInfoFromAPI(config, apiToken, config.commitRefName);
+		const initialCommitsWithBuilds = [];
+		if (gitInfo.lastBuildOnBranch) {
+			const lastCommit = gitInfo.lastBuildOnBranch.commit;
+			if (commitExists(lastCommit)) {
+				log.process("info", "general", `📌 Seeding with last build on branch: ${lastCommit.slice(0, 7)}`);
+				initialCommitsWithBuilds.push(lastCommit);
+			} else log.process("info", "general", `📌 Last build on branch ${lastCommit.slice(0, 7)} not in local git (rebase?), skipping`);
+		}
+		if (config.baseBranch) {
+			const mergeBase = getMergeBaseInfo(config.baseBranch);
+			if (mergeBase) {
+				const mbCommitsWithBuilds = await sendHasBuildsWithCommitsToAPI(config, apiToken, mergeBase.mergeBaseAncestors);
+				for (const c of mbCommitsWithBuilds) if (!initialCommitsWithBuilds.includes(c)) {
+					log.process("info", "general", `📌 Seeding with base branch build: ${c.slice(0, 7)}`);
+					initialCommitsWithBuilds.push(c);
+				}
+			}
+		}
 		log.process("info", "general", "🔍 Resolving ancestor builds from git history...");
-		await sendInitToAPI(config, apiToken, await getParentCommits((commits) => sendHasBuildsWithCommitsToAPI(config, apiToken, commits)));
+		const { parentCommits, visitedCommitsWithoutBuilds } = await getParentCommits((commits) => sendHasBuildsWithCommitsToAPI(config, apiToken, commits), {
+			firstCommittedAtSeconds: gitInfo.firstBuildCreatedAt ?? void 0,
+			initialCommitsWithBuilds: initialCommitsWithBuilds.length > 0 ? initialCommitsWithBuilds : void 0
+		});
+		if (visitedCommitsWithoutBuilds.length > 0) {
+			log.process("info", "general", `🔀 Checking ${Math.min(visitedCommitsWithoutBuilds.length, 100)} commits for squash merges...`);
+			try {
+				const squashResult = await sendFindSquashMergeParentsToAPI(config, apiToken, visitedCommitsWithoutBuilds.slice(0, 100));
+				for (const { buildCommit } of squashResult) if (commitExists(buildCommit) && !parentCommits.includes(buildCommit)) {
+					log.process("info", "general", `🔀 Found squash merge parent: ${buildCommit.slice(0, 7)}`);
+					parentCommits.push(buildCommit);
+				}
+			} catch (err) {
+				log.process("info", "general", `Squash merge detection failed (non-fatal): ${err instanceof Error ? err.message : String(err)}`);
+			}
+		}
+		await sendInitToAPI(config, apiToken, parentCommits, config.baseBranch ? getMergeBaseInfo(config.baseBranch) : null);
 		if (!await checkForCachedBuild(config, apiToken)) {
 			log.process("info", "general", "📂 Creating shot folders");
 			const createShotsStart = process.hrtime();
@@ -2298,6 +2424,90 @@ const platformRunner = async (config, apiToken) => {
 		process.exit(1);
 	}
 };
+const patchBuildRunner = async (patchBuildArg, config, apiToken) => {
+	const { execSync } = await import("node:child_process");
+	const parts = patchBuildArg.split("...");
+	if (parts.length !== 2) {
+		log.process("error", "general", `Invalid --patch-build format. Expected: head...base (e.g. feature...main)`);
+		process.exit(1);
+	}
+	const [headRef, baseRef] = parts;
+	log.process("info", "general", `🩹 Patch build: computing merge base between ${headRef} and ${baseRef}...`);
+	let mergeBaseCommit;
+	try {
+		mergeBaseCommit = execSync(`git merge-base ${headRef} ${baseRef}`, { encoding: "utf-8" }).trim();
+	} catch {
+		log.process("error", "general", `Failed to compute merge base between ${headRef} and ${baseRef}`);
+		process.exit(1);
+	}
+	log.process("info", "general", `📍 Merge base commit: ${mergeBaseCommit.slice(0, 7)}`);
+	const { sendHasBuildsWithCommitsToAPI: checkCommits } = await Promise.resolve().then(() => api_exports);
+	if ((await checkCommits(config, apiToken, [mergeBaseCommit])).length > 0) {
+		log.process("info", "general", `✅ Build already exists for merge base ${mergeBaseCommit.slice(0, 7)}, no patch build needed.`);
+		return;
+	}
+	const originalRef = execSync("git rev-parse HEAD", { encoding: "utf-8" }).trim();
+	const originalBranch = execSync("git rev-parse --abbrev-ref HEAD", { encoding: "utf-8" }).trim();
+	log.process("info", "general", `📦 Saving current state (${originalBranch} @ ${originalRef.slice(0, 7)})...`);
+	let hasStash = false;
+	try {
+		hasStash = !execSync("git stash", { encoding: "utf-8" }).trim().includes("No local changes");
+	} catch {}
+	try {
+		log.process("info", "general", `🔄 Checking out merge base ${mergeBaseCommit.slice(0, 7)}...`);
+		execSync(`git checkout ${mergeBaseCommit}`, { stdio: "pipe" });
+		log.process("info", "general", "📦 Installing dependencies...");
+		try {
+			execSync("pnpm install --frozen-lockfile", {
+				stdio: "pipe",
+				timeout: 12e4
+			});
+		} catch {
+			try {
+				execSync("npm ci", {
+					stdio: "pipe",
+					timeout: 12e4
+				});
+			} catch {
+				execSync("npm install", {
+					stdio: "pipe",
+					timeout: 12e4
+				});
+			}
+		}
+		log.process("info", "general", "📖 Building Storybook...");
+		if (config.storybookShots?.storybookUrl) try {
+			execSync("npx storybook build", {
+				stdio: "pipe",
+				timeout: 3e5
+			});
+		} catch {
+			log.process("info", "general", "Storybook build command failed, trying build-storybook...");
+			execSync("npx build-storybook", {
+				stdio: "pipe",
+				timeout: 3e5
+			});
+		}
+		const originalCommitHash = config.commitHash;
+		const originalRefName = config.commitRefName;
+		config.commitHash = mergeBaseCommit;
+		config.commitRefName = baseRef;
+		log.process("info", "general", "🚀 Running Third Eye for merge base...");
+		await platformRunner(config, apiToken);
+		config.commitHash = originalCommitHash;
+		config.commitRefName = originalRefName;
+		log.process("info", "general", `✅ Patch build complete for merge base ${mergeBaseCommit.slice(0, 7)}`);
+	} finally {
+		log.process("info", "general", `🔄 Restoring original state...`);
+		try {
+			if (originalBranch === "HEAD") execSync(`git checkout ${originalRef}`, { stdio: "pipe" });
+			else execSync(`git checkout ${originalBranch}`, { stdio: "pipe" });
+			if (hasStash) execSync("git stash pop", { stdio: "pipe" });
+		} catch (err) {
+			log.process("error", "general", `Failed to restore git state: ${err instanceof Error ? err.message : String(err)}`);
+		}
+	}
+};
 //#endregion
 //#region src/docker-runner/utils.ts
 const executeDockerRun = async ({ version }) => {
@@ -2372,7 +2582,11 @@ const generatePagesFromSitemap = async () => {
 //#endregion
 //#region src/bin.ts
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
-const commandArgs = yargs(hideBin(process.argv)).parseSync()._;
+const args = yargs(hideBin(process.argv)).option("patch-build", {
+	type: "string",
+	describe: "Create a patch build for the merge base. Format: head...base (e.g. feature...main)"
+}).parseSync();
+const commandArgs = args._;
 const version = getVersion();
 if (version) log.process("info", "general", `Version: ${version}`);
 (async () => {
@@ -2401,6 +2615,7 @@ if (version) log.process("info", "general", `Version: ${version}`);
 			log.process("info", "general", `🚀 Starting Lost Pixel in 'platform' mode`);
 			const apiToken = await getPlatformApiToken(config);
 			if (commandArgs.includes("finalize")) await sendFinalizeToAPI(config, apiToken);
+			else if (args["patch-build"]) await patchBuildRunner(args["patch-build"], config, apiToken);
 			else await platformRunner(config, apiToken);
 		} else {
 			log.process("info", "general", `🚀 Starting Lost Pixel in 'generateOnly' mode`);