feat: publish subgraphs as plugins

Author: comatory

cli/src/commands/demo/command.ts

@@ -16,8 +16,10 @@ import {
 import {
   checkDockerReadiness,
   clearScreen,
+  getDemoLogPath,
   prepareSupportingData,
   printLogo,
+  publishAllPlugins,
   resetScreen,
   updateScreenWithUserInfo,
 } from './util.js';
@@ -164,13 +166,20 @@ async function handleStep2(
   {
     onboarding,
     userInfo,
+    supportDir,
+    signal,
   }: {
-    onboarding: {
-      finishedAt?: string;
-    };
+    onboarding: { finishedAt?: string };
     userInfo: UserInfo;
+    supportDir: string;
+    signal: AbortSignal;
   },
 ) {
+  function retryFn() {
+    resetScreen(userInfo);
+    return handleStep2(opts, { onboarding, userInfo, supportDir, signal });
+  }
+
   const graphData = await handleGetFederatedGraphResponse(opts.client, {
     onboarding,
     userInfo,
@@ -199,6 +208,26 @@ async function handleStep2(
     onboarding,
     userInfo,
   });
+
+  const logPath = getDemoLogPath();
+  console.log(`\nPublishing plugins… ${pc.dim(`(logs: ${logPath})`)}`);
+
+  const publishResult = await publishAllPlugins({
+    client: opts.client,
+    supportDir,
+    signal,
+    logPath,
+  });
+
+  if (publishResult.error) {
+    await waitForKeyPress(
+      {
+        r: retryFn,
+        R: retryFn,
+      },
+      'Hit [r] to retry. CTRL+C to quit.',
+    );
+  }
 }
 
 async function handleGetOnboardingResponse(client: BaseCommandOptions['client'], userInfo: UserInfo) {
@@ -252,28 +281,38 @@ async function getUserInfo(client: BaseCommandOptions['client']) {
 
 export default function (opts: BaseCommandOptions) {
   return async function handleCommand() {
-    clearScreen();
-    printHello();
-    await prepareSupportingData();
-    await checkDockerReadiness();
-    const userInfo = await getUserInfo(opts.client);
-    updateScreenWithUserInfo(userInfo);
-
-    await waitForKeyPress(
-      {
-        Enter: () => undefined,
-      },
-      `It is recommended you run this command along the onboarding wizard at ${config.baseURL}/onboarding with the same account.\nPress ENTER to continue…`,
-    );
-
-    resetScreen(userInfo);
-
-    const onboardingCheck = await handleStep1(opts, userInfo);
-
-    if (!onboardingCheck) {
-      return;
+    const controller = new AbortController();
+    const cleanup = () => controller.abort();
+    process.on('SIGINT', cleanup);
+    process.on('SIGTERM', cleanup);
+
+    try {
+      clearScreen();
+      printHello();
+      const supportDir = await prepareSupportingData();
+      await checkDockerReadiness();
+      const userInfo = await getUserInfo(opts.client);
+      updateScreenWithUserInfo(userInfo);
+
+      await waitForKeyPress(
+        {
+          Enter: () => undefined,
+        },
+        `It is recommended you run this command along the onboarding wizard at ${config.baseURL}/onboarding with the same account.\nPress ENTER to continue…`,
+      );
+
+      resetScreen(userInfo);
+
+      const onboardingCheck = await handleStep1(opts, userInfo);
+
+      if (!onboardingCheck) {
+        return;
+      }
+
+      await handleStep2(opts, { onboarding: onboardingCheck, userInfo, supportDir, signal: controller.signal });
+    } finally {
+      process.off('SIGINT', cleanup);
+      process.off('SIGTERM', cleanup);
     }
-
-    await handleStep2(opts, { onboarding: onboardingCheck, userInfo });
   };
 }

cli/src/commands/demo/util.ts

@@ -1,11 +1,14 @@
 import fs from 'node:fs/promises';
+import { createWriteStream, existsSync, mkdirSync, type WriteStream } from 'node:fs';
 import path from 'node:path';
 import { program } from 'commander';
-import { execa } from 'execa';
+import { execa, type ResultPromise } from 'execa';
 import ora from 'ora';
 import pc from 'picocolors';
 import { z } from 'zod';
 import { config, cacheDir } from '../../core/config.js';
+import { getDefaultPlatforms, publishPluginPipeline, readPluginFiles } from '../../core/plugin-publish.js';
+import type { BaseCommandOptions } from '../../core/types/types.js';
 import { visibleLength } from '../../utils.js';
 import type { UserInfo } from './types.js';
 
@@ -245,3 +248,77 @@ export async function checkDockerReadiness(): Promise<void> {
 
   spinner.succeed('Docker is ready.');
 }
+
+/**
+ * Returns the path to the demo log file at ~/.cache/cosmo/demo/demo.log.
+ * Creates the parent directory if needed.
+ */
+export function getDemoLogPath(): string {
+  const cosmoDir = path.join(cacheDir, 'demo');
+  if (!existsSync(cosmoDir)) {
+    mkdirSync(cosmoDir, { recursive: true });
+  }
+  return path.join(cosmoDir, 'demo.log');
+}
+
+function pipeToLog(logStream: WriteStream, proc: ResultPromise) {
+  proc.stdout?.pipe(logStream, { end: false });
+  proc.stderr?.pipe(logStream, { end: false });
+}
+
+/**
+ * Publishes demo plugins sequentially.
+ * Returns [error] on first failure; spinner shows which plugin failed.
+ */
+export async function publishAllPlugins({
+  client,
+  supportDir,
+  signal,
+  logPath,
+}: {
+  client: BaseCommandOptions['client'];
+  supportDir: string;
+  signal: AbortSignal;
+  logPath: string;
+}) {
+  const pluginNames = config.demoPluginNames;
+  const namespace = config.demoNamespace;
+  const labels = [config.demoLabelMatcher];
+  // The demo router always runs in a Linux Docker container, so we need
+  // linux builds for both architectures regardless of the host OS.
+  const platforms = [...new Set([...getDefaultPlatforms(), 'linux/amd64', 'linux/arm64'])];
+  const logStream = createWriteStream(logPath, { flags: 'w' });
+
+  try {
+    for (let i = 0; i < pluginNames.length; i++) {
+      const pluginName = pluginNames[i];
+      const pluginDir = path.join(supportDir, 'plugins', pluginName);
+
+      const spinner = ora(`Publishing plugin ${pc.bold(pluginName)} (${i + 1}/${pluginNames.length})…`).start();
+
+      const files = await readPluginFiles(pluginDir);
+      const result = await publishPluginPipeline({
+        client,
+        pluginDir,
+        pluginName,
+        namespace,
+        labels,
+        platforms,
+        files,
+        cancelSignal: signal,
+        onProcess: (proc) => pipeToLog(logStream, proc),
+      });
+
+      if (result.error) {
+        spinner.fail(`Failed to publish plugin ${pc.bold(pluginName)}: ${result.error.message}`);
+        return { error: result.error };
+      }
+
+      spinner.succeed(`Plugin ${pc.bold(pluginName)} published.`);
+    }
+  } finally {
+    logStream.end();
+  }
+
+  return { error: null };
+}