feat: Flag to allow schema changes that require table replacement

.circleci/config.yml

@@ -1283,14 +1283,22 @@ jobs:
     environment:
       TEST_SUITE: src/__tests__/auth_6.test.ts
       CLI_REGION: us-east-2
+  api_5-amplify_e2e_tests:
+    working_directory: ~/repo
+    docker: *ref_1
+    resource_class: large
+    steps: *ref_5
+    environment:
+      TEST_SUITE: src/__tests__/api_5.test.ts
+      CLI_REGION: us-west-2
   api_4-amplify_e2e_tests:
     working_directory: ~/repo
     docker: *ref_1
     resource_class: large
     steps: *ref_5
     environment:
       TEST_SUITE: src/__tests__/api_4.test.ts
-      CLI_REGION: us-west-2
+      CLI_REGION: eu-west-2
   schema-iterative-update-4-amplify_e2e_tests_pkg_linux:
     working_directory: ~/repo
     docker: *ref_1
@@ -2071,6 +2079,16 @@ jobs:
       TEST_SUITE: src/__tests__/auth_6.test.ts
       CLI_REGION: us-east-2
     steps: *ref_6
+  api_5-amplify_e2e_tests_pkg_linux:
+    working_directory: ~/repo
+    docker: *ref_1
+    resource_class: large
+    environment:
+      AMPLIFY_DIR: /home/circleci/repo/out
+      AMPLIFY_PATH: /home/circleci/repo/out/amplify-pkg-linux
+      TEST_SUITE: src/__tests__/api_5.test.ts
+      CLI_REGION: us-west-2
+    steps: *ref_6
   api_4-amplify_e2e_tests_pkg_linux:
     working_directory: ~/repo
     docker: *ref_1
@@ -2079,7 +2097,7 @@ jobs:
       AMPLIFY_DIR: /home/circleci/repo/out
       AMPLIFY_PATH: /home/circleci/repo/out/amplify-pkg-linux
       TEST_SUITE: src/__tests__/api_4.test.ts
-      CLI_REGION: us-west-2
+      CLI_REGION: eu-west-2
     steps: *ref_6
 workflows:
   version: 2
@@ -2190,11 +2208,11 @@ workflows:
             - tags-amplify_e2e_tests
             - schema-iterative-update-locking-amplify_e2e_tests
             - function_7-amplify_e2e_tests
-            - api_4-amplify_e2e_tests
-            - amplify-app-amplify_e2e_tests
+            - api_5-amplify_e2e_tests
             - init-amplify_e2e_tests
             - s3-sse-amplify_e2e_tests
             - function_6-amplify_e2e_tests
+            - api_4-amplify_e2e_tests
             - schema-predictions-amplify_e2e_tests
             - amplify-configure-amplify_e2e_tests
             - pull-amplify_e2e_tests
@@ -2220,11 +2238,11 @@ workflows:
             - tags-amplify_e2e_tests_pkg_linux
             - schema-iterative-update-locking-amplify_e2e_tests_pkg_linux
             - function_7-amplify_e2e_tests_pkg_linux
-            - api_4-amplify_e2e_tests_pkg_linux
-            - amplify-app-amplify_e2e_tests_pkg_linux
+            - api_5-amplify_e2e_tests_pkg_linux
             - init-amplify_e2e_tests_pkg_linux
             - s3-sse-amplify_e2e_tests_pkg_linux
             - function_6-amplify_e2e_tests_pkg_linux
+            - api_4-amplify_e2e_tests_pkg_linux
             - schema-predictions-amplify_e2e_tests_pkg_linux
             - amplify-configure-amplify_e2e_tests_pkg_linux
             - pull-amplify_e2e_tests_pkg_linux
@@ -2534,7 +2552,7 @@ workflows:
           filters: *ref_10
           requires:
             - schema-auth-10-amplify_e2e_tests
-      - api_4-amplify_e2e_tests:
+      - api_5-amplify_e2e_tests:
           context: *ref_8
           post-steps: *ref_9
           filters: *ref_10
@@ -2606,6 +2624,12 @@ workflows:
           filters: *ref_10
           requires:
             - hostingPROD-amplify_e2e_tests
+      - api_4-amplify_e2e_tests:
+          context: *ref_8
+          post-steps: *ref_9
+          filters: *ref_10
+          requires:
+            - amplify-app-amplify_e2e_tests
       - schema-auth-5-amplify_e2e_tests:
           context: *ref_8
           post-steps: *ref_9
@@ -3026,7 +3050,7 @@ workflows:
           filters: *ref_13
           requires:
             - schema-auth-10-amplify_e2e_tests_pkg_linux
-      - api_4-amplify_e2e_tests_pkg_linux:
+      - api_5-amplify_e2e_tests_pkg_linux:
           context: *ref_11
           post-steps: *ref_12
           filters: *ref_13
@@ -3102,6 +3126,12 @@ workflows:
           filters: *ref_13
           requires:
             - hostingPROD-amplify_e2e_tests_pkg_linux
+      - api_4-amplify_e2e_tests_pkg_linux:
+          context: *ref_11
+          post-steps: *ref_12
+          filters: *ref_13
+          requires:
+            - amplify-app-amplify_e2e_tests_pkg_linux
       - schema-auth-5-amplify_e2e_tests_pkg_linux:
           context: *ref_11
           post-steps: *ref_12

packages/amplify-category-api/amplify-plugin.json

@@ -1,18 +1,9 @@
 {
-    "name": "api",
-    "type": "category",
-    "commands": [
-        "add-graphql-datasource",
-        "add",
-        "console",
-        "gql-compile",
-        "push",
-        "remove",
-        "update",
-        "help"
-    ],
-    "commandAliases":{
-        "configure": "update"
-    },
-    "eventHandlers": []
-}
+  "name": "api",
+  "type": "category",
+  "commands": ["add-graphql-datasource", "add", "console", "gql-compile", "push", "rebuild", "remove", "update", "help"],
+  "commandAliases": {
+    "configure": "update"
+  },
+  "eventHandlers": []
+}

packages/amplify-category-api/package.json

@@ -64,6 +64,7 @@
     "@octokit/rest": "^18.0.9",
     "@graphql-tools/merge": "^6.0.18",
     "amplify-cli-core": "1.27.0",
+    "amplify-prompts": "1.1.2",
     "amplify-headless-interface": "1.10.0",
     "amplify-util-headless-input": "1.5.3",
     "chalk": "^4.1.1",

packages/amplify-category-api/src/__tests__/commands/api/rebuild.test.ts

@@ -0,0 +1,64 @@
+import { $TSContext, FeatureFlags, stateManager } from 'amplify-cli-core';
+import { printer, prompter } from 'amplify-prompts';
+import { mocked } from 'ts-jest/utils';
+import { run } from '../../../commands/api/rebuild';
+
+jest.mock('amplify-cli-core');
+jest.mock('amplify-prompts');
+
+const FeatureFlags_mock = mocked(FeatureFlags);
+const stateManager_mock = mocked(stateManager);
+const printer_mock = mocked(printer);
+const prompter_mock = mocked(prompter);
+
+FeatureFlags_mock.getBoolean.mockReturnValue(true);
+
+beforeEach(jest.clearAllMocks);
+
+const pushResourcesMock = jest.fn();
+
+const context_stub = {
+  amplify: {
+    constructExeInfo: jest.fn(),
+    pushResources: pushResourcesMock,
+  },
+  parameters: {
+    first: 'resourceName',
+  },
+} as unknown as $TSContext;
+
+it('prints error if iterative updates not enabled', async () => {
+  FeatureFlags_mock.getBoolean.mockReturnValueOnce(false);
+
+  await run(context_stub);
+
+  expect(printer_mock.error.mock.calls.length).toBe(1);
+  expect(pushResourcesMock.mock.calls.length).toBe(0);
+});
+
+it('exits early if no api in project', async () => {
+  stateManager_mock.getMeta.mockReturnValueOnce({
+    api: {},
+  });
+
+  await run(context_stub);
+
+  expect(printer_mock.info.mock.calls.length).toBe(1);
+  expect(pushResourcesMock.mock.calls.length).toBe(0);
+});
+
+it('asks for strong confirmation before continuing', async () => {
+  stateManager_mock.getMeta.mockReturnValueOnce({
+    api: {
+      testapiname: {
+        service: 'AppSync',
+      },
+    },
+  });
+
+  await run(context_stub);
+
+  expect(prompter_mock.input.mock.calls.length).toBe(1);
+  expect(pushResourcesMock.mock.calls.length).toBe(1);
+  expect(pushResourcesMock.mock.calls[0][4]).toBe(true); // rebuild flag is set
+});

packages/amplify-category-api/src/commands/api.js

@@ -41,6 +41,11 @@ module.exports = {
         name: 'console',
         description: 'Opens the web console for the selected api service',
       },
+      {
+        name: 'rebuild',
+        description:
+          'Removes and recreates all DynamoDB tables backing a GraphQL API. Useful for resetting test data during the development phase of an app',
+      },
     ];
 
     context.amplify.showHelp(header, commands);

packages/amplify-category-api/src/commands/api/rebuild.ts

@@ -0,0 +1,40 @@
+import { $TSContext, FeatureFlags, stateManager } from 'amplify-cli-core';
+import { printer, prompter, exact } from 'amplify-prompts';
+
+const subcommand = 'rebuild';
+const category = 'api';
+
+export const name = subcommand;
+
+const rebuild = true;
+
+export const run = async (context: $TSContext) => {
+  if (!FeatureFlags.getBoolean('graphqlTransformer.enableIterativeGSIUpdates')) {
+    printer.error('Iterative GSI Updates must be enabled to rebuild an API. See https://docs.amplify.aws/cli/reference/feature-flags/');
+    return;
+  }
+  const apiNames = Object.entries(stateManager.getMeta()?.api || {})
+    .filter(([_, meta]) => (meta as any).service === 'AppSync')
+    .map(([name]) => name);
+  if (apiNames.length === 0) {
+    printer.info('No GraphQL API configured in the project. Only GraphQL APIs can be rebuilt. To add a GraphQL API run `amplify add api`.');
+    return;
+  }
+  if (apiNames.length > 1) {
+    // this condition should never hit as we have upstream defensive logic to prevent multiple GraphQL APIs. But just to cover all the bases
+    printer.error(
+      'You have multiple GraphQL APIs in the project. Only one GraphQL API is allowed per project. Run `amplify remove api` to remove an API.',
+    );
+    return;
+  }
+  const apiName = apiNames[0];
+  printer.warn(`This will recreate all tables backing models in your GraphQL API ${apiName}.`);
+  printer.warn('ALL EXISTING DATA IN THESE TABLES WILL BE LOST.');
+  await prompter.input('Type the name of the API to confirm you want to continue', {
+    validate: exact(apiName, 'Input does not match the GraphQL API name'),
+  });
+  const { amplify, parameters } = context;
+  const resourceName = parameters.first;
+  amplify.constructExeInfo(context);
+  return amplify.pushResources(context, category, resourceName, undefined, rebuild);
+};

packages/amplify-category-api/tsconfig.json

@@ -14,7 +14,9 @@
     "src/__tests__"
   ],
   "references": [
+    {"path": "../amplify-cli-core"},
     {"path": "../amplify-headless-interface"},
+    {"path": "../amplify-prompts"},
     {"path": "../graphql-transformer-core"},
     {"path": "../amplify-util-headless-input"},
   ]

packages/amplify-category-function/src/index.ts

@@ -28,6 +28,7 @@ export { hashLayerResource } from './provider-utils/awscloudformation/utils/laye
 export { migrateLegacyLayer } from './provider-utils/awscloudformation/utils/layerMigrationUtils';
 export { packageResource } from './provider-utils/awscloudformation/utils/package';
 export { updateDependentFunctionsCfn } from './provider-utils/awscloudformation/utils/updateDependentFunctionCfn';
+export { loadFunctionParameters } from './provider-utils/awscloudformation/utils/loadFunctionParameters';
 
 export async function add(context, providerName, service, parameters) {
   const options = {

packages/amplify-cli-core/src/index.ts

@@ -248,6 +248,7 @@ interface AmplifyToolkit {
     category?: string,
     resourceName?: string,
     filteredResources?: { category: string; resourceName: string }[],
+    rebuild?: boolean,
   ) => $TSAny;
   storeCurrentCloudBackend: () => $TSAny;
   readJsonFile: () => $TSAny;
@@ -316,9 +317,7 @@ interface AmplifyToolkit {
   leaveBreadcrumbs: (category: string, resourceName: string, breadcrumbs: unknown) => void;
   readBreadcrumbs: (category: string, resourceName: string) => $TSAny;
   loadRuntimePlugin: (context: $TSContext, pluginId: string) => Promise<$TSAny>;
-  getImportedAuthProperties: (
-    context: $TSContext,
-  ) => {
+  getImportedAuthProperties: (context: $TSContext) => {
     imported: boolean;
     userPoolId?: string;
     authRoleArn?: string;

packages/amplify-cli/src/extensions/amplify-helpers/push-resources.ts

@@ -11,6 +11,7 @@ export async function pushResources(
   category?: string,
   resourceName?: string,
   filteredResources?: { category: string; resourceName: string }[],
+  rebuild: boolean = false,
 ) {
   if (context.parameters.options['iterative-rollback']) {
     // validate --iterative-rollback with --force
@@ -49,16 +50,21 @@ export async function pushResources(
     }
   }
 
-  const hasChanges = await showResourceTable(category, resourceName, filteredResources);
+  let hasChanges = false;
+  if (!rebuild) {
+    // status table does not have a way to show resource in "rebuild" state so skipping it to avoid confusion
+    hasChanges = await showResourceTable(category, resourceName, filteredResources);
+  }
 
   // no changes detected
-  if (!hasChanges && !context.exeInfo.forcePush) {
+  if (!hasChanges && !context.exeInfo.forcePush && !rebuild) {
     context.print.info('\nNo changes detected');
 
     return context;
   }
 
-  let continueToPush = context.exeInfo && context.exeInfo.inputParams && context.exeInfo.inputParams.yes;
+  // rebuild has an upstream confirmation prompt so no need to prompt again here
+  let continueToPush = (context.exeInfo && context.exeInfo.inputParams && context.exeInfo.inputParams.yes) || rebuild;
 
   if (!continueToPush) {
     if (context.exeInfo.iterativeRollback) {
@@ -68,18 +74,11 @@ export async function pushResources(
   }
 
   if (continueToPush) {
-    try {
-      // Get current-cloud-backend's amplify-meta
-      const currentAmplifyMeta = stateManager.getCurrentMeta();
-
-      await providersPush(context, category, resourceName, filteredResources);
-      await onCategoryOutputsChange(context, currentAmplifyMeta);
-    } catch (err) {
-      // Handle the errors and print them nicely for the user.
-      context.print.error(`\n${err.message}`);
+    // Get current-cloud-backend's amplify-meta
+    const currentAmplifyMeta = stateManager.getCurrentMeta();
 
-      throw err;
-    }
+    await providersPush(context, rebuild, category, resourceName, filteredResources);
+    await onCategoryOutputsChange(context, currentAmplifyMeta);
   } else {
     // there's currently no other mechanism to stop the execution of the postPush workflow in this case, so exiting here
     exitOnNextTick(1);
@@ -88,15 +87,21 @@ export async function pushResources(
   return continueToPush;
 }
 
-async function providersPush(context: $TSContext, category, resourceName, filteredResources) {
+async function providersPush(
+  context: $TSContext,
+  rebuild: boolean = false,
+  category?: string,
+  resourceName?: string,
+  filteredResources?: { category: string; resourceName: string }[],
+) {
   const { providers } = getProjectConfig();
   const providerPlugins = getProviderPlugins(context);
   const providerPromises: (() => Promise<$TSAny>)[] = [];
 
   for (const provider of providers) {
     const providerModule = require(providerPlugins[provider]);
     const resourceDefiniton = await context.amplify.getResourceStatus(category, resourceName, provider, filteredResources);
-    providerPromises.push(providerModule.pushResources(context, resourceDefiniton));
+    providerPromises.push(providerModule.pushResources(context, resourceDefiniton, rebuild));
   }
 
   await Promise.all(providerPromises);