selector.go

package script

import (
	"fmt"
	"go/ast"
	"reflect"
	"strings"
	"unicode"
)

const GoExclusiveMethodSuffix = "Go"

type selector struct {
	x      Expr
	method string
}

// compiles a selector statement like x.sel
func (w *World) compileSelectorStmt(n *ast.SelectorExpr) Expr {
	x := w.compileExpr(n.X)
	t := x.Type()

	if t == nil {
		panic(err(n.Pos(), "void does not have member", n.Sel.Name))
	}

	sel := strings.ToLower(n.Sel.Name)
	N := ""
	for i := 0; i < t.NumMethod(); i++ {
		name := t.Method(i).Name
		if strings.ToLower(name) == sel && unicode.IsUpper(rune(name[0])) && !strings.HasSuffix(name, GoExclusiveMethodSuffix) {
			N = t.Method(i).Name
			break
		}
	}
	if N == "" {
		panic(err(n.Pos(), t, "has no method", n.Sel.Name))
	}
	return &selector{x, N}
}

func (e *selector) Eval() interface{} {
	obj := reflect.ValueOf(e.x.Eval())
	meth := obj.MethodByName(e.method)
	if meth.Kind() == 0 {
		panic(fmt.Sprint(e.x, " has no method ", e.method))
	}
	return meth.Interface()
}

func (e *selector) Type() reflect.Type {
	return reflect.New(e.x.Type()).Elem().MethodByName(e.method).Type()
}

func (e *selector) Child() []Expr {
	return []Expr{e.x}
}

func (e *selector) Fix() Expr {
	return &selector{x: e.x.Fix(), method: e.method}
}