mirror of https://github.com/joho/godotenv.git
119 lines
2.5 KiB
Go
119 lines
2.5 KiB
Go
package godotenv
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func Load(filenames ...string) (err error) {
|
|
for _, filename := range filenames {
|
|
err = loadFile(filename)
|
|
if err != nil {
|
|
return // return early on a spazout
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func loadFile(filename string) (err error) {
|
|
file, err := os.Open(filename)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
bufferSize := 20
|
|
lines := make([]string, bufferSize)
|
|
lineReader := bufio.NewReaderSize(file, bufferSize)
|
|
for line, isPrefix, e := lineReader.ReadLine(); e == nil; line, isPrefix, e = lineReader.ReadLine() {
|
|
fullLine := string(line)
|
|
if isPrefix {
|
|
for {
|
|
line, isPrefix, _ = lineReader.ReadLine()
|
|
fullLine += string(line)
|
|
if !isPrefix {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
// add a line to the game/parse
|
|
lines = append(lines, string(line))
|
|
}
|
|
|
|
for _, fullLine := range lines {
|
|
key, value, err := parseLine(fullLine)
|
|
|
|
if err == nil {
|
|
os.Setenv(key, value)
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func parseLine(line string) (key string, value string, err error) {
|
|
if len(line) == 0 {
|
|
err = errors.New("zero length string")
|
|
return
|
|
}
|
|
|
|
splitString := strings.Split(line, "=")
|
|
|
|
if len(splitString) != 2 {
|
|
// try yaml mode!
|
|
splitString = strings.Split(line, ":")
|
|
}
|
|
|
|
if len(splitString) != 2 {
|
|
err = errors.New("Can't separate key from value")
|
|
return
|
|
}
|
|
|
|
key = splitString[0]
|
|
if strings.HasPrefix(key, "export") {
|
|
key = strings.TrimPrefix(key, "export")
|
|
}
|
|
key = strings.Trim(key, " ")
|
|
|
|
value = splitString[1]
|
|
|
|
// ditch the comments (but keep quoted hashes)
|
|
if strings.Contains(value, "#") {
|
|
segmentsBetweenHashes := strings.Split(value, "#")
|
|
quotesAreOpen := false
|
|
segmentsToKeep := make([]string, 0)
|
|
for _, segment := range segmentsBetweenHashes {
|
|
if strings.Count(segment, "\"") == 1 || strings.Count(segment, "'") == 1 {
|
|
if quotesAreOpen {
|
|
quotesAreOpen = false
|
|
segmentsToKeep = append(segmentsToKeep, segment)
|
|
} else {
|
|
quotesAreOpen = true
|
|
}
|
|
}
|
|
|
|
if len(segmentsToKeep) == 0 || quotesAreOpen {
|
|
segmentsToKeep = append(segmentsToKeep, segment)
|
|
}
|
|
}
|
|
|
|
value = strings.Join(segmentsToKeep, "#")
|
|
}
|
|
|
|
// check if we've got quoted values
|
|
if strings.Count(value, "\"") == 2 || strings.Count(value, "'") == 2 {
|
|
// pull the quotes off the edges
|
|
value = strings.Trim(value, "\"' ")
|
|
|
|
// expand quotes
|
|
value = strings.Replace(value, "\\\"", "\"", -1)
|
|
// expand newlines
|
|
value = strings.Replace(value, "\\n", "\n", -1)
|
|
}
|
|
// trim
|
|
value = strings.Trim(value, " ")
|
|
|
|
return
|
|
}
|