forked from candid82/joker
-
Notifications
You must be signed in to change notification settings - Fork 0
/
os_native.go
87 lines (79 loc) · 1.86 KB
/
os_native.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
package os
import (
"bytes"
"io/ioutil"
"os"
"os/exec"
"strings"
. "github.com/candid82/joker/core"
)
func env() Object {
res := EmptyArrayMap()
for _, v := range os.Environ() {
parts := strings.Split(v, "=")
res.Add(String{S: parts[0]}, String{S: parts[1]})
}
return res
}
func commandArgs() Object {
res := EmptyVector
for _, arg := range os.Args {
res = res.Conjoin(String{S: arg})
}
return res
}
func sh(name string, args []string) Object {
cmd := exec.Command(name, args...)
stdoutReader, err := cmd.StdoutPipe()
PanicOnErr(err)
stderrReader, err := cmd.StderrPipe()
PanicOnErr(err)
if err = cmd.Start(); err != nil {
panic(RT.NewError(err.Error()))
}
buf := new(bytes.Buffer)
buf.ReadFrom(stdoutReader)
stdoutString := buf.String()
buf = new(bytes.Buffer)
buf.ReadFrom(stderrReader)
stderrString := buf.String()
err = cmd.Wait()
res := EmptyArrayMap()
res.Add(MakeKeyword("success"), Bool{B: err == nil})
if err != nil {
res.Add(MakeKeyword("err-msg"), String{S: err.Error()})
}
res.Add(MakeKeyword("out"), String{S: stdoutString})
res.Add(MakeKeyword("err"), String{S: stderrString})
return res
}
func mkdir(name string, perm int) Object {
err := os.Mkdir(name, os.FileMode(perm))
PanicOnErr(err)
return NIL
}
func readDir(dirname string) Object {
files, err := ioutil.ReadDir(dirname)
PanicOnErr(err)
res := EmptyVector
name := MakeKeyword("name")
size := MakeKeyword("size")
mode := MakeKeyword("mode")
isDir := MakeKeyword("dir?")
modTime := MakeKeyword("modtime")
for _, f := range files {
m := EmptyArrayMap()
m.Add(name, MakeString(f.Name()))
m.Add(size, MakeInt(int(f.Size())))
m.Add(mode, MakeInt(int(f.Mode())))
m.Add(isDir, MakeBool(f.IsDir()))
m.Add(modTime, MakeInt(int(f.ModTime().Unix())))
res = res.Conjoin(m)
}
return res
}
func getwd() string {
res, err := os.Getwd()
PanicOnErr(err)
return res
}