@aws-amplify/amplify-category-api/resources/awscloudformation/lambdas/pipeline.js

Amplify CLI API Category Plugin

const { CloudFormation } = require('@aws-sdk/client-cloudformation');
const { CodePipeline } = require('@aws-sdk/client-codepipeline');

const stageName = 'Source';
const actionName = 'Source';

const codePipeline = new CodePipeline();
const cloudFormation = new CloudFormation();

exports.handler = async function({ RequestType, ResourceProperties, StackId }) {
  const { pipelineName, artifactBucketName, artifactKey, deploymentMechanism } = ResourceProperties;

  console.log('Properties', ResourceProperties);

  switch (RequestType) {
    case 'Delete':
      return { IsComplete: true };
    case 'Update':
      const [, StackName] = StackId.split('/');
      const { Stacks } = await cloudFormation.describeStacks({ StackName });
      const [{ StackStatus }] = Stacks;

      if (StackStatus.includes('ROLLBACK')) {
        return { IsComplete: true };
      }
  }

  let stages;

  try {
    const { pipeline } = await codePipeline.getPipeline({ name: pipelineName });

    ({ stages } = pipeline);
  } catch (error) {
    const { code } = error;

    switch (code) {
      case 'PipelineNotFoundException':
        console.warn(error);

        return {
          IsComplete: false,
        };
      default:
        throw error;
    }
  }

  const stage = stages.find(({ name }) => name === stageName);

  if (stage === undefined) {
    throw new Error(`There is no stage named "${stageName}" in the "${pipelineName}" pipeline`);
  }

  const action = stage.actions.find(({ name }) => name === actionName);

  if (action === undefined) {
    throw new Error(`There is no action named "${actionName}" in the "${stageName}" stage of the "${pipelineName}" pipeline`);
  }

  const {
    configuration,
    configuration: { S3Bucket, S3ObjectKey },
  } = action;

  if (deploymentMechanism === 'FULLY_MANAGED') {
    if (S3Bucket !== artifactBucketName || S3ObjectKey !== artifactKey) {
      console.warn(
        `Bucket "${artifactBucketName}" and key "${artifactKey}" dont match the "${actionName}" action configuration ${JSON.stringify(
          configuration,
        )}`,
      );

      return {
        IsComplete: false,
      };
    }
  }

  let execution;

  try {
    const { pipelineExecutionSummaries } = await codePipeline.listPipelineExecutions({ pipelineName });

    [execution] = pipelineExecutionSummaries;
  } catch (error) {
    console.warn(error);

    return {
      IsComplete: false,
    };
  }

  console.log(execution);

  const { status } = execution || {};

  let IsComplete = false;

  switch (status) {
    case 'Failed':
    case 'Stopped':
      throw new Error("The execution didn't succeed");
    case 'Succeeded':
      IsComplete = true;
  }

  return { IsComplete };
};