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
143
144
145
146
147
148
149
150
151
152
153
154
|
// Copyright 2017 The Bazel Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// The starlark command interprets a Starlark file.
// With no arguments, it starts a read-eval-print loop (REPL).
package main // import "go.starlark.net/cmd/starlark"
import (
"flag"
"fmt"
"log"
"os"
"runtime"
"runtime/pprof"
"strings"
"go.starlark.net/internal/compile"
"go.starlark.net/lib/json"
"go.starlark.net/lib/math"
"go.starlark.net/lib/time"
"go.starlark.net/repl"
"go.starlark.net/resolve"
"go.starlark.net/starlark"
"golang.org/x/term"
)
// flags
var (
cpuprofile = flag.String("cpuprofile", "", "gather Go CPU profile in this file")
memprofile = flag.String("memprofile", "", "gather Go memory profile in this file")
profile = flag.String("profile", "", "gather Starlark time profile in this file")
showenv = flag.Bool("showenv", false, "on success, print final global environment")
execprog = flag.String("c", "", "execute program `prog`")
)
func init() {
flag.BoolVar(&compile.Disassemble, "disassemble", compile.Disassemble, "show disassembly during compilation of each function")
// non-standard dialect flags
flag.BoolVar(&resolve.AllowSet, "set", resolve.AllowSet, "allow set data type")
flag.BoolVar(&resolve.AllowRecursion, "recursion", resolve.AllowRecursion, "allow while statements and recursive functions")
flag.BoolVar(&resolve.AllowGlobalReassign, "globalreassign", resolve.AllowGlobalReassign, "allow reassignment of globals, and if/for/while statements at top level")
// flags that are now standard
flag.BoolVar(&resolve.AllowFloat, "float", resolve.AllowFloat, "obsolete; no effect")
flag.BoolVar(&resolve.AllowLambda, "lambda", resolve.AllowLambda, "obsolete; no effect")
}
func main() {
os.Exit(doMain())
}
func doMain() int {
log.SetPrefix("starlark: ")
log.SetFlags(0)
flag.Parse()
if *cpuprofile != "" {
f, err := os.Create(*cpuprofile)
check(err)
err = pprof.StartCPUProfile(f)
check(err)
defer func() {
pprof.StopCPUProfile()
err := f.Close()
check(err)
}()
}
if *memprofile != "" {
f, err := os.Create(*memprofile)
check(err)
defer func() {
runtime.GC()
err := pprof.Lookup("heap").WriteTo(f, 0)
check(err)
err = f.Close()
check(err)
}()
}
if *profile != "" {
f, err := os.Create(*profile)
check(err)
err = starlark.StartProfile(f)
check(err)
defer func() {
err := starlark.StopProfile()
check(err)
}()
}
thread := &starlark.Thread{Load: repl.MakeLoad()}
globals := make(starlark.StringDict)
// Ideally this statement would update the predeclared environment.
// TODO(adonovan): plumb predeclared env through to the REPL.
starlark.Universe["json"] = json.Module
starlark.Universe["time"] = time.Module
starlark.Universe["math"] = math.Module
switch {
case flag.NArg() == 1 || *execprog != "":
var (
filename string
src interface{}
err error
)
if *execprog != "" {
// Execute provided program.
filename = "cmdline"
src = *execprog
} else {
// Execute specified file.
filename = flag.Arg(0)
}
thread.Name = "exec " + filename
globals, err = starlark.ExecFile(thread, filename, src, nil)
if err != nil {
repl.PrintError(err)
return 1
}
case flag.NArg() == 0:
stdinIsTerminal := term.IsTerminal(int(os.Stdin.Fd()))
if stdinIsTerminal {
fmt.Println("Welcome to Starlark (go.starlark.net)")
}
thread.Name = "REPL"
repl.REPL(thread, globals)
if stdinIsTerminal {
fmt.Println()
}
default:
log.Print("want at most one Starlark file name")
return 1
}
// Print the global environment.
if *showenv {
for _, name := range globals.Keys() {
if !strings.HasPrefix(name, "_") {
fmt.Fprintf(os.Stderr, "%s = %s\n", name, globals[name])
}
}
}
return 0
}
func check(err error) {
if err != nil {
log.Fatal(err)
}
}
|
