Reworked project parser *working*

v2/pkg/parser/boundStructs.go

@@ -0,0 +1,155 @@
+package parser
+
+import (
+	"go/ast"
+
+	"golang.org/x/tools/go/packages"
+)
+
+func (p *Parser) getImportByName(pkg *packages.Package, importName string) *packages.Package {
+	// Find package path
+	for _, imp := range pkg.Imports {
+		if imp.Name == importName {
+			return imp
+		}
+	}
+	return nil
+}
+
+func (p *Parser) findBoundStructsInPackage(pkg *packages.Package, applicationVariableName string) []*StructReference {
+
+	var boundStructs []*StructReference
+
+	// Iterate through the whole package looking for the bound structs
+	for _, fileAst := range pkg.Syntax {
+		ast.Inspect(fileAst, func(n ast.Node) bool {
+			// Parse Call expressions looking for bind calls
+			callExpr, ok := n.(*ast.CallExpr)
+			if !ok {
+				return true
+			}
+			// Check this is the right kind of expression (something.something())
+			f, ok := callExpr.Fun.(*ast.SelectorExpr)
+			if !ok {
+				return true
+			}
+			ident, ok := f.X.(*ast.Ident)
+			if !ok {
+				return true
+			}
+
+			if ident.Name != applicationVariableName {
+				return true
+			}
+
+			if f.Sel.Name != "Bind" {
+				return true
+			}
+
+			if len(callExpr.Args) != 1 {
+				return true
+			}
+
+			// Work out what was bound
+			switch boundItem := callExpr.Args[0].(type) {
+
+			// // app.Bind( someFunction() )
+			// case *ast.CallExpr:
+			// 	switch fn := boundItem.Fun.(type) {
+			// 	case *ast.Ident:
+			// 		boundStructs = append(boundStructs, newStruct(pkg.Name, fn.Name))
+			// 		println("Found bound function:", fn.Name)
+			// 	case *ast.SelectorExpr:
+			// 		ident, ok := fn.X.(*ast.Ident)
+			// 		if !ok {
+			// 			return true
+			// 		}
+			// 		packageName := ident.Name
+			// 		functionName := fn.Sel.Name
+			// 		println("Found bound function:", packageName+"."+functionName)
+
+			// 		strct := p.getFunctionReturnType(packageName, functionName)
+			// 		if strct == nil {
+			// 			// Unable to resolve function
+			// 			return true
+			// 		}
+			// 		boundStructs = append(boundStructs, strct)
+			// 	}
+
+			// Binding struct pointer literals
+			case *ast.UnaryExpr:
+
+				if boundItem.Op.String() != "&" {
+					return true
+				}
+
+				cl, ok := boundItem.X.(*ast.CompositeLit)
+				if !ok {
+					return true
+				}
+
+				switch boundStructExp := cl.Type.(type) {
+
+				// app.Bind( &myStruct{} )
+				case *ast.Ident:
+					boundStruct := newStructReference(pkg.Name, boundStructExp.Name)
+					boundStructs = append(boundStructs, boundStruct)
+
+				// app.Bind( &mypackage.myStruct{} )
+				case *ast.SelectorExpr:
+					var structName = ""
+					var packageName = ""
+					switch x := boundStructExp.X.(type) {
+					case *ast.Ident:
+						packageName = x.Name
+					default:
+						// TODO: Save these warnings
+						// println("Identifier in binding not supported:")
+						return true
+					}
+					structName = boundStructExp.Sel.Name
+					referencedPackage := p.getImportByName(pkg, packageName)
+					boundStruct := newStructReference(referencedPackage.Name, structName)
+					boundStructs = append(boundStructs, boundStruct)
+				}
+
+			// Binding struct literals
+			case *ast.CompositeLit:
+				switch literal := boundItem.Type.(type) {
+
+				// app.Bind( myStruct{} )
+				case *ast.Ident:
+					structName := literal.Name
+					boundStruct := newStructReference(pkg.Name, structName)
+					boundStructs = append(boundStructs, boundStruct)
+
+				// app.Bind( mypackage.myStruct{} )
+				case *ast.SelectorExpr:
+					var structName = ""
+					var packageName = ""
+					switch x := literal.X.(type) {
+					case *ast.Ident:
+						packageName = x.Name
+					default:
+						// TODO: Save these warnings
+						// println("Identifier in binding not supported:")
+						return true
+					}
+					structName = literal.Sel.Name
+
+					referencedPackage := p.getImportByName(pkg, packageName)
+					boundStruct := newStructReference(referencedPackage.Name, structName)
+					boundStructs = append(boundStructs, boundStruct)
+				}
+
+			default:
+				// TODO: Save these warnings
+				// println("Unsupported bind expression:")
+				// spew.Dump(boundItem)
+			}
+
+			return true
+		})
+	}
+	return boundStructs
+}