add osx support

Author: calebdoxsey

darwin_dao.go

@@ -3,24 +3,65 @@
 package penv
 
 import (
+	"fmt"
+	"io/ioutil"
 	"os"
+	"os/exec"
 	"path/filepath"
 	"runtime"
+	"strings"
 )
 
-var darwinPlist = `<?xml version="1.0" encoding="UTF-8"?>
+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>Program</key>
-   <string>` + filepath.Join(os.Getenv("HOME"), ".config", "penv.sh") + `</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,
+			)
+		},
+	}
+)
+
 // DarwinDAO is the data access object for OSX
 type DarwinDAO struct {
 }
@@ -33,10 +74,30 @@ func init() {
 
 // Load loads the environment
 func (dao *DarwinDAO) Load() (*Environment, error) {
-	panic("NOT IMPLEMENTED")
+	return darwinShell.Load()
 }
 
 // Save saves the environment
 func (dao *DarwinDAO) Save(env *Environment) error {
-	panic("NOT IMPLEMENTED")
+	err := darwinShell.Save(env)
+	if err != nil {
+		return err
+	}
+
+	plistName := filepath.Join(os.Getenv("HOME"), "Library", "LaunchAgents", "penv.plist")
+
+	_, err = ioutil.WriteFile(
+		plistName,
+		darwinPlist,
+	)
+	if err != nil {
+		return err
+	}
+
+	err = exec.Command("launchctl", "load", plistName).Run()
+	if err != nil {
+		return err
+	}
+
+	return nil
 }