dotenv.go

package dotenv

import (
	"bufio"
	"io"
	"os"
)

// Read reads the .env file and returns the values as a map.
func Read(path string) (map[string]string, error) {
	result := make(map[string]string)
	file, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	reader := bufio.NewReader(file)
	for {
		key, err := reader.ReadBytes('=')
		if err != nil {
			break
		}
		value, err := reader.ReadBytes('\n')
		if err != nil {
			if err == io.EOF {
				result[string(key[:len(key)-1])] = string(value)
			}
			break
		}
		result[string(key[:len(key)-1])] = string(value[:len(value)-1])
	}
	return result, nil
}

// Apply reads the .env file and sets the values in the environment.
func Apply(path string) error {
	file, err := os.Open(path)
	if err != nil {
		return err
	}
	reader := bufio.NewReader(file)
	for {
		key, err := reader.ReadBytes('=')
		if err != nil {
			break
		}
		value, err := reader.ReadBytes('\n')
		if err != nil {
			if err == io.EOF {
				os.Setenv(string(key[:len(key)-1]), string(value))
			}
			break
		}
		os.Setenv(string(key[:len(key)-1]), string(value[:len(value)-1]))
	}
	return nil
}