Update lambda Function arn at cloudfront using a script

index.js

var cloudfront = require("aws-sdk/clients/cloudfront");
var cloudfrontClient = new cloudfront({ apiVersion: "2020-05-31" });

const distributionId = process.argv[2];
const lambdaEdgeEvent = process.argv[3];
const lambdaEdgeArn = process.argv[4];

function updateLambdaFunction(lambdaFunctionAssociations) {
    if (lambdaFunctionAssociations["Quantity"] < 0) return [];
    lambdaFunctionAssociations["Items"].forEach(function (item) {
        if (item["EventType"] === lambdaEdgeEvent) {
            item["LambdaFunctionARN"] = lambdaEdgeArn;
        }
    });
    return lambdaFunctionAssociations;
}

cloudfrontClient.getDistributionConfig(
    {
        Id: distributionId,
    },
    function (err, data) {
        if (!err) {
            lambdaFunctionAssociations =
                data["DistributionConfig"]["DefaultCacheBehavior"][
                "LambdaFunctionAssociations"
                ];
            if (lambdaFunctionAssociations) {
                data["DistributionConfig"]["DefaultCacheBehavior"][
                    "LambdaFunctionAssociations"
                ] = updateLambdaFunction(lambdaFunctionAssociations);
            }

            if (
                data["DistributionConfig"]["CacheBehaviors"] &&
                data["DistributionConfig"]["CacheBehaviors"]["Quantity"] > 0
            ) {
                data["DistributionConfig"]["CacheBehaviors"].Items.forEach(function (
                    item
                ) {
                    item.LambdaFunctionAssociations = updateLambdaFunction(
                        item.LambdaFunctionAssociations
                    );
                });
            } 
            cloudfrontClient.updateDistribution({
                Id: distributionId,
                DistributionConfig: data['DistributionConfig'],
                IfMatch: data['ETag']
            }, function (err, _) {
                if (!err) {
                    console.log('Updated Successfully');
                } else {
                    console.log(err);
                }
            })
        } else {
            console.log(err);
        }
    }
);

usage
node index.js cloudfront-id lambda edge event lambda-arn