Allow env variables to be set via 'daemon env' command

Finally an easy way to use it with an SSH_AGENT

Author: vHanda

common/config/v1.go

@@ -11,6 +11,7 @@ import (
 
 type ConfigV1 struct {
 	Repos []string `json:"repos"`
+	Envs  []string `json:"envs"`
 }
 
 func ReadV1() (*ConfigV1, error) {

common/config/v1_test.go

@@ -21,7 +21,10 @@ func setup(t *testing.T, name string) {
 func Test_SimpleWriteReadV1(t *testing.T) {
 	setup(t, "SimpleWriteRead")
 
-	c := &ConfigV1{Repos: []string{"/home/xyz/hello"}}
+	c := &ConfigV1{
+		Repos: []string{"/home/xyz/hello"},
+		Envs:  []string{"SSH_AUTH_SOCK=/private/tmp/com.apple.launchd.74ZznY1v1F/Listeners"},
+	}
 	err := WriteV1(c)
 	assert.NilError(t, err)
 

daemon.go

@@ -3,8 +3,10 @@ package main
 import (
 	"errors"
 	"fmt"
+	"log"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/GitJournal/git-auto-sync/common"
 	cfg "github.com/GitJournal/git-auto-sync/common/config"
@@ -183,6 +185,62 @@ func daemonRm(ctx *cli.Context) error {
 	return nil
 }
 
+func daemonEnv(ctx *cli.Context) error {
+	vars := ctx.Args().Slice()
+
+	for _, v := range vars {
+		if !strings.Contains(v, "=") {
+			log.Fatalln("Env variables must be in the format 'key=value'")
+		}
+	}
+
+	config, err := cfg.Read()
+	if err != nil {
+		return tracerr.Wrap(err)
+	}
+
+	envMap := toEnvMap(config.Envs)
+	newMap := toEnvMap(vars)
+
+	for k, v := range newMap {
+		envMap[k] = v
+	}
+
+	config.Envs = toEnvStrings(envMap)
+	err = cfg.Write(config)
+	if err != nil {
+		return tracerr.Wrap(err)
+	}
+
+	fmt.Println(strings.Join(config.Envs, "\n"))
+
+	return nil
+}
+
 func remove(slice []string, s int) []string {
 	return append(slice[:s], slice[s+1:]...)
 }
+
+func toEnvMap(envs []string) map[string]string {
+	m := map[string]string{}
+	for _, e := range envs {
+		parts := strings.Split(e, "=")
+		if len(parts) > 1 {
+			m[parts[0]] = strings.Join(parts[1:], "=")
+		} else {
+			m[e] = ""
+		}
+	}
+
+	return m
+}
+
+func toEnvStrings(m map[string]string) []string {
+	vals := []string{}
+	for k, v := range m {
+		x := fmt.Sprintf("%s=%s", k, v)
+		vals = append(vals, x)
+	}
+
+	return vals
+}

main.go

@@ -141,13 +141,17 @@ func main() {
 						Usage:  "Add a repo for auto-sync",
 						Action: daemonAdd,
 					},
-
 					{
 						Name:    "remove",
 						Aliases: []string{"rm"},
 						Usage:   "Remove a repo from auto-sync",
 						Action:  daemonRm,
 					},
+					{
+						Name:   "env",
+						Usage:  "Set an environment variable",
+						Action: daemonEnv,
+					},
 				},
 			},
 		},