moxa/interp/type.go

package interp

import (
	"reflect"
	"strconv"
)

// Cat defines interpreter type categories
type Cat uint

// Types for go language
const (
	NilT Cat = iota
	AliasT
	ArrayT
	BinT
	BinPkgT
	BoolT
	BuiltinT
	ByteT
	ChanT
	Complex64T
	Complex128T
	ErrorT
	Float32T
	Float64T
	FuncT
	InterfaceT
	IntT
	Int8T
	Int16T
	Int32T
	Int64T
	MapT
	PtrT
	RuneT
	SrcPkgT
	StringT
	StructT
	UintT
	Uint8T
	Uint16T
	Uint32T
	Uint64T
	UintptrT
	ValueT
	MaxT
)

var cats = [...]string{
	NilT:        "NilT",
	AliasT:      "AliasT",
	ArrayT:      "ArrayT",
	BinT:        "BinT",
	BinPkgT:     "BinPkgT",
	ByteT:       "ByteT",
	BoolT:       "BoolT",
	BuiltinT:    "BuiltinT",
	ChanT:       "ChanT",
	Complex64T:  "Complex64T",
	Complex128T: "Complex128T",
	ErrorT:      "ErrorT",
	Float32T:    "Float32",
	Float64T:    "Float64T",
	FuncT:       "FuncT",
	InterfaceT:  "InterfaceT",
	IntT:        "IntT",
	Int8T:       "Int8T",
	Int16T:      "Int16T",
	Int32T:      "Int32T",
	Int64T:      "Int64T",
	MapT:        "MapT",
	PtrT:        "PtrT",
	RuneT:       "RuneT",
	SrcPkgT:     "SrcPkgT",
	StringT:     "StringT",
	StructT:     "StructT",
	UintT:       "UintT",
	Uint8T:      "Uint8T",
	Uint16T:     "Uint16T",
	Uint32T:     "Uint32T",
	Uint64T:     "Uint64T",
	UintptrT:    "UintptrT",
	ValueT:      "ValueT",
}

func (c Cat) String() string {
	if c < Cat(len(cats)) {
		return cats[c]
	}
	return "Cat(" + strconv.Itoa(int(c)) + ")"
}

// StructField type defines a field in a struct
type StructField struct {
	name  string
	embed bool
	typ   *Type
}

// Type defines the internal representation of types in the interpreter
type Type struct {
	cat        Cat           // Type category
	field      []StructField // Array of struct fields if StrucT or InterfaceT
	key        *Type         // Type of key element if MapT or nil
	val        *Type         // Type of value element if ChanT, MapT, PtrT, AliasT or ArrayT
	arg        []*Type       // Argument types if FuncT or nil
	ret        []*Type       // Return types if FuncT or nil
	method     []*Node       // Associated methods or nil
	name       string        // name of type within its package for a defined type
	pkgPath    string        // for a defined type, the package import path
	size       int           // Size of array if ArrayT
	rtype      reflect.Type  // Reflection type if ValueT, or nil
	variadic   bool          // true if type is variadic
	incomplete bool          // true if type must be parsed again (out of order declarations)
	untyped    bool          // true for a literal value (string or number)
	node       *Node         // root AST node of type definition
	scope      *Scope        // type declaration scope (in case of re-parse incomplete type)
}

// nodeType returns a type definition for the corresponding AST subtree
func nodeType(interp *Interpreter, scope *Scope, n *Node) (*Type, error) {
	var err CfgError

	if n.typ != nil && !n.typ.incomplete {
		return n.typ, err
	}

	var t = &Type{node: n, scope: scope}

	switch n.kind {
	case Address, StarExpr:
		t.cat = PtrT
		t.val, err = nodeType(interp, scope, n.child[0])
		t.incomplete = t.val.incomplete

	case ArrayType:
		t.cat = ArrayT
		if len(n.child) > 1 {
			if n.child[0].rval.IsValid() {
				t.size = int(n.child[0].rval.Int())
			} else {
				if sym, _, ok := scope.lookup(n.child[0].ident); ok {
					// Resolve symbol to get size value
					if sym.typ != nil && sym.typ.cat == IntT {
						if v, ok := sym.rval.Interface().(int); ok {
							t.size = v
						} else {
							t.incomplete = true
						}
					} else {
						t.incomplete = true
					}
				} else {
					// Evaluate constant array size expression
					_, err = interp.Cfg(n.child[0])
					if err != nil {
						return nil, err
					}
					t.incomplete = true
				}
			}
			t.val, err = nodeType(interp, scope, n.child[1])
			t.incomplete = t.incomplete || t.val.incomplete
		} else {
			t.val, err = nodeType(interp, scope, n.child[0])
			t.incomplete = t.val.incomplete
		}

	case BasicLit:
		switch v := n.rval.Interface().(type) {
		case bool:
			t.cat = BoolT
			t.name = "bool"
		case byte:
			t.cat = ByteT
			t.name = "byte"
			t.untyped = true
		case complex64:
			t.cat = Complex64T
			t.name = "complex64"
		case complex128:
			t.cat = Complex128T
			t.name = "complex128"
			t.untyped = true
		case float32:
			t.cat = Float32T
			t.name = "float32"
			t.untyped = true
		case float64:
			t.cat = Float64T
			t.name = "float64"
			t.untyped = true
		case int:
			if isShiftOperand(n) && v >= 0 {
				t.cat = UintT
				t.name = "uint"
				n.rval = reflect.ValueOf(uint(v))
			} else {
				t.cat = IntT
				t.name = "int"
			}
			t.untyped = true
		case rune:
			t.cat = RuneT
			t.name = "rune"
			t.untyped = true
		case string:
			t.cat = StringT
			t.name = "string"
			t.untyped = true
		default:
			err = n.cfgError("missing support for type %T: %v", v, n.rval)
		}

	case UnaryExpr:
		t, err = nodeType(interp, scope, n.child[0])

	case BinaryExpr:
		if a := n.anc; a.kind == Define && len(a.child) > a.nleft+a.nright {
			t, err = nodeType(interp, scope, a.child[a.nleft])
		} else {
			t, err = nodeType(interp, scope, n.child[0])
			if err != nil {
				return nil, err
			}
			if t.untyped {
				var t1 *Type
				t1, err = nodeType(interp, scope, n.child[1])
				if !(t1.untyped && isInt(t1.TypeOf()) && isFloat(t.TypeOf())) {
					t = t1
				}
			}
		}

	case CallExpr:
		t, err = nodeType(interp, scope, n.child[0])
		if err != nil {
			return nil, err
		}
		switch t.cat {
		case ValueT:
			if t.rtype.NumOut() == 1 {
				t = &Type{cat: ValueT, rtype: t.rtype.Out(0)}
			}
		default:
			if len(t.ret) == 1 {
				t = t.ret[0]
			}
		}

	case CompositeLitExpr:
		t, err = nodeType(interp, scope, n.child[0])

	case ChanType:
		t.cat = ChanT
		t.val, err = nodeType(interp, scope, n.child[0])
		t.incomplete = t.val.incomplete

	case Ellipsis:
		t, err = nodeType(interp, scope, n.child[0])
		t.variadic = true

	case FuncLit:
		t, err = nodeType(interp, scope, n.child[2])

	case FuncType:
		t.cat = FuncT
		// Handle input parameters
		for _, arg := range n.child[0].child {
			cl := len(arg.child) - 1
			typ, err := nodeType(interp, scope, arg.child[cl])
			if err != nil {
				return nil, err
			}
			t.arg = append(t.arg, typ)
			for i := 1; i < cl; i++ {
				// Several arguments may be factorized on the same field type
				t.arg = append(t.arg, typ)
			}
			t.incomplete = t.incomplete || typ.incomplete
		}
		if len(n.child) == 2 {
			// Handle returned values
			for _, ret := range n.child[1].child {
				cl := len(ret.child) - 1
				typ, err := nodeType(interp, scope, ret.child[cl])
				if err != nil {
					return nil, err
				}
				t.ret = append(t.ret, typ)
				for i := 1; i < cl; i++ {
					// Several arguments may be factorized on the same field type
					t.ret = append(t.ret, typ)
				}
				t.incomplete = t.incomplete || typ.incomplete
			}
		}

	case Ident:
		if sym, _, found := scope.lookup(n.ident); found {
			t = sym.typ
		} else {
			t.incomplete = true
			scope.sym[n.ident] = &Symbol{kind: Typ, typ: t}
		}

	case InterfaceType:
		t.cat = InterfaceT
		for _, field := range n.child[0].child {
			if len(field.child) == 1 {
				typ, err := nodeType(interp, scope, field.child[0])
				if err != nil {
					return nil, err
				}
				t.field = append(t.field, StructField{name: fieldName(field.child[0]), embed: true, typ: typ})
				t.incomplete = t.incomplete || typ.incomplete
			} else {
				typ, err := nodeType(interp, scope, field.child[1])
				if err != nil {
					return nil, err
				}
				t.field = append(t.field, StructField{name: field.child[0].ident, typ: typ})
				t.incomplete = t.incomplete || typ.incomplete
			}
		}

	case MapType:
		t.cat = MapT
		if t.key, err = nodeType(interp, scope, n.child[0]); err != nil {
			return nil, err
		}
		if t.val, err = nodeType(interp, scope, n.child[1]); err != nil {
			return nil, err
		}
		t.incomplete = t.key.incomplete || t.val.incomplete

	case ParenExpr:
		t, err = nodeType(interp, scope, n.child[0])

	case SelectorExpr:
		pkg, name := n.child[0].ident, n.child[1].ident
		if sym, _, found := scope.lookup(pkg); found {
			if sym.typ == nil {
				t.incomplete = true
				break
			}
			switch sym.typ.cat {
			case BinPkgT:
				pkg := interp.binValue[sym.path]
				if v, ok := pkg[name]; ok {
					t.cat = ValueT
					t.rtype = v.Type()
					if isBinType(v) {
						t.rtype = t.rtype.Elem()
					}
				} else {
					t.incomplete = true
				}

			case SrcPkgT:
				spkg := interp.scope[pkg]
				if st, ok := spkg.sym[name]; ok {
					t = st.typ
				} else {
					t.incomplete = true
				}
			}
		} else {
			err = n.cfgError("undefined package: %s", pkg)
		}

	case StructType:
		t.cat = StructT
		var incomplete bool
		for _, c := range n.child[0].child {
			if len(c.child) == 1 {
				typ, err := nodeType(interp, scope, c.child[0])
				if err != nil {
					return nil, err
				}
				t.field = append(t.field, StructField{name: fieldName(c.child[0]), embed: true, typ: typ})
				incomplete = incomplete || typ.incomplete
			} else {
				l := len(c.child)
				typ, err := nodeType(interp, scope, c.child[l-1])
				if err != nil {
					return nil, err
				}
				incomplete = incomplete || typ.incomplete
				for _, d := range c.child[:l-1] {
					t.field = append(t.field, StructField{name: d.ident, typ: typ})
				}
			}
		}
		t.incomplete = incomplete

	default:
		err = n.cfgError("type definition not implemented: %s", n.kind)
	}

	return t, err
}

// fieldName returns an implicit struct field name according to node kind
func fieldName(n *Node) string {
	switch n.kind {
	case SelectorExpr:
		return fieldName(n.child[1])
	case StarExpr:
		return fieldName(n.child[0])
	case Ident:
		return n.ident
	default:
		return ""
	}
}

var zeroValues [MaxT]reflect.Value

func init() {
	zeroValues[BoolT] = reflect.ValueOf(false)
	zeroValues[ByteT] = reflect.ValueOf(byte(0))
	zeroValues[Complex64T] = reflect.ValueOf(complex64(0))
	zeroValues[Complex128T] = reflect.ValueOf(complex128(0))
	zeroValues[ErrorT] = reflect.ValueOf(new(error)).Elem()
	zeroValues[Float32T] = reflect.ValueOf(float32(0))
	zeroValues[Float64T] = reflect.ValueOf(float64(0))
	zeroValues[IntT] = reflect.ValueOf(int(0))
	zeroValues[Int8T] = reflect.ValueOf(int8(0))
	zeroValues[Int16T] = reflect.ValueOf(int16(0))
	zeroValues[Int32T] = reflect.ValueOf(int32(0))
	zeroValues[Int64T] = reflect.ValueOf(int64(0))
	zeroValues[RuneT] = reflect.ValueOf(rune(0))
	zeroValues[StringT] = reflect.ValueOf("")
	zeroValues[UintT] = reflect.ValueOf(uint(0))
	zeroValues[Uint8T] = reflect.ValueOf(uint8(0))
	zeroValues[Uint16T] = reflect.ValueOf(uint16(0))
	zeroValues[Uint32T] = reflect.ValueOf(uint32(0))
	zeroValues[Uint64T] = reflect.ValueOf(uint64(0))
	zeroValues[UintptrT] = reflect.ValueOf(uintptr(0))
}

// if type is incomplete, re-parse it.
func (t *Type) finalize() (*Type, error) {
	var err CfgError
	if t.incomplete {
		t, err = nodeType(t.node.interp, t.scope, t.node)
		t.node.typ = t
		if t.incomplete && err == nil {
			err = t.node.cfgError("incomplete type")
		}
	}
	return t, err
}

// id returns a unique type identificator string
func (t *Type) id() string {
	// TODO: if res is nil, build identity from String()

	res := ""
	switch t.cat {
	case ValueT:
		res = t.rtype.PkgPath() + "." + t.rtype.Name()
	case PtrT:
		res = "*" + t.val.id()
	default:
		res = t.pkgPath + "." + t.name
	}
	return res
}

// zero instantiates and return a zero value object for the given type during execution
func (t *Type) zero() (v reflect.Value, err error) {
	if t, err = t.finalize(); err != nil {
		return v, err
	}
	switch t.cat {
	case AliasT:
		v, err = t.val.zero()

	case ArrayT, PtrT, StructT:
		v = reflect.New(t.TypeOf()).Elem()

	case ValueT:
		v = reflect.New(t.rtype).Elem()

	default:
		v = zeroValues[t.cat]
	}
	return v, err
}

// fieldIndex returns the field index from name in a struct, or -1 if not found
func (t *Type) fieldIndex(name string) int {
	if t.cat == PtrT {
		return t.val.fieldIndex(name)
	}
	for i, field := range t.field {
		if name == field.name {
			return i
		}
	}
	return -1
}

// fieldSeq returns the field type from the list of field indexes
func (t *Type) fieldSeq(seq []int) *Type {
	ft := t
	for _, i := range seq {
		if ft.cat == PtrT {
			ft = ft.val
		}
		ft = ft.field[i].typ
	}
	return ft
}

// lookupField returns a list of indices, i.e. a path to access a field in a struct object
func (t *Type) lookupField(name string) []int {
	if fi := t.fieldIndex(name); fi >= 0 {
		return []int{fi}
	}

	for i, f := range t.field {
		switch f.typ.cat {
		case PtrT, StructT:
			if index2 := f.typ.lookupField(name); len(index2) > 0 {
				return append([]int{i}, index2...)
			}
		}
	}

	return nil
}

// lookupBinField returns a structfield and a path to access an embedded binary field in a struct object
func (t *Type) lookupBinField(name string) (reflect.StructField, []int, bool) {
	if t.cat == PtrT {
		return t.val.lookupBinField(name)
	}
	var index []int
	s, ok := t.TypeOf().FieldByName(name)
	if !ok {
		for i, f := range t.field {
			if f.embed {
				if s2, index2, ok2 := f.typ.lookupBinField(name); ok2 {
					index = append([]int{i}, index2...)
					return s2, index, ok2
				}
			}
		}
	}
	return s, index, ok
}

// getMethod returns a pointer to the method definition
func (t *Type) getMethod(name string) *Node {
	for _, m := range t.method {
		if name == m.ident {
			return m
		}
	}
	return nil
}

// lookupMethod returns a pointer to method definition associated to type t
// and the list of indices to access the right struct field, in case of an embedded method
func (t *Type) lookupMethod(name string) (*Node, []int) {
	if t.cat == PtrT {
		return t.val.lookupMethod(name)
	}
	var index []int
	m := t.getMethod(name)
	if m == nil {
		for i, f := range t.field {
			if f.embed {
				if n, index2 := f.typ.lookupMethod(name); n != nil {
					index = append([]int{i}, index2...)
					return n, index
				}
			}
		}
	}
	return m, index
}

// lookupBinMethod returns a method and a path to access a field in a struct object (the receiver)
func (t *Type) lookupBinMethod(name string) (reflect.Method, []int, bool) {
	if t.cat == PtrT {
		return t.val.lookupBinMethod(name)
	}
	var index []int
	m, ok := t.TypeOf().MethodByName(name)
	if !ok {
		for i, f := range t.field {
			if f.embed {
				if m2, index2, ok2 := f.typ.lookupBinMethod(name); ok2 {
					index = append([]int{i}, index2...)
					return m2, index, ok2
				}
			}
		}
	}
	return m, index, ok
}

func exportName(s string) string {
	if canExport(s) {
		return s
	}
	return "X" + s
}

// TypeOf returns the reflection type of dynamic interpreter type t.
func (t *Type) TypeOf() reflect.Type {
	if t.rtype != nil {
		return t.rtype
	}

	if t.incomplete {
		t, _ = t.finalize()
	}

	switch t.cat {
	case ArrayT:
		if t.size > 0 {
			t.rtype = reflect.ArrayOf(t.size, t.val.TypeOf())
		} else {
			t.rtype = reflect.SliceOf(t.val.TypeOf())
		}
	case ChanT:
		t.rtype = reflect.ChanOf(reflect.BothDir, t.val.TypeOf())
	case ErrorT:
		t.rtype = reflect.TypeOf(new(error)).Elem()
	case FuncT:
		in := make([]reflect.Type, len(t.arg))
		out := make([]reflect.Type, len(t.ret))
		for i, v := range t.arg {
			in[i] = v.TypeOf()
		}
		for i, v := range t.ret {
			out[i] = v.TypeOf()
		}
		t.rtype = reflect.FuncOf(in, out, false)
	case InterfaceT:
		t.rtype = reflect.TypeOf(new(interface{})).Elem()
	case MapT:
		t.rtype = reflect.MapOf(t.key.TypeOf(), t.val.TypeOf())
	case PtrT:
		t.rtype = reflect.PtrTo(t.val.TypeOf())
	case StructT:
		var fields []reflect.StructField
		for _, f := range t.field {
			field := reflect.StructField{Name: exportName(f.name), Type: f.typ.TypeOf()}
			fields = append(fields, field)
		}
		t.rtype = reflect.StructOf(fields)
	default:
		if z, _ := t.zero(); z.IsValid() {
			t.rtype = z.Type()
		}
	}
	return t.rtype
}

func (t *Type) frameType() reflect.Type {
	var r reflect.Type
	switch t.cat {
	case ArrayT:
		if t.size > 0 {
			r = reflect.ArrayOf(t.size, t.val.frameType())
		} else {
			r = reflect.SliceOf(t.val.frameType())
		}
	//case ChanT:
	//	r = reflect.ChanOf(reflect.BothDir, t.val.frameType())
	//case ErrorT:
	//	r = reflect.TypeOf(new(error)).Elem()
	case FuncT:
		r = reflect.TypeOf((*Node)(nil))
	case InterfaceT:
		r = reflect.TypeOf((*valueInterface)(nil)).Elem()
	//case MapT:
	//	r = reflect.MapOf(t.key.frameType(), t.val.frameType())
	//case PtrT:
	//	r = reflect.PtrTo(t.val.frameType())
	//case StructT:
	//	var fields []reflect.StructField
	//	for _, f := range t.field {
	//		field := reflect.StructField{Name: exportName(f.name), Type: f.typ.frameType()}
	//		fields = append(fields, field)
	//	}
	//	r = reflect.StructOf(fields)
	default:
		//	if z, _ := t.zero(); z.IsValid() {
		//		r = z.Type()
		//	}
		r = t.TypeOf()
	}
	return r
}

func (t *Type) implements(it *Type) bool {
	if t.cat == ValueT {
		return t.TypeOf().Implements(it.TypeOf())
	}
	// TODO: implement method check for interpreted types
	return true
}

func defRecvType(n *Node) *Type {
	if n.kind != FuncDecl || len(n.child[0].child) == 0 {
		return nil
	}
	if r := n.child[0].child[0].lastChild(); r != nil {
		return r.typ
	}
	return nil
}

func isShiftOperand(n *Node) bool {
	switch n.anc.action {
	case Shl, Shr, ShlAssign, ShrAssign:
		return n.anc.lastChild() == n
	}
	return false
}

func isInterface(t *Type) bool { return t.cat == InterfaceT || t.TypeOf().Kind() == reflect.Interface }

func isStruct(t *Type) bool { return t.TypeOf().Kind() == reflect.Struct }

func isInt(t reflect.Type) bool {
	switch t.Kind() {
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
		return true
	}
	return false
}

func isUint(t reflect.Type) bool {
	switch t.Kind() {
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
		return true
	}
	return false
}

func isComplex(t reflect.Type) bool {
	switch t.Kind() {
	case reflect.Complex64, reflect.Complex128:
		return true
	}
	return false
}

func isFloat(t reflect.Type) bool {
	switch t.Kind() {
	case reflect.Float32, reflect.Float64:
		return true
	}
	return false
}

func isByteArray(t reflect.Type) bool {
	k := t.Kind()
	return (k == reflect.Array || k == reflect.Slice) && t.Elem().Kind() == reflect.Uint8
}

func isFloat32(t reflect.Type) bool { return t.Kind() == reflect.Float32 }
func isFloat64(t reflect.Type) bool { return t.Kind() == reflect.Float64 }
func isNumber(t reflect.Type) bool  { return isInt(t) || isFloat(t) || isComplex(t) }
func isString(t reflect.Type) bool  { return t.Kind() == reflect.String }