Initial implementation of simplified invokes for dotnet and nodejs an…

…d generalized outputs for function schemas
pulumi · Nov 24, 2022 · 02d53b5 · 02d53b5
1 parent 45dddd8
commit 02d53b5
Show file tree

Hide file tree

Showing 99 changed files with 4,347 additions and 491 deletions.
diff --git a/...en-dotnet-nodejs--initial-implementation-of-simplified-invokes-for-dotnet-and-nodejs.yaml b/...en-dotnet-nodejs--initial-implementation-of-simplified-invokes-for-dotnet-and-nodejs.yaml
@@ -0,0 +1,4 @@
+changes:
+- type: feat
+  scope: sdkgen/dotnet,nodejs
+  description: Initial implementation of simplified invokes for dotnet and nodejs
diff --git a/pkg/codegen/docs/gen.go b/pkg/codegen/docs/gen.go
@@ -1680,11 +1680,14 @@ func (mod *modContext) getTypes(member interface{}, types nestedTypeUsageInfo) {
 			mod.getTypes(m.Function, types)
 		}
 	case *schema.Function:
-		if t.Inputs != nil {
+		if t.Inputs != nil && !t.MultiArgumentInputs {
 			mod.getNestedTypes(t.Inputs, types, true)
 		}
-		if t.Outputs != nil {
-			mod.getNestedTypes(t.Outputs, types, false)
+
+		if t.ReturnType != nil {
+			if objectType, ok := t.ReturnType.(*schema.ObjectType); ok && objectType != nil {
+				mod.getNestedTypes(objectType, types, false)
+			}
 		}
 	}
 }

diff --git a/pkg/codegen/docs/gen_function.go b/pkg/codegen/docs/gen_function.go
@@ -433,8 +433,11 @@ func (mod *modContext) genFunction(f *schema.Function) functionDocArgs {
 		if f.Inputs != nil {
 			inputProps[lang] = mod.getProperties(f.Inputs.Properties, lang, true, false, false)
 		}
-		if f.Outputs != nil {
-			outputProps[lang] = mod.getProperties(f.Outputs.Properties, lang, false, false, false)
+		if f.ReturnType != nil {
+			if objectObject, ok := f.ReturnType.(*schema.ObjectType); ok {
+				outputProps[lang] = mod.getProperties(objectObject.Properties,
+					lang, false, false, false)
+			}
 		}
 	}
 

diff --git a/pkg/codegen/docs/gen_method.go b/pkg/codegen/docs/gen_method.go
@@ -77,9 +77,11 @@ func (mod *modContext) genMethod(r *schema.Resource, m *schema.Method) methodDoc
 				inputProps[lang] = props
 			}
 		}
-		if f.Outputs != nil {
-			outputProps[lang] = mod.getPropertiesWithIDPrefixAndExclude(f.Outputs.Properties, lang, false, false, false,
-				fmt.Sprintf("%s_result_", m.Name), nil)
+		if f.ReturnType != nil {
+			if objectType, ok := f.ReturnType.(*schema.ObjectType); ok && objectType != nil {
+				outputProps[lang] = mod.getPropertiesWithIDPrefixAndExclude(objectType.Properties, lang, false, false, false,
+					fmt.Sprintf("%s_result_", m.Name), nil)
+			}
 		}
 	}
 
@@ -321,7 +323,7 @@ func (mod *modContext) getMethodResult(r *schema.Resource, m *schema.Method) map
 
 	var resultTypeName string
 	for _, lang := range dctx.supportedLanguages {
-		if m.Function.Outputs != nil && len(m.Function.Outputs.Properties) > 0 {
+		if m.Function.ReturnType != nil {
 			resultTypeName = dctx.getLanguageDocHelper(lang).GetMethodResultName(mod.pkg, mod.mod, r, m)
 		}
 		resourceMap[lang] = propertyType{

diff --git a/pkg/codegen/dotnet/doc.go b/pkg/codegen/dotnet/doc.go
@@ -110,8 +110,24 @@ func (d DocLanguageHelper) GetMethodName(m *schema.Method) string {
 func (d DocLanguageHelper) GetMethodResultName(pkg *schema.Package, modName string, r *schema.Resource,
 	m *schema.Method) string {
 
+	var returnType *schema.ObjectType
+	if m.Function.ReturnType != nil {
+		if objectType, ok := m.Function.ReturnType.(*schema.ObjectType); ok {
+			returnType = objectType
+		} else {
+			typeDetails := map[*schema.ObjectType]*typeDetails{}
+			mod := &modContext{
+				pkg:         pkg,
+				mod:         modName,
+				typeDetails: typeDetails,
+				namespaces:  d.Namespaces,
+			}
+			return mod.typeString(m.Function.ReturnType, "", false, false, false)
+		}
+	}
+
 	if info, ok := pkg.Language["csharp"].(CSharpPackageInfo); ok {
-		if info.LiftSingleValueMethodReturns && m.Function.Outputs != nil && len(m.Function.Outputs.Properties) == 1 {
+		if info.LiftSingleValueMethodReturns && returnType != nil && len(returnType.Properties) == 1 {
 			typeDetails := map[*schema.ObjectType]*typeDetails{}
 			mod := &modContext{
 				pkg:           pkg,
@@ -120,7 +136,7 @@ func (d DocLanguageHelper) GetMethodResultName(pkg *schema.Package, modName stri
 				namespaces:    d.Namespaces,
 				rootNamespace: info.GetRootNamespace(),
 			}
-			return mod.typeString(m.Function.Outputs.Properties[0].Type, "", false, false, false)
+			return mod.typeString(returnType.Properties[0].Type, "", false, false, false)
 		}
 	}
 	return fmt.Sprintf("%s.%sResult", resourceName(r), d.GetMethodName(m))