@datadog/datadog-ci 0.17.12 → 0.17.13

package/dist/commands/lambda/__tests__/prompt.test.js

@@ -0,0 +1,216 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+jest.mock('inquirer');
+const chalk_1 = require("chalk");
+const inquirer_1 = require("inquirer");
+const constants_1 = require("../constants");
+const prompt_1 = require("../prompt");
+const fixtures_1 = require("./fixtures");
+describe('prompt', () => {
+    describe('confirmationQuestion', () => {
+        test('returns question with provided message', () => {
+            const message = 'Do you want to continue?';
+            const question = prompt_1.confirmationQuestion(message);
+            expect(question.message).toBe(message);
+        });
+    });
+    describe('datadogApiKeyTypeQuestion', () => {
+        test('returns question with message pointing to the correct given site', () => {
+            const site = 'datadoghq.com';
+            const question = prompt_1.datadogApiKeyTypeQuestion(site);
+            expect(question.message).toBe(`Which type of Datadog API Key you want to set? \nLearn more at ${chalk_1.blueBright(`https://app.${site}/organization-settings/api-keys`)}`);
+        });
+    });
+    describe('datadogEnvVarsQuestions', () => {
+        test('returns correct message when user selects DATADOG_API_KEY', () => {
+            const datadogApiKeyType = {
+                envVar: constants_1.CI_API_KEY_ENV_VAR,
+                message: 'API Key:',
+            };
+            const question = prompt_1.datadogEnvVarsQuestions(datadogApiKeyType);
+            expect(question.message).toBe('API Key:');
+            expect(question.name).toBe(constants_1.CI_API_KEY_ENV_VAR);
+        });
+        test('returns correct message when user selects DATADOG_KMS_API_KEY', () => {
+            const datadogApiKeyType = {
+                envVar: constants_1.CI_KMS_API_KEY_ENV_VAR,
+                message: 'KMS API Key:',
+            };
+            const question = prompt_1.datadogEnvVarsQuestions(datadogApiKeyType);
+            expect(question.message).toBe('KMS API Key:');
+            expect(question.name).toBe(constants_1.CI_KMS_API_KEY_ENV_VAR);
+        });
+        test('returns correct message when user selects DATADOG_API_KEY_SECRET_ARN', () => {
+            const datadogApiKeyType = {
+                envVar: constants_1.CI_API_KEY_SECRET_ARN_ENV_VAR,
+                message: 'API Key Secret ARN:',
+            };
+            const question = prompt_1.datadogEnvVarsQuestions(datadogApiKeyType);
+            expect(question.message).toBe('API Key Secret ARN:');
+            expect(question.name).toBe(constants_1.CI_API_KEY_SECRET_ARN_ENV_VAR);
+        });
+    });
+    describe('functionSelectionQuestion', () => {
+        test('returns question with the provided function names being its choices', () => {
+            const functionNames = ['my-func', 'my-func-2', 'my-third-func'];
+            const question = prompt_1.functionSelectionQuestion(functionNames);
+            expect(question.choices).toBe(functionNames);
+        });
+    });
+    describe('requestAWSCrendentials', () => {
+        const OLD_ENV = process.env;
+        beforeEach(() => {
+            jest.resetModules();
+            process.env = {};
+        });
+        afterAll(() => {
+            process.env = OLD_ENV;
+        });
+        test('sets the AWS credentials as environment variables', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+            }));
+            yield prompt_1.requestAWSCredentials();
+            expect(process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]).toBe(fixtures_1.mockAwsAccessKeyId);
+            expect(process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]).toBe(fixtures_1.mockAwsSecretAccessKey);
+            expect(process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR]).toBe('sa-east-1');
+        }));
+        test('sets the AWS credentials with session token as environment variables', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+                [constants_1.AWS_SESSION_TOKEN_ENV_VAR]: 'some-session-token',
+            }));
+            yield prompt_1.requestAWSCredentials();
+            expect(process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]).toBe(fixtures_1.mockAwsAccessKeyId);
+            expect(process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]).toBe(fixtures_1.mockAwsSecretAccessKey);
+            expect(process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR]).toBe('sa-east-1');
+            expect(process.env[constants_1.AWS_SESSION_TOKEN_ENV_VAR]).toBe('some-session-token');
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestAWSCredentials();
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't set AWS Credentials. Unexpected error");
+        }));
+    });
+    describe('requestChangesConfirmation', () => {
+        test('returns boolean when users responds to confirmation question', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({
+                confirmation: true,
+            }));
+            const confirmation = yield prompt_1.requestChangesConfirmation('Do you want to continue?');
+            expect(confirmation).toBe(true);
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestChangesConfirmation('Do you wanna continue?');
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't receive confirmation. Unexpected error");
+        }));
+    });
+    describe('requestDatadogEnvVars', () => {
+        const OLD_ENV = process.env;
+        beforeEach(() => {
+            jest.resetModules();
+            process.env = {};
+        });
+        afterAll(() => {
+            process.env = OLD_ENV;
+        });
+        test('sets the Datadog Environment Variables as provided/selected by user', () => __awaiter(void 0, void 0, void 0, function* () {
+            const site = 'datadoghq.com';
+            inquirer_1.prompt.mockImplementation((question) => {
+                switch (question.name) {
+                    case constants_1.CI_API_KEY_ENV_VAR:
+                        return Promise.resolve({
+                            [constants_1.CI_API_KEY_ENV_VAR]: fixtures_1.mockDatadogApiKey,
+                        });
+                    case constants_1.CI_SITE_ENV_VAR:
+                        return Promise.resolve({
+                            [constants_1.CI_SITE_ENV_VAR]: 'datadoghq.com',
+                        });
+                    case 'type':
+                        return Promise.resolve({
+                            type: {
+                                envVar: constants_1.CI_API_KEY_ENV_VAR,
+                                message: 'API Key:',
+                            },
+                        });
+                    default:
+                }
+            });
+            yield prompt_1.requestDatadogEnvVars();
+            expect(process.env[constants_1.CI_SITE_ENV_VAR]).toBe(site);
+            expect(process.env[constants_1.CI_API_KEY_ENV_VAR]).toBe(fixtures_1.mockDatadogApiKey);
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestDatadogEnvVars();
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't set Datadog Environment Variables. Unexpected error");
+        }));
+    });
+    describe('requestFunctionSelection', () => {
+        const selectedFunctions = ['my-func', 'my-func-2', 'my-third-func'];
+        test('returns the selected functions', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({ functions: selectedFunctions }));
+            const functions = yield prompt_1.requestFunctionSelection(selectedFunctions);
+            expect(functions).toBe(selectedFunctions);
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestFunctionSelection(selectedFunctions);
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't receive selected functions. Unexpected error");
+        }));
+    });
+});

package/dist/commands/lambda/__tests__/uninstrument.test.js

@@ -31,10 +31,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
 // tslint:disable: no-string-literal
 jest.mock('fs');
 jest.mock('aws-sdk');
+jest.mock('../prompt');
 const aws_sdk_1 = require("aws-sdk");
 const chalk_1 = require("chalk");
 const fs = __importStar(require("fs"));
 const constants_1 = require("../constants");
+const prompt_1 = require("../prompt");
 const uninstrument_1 = require("../uninstrument");
 const fixtures_1 = require("./fixtures");
 describe('uninstrument', () => {
@@ -170,7 +172,7 @@ UpdateFunctionConfiguration -> arn:aws:lambda:us-east-1:000000000000:function:un
             const code = yield command['execute']();
             const output = command.context.stdout.toString();
             expect(code).toBe(1);
-            expect(output).toMatch(`${chalk_1.red('[Error]')} Couldn't fetch lambda functions. Lambda failed\n`);
+            expect(output).toMatch(`${chalk_1.red('[Error]')} Couldn't fetch Lambda functions. Lambda failed\n`);
         }));
         test("aborts early when function regions can't be found", () => __awaiter(void 0, void 0, void 0, function* () {
             ;
@@ -193,7 +195,7 @@ UpdateFunctionConfiguration -> arn:aws:lambda:us-east-1:000000000000:function:un
             const output = context.stdout.toString();
             expect(code).toBe(1);
             expect(output).toMatchInlineSnapshot(`
-        "No functions specified for un-instrumentation.
+        "${chalk_1.red('[Error]')} No functions specified for un-instrumentation.
         "
       `);
         }));
@@ -206,7 +208,7 @@ UpdateFunctionConfiguration -> arn:aws:lambda:us-east-1:000000000000:function:un
             yield command['execute']();
             const output = command.context.stdout.toString();
             expect(output).toMatchInlineSnapshot(`
-        "No functions specified for un-instrumentation.
+        "${chalk_1.red('[Error]')} No functions specified for un-instrumentation.
         "
       `);
         }));
@@ -263,7 +265,311 @@ UpdateFunctionConfiguration -> arn:aws:lambda:us-east-1:000000000000:function:un
             const code = yield command['execute']();
             const output = command.context.stdout.toString();
             expect(code).toBe(1);
-            expect(output).toMatch("Fetching lambda functions, this might take a while.\nCouldn't fetch lambda functions. Error: Max retry count exceeded.\n");
+            expect(output).toMatch(`Fetching Lambda functions, this might take a while.\n${chalk_1.red('[Error]')} Couldn't fetch Lambda functions. Error: Max retry count exceeded.\n`);
+        }));
+        test('uninstrument multiple functions interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                            [constants_1.LOG_LEVEL_ENV_VAR]: 'debug',
+                            [constants_1.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                            [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                            [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                            [constants_1.TRACE_ENABLED_ENV_VAR]: 'true',
+                            [constants_1.VERSION_ENV_VAR]: '0.2',
+                            USER_VARIABLE: 'shouldnt be deleted by uninstrumentation',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                    FunctionName: 'lambda-hello-world',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python38:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.8',
+                },
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                            [constants_1.LOG_LEVEL_ENV_VAR]: 'debug',
+                            [constants_1.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                            [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                            [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                            [constants_1.TRACE_ENABLED_ENV_VAR]: 'true',
+                            [constants_1.VERSION_ENV_VAR]: '0.2',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+                    FunctionName: 'lambda-hello-world-2',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python39:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.9',
+                },
+            }));
+            prompt_1.requestAWSCredentials.mockImplementation(() => {
+                process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR] = fixtures_1.mockAwsAccessKeyId;
+                process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR] = fixtures_1.mockAwsSecretAccessKey;
+                process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR] = 'sa-east-1';
+            });
+            prompt_1.requestFunctionSelection.mockImplementation(() => [
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+            ]);
+            prompt_1.requestChangesConfirmation.mockImplementation(() => true);
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(0);
+            expect(output).toMatchInlineSnapshot(`
+"${chalk_1.bold(chalk_1.yellow('[!]'))} No existing AWS credentials found, let's set them up!
+Fetching Lambda functions, this might take a while.\n
+${chalk_1.bold(chalk_1.yellow('[!]'))} Functions to be updated:
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world')}
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2')}\n
+Will apply the following updates:
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {
+      \\"USER_VARIABLE\\": \\"shouldnt be deleted by uninstrumentation\\"
+    }
+  },
+  \\"Layers\\": []
+}
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {}
+  },
+  \\"Layers\\": []
+}
+${chalk_1.yellow('[!]')} Confirmation needed.
+${chalk_1.yellow('[!]')} Uninstrumenting functions.
+"
+`);
+        }));
+        test('uninstrument multiple specified functions interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                            [constants_1.LOG_LEVEL_ENV_VAR]: 'debug',
+                            [constants_1.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                            [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                            [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                            [constants_1.TRACE_ENABLED_ENV_VAR]: 'true',
+                            [constants_1.VERSION_ENV_VAR]: '0.2',
+                            USER_VARIABLE: 'shouldnt be deleted by uninstrumentation',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                    FunctionName: 'lambda-hello-world',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python38:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.8',
+                },
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                            [constants_1.LOG_LEVEL_ENV_VAR]: 'debug',
+                            [constants_1.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                            [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                            [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                            [constants_1.TRACE_ENABLED_ENV_VAR]: 'true',
+                            [constants_1.VERSION_ENV_VAR]: '0.2',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+                    FunctionName: 'lambda-hello-world-2',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python39:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.9',
+                },
+            }));
+            prompt_1.requestAWSCredentials.mockImplementation(() => {
+                process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR] = fixtures_1.mockAwsAccessKeyId;
+                process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR] = fixtures_1.mockAwsSecretAccessKey;
+                process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR] = 'sa-east-1';
+            });
+            prompt_1.requestFunctionSelection.mockImplementation(() => [
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+            ]);
+            prompt_1.requestChangesConfirmation.mockImplementation(() => true);
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run([
+                'lambda',
+                'uninstrument',
+                '-i',
+                '-f',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                '-f',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+            ], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(0);
+            expect(output).toMatchInlineSnapshot(`
+"${chalk_1.bold(chalk_1.yellow('[!]'))} No existing AWS credentials found, let's set them up!\n
+${chalk_1.bold(chalk_1.yellow('[!]'))} Functions to be updated:
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world')}
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2')}\n
+Will apply the following updates:
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {
+      \\"USER_VARIABLE\\": \\"shouldnt be deleted by uninstrumentation\\"
+    }
+  },
+  \\"Layers\\": []
+}
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {}
+  },
+  \\"Layers\\": []
+}
+${chalk_1.yellow('[!]')} Confirmation needed.
+${chalk_1.yellow('[!]')} Uninstrumenting functions.
+"
+`);
+        }));
+        test('aborts if a problem occurs while setting the AWS credentials interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            prompt_1.requestAWSCredentials.mockImplementation(() => Promise.reject('Unexpected error'));
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatchInlineSnapshot(`
+"${chalk_1.bold(chalk_1.yellow('[!]'))} No existing AWS credentials found, let's set them up!
+${chalk_1.red('[Error]')} Unexpected error
+"
+`);
+        }));
+        test('aborts if there are no functions to uninstrument in the user AWS account', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env = {
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+            };
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({}));
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatchInlineSnapshot(`
+"Fetching Lambda functions, this might take a while.
+${chalk_1.red('[Error]')} Couldn't find any Lambda functions in the specified region.
+"
+`);
+        }));
+        test('aborts early when the aws-sdk throws an error while uninstrumenting interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env = {
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+            };
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => ({
+                listFunctions: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('Lambda failed') })),
+            }));
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatchInlineSnapshot(`
+"Fetching Lambda functions, this might take a while.
+${chalk_1.red('[Error]')} Couldn't fetch Lambda functions. Error: Max retry count exceeded.
+"
+`);
         }));
     });
     describe('printPlannedActions', () => {

package/dist/commands/lambda/constants.d.ts

@@ -29,6 +29,7 @@ export declare const HANDLER_LOCATION: {
     'python3.9': string;
 };
 export declare const SITES: string[];
+export declare const AWS_REGIONS: string[];
 export declare const DEFAULT_LAYER_AWS_ACCOUNT = "464622532012";
 export declare const GOVCLOUD_LAYER_AWS_ACCOUNT = "002406178527";
 export declare const SUBSCRIPTION_FILTER_NAME = "datadog-ci-filter";
@@ -46,10 +47,19 @@ export declare const SERVICE_ENV_VAR = "DD_SERVICE";
 export declare const VERSION_ENV_VAR = "DD_VERSION";
 export declare const ENVIRONMENT_ENV_VAR = "DD_ENV";
 export declare const EXTRA_TAGS_ENV_VAR = "DD_TAGS";
+export declare const CAPTURE_LAMBDA_PAYLOAD_ENV_VAR = "DD_CAPTURE_LAMBDA_PAYLOAD";
 export declare const CI_SITE_ENV_VAR = "DATADOG_SITE";
 export declare const CI_API_KEY_ENV_VAR = "DATADOG_API_KEY";
 export declare const CI_API_KEY_SECRET_ARN_ENV_VAR = "DATADOG_API_KEY_SECRET_ARN";
 export declare const CI_KMS_API_KEY_ENV_VAR = "DATADOG_KMS_API_KEY";
+export declare const AWS_ACCESS_KEY_ID_ENV_VAR = "AWS_ACCESS_KEY_ID";
+export declare const AWS_SECRET_ACCESS_KEY_ENV_VAR = "AWS_SECRET_ACCESS_KEY";
+export declare const AWS_DEFAULT_REGION_ENV_VAR = "AWS_DEFAULT_REGION";
+export declare const AWS_SESSION_TOKEN_ENV_VAR = "AWS_SESSION_TOKEN";
 export declare const LIST_FUNCTIONS_MAX_RETRY_COUNT = 2;
 export declare const MAX_LAMBDA_STATE_CHECK_ATTEMPTS = 3;
 export declare const EXTRA_TAGS_REG_EXP: RegExp;
+export declare const AWS_ACCESS_KEY_ID_REG_EXP: RegExp;
+export declare const AWS_SECRET_ACCESS_KEY_REG_EXP: RegExp;
+export declare const DATADOG_API_KEY_REG_EXP: RegExp;
+export declare const DATADOG_APP_KEY_REG_EXP: RegExp;

package/dist/commands/lambda/constants.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.EXTRA_TAGS_REG_EXP = exports.MAX_LAMBDA_STATE_CHECK_ATTEMPTS = exports.LIST_FUNCTIONS_MAX_RETRY_COUNT = exports.CI_KMS_API_KEY_ENV_VAR = exports.CI_API_KEY_SECRET_ARN_ENV_VAR = exports.CI_API_KEY_ENV_VAR = exports.CI_SITE_ENV_VAR = exports.EXTRA_TAGS_ENV_VAR = exports.ENVIRONMENT_ENV_VAR = exports.VERSION_ENV_VAR = exports.SERVICE_ENV_VAR = exports.LAMBDA_HANDLER_ENV_VAR = exports.LOG_LEVEL_ENV_VAR = exports.FLUSH_TO_LOG_ENV_VAR = exports.MERGE_XRAY_TRACES_ENV_VAR = exports.TRACE_ENABLED_ENV_VAR = exports.SITE_ENV_VAR = exports.KMS_API_KEY_ENV_VAR = exports.API_KEY_SECRET_ARN_ENV_VAR = exports.API_KEY_ENV_VAR = exports.TAG_VERSION_NAME = exports.SUBSCRIPTION_FILTER_NAME = exports.GOVCLOUD_LAYER_AWS_ACCOUNT = exports.DEFAULT_LAYER_AWS_ACCOUNT = exports.SITES = exports.HANDLER_LOCATION = exports.RUNTIME_LOOKUP = exports.RuntimeType = exports.ARM_LAYER_SUFFIX = exports.ARM64_ARCHITECTURE = exports.ARM_RUNTIMES = exports.RUNTIME_LAYER_LOOKUP = exports.EXTENSION_LAYER_KEY = exports.DD_LAMBDA_EXTENSION_LAYER_NAME = void 0;
+exports.DATADOG_APP_KEY_REG_EXP = exports.DATADOG_API_KEY_REG_EXP = exports.AWS_SECRET_ACCESS_KEY_REG_EXP = exports.AWS_ACCESS_KEY_ID_REG_EXP = exports.EXTRA_TAGS_REG_EXP = exports.MAX_LAMBDA_STATE_CHECK_ATTEMPTS = exports.LIST_FUNCTIONS_MAX_RETRY_COUNT = exports.AWS_SESSION_TOKEN_ENV_VAR = exports.AWS_DEFAULT_REGION_ENV_VAR = exports.AWS_SECRET_ACCESS_KEY_ENV_VAR = exports.AWS_ACCESS_KEY_ID_ENV_VAR = exports.CI_KMS_API_KEY_ENV_VAR = exports.CI_API_KEY_SECRET_ARN_ENV_VAR = exports.CI_API_KEY_ENV_VAR = exports.CI_SITE_ENV_VAR = exports.CAPTURE_LAMBDA_PAYLOAD_ENV_VAR = exports.EXTRA_TAGS_ENV_VAR = exports.ENVIRONMENT_ENV_VAR = exports.VERSION_ENV_VAR = exports.SERVICE_ENV_VAR = exports.LAMBDA_HANDLER_ENV_VAR = exports.LOG_LEVEL_ENV_VAR = exports.FLUSH_TO_LOG_ENV_VAR = exports.MERGE_XRAY_TRACES_ENV_VAR = exports.TRACE_ENABLED_ENV_VAR = exports.SITE_ENV_VAR = exports.KMS_API_KEY_ENV_VAR = exports.API_KEY_SECRET_ARN_ENV_VAR = exports.API_KEY_ENV_VAR = exports.TAG_VERSION_NAME = exports.SUBSCRIPTION_FILTER_NAME = exports.GOVCLOUD_LAYER_AWS_ACCOUNT = exports.DEFAULT_LAYER_AWS_ACCOUNT = exports.AWS_REGIONS = exports.SITES = exports.HANDLER_LOCATION = exports.RUNTIME_LOOKUP = exports.RuntimeType = exports.ARM_LAYER_SUFFIX = exports.ARM64_ARCHITECTURE = exports.ARM_RUNTIMES = exports.RUNTIME_LAYER_LOOKUP = exports.EXTENSION_LAYER_KEY = exports.DD_LAMBDA_EXTENSION_LAYER_NAME = void 0;
 exports.DD_LAMBDA_EXTENSION_LAYER_NAME = 'Datadog-Extension';
 exports.EXTENSION_LAYER_KEY = 'extension';
 exports.RUNTIME_LAYER_LOOKUP = {
@@ -43,7 +43,32 @@ exports.SITES = [
     'datadoghq.eu',
     'us3.datadoghq.com',
     'us5.datadoghq.com',
-    'ddog-gov.,com',
+    'ddog-gov.com',
+];
+exports.AWS_REGIONS = [
+    'us-east-1',
+    'us-east-2',
+    'us-west-1',
+    'us-west-2',
+    'af-south-1',
+    'ap-east-1',
+    'ap-south-1',
+    'ap-northeast-3',
+    'ap-northeast-2',
+    'ap-southeast-1',
+    'ap-southeast-2',
+    'ap-northeast-1',
+    'ca-central-1',
+    'eu-central-1',
+    'eu-west-1',
+    'eu-west-2',
+    'eu-south-1',
+    'eu-west-3',
+    'eu-north-1',
+    'me-south-1',
+    'sa-east-1',
+    'us-gov-east-1',
+    'us-gov-west-1',
 ];
 exports.DEFAULT_LAYER_AWS_ACCOUNT = '464622532012';
 exports.GOVCLOUD_LAYER_AWS_ACCOUNT = '002406178527';
@@ -63,11 +88,16 @@ exports.SERVICE_ENV_VAR = 'DD_SERVICE';
 exports.VERSION_ENV_VAR = 'DD_VERSION';
 exports.ENVIRONMENT_ENV_VAR = 'DD_ENV';
 exports.EXTRA_TAGS_ENV_VAR = 'DD_TAGS';
+exports.CAPTURE_LAMBDA_PAYLOAD_ENV_VAR = 'DD_CAPTURE_LAMBDA_PAYLOAD';
 // Environment variables used by Datadog CI
 exports.CI_SITE_ENV_VAR = 'DATADOG_SITE';
 exports.CI_API_KEY_ENV_VAR = 'DATADOG_API_KEY';
 exports.CI_API_KEY_SECRET_ARN_ENV_VAR = 'DATADOG_API_KEY_SECRET_ARN';
 exports.CI_KMS_API_KEY_ENV_VAR = 'DATADOG_KMS_API_KEY';
+exports.AWS_ACCESS_KEY_ID_ENV_VAR = 'AWS_ACCESS_KEY_ID';
+exports.AWS_SECRET_ACCESS_KEY_ENV_VAR = 'AWS_SECRET_ACCESS_KEY';
+exports.AWS_DEFAULT_REGION_ENV_VAR = 'AWS_DEFAULT_REGION';
+exports.AWS_SESSION_TOKEN_ENV_VAR = 'AWS_SESSION_TOKEN';
 exports.LIST_FUNCTIONS_MAX_RETRY_COUNT = 2;
 exports.MAX_LAMBDA_STATE_CHECK_ATTEMPTS = 3;
 // DD_TAGS Regular Expression
@@ -75,3 +105,7 @@ exports.MAX_LAMBDA_STATE_CHECK_ATTEMPTS = 3;
 // matches a list of <key>:<value> separated by commas
 // such as: layer:api,team:intake
 exports.EXTRA_TAGS_REG_EXP = /^(([a-zA-Z]+)\w+:[\w\-/\.]+)+((\,)([a-zA-Z]+)\w+:[\w\-/\.]+)*$/g;
+exports.AWS_ACCESS_KEY_ID_REG_EXP = /(?<![A-Z0-9])[A-Z0-9]{20}(?![A-Z0-9])/g;
+exports.AWS_SECRET_ACCESS_KEY_REG_EXP = /(?<![A-Za-z0-9/+=])[A-Za-z0-9/+=]{40}(?![A-Za-z0-9/+=])/g;
+exports.DATADOG_API_KEY_REG_EXP = /(?<![a-f0-9])[a-f0-9]{32}(?![a-f0-9])/g;
+exports.DATADOG_APP_KEY_REG_EXP = /(?<![a-f0-9])[a-f0-9]{40}(?![a-f0-9])/g;