-
Notifications
You must be signed in to change notification settings - Fork 1
/
credentials.js
56 lines (51 loc) · 1.61 KB
/
credentials.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
const { IAM, STS, CloudFormation } = require('aws-sdk');
const prompts = require('prompts');
async function getMFADevice(region) {
const iam = new IAM({
region
});
const mfaDeviceResponse = await iam.listMFADevices().promise();
return mfaDeviceResponse.MFADevices[0].SerialNumber;
}
async function getSessionToken(region, mfaDevice, code) {
const sessionTokenParams = {
DurationSeconds: 28800, // 8hours
SerialNumber: mfaDevice,
TokenCode: code
};
const sts = new STS({
region
})
const sessionTokenResponse = await sts.getSessionToken(sessionTokenParams).promise();
return sessionTokenResponse.Credentials;
}
async function getSessionnCredentials(region, mfaEnabled) {
const mfaDevice = await getMFADevice(region);
if (mfaDevice && mfaEnabled) {
var mfaDeviceName = mfaDevice.split(/\//).pop();
const tokenResponse = await prompts({
type: 'text',
name: 'token',
message: `Enter code from ${mfaDeviceName}`,
});
return await getSessionToken(region, mfaDevice, tokenResponse.token);
} else {
return {};
}
}
function getEnvWithSessionCredentials(credentials) {
if (credentials && credentials.SessionToken) {
return {
AWS_ACCESS_KEY_ID: credentials.AccessKeyId,
AWS_SECRET_ACCESS_KEY: credentials.SecretAccessKey,
AWS_SESSION_TOKEN: credentials.SessionToken,
...process.env,
}
} else {
return process.env
}
}
module.exports = {
getSessionnCredentials,
getEnvWithSessionCredentials,
};