summaryrefslogtreecommitdiff
path: root/cmds/env.go
blob: 5f09df6dffb88f7f99377bfc9442dc4feb0efd3d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package cmds

import (
	"errors"
	"fmt"
	"os"
	"strings"

	"gosh/global"
)

// printEnv: prints all environment variables with shell options.
func printEnv() {
	vars := os.Environ()
	for i := range vars {
		fmt.Println(vars[i])
	}
	if len(global.Options) > 0 {
		fmt.Printf("SH_OPTS=%v\n", global.Options)
	}
}

// set: sets the shell options
func set(args []string) {
	for i := 1; i < len(args); i++ {
		if i-1 >= len(global.Options) {
			global.Options = append(global.Options, args[i])
			continue
		}
		global.Options[i-1] = args[i]
	}

	if len(args)-1 < len(global.Options) {
		global.Options = global.Options[:len(args)-1]
	}
}

// unset: unsets an environment variable
func unset(args []string) error {
	if len(args) == 1 || len(args) >= 3 {
		return errors.New("usage: unset [name]")
	}
	os.Unsetenv(args[1])
	return nil
}

// export: exports a key-value pair to the environment
// in the form of `name=value`
func export(args []string) error {
	if len(args) == 1 || len(args) >= 3 {
		return errors.New("usage: export [name=value]")
	}
	tmp := strings.Split(args[1], "=")
	if len(tmp) != 2 {
		return errors.New("usage: export [name=value]")
	}
	return os.Setenv(tmp[0], tmp[1])
}