Ejemplo n.º 1
0
Archivo: parser.go Proyecto: h12w/gombi
func (p *parser) parseParamDecl(n *parse.Node, scope *ast.Scope) *ast.Field {
	field := ast.Field{}
	if n.Child(0).Is(type_) {
		field.Type = p.parseType(n.Child(0))
	} else {
		field.Names = p.parseIdentList(n.Child(0))
		if n.Child(1).Is(type_) {
			field.Type = p.parseType(n.Child(1))
		} else {
			// TODO ...
			field.Type = p.parseType(n.Child(2))
		}
	}
	p.declare(field, nil, scope, ast.Var, field.Names...)
	if field.Type != nil {
		p.resolve(field.Type)
	}
	return &field
}
Ejemplo n.º 2
0
// Struct creates a struct{} expression. The arguments are a series
// of name/type/tag tuples. Name must be of type *ast.Ident, type
// must be of type ast.Expr, and tag must be of type *ast.BasicLit,
// The number of arguments must be a multiple of 3, or a run-time
// panic will occur.
func Struct(args ...ast.Expr) *ast.StructType {
	fields := new(ast.FieldList)
	if len(args)%3 != 0 {
		panic("Number of args to FieldList must be a multiple of 3, got " + strconv.Itoa(len(args)))
	}
	for i := 0; i < len(args); i += 3 {
		var field ast.Field
		name, typ, tag := args[i], args[i+1], args[i+2]
		if name != nil {
			field.Names = []*ast.Ident{name.(*ast.Ident)}
		}
		if typ != nil {
			field.Type = typ
		}
		if tag != nil {
			field.Tag = tag.(*ast.BasicLit)
		}
		fields.List = append(fields.List, &field)
	}
	return &ast.StructType{Fields: fields}
}
Ejemplo n.º 3
0
Archivo: method.go Proyecto: nelsam/hel
func mockField(idx int, f *ast.Field) *ast.Field {
	if f.Names == nil {
		if idx < 0 {
			return f
		}
		// Edit the field directly to ensure the same name is used in the mock
		// struct.
		f.Names = []*ast.Ident{{Name: fmt.Sprintf(inputFmt, idx)}}
		return f
	}

	// Here, we want a copy, so that we can use altered names without affecting
	// field names in the mock struct.
	newField := &ast.Field{Type: f.Type}
	for _, n := range f.Names {
		name := n.Name
		if name == receiverName {
			name += "_"
		}
		newField.Names = append(newField.Names, &ast.Ident{Name: name})
	}
	return newField
}