devlensio 0.6.2 → 1.0.1

package/extractors/go/calls.go

@@ -0,0 +1,289 @@
+// calls.go — CALLS resolution ladder.
+//
+// 1. types.Info.Selections[selExpr] → *types.Func → local package → exact
+//    METHOD/FUNCTION node; std → skip; external → lazy [mod]/pkg::member
+//    (gated), receiver package from the selection's receiver type.
+// 2. types.Info.Uses[ident] → *types.Func → same by name.
+// 3. Name-based fallback (no type info / external receivers): plain name →
+//    same-package func; alias.member → symbol maps → local method/func or
+//    [mod] lazy member (python's proven chain rule).
+//
+// metadata.calls (sorted unique strings) + metadata.resolvedCalls written on
+// every caller node (contract compliance + LLM context).
+
+package main
+
+import (
+	"go/ast"
+	"go/token"
+	"go/types"
+	"sort"
+	"strings"
+)
+
+type resolvedCall struct {
+	Target string `json:"target"`
+	Method string `json:"method,omitempty"`
+	Class  string `json:"class,omitempty"`
+}
+
+// detectCalls — CALLS edges for every non-test function/method body.
+func detectCalls(pr *parsedRepo, ti *TypeInfo, l *LookupMaps, opts *Options) []map[string]any {
+	edges := []map[string]any{}
+
+	for _, pf := range pr.files {
+		info := ti.Infos[pf.RelPath]
+		for _, fn := range pf.Funcs {
+			if fn.IsTest {
+				continue // test funcs are leaf metadata, not callers
+			}
+			callerID := nodeIDForFunc(pf.RelPath, fn)
+			callStrings := make([]string, 0, len(fn.Calls))
+			var details []resolvedCall
+			for _, c := range fn.Calls {
+				callStrings = append(callStrings, c.Str)
+				target, method, class, e := resolveCall(c, fn, pf, info, l)
+				if e != nil {
+					edges = append(edges, e)
+				}
+				if target != "" {
+					details = append(details, resolvedCall{Target: target, Method: method, Class: class})
+				}
+			}
+			// metadata enrichment on the (already created) caller node
+			if n, ok := l.NodeByID[callerID]; ok {
+				m := n["metadata"].(map[string]any)
+				m["calls"] = sortedUnique(callStrings)
+				if len(details) > 0 {
+					m["resolvedCalls"] = details
+				}
+			}
+		}
+	}
+	return edges
+}
+
+// nodeIDForFunc — matches the id scheme in lookup.go.
+func nodeIDForFunc(rel string, fn *ParsedFunc) string {
+	if fn.IsMethod {
+		return methodNodeID(rel, fn)
+	}
+	return funcNodeID(rel, fn.Name)
+}
+
+// resolveCall → (target node id, method, class, edge) — edge nil when
+// unresolvable (documented: builtins/locals/dynamic stay metadata-only).
+func resolveCall(c CallSite, fn *ParsedFunc, pf *ParsedFile, info *types.Info,
+	l *LookupMaps) (string, string, string, map[string]any) {
+
+	pkgPath := fn.PkgPath
+	callerID := nodeIDForFunc(pf.RelPath, fn)
+
+	// ── 1. type-info resolution (exact) ──
+	if info != nil {
+		if selExpr, ok := c.Node.(*ast.SelectorExpr); ok {
+			if sel, ok := info.Selections[selExpr]; ok {
+				if obj, ok := sel.Obj().(*types.Func); ok && obj != nil {
+					if id, class, ok := funcToNode(obj, l); ok {
+						return id, c.Sel, class, callEdge(callerID, id, c.Sel, class)
+					}
+					if obj.Pkg() != nil {
+						path := obj.Pkg().Path()
+						if !isStdImport(path) {
+							if n := l.tp.memberNode(path, c.Sel); n != nil {
+								return n["id"].(string), c.Sel, "", callEdge(callerID, n["id"].(string), c.Sel, "")
+							}
+						}
+					}
+				}
+			}
+		} else if idExpr, ok := c.Node.(*ast.Ident); ok {
+			if obj, ok := info.Uses[idExpr].(*types.Func); ok && obj != nil {
+				if id, class, ok := funcToNode(obj, l); ok {
+					return id, "", class, callEdge(callerID, id, "", class)
+				}
+				if obj.Pkg() != nil {
+					path := obj.Pkg().Path()
+					if !isStdImport(path) {
+						if n := l.tp.memberNode(path, obj.Name()); n != nil {
+							return n["id"].(string), obj.Name(), "", callEdge(callerID, n["id"].(string), obj.Name(), "")
+						}
+					}
+				}
+			}
+		}
+	}
+
+	// ── 1.5 receiver-type-from-declaration (params / typed vars) ──
+	// Handlers receive framework types by signature: func(c *gin.Context) —
+	// the checker can't resolve external types, but the DECLARED type can.
+	if c.IsSel {
+		if impPath, typeName, local := receiverDeclaredType(c, fn, pf, l); typeName != "" {
+			if local {
+				if id, ok := l.MethodNodeByPkgTypeName[impPath+"::"+typeName+"."+c.Sel]; ok {
+					return id, c.Sel, typeName, callEdge(callerID, id, c.Sel, typeName)
+				}
+			} else if n := l.tp.memberNode(impPath, typeName+"."+c.Sel); n != nil {
+				return n["id"].(string), c.Sel, typeName, callEdge(callerID, n["id"].(string), c.Sel, typeName)
+			}
+		}
+	}
+
+	// ── 2. name-based fallback ──
+	if c.IsSel {
+		// alias.member: receiver is an import alias → local member or [mod]
+		alias := firstSegment(c.Receiver)
+		if impPath, ok := l.SymbolMaps[pf.RelPath][alias]; ok {
+			if _, local := l.PkgNodesByPath[impPath]; local {
+				if id, ok := l.MethodNodeByPkgTypeName[impPath+"::"+c.Sel]; ok {
+					return id, c.Sel, "", callEdge(callerID, id, c.Sel, "")
+				}
+				if id, ok := l.FuncNodeByPkgName[impPath+"::"+c.Sel]; ok {
+					return id, c.Sel, "", callEdge(callerID, id, c.Sel, "")
+				}
+				return "", "", "", nil
+			}
+			if n := l.tp.memberNode(impPath, c.Sel); n != nil {
+				return n["id"].(string), c.Sel, "", callEdge(callerID, n["id"].(string), c.Sel, "")
+			}
+			return "", "", "", nil
+		}
+		// receiver names a local TYPE (User.Create) → method on that type
+		if id, ok := l.MethodNodeByPkgTypeName[pkgPath+"::"+c.Receiver+"."+c.Sel]; ok {
+			return id, c.Sel, c.Receiver, callEdge(callerID, id, c.Sel, c.Receiver)
+		}
+		// receiver is a plain variable whose type we can't see — unresolvable
+		// tier (documented; type info covers the exact cases)
+		return "", "", "", nil
+	}
+
+	// plain name: same package first
+	if id, ok := l.FuncNodeByPkgName[pkgPath+"::"+c.Str]; ok {
+		return id, "", "", callEdge(callerID, id, "", "")
+	}
+	return "", "", "", nil
+}
+
+func callEdge(from, to, method, class string) map[string]any {
+	meta := map[string]any{}
+	if method != "" {
+		meta["method"] = method
+	}
+	if class != "" {
+		meta["class"] = class
+	}
+	return edgeWithMeta(from, to, EdgeCalls, meta)
+}
+
+// receiverDeclaredType — the receiver's type from its DECLARATION: function
+// parameters first (func(c *gin.Context) — the documented handler signature),
+// then package-level typed vars (var DB *gorm.DB). Returns (importPath,
+// typeName, local) — local=false means external (memberNode candidate).
+func receiverDeclaredType(c CallSite, fn *ParsedFunc, pf *ParsedFile, l *LookupMaps) (string, string, bool) {
+	recvName := firstSegment(c.Receiver)
+	if recvName == "" {
+		return "", "", false
+	}
+	// 1. function parameters
+	if fn.Decl != nil && fn.Decl.Type != nil && fn.Decl.Type.Params != nil {
+		for _, field := range fn.Decl.Type.Params.List {
+			for _, name := range field.Names {
+				if name.Name == recvName {
+					return declaredTypeInfo(field.Type, fn.PkgPath, pf, l)
+				}
+			}
+		}
+	}
+	// 2. package-level typed vars in this file (var DB *gorm.DB)
+	for _, decl := range pf.AST.Decls {
+		gd, ok := decl.(*ast.GenDecl)
+		if !ok || gd.Tok != token.VAR {
+			continue
+		}
+		for _, spec := range gd.Specs {
+			vs, ok := spec.(*ast.ValueSpec)
+			if !ok || vs.Type == nil {
+				continue
+			}
+			for _, name := range vs.Names {
+				if name.Name == recvName {
+					return declaredTypeInfo(vs.Type, fn.PkgPath, pf, l)
+				}
+			}
+		}
+	}
+	return "", "", false
+}
+
+// declaredTypeInfo — a declared type expr → (importPath, typeName, local).
+// Ident "User" → local; SelectorExpr "gin.Context" → alias → external path.
+func declaredTypeInfo(expr ast.Expr, pkgPath string, pf *ParsedFile, l *LookupMaps) (string, string, bool) {
+	switch t := expr.(type) {
+	case *ast.StarExpr:
+		return declaredTypeInfo(t.X, pkgPath, pf, l)
+	case *ast.Ident:
+		return pkgPath, t.Name, true // same package — local type
+	case *ast.SelectorExpr:
+		alias := identName(t.X)
+		if impPath, ok := l.SymbolMaps[pf.RelPath][alias]; ok {
+			return impPath, t.Sel.Name, false
+		}
+	case *ast.IndexExpr:
+		return declaredTypeInfo(t.X, pkgPath, pf, l)
+	case *ast.IndexListExpr:
+		return declaredTypeInfo(t.X, pkgPath, pf, l)
+	}
+	return "", "", false
+}
+
+// funcToNode — *types.Func declared in a local package → (node id, typeName).
+func funcToNode(obj *types.Func, l *LookupMaps) (string, string, bool) {
+	if obj.Pkg() == nil {
+		return "", "", false
+	}
+	pkgPath := obj.Pkg().Path()
+	if _, local := l.PkgNodesByPath[pkgPath]; !local {
+		return "", "", false
+	}
+	if recv := obj.Type().(*types.Signature).Recv(); recv != nil {
+		typeName := recvTypeNameFromTypes(recv.Type())
+		if typeName == "" {
+			return "", "", false
+		}
+		id, ok := l.MethodNodeByPkgTypeName[pkgPath+"::"+typeName+"."+obj.Name()]
+		return id, typeName, ok
+	}
+	id, ok := l.FuncNodeByPkgName[pkgPath+"::"+obj.Name()]
+	return id, "", ok
+}
+
+// recvTypeNameFromTypes — receiver type → simple name (strips * and pkg path).
+func recvTypeNameFromTypes(t types.Type) string {
+	switch tt := t.(type) {
+	case *types.Pointer:
+		return recvTypeNameFromTypes(tt.Elem())
+	case *types.Named:
+		return tt.Obj().Name()
+	}
+	return ""
+}
+
+func firstSegment(s string) string {
+	if i := strings.Index(s, "."); i >= 0 {
+		return s[:i]
+	}
+	return s
+}
+
+func sortedUnique(list []string) []string {
+	seen := map[string]bool{}
+	out := []string{}
+	for _, s := range list {
+		if !seen[s] {
+			seen[s] = true
+			out = append(out, s)
+		}
+	}
+	sort.Strings(out)
+	return out
+}

package/extractors/go/contract.go

@@ -0,0 +1,140 @@
+// contract.go — the JSON shape DevLens Engine expects.
+//
+// Every key here is camelCase to match the TypeScript interfaces in the
+// devlens-engine (src/types.ts, src/extractors/types.ts). This file is the
+// single source of truth for the output shape; other modules never spell
+// keys by hand.
+
+package main
+
+import (
+	"crypto/sha256"
+	"encoding/hex"
+	"path"
+)
+
+// ── Node types (subset of the engine's NodeType union used by Go) ──
+const (
+	NodeFile       = "FILE"
+	NodeFunction   = "FUNCTION"
+	NodeMethod     = "METHOD"
+	NodeStruct     = "STRUCT"
+	NodeInterface  = "INTERFACE"
+	NodePackage    = "PACKAGE"
+	NodeTest       = "TEST"
+	NodeRoute      = "ROUTE"
+	NodeThirdParty = "THIRD_PARTY"
+	NodeEnum       = "ENUM"
+)
+
+// ── Edge types (subset of the engine's EdgeType union used by Go) ──
+const (
+	EdgeCalls      = "CALLS"
+	EdgeImports    = "IMPORTS"
+	EdgeExtends    = "EXTENDS"
+	EdgeImplements = "IMPLEMENTS"
+	EdgeHandles    = "HANDLES"
+	EdgeTests      = "TESTS"
+	EdgeReadsFrom  = "READS_FROM"
+	EdgeWritesTo   = "WRITES_TO"
+)
+
+// codeHash = sha256(rawCode) hex-encoded, first 16 chars (contract rule).
+func codeHash(raw string) string {
+	sum := sha256.Sum256([]byte(raw))
+	return hex.EncodeToString(sum[:])[:16]
+}
+
+// ── Fingerprint ──
+
+type Fingerprint struct {
+	Language        string            `json:"language"`
+	ProjectType     string            `json:"projectType"`
+	Framework       string            `json:"framework"`
+	Router          string            `json:"router"`
+	StateManagement []string          `json:"stateManagement"`
+	DataFetching    []string          `json:"dataFetching"`
+	Databases       []string          `json:"databases"`
+	RawDependencies map[string]string `json:"rawDependencies"`
+}
+
+func newFingerprint() *Fingerprint {
+	return &Fingerprint{
+		Language:        "go",
+		ProjectType:     "unknown",
+		Framework:       "unknown",
+		Router:          "none",
+		StateManagement: []string{},
+		DataFetching:    []string{},
+		Databases:       []string{},
+		RawDependencies: map[string]string{},
+	}
+}
+
+// ── Stats / Errors / Result ──
+
+type Stats struct {
+	TotalFiles   int `json:"totalFiles"`
+	TotalNodes   int `json:"totalNodes"`
+	SkippedFiles int `json:"skippedFiles"`
+}
+
+type ExtractorError struct {
+	File  string `json:"file,omitempty"`
+	Error string `json:"error"`
+}
+
+type ExtractorResult struct {
+	Fingerprint *Fingerprint     `json:"fingerprint"`
+	Nodes       []map[string]any `json:"nodes"`
+	Edges       []map[string]any `json:"edges"`
+	Routes      []map[string]any `json:"routes"`
+	Stats       Stats            `json:"stats"`
+	Errors      []ExtractorError `json:"errors"`
+}
+
+// ── Node / edge builders ──
+
+// fileNode — FILE (or TEST) node. Id format: file::rel/path.go (matches
+// src/parser/index.ts). File nodes parent themselves.
+func fileNode(relPath string, endLine int, nodeType string) map[string]any {
+	return map[string]any{
+		"id":         "file::" + relPath,
+		"name":       path.Base(relPath),
+		"type":       nodeType,
+		"filePath":   relPath,
+		"startLine":  1,
+		"endLine":    endLine,
+		"parentFile": "file::" + relPath,
+		"metadata": map[string]any{
+			"nodeCount":    0,
+			"childNodeIds": []any{},
+			"language":     "go",
+		},
+	}
+}
+
+// codeNode — FUNCTION/METHOD/STRUCT/INTERFACE/ENUM/PACKAGE/THIRD_PARTY/ROUTE.
+// rawCode on every code node (summarizer input — engine discards after).
+func codeNode(id, relPath, name, nodeType string, startLine, endLine int, rawCode string, metadata map[string]any) map[string]any {
+	return map[string]any{
+		"id":         id,
+		"name":       name,
+		"type":       nodeType,
+		"filePath":   relPath,
+		"startLine":  startLine,
+		"endLine":    endLine,
+		"parentFile": "file::" + relPath,
+		"codeHash":   codeHash(rawCode),
+		"rawCode":    rawCode,
+		"metadata":   metadata,
+	}
+}
+
+func edge(from, to, etype string) map[string]any {
+	return map[string]any{"from": from, "to": to, "type": etype}
+}
+
+func edgeWithMeta(from, to, etype string, metadata map[string]any) map[string]any {
+	return map[string]any{"from": from, "to": to, "type": etype, "metadata": metadata}
+}

package/extractors/go/extractor.go

@@ -0,0 +1,161 @@
+// extractor.go — pipeline orchestration.
+//
+// Order (mirrors the python/java extractors):
+//   walk → parse (facts) → typeload (Mode A/B) → lookup → imports (+PACKAGE
+//   nodes) → calls → routes → orm → inheritance → tests → enrich →
+//   dedupe → deterministic sort → result.
+// Edge detectors resolve from facts via lookup maps — never re-walk ASTs.
+
+package main
+
+import (
+	"sort"
+)
+
+func runExtractor(repoPath string, opts *Options) *ExtractorResult {
+	if opts == nil {
+		opts = &Options{}
+	}
+	result := &ExtractorResult{
+		Fingerprint: newFingerprint(),
+		Nodes:       []map[string]any{},
+		Edges:       []map[string]any{},
+		Routes:      []map[string]any{},
+		Errors:      []ExtractorError{},
+	}
+	addError := func(file, msg string) {
+		result.Errors = append(result.Errors, ExtractorError{File: file, Error: msg})
+	}
+
+	// ── fingerprint (go.mod only; net-http/sql upgrades come after walk) ──
+	mf := parseGoMod(repoPath)
+	fp := fingerprintFromMod(mf)
+	result.Fingerprint = fp
+
+	// ── walk + parse ──
+	parsed := parseRepo(repoPath, mf, addError)
+
+	// stdlib database/sql + net-http presence upgrade the fingerprint
+	// (both are stdlib — invisible in go.mod requires).
+	if fp.Framework == "unknown" && importsNetHTTP(parsed) {
+		fp.Framework = "net-http"
+		fp.ProjectType = "backend"
+	}
+	if importsDatabaseSQL(parsed) && !contains(fp.Databases, "sql") {
+		fp.Databases = append(fp.Databases, "sql")
+	}
+	if hasMainPackage(parsed) {
+		fp.ProjectType = "backend"
+	}
+
+	// ── type load (in-process go/types; local + std, external degraded) ──
+	ti := loadTypes(parsed, addError)
+
+	// ── lookup maps — ONE shared index, consumed by all detectors ──
+	lookup := buildLookupMaps(parsed, ti)
+
+	// ── edges, in dependency order ──
+	allEdges := []map[string]any{}
+	nodes := []map[string]any{}
+	routes := []map[string]any{}
+
+	// file + test nodes first (parents exist before children)
+	files := collectFileNodes(parsed, lookup)
+	nodes = append(nodes, files...)
+
+	// IMPORTS edges + [mod] third-party nodes (no PACKAGE nodes — connected-only
+	// pruning is engine-side; imports target the package's FILES, JS parity)
+	allEdges = append(allEdges, detectImports(parsed, lookup, opts, fp)...)
+
+	// code nodes (functions/methods/structs/interfaces/enums)
+	codeNodes := collectCodeNodes(parsed, lookup, ti)
+	nodes = append(nodes, codeNodes...)
+
+	// CALLS edges
+	allEdges = append(allEdges, detectCalls(parsed, ti, lookup, opts)...)
+
+	// ROUTE nodes + HANDLES edges + BackendRouteNode entries
+	routeNodes, handlesEdges, backendRoutes := detectRoutes(parsed, lookup, ti, fp)
+	nodes = append(nodes, routeNodes...)
+	allEdges = append(allEdges, handlesEdges...)
+	routes = append(routes, backendRoutes...)
+
+	// EXTENDS / IMPLEMENTS
+	allEdges = append(allEdges, detectInheritance(parsed, lookup, ti, opts)...)
+
+	// READS_FROM / WRITES_TO
+	allEdges = append(allEdges, detectOrmEdges(parsed, lookup, ti, opts)...)
+
+	// TESTS edges
+	allEdges = append(allEdges, detectTests(parsed, lookup)...)
+
+	// third-party nodes (gated) — created during imports/calls, collected last
+	nodes = append(nodes, lookup.thirdPartyNodes...)
+
+	// ── dedupe + deterministic sort ──
+	allEdges = dedupeEdges(allEdges)
+	sort.SliceStable(allEdges, func(i, j int) bool {
+		a, b := allEdges[i], allEdges[j]
+		return edgeKey(a) < edgeKey(b)
+	})
+	sort.SliceStable(nodes, func(i, j int) bool {
+		return nodeKey(nodes[i]) < nodeKey(nodes[j])
+	})
+	sort.SliceStable(routes, func(i, j int) bool {
+		return routeKey(routes[i]) < routeKey(routes[j])
+	})
+	// keep errors in insertion order (they're already deterministic: file walk order)
+
+	result.Nodes = nodes
+	result.Edges = allEdges
+	result.Routes = routes
+	result.Stats = Stats{
+		TotalFiles:   len(parsed.files),
+		TotalNodes:   len(nodes),
+		SkippedFiles: parsed.skipped,
+	}
+	return result
+}
+
+// ── deterministic key helpers ──
+
+func nodeKey(n map[string]any) string {
+	s, _ := n["id"].(string)
+	return s
+}
+
+func edgeKey(e map[string]any) string {
+	from, _ := e["from"].(string)
+	to, _ := e["to"].(string)
+	t, _ := e["type"].(string)
+	return from + "\x00" + t + "\x00" + to
+}
+
+func routeKey(r map[string]any) string {
+	m, _ := r["method"].(string)
+	p, _ := r["path"].(string)
+	return m + "\x00" + p
+}
+
+func dedupeEdges(edges []map[string]any) []map[string]any {
+	seen := map[string]bool{}
+	out := []map[string]any{}
+	for _, e := range edges {
+		k := edgeKey(e)
+		if seen[k] {
+			continue
+		}
+		seen[k] = true
+		out = append(out, e)
+	}
+	return out
+}
+
+func contains(list []string, s string) bool {
+	for _, v := range list {
+		if v == s {
+			return true
+		}
+	}
+	return false
+}