Function ParseFunction

parser.go:11–33

(x string)

)

func ParseFunction(x string) (function, error) {
	function := function{
		Src: x,
	}
	code := "package p;" + x
	file, err := parser.ParseFile(token.NewFileSet(), "", code, 0)
	if err != nil {
		return function, err
	}
	funcDecl := file.Decls[0].(*ast.FuncDecl)
	function.Name = funcDecl.Name.Name
	returnVariables := funcDecl.Type.Results
	if returnVariables == nil {
		return function, nil
	}
	for _, variable := range returnVariables.List {
		ident, ok := variable.Type.(*ast.Ident)
		if ok {
			function.returnVariables = append(function.returnVariables, ident.Name)
		}
	}
	return function, nil
}

// ParseStatement is a modified version of go/parser.ParseExpr
func ParseStatement(x string) (*AstVisitor, error) {