Infer symbol relationship for object literals property in return statement

snapshots/input/syntax/src/infer-relationship.ts

@@ -0,0 +1,18 @@
+interface Configuration {
+  property: number
+}
+
+export function returnStatement(): Configuration {
+  return {
+    property: 42,
+  }
+}
+
+export function returnStatementInsideArgumentExpression(): Configuration[] {
+  return [1].map<Configuration>(number => {
+    const incremented = number + 1
+    return {
+      property: incremented,
+    }
+  })
+}

snapshots/output/syntax/src/infer-relationship.ts

@@ -0,0 +1,44 @@
+  interface Configuration {
+// definition syntax 1.0.0 src/`infer-relationship.ts`/
+//documentation ```ts\nmodule "infer-relationship.ts"\n```
+//          ^^^^^^^^^^^^^ definition syntax 1.0.0 src/`infer-relationship.ts`/Configuration#
+//          documentation ```ts\ninterface Configuration\n```
+    property: number
+//  ^^^^^^^^ definition syntax 1.0.0 src/`infer-relationship.ts`/Configuration#property.
+//  documentation ```ts\n(property) property: number\n```
+  }
+
+  export function returnStatement(): Configuration {
+//                ^^^^^^^^^^^^^^^ definition syntax 1.0.0 src/`infer-relationship.ts`/returnStatement().
+//                documentation ```ts\nfunction returnStatement(): Configuration\n```
+//                                   ^^^^^^^^^^^^^ reference syntax 1.0.0 src/`infer-relationship.ts`/Configuration#
+    return {
+      property: 42,
+//    ^^^^^^^^ definition syntax 1.0.0 src/`infer-relationship.ts`/property0:
+//    documentation ```ts\n(property) property: number\n```
+//    relationship implementation reference scip-typescript npm syntax 1.0.0 src/`infer-relationship.ts`/Configuration#property.
+    }
+  }
+
+  export function returnStatementInsideArgumentExpression(): Configuration[] {
+//                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ definition syntax 1.0.0 src/`infer-relationship.ts`/returnStatementInsideArgumentExpression().
+//                documentation ```ts\nfunction returnStatementInsideArgumentExpression(): Configuration[]\n```
+//                                                           ^^^^^^^^^^^^^ reference syntax 1.0.0 src/`infer-relationship.ts`/Configuration#
+    return [1].map<Configuration>(number => {
+//             ^^^ reference typescript 4.8.4 lib/`lib.es5.d.ts`/Array#map().
+//                 ^^^^^^^^^^^^^ reference syntax 1.0.0 src/`infer-relationship.ts`/Configuration#
+//                                ^^^^^^ definition local 3
+//                                documentation ```ts\n(parameter) number: number\n```
+      const incremented = number + 1
+//          ^^^^^^^^^^^ definition local 6
+//          documentation ```ts\nvar incremented: number\n```
+//                        ^^^^^^ reference local 3
+      return {
+        property: incremented,
+//      ^^^^^^^^ definition syntax 1.0.0 src/`infer-relationship.ts`/property1:
+//      documentation ```ts\n(property) property: number\n```
+//                ^^^^^^^^^^^ reference local 6
+      }
+    })
+  }
+

snapshots/output/syntax/src/interface.ts

@@ -26,9 +26,11 @@
       property: 'a',
 //    ^^^^^^^^ definition syntax 1.0.0 src/`interface.ts`/property0:
 //    documentation ```ts\n(property) property: string\n```
+//    relationship implementation reference scip-typescript npm syntax 1.0.0 src/`interface.ts`/Interface#property.
       methodSignature(param: string): string {
 //    ^^^^^^^^^^^^^^^ definition local 4
 //    documentation ```ts\n(method) methodSignature(param: string): string\n```
+//    relationship implementation reference scip-typescript npm syntax 1.0.0 src/`interface.ts`/Interface#methodSignature().
 //                    ^^^^^ definition local 5
 //                    documentation ```ts\n(parameter) param: string\n```
         return param
@@ -37,6 +39,7 @@
       methodSignature2: (param: string): string => {
 //    ^^^^^^^^^^^^^^^^ definition syntax 1.0.0 src/`interface.ts`/methodSignature20:
 //    documentation ```ts\n(property) methodSignature2: (param: string) => string\n```
+//    relationship implementation reference scip-typescript npm syntax 1.0.0 src/`interface.ts`/Interface#methodSignature2.
 //                       ^^^^^ definition local 7
 //                       documentation ```ts\n(parameter) param: string\n```
         return param

src/FileIndexer.ts

@@ -38,7 +38,7 @@ export class FileIndexer {
   }
   public index(): void {
     // Uncomment below if you want to skip certain files for local development.
-    // if (!this.sourceFile.fileName.includes('ClassWithPrivate')) {
+    // if (!this.sourceFile.fileName.includes('infer-relationship')) {
     //   return
     // }
     this.emitSourceFileOccurrence()
@@ -561,7 +561,10 @@ export class FileIndexer {
         onAncestor(declaration)
       }
       if (ts.isObjectLiteralExpression(declaration)) {
-        const tpe = this.inferredTypeOfObjectLiteral(declaration)
+        const tpe = this.inferredTypeOfObjectLiteral(
+          declaration.parent,
+          declaration
+        )
         for (const symbolDeclaration of tpe.symbol?.declarations || []) {
           loop(symbolDeclaration)
         }
@@ -592,19 +595,32 @@ export class FileIndexer {
   // `SomeInterface`. The object literal could satisfy many types, but in this
   // particular location must only satisfy `SomeInterface`.
   private inferredTypeOfObjectLiteral(
-    node: ts.ObjectLiteralExpression
+    node: ts.Node,
+    literal: ts.ObjectLiteralExpression
   ): ts.Type {
-    if (ts.isVariableDeclaration(node.parent)) {
+    if (ts.isReturnStatement(node) || ts.isBlock(node)) {
+      return this.inferredTypeOfObjectLiteral(node.parent, literal)
+    }
+
+    if (ts.isVariableDeclaration(node)) {
       // Example, return `SomeInterface` from `const x: SomeInterface = {y: 42}`.
-      return this.checker.getTypeAtLocation(node.parent.name)
+      return this.checker.getTypeAtLocation(node.name)
+    }
+
+    if (ts.isFunctionLike(node)) {
+      const functionType = this.checker.getTypeAtLocation(node)
+      const callSignatures = functionType.getCallSignatures()
+      if (callSignatures.length > 0) {
+        return callSignatures[0].getReturnType()
+      }
     }
 
-    if (ts.isCallOrNewExpression(node.parent)) {
+    if (ts.isCallOrNewExpression(node)) {
       // Example: return the type of the second parameter of `someMethod` from
       // the expression `someMethod(someParameter, {y: 42})`.
-      const signature = this.checker.getResolvedSignature(node.parent)
-      for (const [index, argument] of (node.parent.arguments || []).entries()) {
-        if (argument === node) {
+      const signature = this.checker.getResolvedSignature(node)
+      for (const [index, argument] of (node.arguments || []).entries()) {
+        if (argument === literal) {
           const parameterSymbol = signature?.getParameters()[index]
           if (parameterSymbol) {
             return this.checker.getTypeOfSymbolAtLocation(parameterSymbol, node)
@@ -613,7 +629,7 @@ export class FileIndexer {
       }
     }
 
-    return this.checker.getTypeAtLocation(node)
+    return this.checker.getTypeAtLocation(literal)
   }
 }