create.ts

import { ux } from '@oclif/core';
import * as fs from 'fs';
import * as inquirer from 'inquirer';
import BaseCommand from '../../base';
import { fetchRemoteTypes } from '../../utils/extension-utils';
import * as questions from '../../utils/questions';
import templates from '../../utils/templates';

export default class Create extends BaseCommand {
  static description = 'Create an example extension';

  static flags = {
    ...BaseCommand.flags,
  };

  async run() {
    const extensionAnswers = await inquirer.prompt(
      questions.extensionQuestions
    );
    const directoryName = extensionAnswers.identifier.match(
      /^[^.]+\.([^.]+)$/
    )[1];

    // Check if the extension already exists.
    if (fs.existsSync(directoryName)) {
      throw new Error(`A directory named '${directoryName}' already exists.`);
    }

    const ahaExtensionSchema: { [k: string]: any } = {};
    ahaExtensionSchema.contributes = {};

    const createContributions = (
      await inquirer.prompt(questions.addContributionsQuestion)
    ).createContributions;

    if (createContributions) {
      do {
        process.stdout.write('\n');

        const contribution = await questions.getContributionFromQuestions();

        ahaExtensionSchema.contributes[contribution.type] =
          ahaExtensionSchema.contributes[contribution.type] || {};
        ahaExtensionSchema.contributes[contribution.type][contribution.name] =
          contribution.contribution;
      } while (
        (await inquirer.prompt(questions.addAnotherContributionQuestion))
          .add === 'yes'
      );
    } else {
      ahaExtensionSchema.contributes = {
        views: {
          samplePage: {
            title: 'Sample Page',
            entryPoint: 'src/views/samplePage.js',
            host: 'page',
            location: {
              menu: 'Work',
            },
          },
        },
        commands: {
          sampleCommand: {
            title: 'Sample Command',
            entryPoint: 'src/commands/sampleCommand.js',
          },
        },
      };
    }

    // Create the extension and template files.
    ux.action.start('Creating extension');

    fs.mkdirSync(directoryName);
    fs.writeFileSync(
      `${directoryName}/package.json`,
      templates.packageTemplate(
        extensionAnswers.identifier,
        extensionAnswers.name,
        extensionAnswers.author,
        ahaExtensionSchema
      )
    );

    fs.writeFileSync(
      `${directoryName}/README.md`,
      templates.readmeTemplate(extensionAnswers.name)
    );

    fs.writeFileSync(
      `${directoryName}/tsconfig.json`,
      templates.tsconfigTemplate()
    );
    fs.mkdirSync(`${directoryName}/.vscode`);
    fs.writeFileSync(
      `${directoryName}/.vscode/settings.json`,
      templates.vscodeTemplate()
    );
    fs.writeFileSync(
      `${directoryName}/.gitignore`,
      templates.gitignoreTemplate()
    );

    templates.writeContributionTemplates(
      fs,
      directoryName,
      extensionAnswers.identifier,
      ahaExtensionSchema.contributes
    );

    await fetchRemoteTypes(directoryName);

    ux.action.stop(`Extension created in directory '${directoryName}'`);
  }
}