113 lines
3.1 KiB
JavaScript
113 lines
3.1 KiB
JavaScript
|
const fs = require('fs')
|
||
|
|
const path = require('path')
|
||
|
|
const os = require('os')
|
||
|
|
|
||
|
|
function log (message) {
|
||
|
|
console.log(`[dotenv][DEBUG] ${message}`)
|
||
|
|
}
|
||
|
|
|
||
|
|
const NEWLINE = '\n'
|
||
|
|
const RE_INI_KEY_VAL = /^\s*([\w.-]+)\s*=\s*("[^"]*"|'[^']*'|[^#]*)?(\s*|\s*#.*)?$/
|
||
|
|
const RE_NEWLINES = /\\n/g
|
||
|
|
const NEWLINES_MATCH = /\r\n|\n|\r/
|
||
|
|
|
||
|
|
// Parses src into an Object
|
||
|
|
function parse (src, options) {
|
||
|
|
const debug = Boolean(options && options.debug)
|
||
|
|
const obj = {}
|
||
|
|
|
||
|
|
// convert Buffers before splitting into lines and processing
|
||
|
|
src.toString().split(NEWLINES_MATCH).forEach(function (line, idx) {
|
||
|
|
// matching "KEY' and 'VAL' in 'KEY=VAL'
|
||
|
|
const keyValueArr = line.match(RE_INI_KEY_VAL)
|
||
|
|
// matched?
|
||
|
|
if (keyValueArr != null) {
|
||
|
|
const key = keyValueArr[1]
|
||
|
|
// default undefined or missing values to empty string
|
||
|
|
let val = (keyValueArr[2] || '')
|
||
|
|
const end = val.length - 1
|
||
|
|
const isDoubleQuoted = val[0] === '"' && val[end] === '"'
|
||
|
|
const isSingleQuoted = val[0] === "'" && val[end] === "'"
|
||
|
|
|
||
|
|
// if single or double quoted, remove quotes
|
||
|
|
if (isSingleQuoted || isDoubleQuoted) {
|
||
|
|
val = val.substring(1, end)
|
||
|
|
|
||
|
|
// if double quoted, expand newlines
|
||
|
|
if (isDoubleQuoted) {
|
||
|
|
val = val.replace(RE_NEWLINES, NEWLINE)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// remove surrounding whitespace
|
||
|
|
val = val.trim()
|
||
|
|
}
|
||
|
|
|
||
|
|
obj[key] = val
|
||
|
|
} else if (debug) {
|
||
|
|
const trimmedLine = line.trim()
|
||
|
|
|
||
|
|
// ignore empty and commented lines
|
||
|
|
if (trimmedLine.length && trimmedLine[0] !== '#') {
|
||
|
|
log(`Failed to match key and value when parsing line ${idx + 1}: ${line}`)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
return obj
|
||
|
|
}
|
||
|
|
|
||
|
|
function resolveHome (envPath) {
|
||
|
|
return envPath[0] === '~' ? path.join(os.homedir(), envPath.slice(1)) : envPath
|
||
|
|
}
|
||
|
|
|
||
|
|
// Populates process.env from .env file
|
||
|
|
function config (options) {
|
||
|
|
let dotenvPath = path.resolve(process.cwd(), '.env')
|
||
|
|
let encoding = 'utf8'
|
||
|
|
const debug = Boolean(options && options.debug)
|
||
|
|
const override = Boolean(options && options.override)
|
||
|
|
|
||
|
|
if (options) {
|
||
|
|
if (options.path != null) {
|
||
|
|
dotenvPath = resolveHome(options.path)
|
||
|
|
}
|
||
|
|
if (options.encoding != null) {
|
||
|
|
encoding = options.encoding
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
// specifying an encoding returns a string instead of a buffer
|
||
|
|
const parsed = parse(fs.readFileSync(dotenvPath, { encoding }), { debug })
|
||
|
|
|
||
|
|
Object.keys(parsed).forEach(function (key) {
|
||
|
|
if (!Object.prototype.hasOwnProperty.call(process.env, key)) {
|
||
|
|
process.env[key] = parsed[key]
|
||
|
|
} else {
|
||
|
|
if (override === true) {
|
||
|
|
process.env[key] = parsed[key]
|
||
|
|
}
|
||
|
|
|
||
|
|
if (debug) {
|
||
|
|
if (override === true) {
|
||
|
|
log(`"${key}" is already defined in \`process.env\` and WAS overwritten`)
|
||
|
|
} else {
|
||
|
|
log(`"${key}" is already defined in \`process.env\` and was NOT overwritten`)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
return { parsed }
|
||
|
|
} catch (e) {
|
||
|
|
if (debug) {
|
||
|
|
log(`Failed to load ${dotenvPath} ${e.message}`)
|
||
|
|
}
|
||
|
|
|
||
|
|
return { error: e }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports.config = config
|
||
|
|
module.exports.parse = parse
|