aboutsummaryrefslogtreecommitdiffstats
path: root/client/go/internal/admin/vespa-wrapper/startcbinary/progspec.go
blob: b0dcc402893ac774e5f1171895ca04e7f70bf614 (plain) (blame)
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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// Author: arnej

package startcbinary

import (
	"os"
	"strings"

	"github.com/vespa-engine/vespa/client/go/internal/admin/envvars"
	"github.com/vespa-engine/vespa/client/go/internal/admin/prog"
)

func NewProgSpec(argv []string) *prog.Spec {
	progName := argv[0]
	binProg := progName + "-bin"
	p := prog.NewSpec(argv)
	p.Program = binProg
	p.Args[0] = binProg
	return p
}

func prependPath(dirName string, p *prog.Spec) {
	pathList := []string{dirName}
	oldPath := p.Getenv(envvars.PATH)
	if oldPath == "" {
		oldPath = "/usr/bin"
	}
	for _, part := range strings.Split(oldPath, ":") {
		if part != dirName {
			pathList = append(pathList, part)
		}
	}
	newPath := strings.Join(pathList, ":")
	p.Setenv(envvars.PATH, newPath)
	os.Setenv(envvars.PATH, newPath)
}