feat: nested structs and chained completion (#11)

pkg/analysis/builtins.go

@@ -141,6 +141,19 @@ func LoadBuiltinsFromSource(ctx context.Context, contents []byte, path string) (
 		members = append(members, t.Members...)
 	}
 
+	funcMap := make(map[string]query.Signature, len(functions))
+	for _, fn := range functions {
+		funcMap[fn.Name] = fn
+	}
+	for i, s := range symbols {
+		if s.Kind == protocol.SymbolKindFunction {
+			if f, found := funcMap[s.Name]; found {
+				symbols[i].Type = f.ReturnType
+				symbols[i].Aux = &f
+			}
+		}
+	}
+
 	doc.Close()
 
 	// NewDocument returns these symbols with a location of __init__.py, which isn't helpful to anyone

pkg/analysis/builtins_test.go

@@ -36,6 +36,7 @@ type testType int
 const (
 	testTypeFunctions testType = iota
 	testTypeSymbols
+	fixtureFileName = "autokitteh-starlark.test"
 )
 
 func TestLoadBuiltinsFromFile(t *testing.T) {
@@ -285,17 +286,14 @@ func (f *fixture) Document(name string, content string) document.Document {
 }
 
 func (f *fixture) MainDoc(content string) document.Document {
-	return f.Document("Tiltfile.test", content)
+	return f.Document(fixtureFileName, content)
 }
 
 func (f *fixture) ParseBuiltins(content string) {
 	builtins, err := LoadBuiltinsFromSource(f.ctx, []byte(content), "__test__")
 	require.NoError(f.t, err)
 	f.a.builtins.Update(builtins)
 	f.builtins = f.a.builtins
-
-	// f.a.builtins = builtins
-	// f.builtins = builtins
 }
 
 func newFixture(t *testing.T) *fixture {