46 lines
745 B
Go
46 lines
745 B
Go
package main
|
|
|
|
type Callable interface {
|
|
arity() int
|
|
call(i *Interpreter, args ...any) (ret any)
|
|
}
|
|
|
|
type Function struct {
|
|
name Token
|
|
args []Token
|
|
body *BlockStmt
|
|
closure *Environment
|
|
}
|
|
|
|
func (f *Function) call(i *Interpreter, args ...any) (ret any) {
|
|
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
re, ok := err.(Return)
|
|
|
|
if !ok {
|
|
panic(err)
|
|
}
|
|
|
|
ret = re.val
|
|
}
|
|
}()
|
|
|
|
env := newEnvironment(f.closure)
|
|
|
|
for idx, arg := range f.args {
|
|
env.define(arg.lexeme, args[idx])
|
|
}
|
|
|
|
i.executeBlock(f.body, env)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (f *Function) arity() int {
|
|
return len(f.args)
|
|
}
|
|
|
|
func newFunction(name Token, args []Token, body *BlockStmt, env *Environment) Callable {
|
|
return &Function{name, args, body, env}
|
|
}
|