-
-
Notifications
You must be signed in to change notification settings - Fork 49
/
index.js
147 lines (127 loc) · 3.75 KB
/
index.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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
'use strict'
const dotenv = require('dotenv')
const dotenvExpand = require('dotenv-expand')
const chalk = require('chalk')
const fs = require('fs')
const errorTypes = {
HALT: 'HALT',
}
class ServerlessPlugin {
constructor(serverless, options) {
this.serverless = serverless
this.serverless.service.provider.environment =
this.serverless.service.provider.environment || {}
this.config =
this.serverless.service.custom && this.serverless.service.custom['dotenv']
this.logging =
this.config && typeof this.config.logging !== 'undefined'
? this.config.logging
: true
this.required = (this.config && this.config.required) || {}
this.loadEnv(this.getEnvironment(options))
}
log(...args) {
if (this.logging) {
this.serverless.cli.log(...args)
}
}
/**
* @param {Object} options
* @returns {string}
*/
getEnvironment(options) {
return process.env.NODE_ENV || options.env || options.stage || 'development'
}
/**
* @param {string} env
* @returns {string[]}
*/
resolveEnvFileNames(env) {
if (this.config && this.config.path) {
if (Array.isArray(this.config.path)) {
return this.config.path
}
return [this.config.path]
}
// https://github.com/bkeepers/dotenv#what-other-env-files-can-i-use
const dotenvFiles = [
`.env.${env}.local`,
`.env.${env}`,
// Don't include `.env.local` for `test` environment
// since normally you expect tests to produce the same
// results for everyone
env !== 'test' && `.env.local`,
`.env`,
]
const basePath =
this.config && this.config.basePath ? this.config.basePath : ''
const filesNames = dotenvFiles.map((file) => basePath + file)
return filesNames.filter((fileName) => fs.existsSync(fileName))
}
/**
* @param {string} env
*/
loadEnv(env) {
const envFileNames = this.resolveEnvFileNames(env)
try {
const envVarsArray = envFileNames.map(
(fileName) => dotenvExpand(dotenv.config({ path: fileName })).parsed,
)
const envVars = envVarsArray.reduce(
(acc, curr) => ({ ...acc, ...curr }),
{},
)
let include = false
let exclude = false
if (this.config && this.config.include) {
include = this.config.include
}
if (this.config && this.config.exclude && !include) {
// Don't allow both include and exclude to be specified
exclude = this.config.exclude
}
if (envFileNames.length > 0) {
this.log(
'DOTENV: Loading environment variables from ' +
envFileNames.reverse().join(', ') +
':',
)
if (include) {
Object.keys(envVars)
.filter((key) => !include.includes(key))
.forEach((key) => {
delete envVars[key]
})
}
if (exclude) {
Object.keys(envVars)
.filter((key) => exclude.includes(key))
.forEach((key) => {
delete envVars[key]
})
}
Object.keys(envVars).forEach((key) => {
this.log('\t - ' + key)
this.serverless.service.provider.environment[key] = envVars[key]
})
} else {
const errorMsg = 'DOTENV: Could not find .env file.'
this.log(errorMsg)
if (this.required.file === true) {
throw Object.assign(new Error(errorMsg), { type: errorTypes.HALT })
}
}
} catch (e) {
if (e.type === errorTypes.HALT) {
throw e
}
console.error(
chalk.red(
'\n Serverless Plugin Error --------------------------------------\n',
),
)
console.error(chalk.red(' ' + e.message))
}
}
}
module.exports = ServerlessPlugin