aws-cdk-lib/aws-ecs/lib/drain-hook/instance-drain-hook.js

Version 2 of the AWS Cloud Development Kit library

"use strict";var __createBinding=exports&&exports.__createBinding||(Object.create?(function(o,m,k,k2){k2===void 0&&(k2=k);var desc=Object.getOwnPropertyDescriptor(m,k);(!desc||("get"in desc?!m.__esModule:desc.writable||desc.configurable))&&(desc={enumerable:!0,get:function(){return m[k]}}),Object.defineProperty(o,k2,desc)}):(function(o,m,k,k2){k2===void 0&&(k2=k),o[k2]=m[k]})),__setModuleDefault=exports&&exports.__setModuleDefault||(Object.create?(function(o,v){Object.defineProperty(o,"default",{enumerable:!0,value:v})}):function(o,v){o.default=v}),__importStar=exports&&exports.__importStar||(function(){var ownKeys=function(o){return ownKeys=Object.getOwnPropertyNames||function(o2){var ar=[];for(var k in o2)Object.prototype.hasOwnProperty.call(o2,k)&&(ar[ar.length]=k);return ar},ownKeys(o)};return function(mod){if(mod&&mod.__esModule)return mod;var result={};if(mod!=null)for(var k=ownKeys(mod),i=0;i<k.length;i++)k[i]!=="default"&&__createBinding(result,mod,k[i]);return __setModuleDefault(result,mod),result}})();Object.defineProperty(exports,"__esModule",{value:!0}),exports.InstanceDrainHook=void 0;var fs=()=>{var tmp=__importStar(require("fs"));return fs=()=>tmp,tmp},path=()=>{var tmp=__importStar(require("path"));return path=()=>tmp,tmp},constructs_1=()=>{var tmp=require("constructs");return constructs_1=()=>tmp,tmp},autoscaling=()=>{var tmp=__importStar(require("../../../aws-autoscaling"));return autoscaling=()=>tmp,tmp},hooks=()=>{var tmp=__importStar(require("../../../aws-autoscaling-hooktargets"));return hooks=()=>tmp,tmp},iam=()=>{var tmp=__importStar(require("../../../aws-iam"));return iam=()=>tmp,tmp},lambda=()=>{var tmp=__importStar(require("../../../aws-lambda"));return lambda=()=>tmp,tmp},cdk=()=>{var tmp=__importStar(require("../../../core"));return cdk=()=>tmp,tmp};class InstanceDrainHook extends constructs_1().Construct{constructor(scope,id,props){super(scope,id);const drainTime=props.drainTime||cdk().Duration.minutes(5),fn=new(lambda()).Function(this,"Function",{code:lambda().Code.fromInline(fs().readFileSync(path().join(__dirname,"..","..","..","custom-resource-handlers","dist","aws-ecs","lambda-source","index.py"),{encoding:"utf-8"})),handler:"index.lambda_handler",runtime:lambda().Runtime.determineLatestPythonRuntime(this),timeout:cdk().Duration.seconds(Math.min(drainTime.toSeconds()+10,900)),environment:{CLUSTER:props.cluster.clusterRef.clusterName}});props.autoScalingGroup.addLifecycleHook("DrainHook",{lifecycleTransition:autoscaling().LifecycleTransition.INSTANCE_TERMINATING,defaultResult:autoscaling().DefaultResult.CONTINUE,notificationTarget:new(hooks()).FunctionHook(fn,props.topicEncryptionKey),heartbeatTimeout:drainTime}),fn.addToRolePolicy(new(iam()).PolicyStatement({actions:["ec2:DescribeInstances","ec2:DescribeInstanceAttribute","ec2:DescribeInstanceStatus","ec2:DescribeHosts"],resources:["*"]})),fn.addToRolePolicy(new(iam()).PolicyStatement({actions:["autoscaling:CompleteLifecycleAction"],resources:[props.autoScalingGroup.autoScalingGroupArn]})),fn.addToRolePolicy(new(iam()).PolicyStatement({actions:["ecs:DescribeContainerInstances","ecs:DescribeTasks"],resources:["*"],conditions:{ArnEquals:{"ecs:cluster":props.cluster.clusterRef.clusterArn}}})),fn.addToRolePolicy(new(iam()).PolicyStatement({actions:["ecs:ListContainerInstances","ecs:SubmitContainerStateChange","ecs:SubmitTaskStateChange"],resources:[props.cluster.clusterRef.clusterArn]})),fn.addToRolePolicy(new(iam()).PolicyStatement({actions:["ecs:UpdateContainerInstancesState","ecs:ListTasks"],conditions:{ArnEquals:{"ecs:cluster":props.cluster.clusterRef.clusterArn}},resources:["*"]}))}}exports.InstanceDrainHook=InstanceDrainHook;