darwin_dao.go

//+build darwin

package penv

import (
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"runtime"
)

var (
	darwinPlist = `<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
   <key>Label</key>
   <string>com.github.badgerodon.penv</string>
   <key>ProgramArguments</key>
   <array>
     <string>bash</string>
     <string>` + filepath.Join(os.Getenv("HOME"), ".config", "penv.sh") + `</string>
   </array>
   <key>RunAtLoad</key>
   <true/>
</dict>
</plist>`

	darwinShell = &shell{
		configFileName: filepath.Join(os.Getenv("HOME"), ".config", "penv.sh"),
		commentSigil:   " #",
		quote: func(value string) string {
			r := strings.NewReplacer(
				"\\", "\\\\",
				"'", "\\'",
				"\n", `'"\n"'`,
				"\r", `'"\r"'`,
			)
			return "'" + r.Replace(value) + "'"
		},
		mkSet: func(sh *shell, nv NameValue) string {
			return fmt.Sprintf(
				"launchctl setenv %s %s",
				nv.Name, sh.quote(nv.Value),
			)
		},
		mkAppend: func(sh *shell, nv NameValue) string {
			return fmt.Sprintf(
				"launchctl setenv %s ${%s}${%s:+:}%s",
				nv.Name, nv.Name, nv.Name, sh.quote(nv.Value),
			)
		},
		mkUnset: func(sh *shell, nv NameValue) string {
			return fmt.Sprintf(
				"launchctl unsetenv %s",
				nv.Name,
			)
		},
	}
)

// DarwinPlistDAO is the data access object for OSX
type DarwinPlistDAO struct{}

// Load loads the environment
func (dao *DarwinPlistDAO) Load() (*Environment, error) {
	return darwinShell.Load()
}

// Save saves the environment
func (dao *DarwinPlistDAO) Save(env *Environment) error {
	err := darwinShell.Save(env)
	if err != nil {
		return err
	}

	plistDir := filepath.Join(os.Getenv("HOME"), "Library", "LaunchAgents")

	err = os.MkdirAll(plistDir, os.ModePerm)
	if err != nil {
		return err
	}

	plistName := filepath.Join(plistDir, "penv.plist")

	err = ioutil.WriteFile(plistName, []byte(darwinPlist), 0777)
	if err != nil {
		return err
	}

	exec.Command("launchctl", "unload", plistName).Run()
	err = exec.Command("launchctl", "load", plistName).Run()
	if err != nil {
		return err
	}

	return nil
}

var (
	// DarwinPlistDAOInstance env with plist
	DarwinPlistDAOInstance = &DarwinPlistDAO{}
)

func init() {
	if runtime.GOOS == "darwin" {
		RegisterDAO(ProfileDAOInstance)
		RegisterDAO(DarwinPlistDAOInstance)
	}
}