elvish/eval/builtin_special.go

211 lines
5.4 KiB
Go
Raw Normal View History

package eval
// Builtin special forms.
import (
"os"
2016-01-25 01:10:54 +08:00
"github.com/elves/elvish/parse"
)
2016-02-08 06:23:16 +08:00
type compileBuiltin func(*compiler, *parse.Form) Op
2016-01-23 01:05:15 +08:00
var builtinSpecials map[string]compileBuiltin
2016-02-08 06:23:16 +08:00
// BuiltinSpecialNames contains all names of builtin special forms. It is
// useful for the syntax highlighter.
2015-07-06 23:58:51 +08:00
var BuiltinSpecialNames []string
func init() {
// Needed to avoid initialization loop
2016-01-23 01:05:15 +08:00
builtinSpecials = map[string]compileBuiltin{
"del": compileDel,
2016-01-23 08:24:47 +08:00
"fn": compileFn,
//"use": compileUse,
}
2016-02-08 06:23:16 +08:00
for k := range builtinSpecials {
2015-07-06 23:58:51 +08:00
BuiltinSpecialNames = append(BuiltinSpecialNames, k)
}
}
2016-02-12 07:30:36 +08:00
func doSet(ec *EvalCtx, variables []Variable, values []Value) {
// TODO Support assignment of mismatched arity in some restricted way -
// "optional" and "rest" arguments and the like
if len(variables) != len(values) {
2016-02-08 06:23:16 +08:00
throw(ErrArityMismatch)
}
for i, variable := range variables {
// TODO Prevent overriding builtin variables e.g. $pid $env
variable.Set(values[i])
}
}
2015-01-22 00:22:55 +08:00
// DelForm = 'del' { VariablePrimary }
2016-02-08 06:23:16 +08:00
func compileDel(cp *compiler, fn *parse.Form) Op {
// Do conventional compiling of all compound expressions, including
// ensuring that variables can be resolved
var names, envNames []string
for _, cn := range fn.Args {
qname := mustString(cp, cn, "should be a literal variable name")
splice, ns, name := parseVariable(qname)
if splice {
cp.errorf(cn.Begin(), "removing spliced variable makes no sense")
}
switch ns {
case "", "local":
if !cp.thisScope()[name] {
2016-01-25 02:04:15 +08:00
cp.errorf(cn.Begin(), "variable $%s not found on current local scope", name)
}
delete(cp.thisScope(), name)
names = append(names, name)
case "env":
envNames = append(envNames, name)
default:
2016-01-25 02:04:15 +08:00
cp.errorf(cn.Begin(), "can only delete a variable in local: or env:")
}
2015-01-22 00:22:55 +08:00
}
2016-02-12 07:30:36 +08:00
return func(ec *EvalCtx) {
2014-09-26 05:26:56 +08:00
for _, name := range names {
delete(ec.local, name)
}
for _, name := range envNames {
2015-02-26 22:17:05 +08:00
// BUG(xiaq): We rely on the fact that os.Unsetenv always returns
// nil.
os.Unsetenv(name)
}
}
}
/*
func stem(fname string) string {
base := path.Base(fname)
ext := path.Ext(base)
return base[0 : len(base)-len(ext)]
}
// UseForm = 'use' StringPrimary.modname Primary.fname
// = 'use' StringPrimary.fname
func compileUse(cp *compiler, fn *parse.Form) op {
var fnameNode *parse.Compound
var fname, modname string
switch len(fn.Args.Nodes) {
case 0:
2016-01-23 01:05:15 +08:00
cp.errorf(fn.Args.Pos, "expect module name or file name")
case 1, 2:
fnameNode = fn.Args.Nodes[0]
2016-01-23 01:05:15 +08:00
_, fname = ensureStringPrimary(cp, fnameNode, "expect string literal")
if len(fn.Args.Nodes) == 2 {
modnameNode := fn.Args.Nodes[1]
_, modname = ensureStringPrimary(
2016-01-23 01:05:15 +08:00
cp, modnameNode, "expect string literal")
if modname == "" {
2016-01-23 01:05:15 +08:00
cp.errorf(modnameNode.Pos, "module name is empty")
}
} else {
modname = stem(fname)
if modname == "" {
2016-01-23 01:05:15 +08:00
cp.errorf(fnameNode.Pos, "stem of file name is empty")
}
}
default:
2016-01-23 01:05:15 +08:00
cp.errorf(fn.Args.Nodes[2].Pos, "superfluous argument")
}
switch {
case strings.HasPrefix(fname, "/"):
// Absolute file name, do nothing
case strings.HasPrefix(fname, "./") || strings.HasPrefix(fname, "../"):
// File name relative to current source
2016-01-23 01:05:15 +08:00
fname = path.Clean(path.Join(cp.dir, fname))
default:
// File name relative to data dir
2016-01-23 01:05:15 +08:00
fname = path.Clean(path.Join(cp.dataDir, fname))
}
src, err := readFileUTF8(fname)
if err != nil {
2016-01-23 01:05:15 +08:00
cp.errorf(fnameNode.Pos, "cannot read module: %s", err.Error())
}
cn, err := parse.Parse(fname, src)
if err != nil {
// TODO(xiaq): Pretty print
2016-01-23 01:05:15 +08:00
cp.errorf(fnameNode.Pos, "cannot parse module: %s", err.Error())
}
2016-01-23 01:05:15 +08:00
newCc := &compiler{
cp.Compiler,
fname, src, path.Dir(fname),
[]staticNS{staticNS{}}, staticNS{},
}
op, err := newCc.compile(cn)
if err != nil {
// TODO(xiaq): Pretty print
2016-01-23 01:05:15 +08:00
cp.errorf(fnameNode.Pos, "cannot compile module: %s", err.Error())
}
2016-01-23 01:05:15 +08:00
cp.mod[modname] = newCc.scopes[0]
return func(ec *evalCtx) exitus {
// TODO(xiaq): Should handle failures when evaluting the module
newEc := &evalCtx{
ec.Evaler,
fname, src, "module " + modname,
ns{}, ns{},
ec.ports,
}
op.f(newEc)
ec.mod[modname] = newEc.local
2015-08-23 22:46:33 +08:00
return ok
}
}
2016-01-23 08:24:47 +08:00
*/
// makeFnOp wraps an op such that a return is converted to an ok.
2016-02-08 06:23:16 +08:00
func makeFnOp(op Op) Op {
2016-02-12 07:30:36 +08:00
return func(ec *EvalCtx) {
2016-02-09 03:27:05 +08:00
ex := ec.PEval(op)
if ex != Return {
2016-01-31 09:11:10 +08:00
// rethrow
throw(ex)
2015-07-08 18:31:40 +08:00
}
}
}
// FnForm = 'fn' StringPrimary LambdaPrimary
//
2016-01-28 05:24:17 +08:00
// fn f []{foobar} is a shorthand for set '&'f = []{foobar}.
2016-02-08 06:23:16 +08:00
func compileFn(cp *compiler, fn *parse.Form) Op {
2016-01-23 08:24:47 +08:00
if len(fn.Args) == 0 {
2016-01-25 02:04:15 +08:00
cp.errorf(fn.End(), "should be followed by function name")
}
2016-01-23 08:24:47 +08:00
fnName := mustString(cp, fn.Args[0], "must be a literal string")
varName := FnPrefix + fnName
2016-01-23 08:24:47 +08:00
if len(fn.Args) == 1 {
2016-01-25 02:04:15 +08:00
cp.errorf(fn.Args[0].End(), "should be followed by a lambda")
}
2016-01-23 08:24:47 +08:00
pn := mustPrimary(cp, fn.Args[1], "should be a lambda")
if pn.Type != parse.Lambda {
2016-01-25 02:04:15 +08:00
cp.errorf(pn.Begin(), "should be a lambda")
2016-01-23 08:24:47 +08:00
}
if len(fn.Args) > 2 {
2016-01-25 02:04:15 +08:00
cp.errorf(fn.Args[2].Begin(), "superfluous argument")
}
2016-01-26 06:45:51 +08:00
cp.registerVariableSet(":" + varName)
2016-01-23 08:24:47 +08:00
op := cp.lambda(pn)
2016-02-12 07:30:36 +08:00
return func(ec *EvalCtx) {
// Initialize the function variable with the builtin nop
// function. This step allows the definition of recursive
// functions; the actual function will never be called.
ec.local[varName] = NewPtrVariable(&BuiltinFn{"<shouldn't be called>", nop})
2016-01-29 20:55:14 +08:00
closure := op(ec)[0].(*Closure)
2015-07-08 18:31:40 +08:00
closure.Op = makeFnOp(closure.Op)
ec.local[varName].Set(closure)
}
}