-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
142 lines (127 loc) · 3.32 KB
/
utils.go
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
// Package utils contains commonly useful functions from Deis testing.
package utils
import (
"bytes"
"crypto/rand"
"fmt"
"io"
"net"
"os"
"os/exec"
"path/filepath"
"strings"
"syscall"
"time"
)
// BuildTag returns the $BUILD_TAG environment variable or `git rev-parse` output.
func BuildTag() string {
var tag string
tag = os.Getenv("BUILD_TAG")
if tag == "" {
out, err := exec.Command("git", "rev-parse", "--short", "HEAD").Output()
if err != nil {
tag = "NOTAG"
}
tag = "git-" + string(out)
}
return strings.TrimSpace(tag)
}
// Chdir sets the current working directory to the relative path specified.
func Chdir(app string) error {
var wd, _ = os.Getwd()
dir, _ := filepath.Abs(filepath.Join(wd, app))
err := os.Chdir(dir)
fmt.Println(dir)
return err
}
// CreateFile creates an empty file at the specified path.
func CreateFile(path string) error {
fo, err := os.Create(path)
if err != nil {
return err
}
defer fo.Close()
return nil
}
// HostAddress returns the host IP for accessing etcd and Deis services.
func HostAddress() string {
IP := os.Getenv("HOST_IPADDR")
if IP == "" {
IP = "172.17.8.100"
}
return IP
}
// Hostname returns the hostname of the machine running the container, *not* the local machine
// We infer the hostname because we don't necessarily know how to log in.
func Hostname() string {
switch HostAddress() {
case "172.17.8.100": return "deis-1"
case "172.17.8.101": return "deis-2"
case "172.17.8.102": return "deis-3"
case "172.21.12.100": return "docker-registry"
default: return "boot2docker"
}
}
// NewID returns the first part of a random RFC 4122 UUID
// See http://play.golang.org/p/4FkNSiUDMg
func NewID() string {
uuid := make([]byte, 16)
io.ReadFull(rand.Reader, uuid)
// variant bits; see section 4.1.1
uuid[8] = uuid[8]&^0xc0 | 0x80
// version 4 (pseudo-random); see section 4.1.3
uuid[6] = uuid[6]&^0xf0 | 0x40
return fmt.Sprintf("%x", uuid[0:4])
}
// RandomPort returns an unused TCP listen port on the host.
func RandomPort() string {
l, _ := net.Listen("tcp", "127.0.0.1:0") // listen on localhost
defer l.Close()
port := l.Addr()
return strings.Split(port.String(), ":")[1]
}
// Rmdir removes a directory and its contents.
func Rmdir(app string) error {
var wd, _ = os.Getwd()
dir, _ := filepath.Abs(filepath.Join(wd, app))
err := os.RemoveAll(dir)
fmt.Println(dir)
return err
}
// RunCommandWithStdoutStderr execs a command and returns its output.
func RunCommandWithStdoutStderr(cmd *exec.Cmd) (bytes.Buffer, bytes.Buffer, error) {
var stdout, stderr bytes.Buffer
stderrPipe, err := cmd.StderrPipe()
stdoutPipe, err := cmd.StdoutPipe()
cmd.Env = os.Environ()
if err != nil {
fmt.Println("error at io pipes")
}
err = cmd.Start()
if err != nil {
fmt.Println("error at command start")
}
go func() {
io.Copy(&stdout, stdoutPipe)
fmt.Println(stdout.String())
}()
go func() {
io.Copy(&stderr, stderrPipe)
fmt.Println(stderr.String())
}()
time.Sleep(2000 * time.Millisecond)
err = cmd.Wait()
if err != nil {
fmt.Println("error at command wait")
}
return stdout, stderr, err
}
func getExitCode(err error) (int, error) {
exitCode := 0
if exiterr, ok := err.(*exec.ExitError); ok {
if procExit := exiterr.Sys().(syscall.WaitStatus); ok {
return procExit.ExitStatus(), nil
}
}
return exitCode, fmt.Errorf("failed to get exit code")
}