feat: tutorialkit eject command

packages/cli/src/commands/create/enterprise.ts

@@ -1,8 +1,7 @@
 import fs from 'node:fs';
 import path from 'node:path';
 import type { CreateOptions } from './options.js';
-import { parseAstroConfig, replaceArgs } from './astro-config.js';
-import { generate } from './babel.js';
+import { parseAstroConfig, replaceArgs, generateAstroConfig } from '../../utils/astro-config.js';
 
 export async function setupEnterpriseConfig(dest: string, flags: CreateOptions) {
   if (!flags.defaults && flags.enterprise === undefined) {
@@ -38,13 +37,7 @@ export async function setupEnterpriseConfig(dest: string, flags: CreateOptions)
       astroConfig,
     );
 
-    const defaultExport = 'export default defineConfig';
-    let output = generate(astroConfig);
-
-    // add a new line
-    output = output.replace(defaultExport, `\n${defaultExport}`);
-
-    fs.writeFileSync(configPath, output);
+    fs.writeFileSync(configPath, generateAstroConfig(astroConfig));
   }
 }
 

packages/cli/src/commands/eject/index.ts

@@ -0,0 +1,132 @@
+import chalk from 'chalk';
+import fs from 'node:fs';
+import path from 'node:path';
+import type { Arguments } from 'yargs-parser';
+import { pkg } from '../../pkg.js';
+import { DEFAULT_VALUES, type EjectOptions } from './options.js';
+import { errorLabel, printHelp } from '../../utils/messages.js';
+import { parseAstroConfig, replaceArgs, generateAstroConfig } from '../../utils/astro-config.js';
+
+export function ejectRoutes(flags: Arguments) {
+  if (flags._[1] === 'help' || flags.help || flags.h) {
+    printHelp({
+      commandName: `${pkg.name} eject`,
+      usage: '[folder] [...options]',
+      tables: {
+        Options: [
+          [
+            '--force',
+            `Overwrite existing files in the target directory without prompting (default ${chalk.yellow(DEFAULT_VALUES.force)})`,
+          ],
+        ],
+      },
+    });
+
+    return 0;
+  }
+
+  try {
+    return _eject(flags);
+  } catch (error) {
+    console.error(`${errorLabel()} Command failed`);
+
+    if (error.stack) {
+      console.error(`\n${error.stack}`);
+    }
+
+    process.exit(1);
+  }
+}
+
+async function _eject(flags: EjectOptions) {
+  let folderPath = flags._[1] !== undefined ? String(flags._[1]) : undefined;
+
+  if (folderPath === undefined) {
+    folderPath = process.cwd();
+  } else {
+    folderPath = path.resolve(process.cwd(), folderPath);
+  }
+
+  /**
+   * First we make sure that the destination has the correct files
+   * and that there won't be any files overriden in the process.
+   *
+   * If they are any and `force` was not specified we abort.
+   */
+  const { astroConfigPath, srcPath, srcDestPath } = validateDestination(folderPath, flags.force);
+
+  /**
+   * We now proceed with the astro configuration.
+   *
+   * There we must disable the default routes so that the
+   * new routes that we're copying will be automatically picked up.
+   */
+  const astroConfig = await parseAstroConfig(astroConfigPath);
+
+  replaceArgs({ defaultRoutes: false }, astroConfig);
+
+  fs.writeFileSync(astroConfigPath, generateAstroConfig(astroConfig));
+
+  /**
+   * We now finalize by copying all the assets from the `default` folder
+   * into the `src` folder.
+   */
+  fs.cpSync(srcPath, srcDestPath, { recursive: true });
+}
+
+function validateDestination(folder: string, force: boolean) {
+  assertExists(folder);
+
+  const astroConfigPath = assertExists(path.join(folder, 'astro.config.ts'));
+  const srcDestPath = assertExists(path.join(folder, 'src'));
+
+  const localAstroIntegration = assertExists(path.resolve(folder, 'node_modules', '@tutorialkit', 'astro'));
+
+  const srcPath = path.join(localAstroIntegration, 'dist', 'default');
+
+  // check that they are no collision
+  if (!force) {
+    walk(srcPath, (relativePath) => {
+      const destination = path.join(srcDestPath, relativePath);
+
+      if (fs.existsSync(destination)) {
+        throw new Error(
+          `Eject aborted because '${destination}' would be overriden by this command. Use --force to ignore this error.`,
+        );
+      }
+    });
+  }
+
+  return {
+    astroConfigPath,
+    srcPath,
+    srcDestPath,
+  };
+}
+
+function assertExists(filePath: string) {
+  if (!fs.existsSync(filePath)) {
+    throw new Error(`${filePath} does not exists!`);
+  }
+
+  return filePath;
+}
+
+function walk(root: string, visit: (relativeFilePath: string) => void) {
+  function traverse(folder: string, pathPrefix: string) {
+    for (const filename of fs.readdirSync(folder)) {
+      const filePath = path.join(folder, filename);
+      const stat = fs.statSync(filePath);
+
+      const relativeFilePath = path.join(pathPrefix, filename);
+
+      if (stat.isDirectory()) {
+        traverse(filePath, relativeFilePath);
+      } else {
+        visit(relativeFilePath);
+      }
+    }
+  }
+
+  traverse(root, '');
+}

packages/cli/src/commands/eject/options.ts

@@ -0,0 +1,8 @@
+export interface EjectOptions {
+  _: Array<string | number>;
+  force?: boolean;
+}
+
+export const DEFAULT_VALUES = {
+  force: false,
+};

packages/cli/src/index.ts

@@ -3,12 +3,13 @@
 import chalk from 'chalk';
 import yargs from 'yargs-parser';
 import { createTutorial } from './commands/create/index.js';
+import { ejectRoutes } from './commands/eject/index.js';
 import { pkg } from './pkg.js';
 import { errorLabel, primaryLabel, printHelp } from './utils/messages.js';
 
-type CLICommand = 'version' | 'help' | 'create';
+type CLICommand = 'version' | 'help' | 'create' | 'eject';
 
-const supportedCommands = new Set(['version', 'help', 'create']);
+const supportedCommands = new Set<string>(['version', 'help', 'create', 'eject'] satisfies CLICommand[]);
 
 cli();
 
@@ -52,6 +53,9 @@ async function runCommand(cmd: CLICommand, flags: yargs.Arguments): Promise<numb
     case 'create': {
       return createTutorial(flags);
     }
+    case 'eject': {
+      return ejectRoutes(flags);
+    }
     default: {
       console.error(`${errorLabel()} Unknown command ${chalk.red(cmd)}`);
       return 1;

packages/cli/src/commands/create/astro-config.ts → packages/cli/src/utils/astro-config.ts

@@ -1,6 +1,6 @@
 import fs from 'node:fs/promises';
-import type { Options } from '../../../../astro/src/index.js';
-import { parse, t, visit } from './babel.js';
+import type { Options } from '../../../astro/src/index.js';
+import { parse, t, visit, generate } from './babel.js';
 
 export async function parseAstroConfig(astroConfigPath: string): Promise<t.File> {
   const source = await fs.readFile(astroConfigPath, { encoding: 'utf-8' });
@@ -17,6 +17,17 @@ export async function parseAstroConfig(astroConfigPath: string): Promise<t.File>
   return result;
 }
 
+export function generateAstroConfig(astroConfig: t.File): string {
+  const defaultExport = 'export default defineConfig';
+
+  let output = generate(astroConfig);
+
+  // add a new line
+  output = output.replace(defaultExport, `\n${defaultExport}`);
+
+  return output;
+}
+
 /**
  * This function modifies the arguments provided to the tutorialkit integration in the astro
  * configuration.
@@ -156,9 +167,9 @@ function updateObject(properties: any, object: t.ObjectExpression | undefined):
 
   object ??= t.objectExpression([]);
 
-  for (const property of properties) {
+  for (const property in properties) {
     const propertyInObject = object.properties.find((prop) => {
-      return prop.type === 'ObjectProperty' && prop.key === property;
+      return prop.type === 'ObjectProperty' && prop.key.type === 'Identifier' && prop.key.name === property;
     }) as t.ObjectProperty | undefined;
 
     if (!propertyInObject) {
@@ -191,6 +202,10 @@ function fromValue(value: any): t.Expression {
     return t.numericLiteral(value);
   }
 
+  if (typeof value === 'boolean') {
+    return t.booleanLiteral(value);
+  }
+
   if (Array.isArray(value)) {
     return t.arrayExpression(value.map(fromValue));
   }

packages/cli/src/utils/messages.ts

@@ -1,5 +1,5 @@
 import chalk from 'chalk';
-import { primaryBlue } from './colors';
+import { primaryBlue } from './colors.js';
 
 interface HelpOptions {
   commandName: string;

packages/cli/src/utils/project.ts

@@ -1,5 +1,5 @@
-import { randomValueFromArray } from './random';
-import { adjectives, nouns } from './words';
+import { randomValueFromArray } from './random.js';
+import { adjectives, nouns } from './words.js';
 
 export function generateProjectName() {
   const adjective = randomValueFromArray(adjectives);