feat: copy onboarding support files to filesystem

Author: comatory

cli/src/commands/demo/command.ts

@@ -1,12 +1,29 @@
+import os from 'node:os';
+import fs from 'node:fs/promises';
+import path from 'node:path';
 import pc from 'picocolors';
 import ora from 'ora';
 import { program } from 'commander';
+import { z } from 'zod';
 import type { WhoAmIResponse } from '@wundergraph/cosmo-connect/dist/platform/v1/platform_pb';
 import { EnumStatusCode } from '@wundergraph/cosmo-connect/dist/common/common_pb';
 import { BaseCommandOptions } from '../../core/types/types.js';
 import { getBaseHeaders, config } from '../../core/config.js';
 import { waitForKeyPress, rainbow, visibleLength } from '../../utils.js';
 
+const ONBOARDING_REPO = 'wundergraph/cosmo-onboarding';
+const ONBOARDING_BRANCH = 'main';
+const ONBOARDING_DIR_PREFIX = 'plugins/';
+
+const GitHubTreeSchema = z.object({
+  tree: z.array(
+    z.object({
+      type: z.string(),
+      path: z.string(),
+    }),
+  ),
+});
+
 type UserInfo = {
   userEmail: WhoAmIResponse['userEmail'];
   organizationName: WhoAmIResponse['organizationName'];
@@ -161,6 +178,63 @@ async function fetchUserInfo(client: BaseCommandOptions['client']) {
   }
 }
 
+async function prepareSupportingData() {
+  const spinner = ora('Preparing supporting data…').start();
+
+  const tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'cosmo-'));
+  const cosmoDir = path.join(tmpDir, 'cosmo');
+  await fs.mkdir(cosmoDir, { recursive: true });
+
+  const treeResponse = await fetch(
+    `https://api.github.com/repos/${ONBOARDING_REPO}/git/trees/${ONBOARDING_BRANCH}?recursive=1`,
+  );
+  if (!treeResponse.ok) {
+    spinner.fail('Failed to fetch repository tree.');
+    program.error(`GitHub API error: ${treeResponse.statusText}`);
+  }
+
+  const parsed = GitHubTreeSchema.safeParse(await treeResponse.json());
+  if (!parsed.success) {
+    spinner.fail('Failed to parse repository tree.');
+    program.error('Unexpected response format from GitHub API. The repository structure may have changed.');
+  }
+
+  const files = parsed.data.tree.filter(
+    (entry) => entry.type === 'blob' && entry.path.startsWith(ONBOARDING_DIR_PREFIX),
+  );
+
+  const results = await Promise.all(
+    files.map(async (file) => {
+      const rawUrl = `https://raw.githubusercontent.com/${ONBOARDING_REPO}/${ONBOARDING_BRANCH}/${file.path}`;
+      try {
+        const response = await fetch(rawUrl);
+        if (!response.ok) {
+          return { path: file.path, error: response.statusText };
+        }
+
+        const content = Buffer.from(await response.arrayBuffer());
+        const destPath = path.join(cosmoDir, file.path);
+        await fs.mkdir(path.dirname(destPath), { recursive: true });
+        await fs.writeFile(destPath, content);
+
+        return { path: file.path, error: null };
+      } catch (err) {
+        return { path: file.path, error: err instanceof Error ? err.message : String(err) };
+      }
+    }),
+  );
+
+  const failed = results.filter((r) => r.error !== null);
+  if (failed.length > 0) {
+    spinner.fail(`Failed to fetch some files from onboarding repository or store them in ${cosmoDir}.`);
+    program.error(failed.map((f) => `  ${f.path}: ${f.error}`).join('\n'));
+  }
+
+  spinner.succeed(`Support files copied to ${pc.bold(cosmoDir)}`);
+
+  return cosmoDir;
+}
+
 async function getUserInfo(client: BaseCommandOptions['client']) {
   const spinner = ora('Retrieving information about you…').start();
   const { userInfo, error } = await fetchUserInfo(client);
@@ -181,6 +255,7 @@ export default function (opts: BaseCommandOptions) {
   return async function handleCommand() {
     clearScreen();
     printHello();
+    await prepareSupportingData();
 
     await waitForKeyPress(
       {