@aws-amplify/amplify-util-uibuilder 1.2.0-uibuilder-ts.0 → 1.2.0

package/src/__tests__/syncAmplifyUiComponents.test.ts

@@ -0,0 +1,150 @@
+import { $TSContext } from 'amplify-cli-core';
+import {
+  generateUiBuilderComponents,
+  getEnvName,
+  getAppId,
+  listUiBuilderComponents,
+  listUiBuilderThemes,
+  getAmplifyUIBuilderService,
+  resolveAppId,
+  generateUiBuilderThemes,
+} from '../commands/utils/syncAmplifyUiBuilderComponents';
+import aws from 'aws-sdk';
+import * as CLICore from 'amplify-cli-core';
+import * as createUiBuilderComponentDependency from '../commands/utils/createUiBuilderComponent';
+const aws_mock = aws as any;
+const createUiBuilderComponentDependency_mock = createUiBuilderComponentDependency as any;
+describe('should sync amplify ui builder components', () => {
+  let context: $TSContext | any;
+  beforeEach(() => {
+    CLICore.FeatureFlags.getBoolean = () => false;
+    CLICore.FeatureFlags.getNumber = () => 0;
+    context = {
+      exeInfo: {
+        projectConfig: {
+          javascript: {
+            config: {
+              SourceDir: 'src',
+            },
+          },
+        },
+      },
+      amplify: {
+        invokePluginMethod: () => ({}),
+      },
+      parameters: {
+        argv: [],
+      },
+      input: {
+        options: {
+          appId: 'testAppId',
+          envName: 'testEnvName',
+        },
+      },
+    };
+
+    aws_mock.AmplifyUIBuilder = jest.fn(() => ({
+      exportComponents: jest.fn(() => ({
+        promise: jest.fn(() => ({
+          entities: [
+            {
+              appId: 'd37nrm8rzt3oek',
+              bindingProperties: {},
+              componentType: 'Box',
+              environmentName: 'staging',
+              id: 's-s4mU579Ycf6JGHwhqT',
+              name: 'aawwdd',
+              overrides: {},
+              properties: {},
+              variants: [],
+            },
+          ],
+        })),
+      })),
+      exportThemes: jest.fn(() => ({
+        promise: jest.fn(() => ({
+          entities: [{}],
+        })),
+      })),
+    }));
+
+    createUiBuilderComponentDependency_mock.createUiBuilderComponent = jest.fn();
+    createUiBuilderComponentDependency_mock.createUiBuilderTheme = jest.fn();
+  });
+
+  it('pulls components from aws-sdk and passes them to createUiBuilderComponent', () => {
+    generateUiBuilderComponents(context, []);
+  });
+
+  it('does not throw an error when createUiBuilderComponent fails', () => {
+    createUiBuilderComponentDependency_mock.createUiBuilderComponent = jest.fn(() => {
+      throw new Error('ahhh!');
+    });
+    expect(async () => generateUiBuilderComponents(context, [])).not.toThrow();
+  });
+
+  it('does not throw an error when createUiBuilderThemes fails', () => {
+    createUiBuilderComponentDependency_mock.createUiBuilderComponent = jest.fn(() => {
+      throw new Error('ahhh!');
+    });
+    expect(async () => generateUiBuilderThemes(context, [])).not.toThrow();
+  });
+
+  it('can getAmplifyUIBuilderService', async () => {
+    process.env.UI_BUILDER_ENDPOINT = 'https://mock-endpoint.com';
+    process.env.UI_BUILDER_REGION = 'mock-region';
+    const service = await getAmplifyUIBuilderService(context, 'testEnv', 'testAppId');
+    expect(Object.keys(service)).toContain('exportComponents');
+    expect(Object.keys(service)).toContain('exportThemes');
+  });
+  it('can listUiBuilderThemes', async () => {
+    const themes = await listUiBuilderThemes(context, 'testEnv');
+    expect(themes.entities).toHaveLength(1);
+  });
+  it('can listUiBuilderComponents', async () => {
+    const components = await listUiBuilderComponents(context, 'testEnv');
+    expect(components.entities).toHaveLength(1);
+  });
+  it('can getAppId', async () => {
+    const appId = await getAppId(context);
+    expect(appId).toBe('testAppId');
+  });
+  it('can getEnvName', () => {
+    const envName = getEnvName(context);
+    expect(envName).toBe('testEnvName');
+  });
+  it('can resolveAppId', async () => {
+    context.amplify.invokePluginMethod = () => 'testAppId';
+    const appId = await resolveAppId(context);
+    expect(appId).toBe('testAppId');
+  });
+  it('can throw on getAppId', async () => {
+    context.input.options.appId = null;
+    context.amplify.invokePluginMethod = () => null;
+    expect(async () => await getAppId(context)).rejects.toThrowError();
+  });
+  it('can generate ui builder components', async () => {
+    createUiBuilderComponentDependency_mock.createUiBuilderComponent = jest.fn().mockImplementation(() => ({}));
+    const components = generateUiBuilderComponents(context, [{}, {}]);
+    expect(components.every(component => component.resultType === 'SUCCESS')).toBeTruthy();
+  });
+  it('can handle failed generation generate ui builder components', async () => {
+    createUiBuilderComponentDependency_mock.createUiBuilderComponent = jest.fn().mockImplementation(() => {
+      throw new Error('ahh!');
+    });
+    const components = generateUiBuilderComponents(context, [{}, {}]);
+    expect(components.every(component => component.resultType === 'FAILURE')).toBeTruthy();
+  });
+  it('can generate ui builder themes', async () => {
+    createUiBuilderComponentDependency_mock.createUiBuilderTheme = jest.fn().mockImplementation(() => ({}));
+    const themes = generateUiBuilderThemes(context, [{}, {}]);
+    expect(themes.every(theme => theme.resultType === 'SUCCESS')).toBeTruthy();
+  });
+  it('can handle failed generation generate ui builder themes', async () => {
+    createUiBuilderComponentDependency_mock.createUiBuilderTheme = jest.fn().mockImplementation(() => {
+      throw new Error('ahh!');
+    });
+    const themes = generateUiBuilderThemes(context, [{}, {}]);
+    expect(themes.every(theme => theme.resultType === 'FAILURE')).toBeTruthy();
+  });
+});

package/src/__tests__/version.test.ts

@@ -0,0 +1,10 @@
+import { printer } from 'amplify-prompts';
+import { run } from '../commands/version';
+jest.mock('amplify-prompts');
+printer.info = jest.fn();
+describe('can run version script', () => {
+  it('runs version script', async () => {
+    await run();
+    expect(printer.info).toBeCalledTimes(1);
+  });
+});

package/{commands/cloneComponentsFromEnv.js → src/commands/cloneComponentsFromEnv.ts}

@@ -1,9 +1,10 @@
-const aws = require('aws-sdk');
-const { extractArgs } = require('./utils/extractArgs');
-const { listUiBuilderComponents } = require('./utils/syncAmplifyUiBuilderComponents');
-const { printer } = require('amplify-prompts');
+import aws from 'aws-sdk';
+import { extractArgs } from './utils/extractArgs';
+import { listUiBuilderComponents } from './utils/syncAmplifyUiBuilderComponents';
+import { printer } from 'amplify-prompts';
+import { $TSAny, $TSContext } from 'amplify-cli-core';
 
-async function run(context) {
+export async function run(context: $TSContext) {
   printer.debug('Running create components command in amplify-util-uibuilder');
   const args = extractArgs(context);
   const sourceEnvName = args.sourceEnvName ? args.sourceEnvName : context.exeInfo.sourceEnvName;
@@ -11,9 +12,11 @@ async function run(context) {
 
   const existingComponents = await listUiBuilderComponents(context, sourceEnvName);
   if (existingComponents.entities.length === 0) {
-    `${existingComponents.entities.length} components exist in source env. Skipping creation of local components.`;
+    printer.debug(`${existingComponents.entities.length} components exist in source env. Skipping creation of local components.`);
+    return;
   }
   const existingComponentsNewEnv = await listUiBuilderComponents(context, newEnvName);
+
   if (existingComponentsNewEnv.entities.length > 0) {
     printer.debug(
       `${existingComponentsNewEnv.entities.length} components already exist in new env. Skipping creation of local components.`,
@@ -21,15 +24,13 @@ async function run(context) {
     return;
   }
 
-  const environmentName = args.environmentName ? args.environmentName : context.exeInfo.localEnvInfo.envName;
-
-  const appId = args.appId ? args.appId : context.exeInfo.teamProviderInfo[environmentName].awscloudformation.AmplifyAppId;
-
-  const awsConfigInfo = await context.amplify.invokePluginMethod(context, 'awscloudformation', undefined, 'loadConfigurationForEnv', [
+  const environmentName = args.environmentName ?? context.exeInfo.localEnvInfo.envName;
+  const appId = args.appId ?? context.exeInfo.teamProviderInfo[environmentName].awscloudformation.AmplifyAppId;
+  const awsConfigInfo = (await context.amplify.invokePluginMethod(context, 'awscloudformation', undefined, 'loadConfigurationForEnv', [
     context,
     newEnvName,
     appId,
-  ]);
+  ])) as $TSAny;
 
   if (process.env.UI_BUILDER_ENDPOINT) {
     awsConfigInfo.endpoint = process.env.UI_BUILDER_ENDPOINT;
@@ -40,13 +41,12 @@ async function run(context) {
   }
 
   const amplifyUIBuilder = new aws.AmplifyUIBuilder(awsConfigInfo);
-
   const components = existingComponents.entities;
   if (!components.length) {
     printer.debug(`No UIBuilder components found in app ${appId} from env ${sourceEnvName}. Skipping component clone process.`);
     return;
   }
-  const promises = components.map(async component => {
+  const promises = components.map(async (component: any) => {
     return await amplifyUIBuilder
       .createComponent({
         appId,
@@ -66,11 +66,7 @@ async function run(context) {
   });
   await Promise.all(promises);
 
-  console.log(
+  printer.info(
     `Successfully cloned ${components.length} UIBuilder components in app ${appId} from env ${sourceEnvName} to env ${newEnvName}.`,
   );
 }
-
-module.exports = {
-  run,
-};

package/{commands/generateComponents.js → src/commands/generateComponents.ts}

@@ -1,15 +1,17 @@
-const { notifyMissingPackages } = require('./utils/notifyMissingPackages');
-const { shouldRenderComponents } = require('./utils/shouldRenderComponents');
-const {
+import { notifyMissingPackages } from './utils/notifyMissingPackages';
+import { shouldRenderComponents } from './utils/shouldRenderComponents';
+import {
   listUiBuilderComponents,
   listUiBuilderThemes,
   generateUiBuilderComponents,
   generateUiBuilderThemes,
-} = require('./utils/syncAmplifyUiBuilderComponents');
-const { generateAmplifyUiBuilderIndexFile } = require('./utils/createUiBuilderComponent');
-const { printer } = require('amplify-prompts');
-const ora = require('ora');
-async function run(context) {
+} from './utils/syncAmplifyUiBuilderComponents';
+import { generateAmplifyUiBuilderIndexFile } from './utils/createUiBuilderComponent';
+import { printer } from 'amplify-prompts';
+import ora from 'ora';
+import { $TSContext } from 'amplify-cli-core';
+
+export async function run(context: $TSContext) {
   printer.debug('Running generate components command in amplify-util-uibuilder');
   if (!(await shouldRenderComponents(context))) {
     return;
@@ -48,7 +50,3 @@ async function run(context) {
 
   notifyMissingPackages(context);
 }
-
-module.exports = {
-  run,
-};

package/src/commands/help.ts

@@ -0,0 +1,5 @@
+import { printer } from 'amplify-prompts';
+
+export async function run() {
+  printer.info('Supported commands: generateComponents, cloneComponentsFromEnv');
+}

package/{commands/utils/createUiBuilderComponent.js → src/commands/utils/createUiBuilderComponent.ts}

@@ -1,14 +1,15 @@
-const { StudioTemplateRendererManager, StudioTemplateRendererFactory } = require('@aws-amplify/codegen-ui');
-const {
+import { StudioTemplateRendererManager, StudioTemplateRendererFactory, StudioComponent } from '@aws-amplify/codegen-ui';
+import {
   AmplifyRenderer,
   ReactThemeStudioTemplateRenderer,
   ReactIndexStudioTemplateRenderer,
   ModuleKind,
   ScriptTarget,
   ScriptKind,
-} = require('@aws-amplify/codegen-ui-react');
-const { getUiBuilderComponentsPath } = require('./getUiBuilderComponentsPath');
-const { printer } = require('amplify-prompts');
+} from '@aws-amplify/codegen-ui-react';
+import { getUiBuilderComponentsPath } from './getUiBuilderComponentsPath';
+import { printer } from 'amplify-prompts';
+import { $TSContext } from 'amplify-cli-core';
 const config = {
   module: ModuleKind.ES2020,
   target: ScriptTarget.ES2020,
@@ -16,9 +17,9 @@ const config = {
   renderTypeDeclarations: true,
 };
 
-const createUiBuilderComponent = (context, schema) => {
+export const createUiBuilderComponent = (context: $TSContext, schema: any) => {
   const uiBuilderComponentsPath = getUiBuilderComponentsPath(context);
-  const rendererFactory = new StudioTemplateRendererFactory(component => new AmplifyRenderer(component, config));
+  const rendererFactory = new StudioTemplateRendererFactory((component: StudioComponent) => new AmplifyRenderer(component, config));
 
   const outputPathDir = uiBuilderComponentsPath;
   const outputConfig = {
@@ -31,9 +32,9 @@ const createUiBuilderComponent = (context, schema) => {
   return schema;
 };
 
-const createUiBuilderTheme = (context, schema) => {
+export const createUiBuilderTheme = (context: $TSContext, schema: any) => {
   const uiBuilderComponentsPath = getUiBuilderComponentsPath(context);
-  const rendererFactory = new StudioTemplateRendererFactory(component => new ReactThemeStudioTemplateRenderer(component, config));
+  const rendererFactory = new StudioTemplateRendererFactory((component: any) => new ReactThemeStudioTemplateRenderer(component, config));
 
   const outputPathDir = uiBuilderComponentsPath;
   const outputConfig = {
@@ -47,14 +48,15 @@ const createUiBuilderTheme = (context, schema) => {
     return schema;
   } catch (e) {
     printer.debug(e);
-    printer.debug({ msg: 'Skipping invalid theme with schema', schema: JSON.stringify(schema, null, 2) });
+    printer.debug('Skipping invalid theme with schema');
+    printer.debug(JSON.stringify(schema, null, 2));
     throw e;
   }
 };
 
-const generateAmplifyUiBuilderIndexFile = (context, schemas) => {
+export const generateAmplifyUiBuilderIndexFile = (context: $TSContext, schemas: any[]) => {
   const uiBuilderComponentsPath = getUiBuilderComponentsPath(context);
-  const rendererFactory = new StudioTemplateRendererFactory(component => new ReactIndexStudioTemplateRenderer(component, config));
+  const rendererFactory = new StudioTemplateRendererFactory((component: any) => new ReactIndexStudioTemplateRenderer(component, config));
 
   const outputPathDir = uiBuilderComponentsPath;
   const outputConfig = {
@@ -64,16 +66,10 @@ const generateAmplifyUiBuilderIndexFile = (context, schemas) => {
   const rendererManager = new StudioTemplateRendererManager(rendererFactory, outputConfig);
 
   try {
-    rendererManager.renderSchemaToTemplate(schemas);
+    return rendererManager.renderSchemaToTemplate(schemas);
   } catch (e) {
     printer.debug(e);
-    printer.debug({ msg: 'Failed to generate component index file' });
+    printer.debug('Failed to generate component index file');
     throw e;
   }
 };
-
-module.exports = {
-  createUiBuilderComponent,
-  createUiBuilderTheme,
-  generateAmplifyUiBuilderIndexFile,
-};

package/{commands/utils/extractArgs.js → src/commands/utils/extractArgs.ts}

@@ -1,4 +1,6 @@
-const extractArgs = context => {
+import { $TSContext } from 'amplify-cli-core';
+
+export const extractArgs = (context: $TSContext) => {
   if (!context.input.options) {
     context.input.options = {};
   }
@@ -21,7 +23,3 @@ const extractArgs = context => {
     newEnvName,
   };
 };
-
-module.exports = {
-  extractArgs,
-};

package/{commands/utils/getUiBuilderComponentsPath.js → src/commands/utils/getUiBuilderComponentsPath.ts}

@@ -1,8 +1,9 @@
-const path = require('path');
-const fs = require('fs-extra');
-const { extractArgs } = require('./extractArgs');
+import path from 'path';
+import fs from 'fs-extra';
+import { extractArgs } from './extractArgs';
+import { $TSContext } from 'amplify-cli-core';
 
-const getUiBuilderComponentsPath = context => {
+export const getUiBuilderComponentsPath = (context: $TSContext) => {
   const args = extractArgs(context);
   const srcDir = args.srcDir ? args.srcDir : context.exeInfo.projectConfig.javascript.config.SourceDir;
   const uiBuilderComponentsPath = path.resolve(path.join('.', srcDir, 'ui-components'));
@@ -13,7 +14,3 @@ const getUiBuilderComponentsPath = context => {
 
   return uiBuilderComponentsPath;
 };
-
-module.exports = {
-  getUiBuilderComponentsPath,
-};

package/src/commands/utils/notifyMissingPackages.ts

@@ -0,0 +1,40 @@
+import { pathManager, JSONUtilities, $TSContext, $TSAny } from 'amplify-cli-core';
+import { printer } from 'amplify-prompts';
+import fs from 'fs-extra';
+import path from 'path';
+import { extractArgs } from './extractArgs';
+const REQUIRED_PACKAGES: { [key: string]: string } = { '@aws-amplify/ui-react': '*', 'aws-amplify': '*' };
+
+export const notifyMissingPackages = (context: $TSContext) => {
+  const args = extractArgs(context);
+  const localEnvFilePath = args.localEnvFilePath ?? pathManager.getLocalEnvFilePath();
+  if (!fs.existsSync(localEnvFilePath)) {
+    printer.debug('localEnvFilePath could not be determined - skipping dependency notification.');
+    return;
+  }
+  const localEnvJson = JSONUtilities.readJson(localEnvFilePath);
+  const packageJsonPath = path.join((localEnvJson as $TSAny).projectPath, 'package.json');
+  if (!fs.existsSync(packageJsonPath)) {
+    printer.debug('package.json file not found - skipping dependency notification.');
+    return;
+  }
+  const packageJson = JSONUtilities.readJson(packageJsonPath) as { dependencies: { [key: string]: string } };
+  Object.keys(REQUIRED_PACKAGES).forEach(packageName => {
+    const packageIsInstalled = Object.keys(packageJson.dependencies).includes(packageName);
+    if (!packageIsInstalled) {
+      printer.warn(`UIBuilder components required "${packageName}" that is not in your package.json. Run \`npm install ${packageName}\``);
+    } else if (!packageVersionMatches(REQUIRED_PACKAGES[packageName], packageJson.dependencies[packageName])) {
+      printer.warn(
+        `UIBuilder components requires version "${REQUIRED_PACKAGES[packageName]}" of "${packageName}". You currently are on version "${packageJson.dependencies[packageName]}".`,
+      );
+    }
+  });
+};
+
+const packageVersionMatches = (requiredVersion: string, existingVersion: string) => {
+  if (requiredVersion === '*') {
+    return true;
+  }
+
+  return requiredVersion === existingVersion;
+};

package/{commands/utils/shouldRenderComponents.js → src/commands/utils/shouldRenderComponents.ts}

@@ -1,15 +1,16 @@
-const { printer } = require('amplify-prompts');
-async function shouldRenderComponents(context) {
+import { $TSContext } from 'amplify-cli-core';
+import { printer } from 'amplify-prompts';
+export async function shouldRenderComponents(context: $TSContext) {
   if (process.env.FORCE_RENDER) {
     printer.debug('Forcing component render since environment variable flag is set.');
     return true;
   }
-  if (context.input.options && context.input.options['no-codegen']) {
+  if (context?.input?.options?.['no-codegen']) {
     printer.debug('Not pulling components because --no-codegen flag is set');
     return false;
   }
 
-  if (!context.exeInfo || !context.exeInfo.projectConfig) {
+  if (!context?.exeInfo?.projectConfig) {
     printer.debug('Not pulling components because there is no projectConfig set for this project');
     return false;
   }
@@ -31,7 +32,3 @@ async function shouldRenderComponents(context) {
 
   return true;
 }
-
-module.exports = {
-  shouldRenderComponents,
-};

package/{commands/utils/syncAmplifyUiBuilderComponents.js → src/commands/utils/syncAmplifyUiBuilderComponents.ts}

@@ -1,16 +1,21 @@
-const aws = require('aws-sdk');
-const { printer } = require('amplify-prompts');
-const { resolveAppId } = require('amplify-provider-awscloudformation');
-const { createUiBuilderComponent, createUiBuilderTheme } = require('./createUiBuilderComponent');
-const { getUiBuilderComponentsPath } = require('./getUiBuilderComponentsPath');
-const { extractArgs } = require('./extractArgs');
-const getEnvName = (context, envName) => {
+import aws from 'aws-sdk';
+import { printer } from 'amplify-prompts';
+import { createUiBuilderComponent, createUiBuilderTheme } from './createUiBuilderComponent';
+import { getUiBuilderComponentsPath } from './getUiBuilderComponentsPath';
+import { extractArgs } from './extractArgs';
+import { $TSAny, $TSContext } from 'amplify-cli-core';
+export const getEnvName = (context: $TSContext, envName?: string) => {
   const args = extractArgs(context);
   return envName ? envName : args.environmentName ? args.environmentName : context.exeInfo.localEnvInfo.envName;
 };
 
-const getAppId = (context, environmentName) => {
-  const appId = extractArgs(context).appId || resolveAppId(context);
+export const resolveAppId = async (context: $TSContext) => {
+  return await context.amplify.invokePluginMethod(context, 'awscloudformation', undefined, 'resolveAppId', [context]);
+};
+
+export const getAppId = async (context: $TSContext) => {
+  const appId = extractArgs(context).appId || (await resolveAppId(context));
+
   if (!appId) {
     throw new Error(
       'Unable to sync Studio components since appId could not be determined. This can happen when you hit the soft limit of number of apps that you can have in Amplify console.',
@@ -19,9 +24,9 @@ const getAppId = (context, environmentName) => {
   return appId;
 };
 
-async function listUiBuilderComponents(context, envName) {
+export async function listUiBuilderComponents(context: $TSContext, envName?: string) {
   const environmentName = getEnvName(context, envName);
-  const appId = getAppId(context, environmentName);
+  const appId = await getAppId(context);
 
   try {
     const amplifyUIBuilder = await getAmplifyUIBuilderService(context, environmentName, appId);
@@ -39,9 +44,9 @@ async function listUiBuilderComponents(context, envName) {
   }
 }
 
-async function listUiBuilderThemes(context, envName) {
+export async function listUiBuilderThemes(context: $TSContext, envName?: string) {
   const environmentName = getEnvName(context, envName);
-  const appId = getAppId(context, environmentName);
+  const appId = await getAppId(context);
 
   try {
     const amplifyUIBuilder = await getAmplifyUIBuilderService(context, environmentName, appId);
@@ -59,7 +64,7 @@ async function listUiBuilderThemes(context, envName) {
   }
 }
 
-function generateUiBuilderComponents(context, componentSchemas) {
+export function generateUiBuilderComponents(context: $TSContext, componentSchemas: any[]) {
   const componentResults = componentSchemas.map(schema => {
     try {
       const component = createUiBuilderComponent(context, schema);
@@ -79,7 +84,7 @@ function generateUiBuilderComponents(context, componentSchemas) {
   return componentResults;
 }
 
-function generateUiBuilderThemes(context, themeSchemas) {
+export function generateUiBuilderThemes(context: $TSContext, themeSchemas: any[]) {
   const themeResults = themeSchemas.map(schema => {
     try {
       const theme = createUiBuilderTheme(context, schema);
@@ -97,12 +102,12 @@ function generateUiBuilderThemes(context, themeSchemas) {
   return themeResults;
 }
 
-const getAmplifyUIBuilderService = async (context, environmentName, appId) => {
-  const awsConfigInfo = await context.amplify.invokePluginMethod(context, 'awscloudformation', undefined, 'loadConfigurationForEnv', [
+export const getAmplifyUIBuilderService = async (context: $TSContext, environmentName: string, appId: string) => {
+  const awsConfigInfo = (await context.amplify.invokePluginMethod(context, 'awscloudformation', undefined, 'loadConfigurationForEnv', [
     context,
     environmentName,
     appId,
-  ]);
+  ])) as $TSAny;
 
   if (process.env.UI_BUILDER_ENDPOINT) {
     awsConfigInfo.endpoint = process.env.UI_BUILDER_ENDPOINT;
@@ -114,11 +119,3 @@ const getAmplifyUIBuilderService = async (context, environmentName, appId) => {
 
   return new aws.AmplifyUIBuilder(awsConfigInfo);
 };
-
-module.exports = {
-  getAmplifyUIBuilderService,
-  generateUiBuilderComponents,
-  generateUiBuilderThemes,
-  listUiBuilderComponents,
-  listUiBuilderThemes,
-};

package/src/commands/version.ts

@@ -0,0 +1,6 @@
+import { printer } from 'amplify-prompts';
+import path from 'path';
+
+export async function run() {
+  printer.info((await import(path.join(__dirname, '..', '..', 'package.json'))).version);
+}

package/{event-handlers/handle-PostEnvAdd.js → src/event-handlers/handle-PostEnvAdd.ts}

@@ -1,6 +1,7 @@
-const { printer } = require('amplify-prompts');
+import { $TSContext } from 'amplify-cli-core';
+import { printer } from 'amplify-prompts';
 
-async function run(context) {
+export async function run(context: $TSContext) {
   try {
     await context.amplify.invokePluginMethod(context, 'ui-builder', undefined, 'executeAmplifyCommand', [
       context,
@@ -11,7 +12,3 @@ async function run(context) {
     printer.debug(e);
   }
 }
-
-module.exports = {
-  run,
-};

package/{event-handlers/handle-PostPull.js → src/event-handlers/handle-PostPull.ts}

@@ -1,6 +1,7 @@
-const { printer } = require('amplify-prompts');
+import { $TSContext } from 'amplify-cli-core';
+import { printer } from 'amplify-prompts';
 
-async function run(context) {
+export async function run(context: $TSContext) {
   try {
     await context.amplify.invokePluginMethod(context, 'ui-builder', undefined, 'executeAmplifyCommand', [context, 'generateComponents']);
   } catch (e) {
@@ -8,7 +9,3 @@ async function run(context) {
     printer.debug(e);
   }
 }
-
-module.exports = {
-  run,
-};

package/{index.js → src/index.ts}

@@ -1,20 +1,16 @@
-const path = require('path');
+import { $TSContext } from 'amplify-cli-core';
+import path from 'path';
 
-async function executeAmplifyCommand(context, cmd) {
+export async function executeAmplifyCommand(context: $TSContext, cmd: string) {
   const commandsDirPath = path.normalize(path.join(__dirname, 'commands'));
   const commandPath = path.join(commandsDirPath, cmd ? cmd : context.input.command);
   const commandModule = require(commandPath);
   await commandModule.run(context);
 }
 
-async function handleAmplifyEvent(context, args) {
+export async function handleAmplifyEvent(context: $TSContext, { event }: { event: string }) {
   const eventHandlersDirPath = path.normalize(path.join(__dirname, 'event-handlers'));
-  const eventHandlerPath = path.join(eventHandlersDirPath, `handle-${args.event}`);
+  const eventHandlerPath = path.join(eventHandlersDirPath, `handle-${event}`);
   const eventHandlerModule = require(eventHandlerPath);
-  await eventHandlerModule.run(context, args);
+  await eventHandlerModule.run(context);
 }
-
-module.exports = {
-  executeAmplifyCommand,
-  handleAmplifyEvent,
-};

package/tsconfig.json

@@ -0,0 +1,9 @@
+{
+	"extends": "../../tsconfig.base.json",
+	"compilerOptions": {
+	  "outDir": "./lib",
+	  "rootDir": "./src",
+	  "strict": true,
+	  "skipLibCheck": true
+	}
+  }