cmd_dotenv.go

package main

import (
	"fmt"
	"io/ioutil"
	"regexp"
	"strings"
)

var DOTENV_REG = regexp.MustCompile("([A-Za-z0-9_]+)=(.*)")
var DOTENV_LF_REG = regexp.MustCompile("\\\\n")
var DOTENV_ESC_REG = regexp.MustCompile("\\\\.")

// `direnv private dotenv [PATH_TO_DOTENV]`
// Transforms a .env file to evaluatable `export KEY=PAIR` statements.
//
// See: https://github.com/bkeepers/dotenv and
//   https://github.com/ddollar/foreman
var CmdDotEnv = &Cmd{
	Name:    "dotenv",
	Desc:    "Transforms a .env file to evaluatable `export KEY=PAIR` statements",
	Args:    []string{"[PATH_TO_DOTENV]"},
	Private: true,
	Fn: func(env Env, args []string) (err error) {
		var target string
		var dotenv = make(Env)

		if len(args) > 1 {
			target = args[1]
		}

		if target == "" {
			target = ".env"
		}

		var data []byte
		if data, err = ioutil.ReadFile(target); err != nil {
			return
		}

		result := DOTENV_REG.FindAllStringSubmatch(string(data), -1)
		for _, match := range result {
			key := match[1]
			value := strings.TrimSpace(match[2])

			if value[0:1] == "'" && value[len(value)-1:len(value)] == "'" {
				value = value[1 : len(value)-1]
			} else if value[0:1] == `"` && value[len(value)-1:len(value)] == `"` {
				value = value[1 : len(value)-1]
				value = DOTENV_LF_REG.ReplaceAllString(value, "\n")
				value = DOTENV_ESC_REG.ReplaceAllStringFunc(value, func(str string) string {
					return str[1:2]
				})
			}

			dotenv[key] = value
		}

		str := EnvToShell(dotenv, BASH)
		fmt.Println(str)

		return
	},
}