Set declaration span only if its not same as own span

Author: sheetalkamat

src/services/findAllReferences.ts

@@ -111,11 +111,6 @@ namespace ts.FindAllReferences {
             case SyntaxKind.ImportClause:
                 return node.parent;
 
-            case SyntaxKind.JsxAttribute:
-                return (node as JsxAttribute).initializer === undefined ?
-                    undefined :
-                    node;
-
             case SyntaxKind.BinaryExpression:
                 return isExpressionStatement(node.parent) ?
                     node.parent :
@@ -129,23 +124,31 @@ namespace ts.FindAllReferences {
                 };
 
             case SyntaxKind.PropertyAssignment:
-            // TODO(shkamat):: Should we show whole object literal instead?
             case SyntaxKind.ShorthandPropertyAssignment:
                 return isArrayLiteralOrObjectLiteralDestructuringPattern(node.parent) ?
                     getDeclarationForDeclarationSpan(
                         findAncestor(node.parent, node =>
                             isBinaryExpression(node) || isForInOrOfStatement(node)
                         ) as BinaryExpression | ForInOrOfStatement
                     ) :
-                    node.kind === SyntaxKind.PropertyAssignment ?
-                        node :
-                        undefined;
+                    node;
 
             default:
                 return node;
         }
     }
 
+    export function setDeclarationSpan(span: DocumentSpan, sourceFile: SourceFile, declaration?: DeclarationNode) {
+        if (declaration) {
+            const declarationSpan = isDeclarationNodeWithStartAndEnd(declaration) ?
+                getTextSpan(declaration.start, sourceFile, declaration.end) :
+                getTextSpan(declaration, sourceFile);
+            if (declarationSpan.start !== span.textSpan.start || declarationSpan.length !== span.textSpan.length) {
+                span.declarationSpan = declarationSpan;
+            }
+        }
+    }
+
     export interface Options {
         readonly findInStrings?: boolean;
         readonly findInComments?: boolean;
@@ -287,11 +290,7 @@ namespace ts.FindAllReferences {
             textSpan: getTextSpan(isComputedPropertyName(node) ? node.expression : node, sourceFile),
             displayParts
         };
-        if (declaration) {
-            result.declarationSpan = isDeclarationNodeWithStartAndEnd(declaration) ?
-                getTextSpan(declaration.start, sourceFile, declaration.end) :
-                getTextSpan(declaration, sourceFile);
-        }
+        setDeclarationSpan(result, sourceFile, declaration);
         return result;
     }
 
@@ -330,11 +329,7 @@ namespace ts.FindAllReferences {
         else {
             const sourceFile = entry.node.getSourceFile();
             const result: DocumentSpan = { textSpan: getTextSpan(entry.node, sourceFile), fileName: sourceFile.fileName };
-            if (entry.declaration) {
-                result.declarationSpan = isDeclarationNodeWithStartAndEnd(entry.declaration) ?
-                    getTextSpan(entry.declaration.start, sourceFile, entry.declaration.end) :
-                    getTextSpan(entry.declaration, sourceFile);
-            }
+            setDeclarationSpan(result, sourceFile, entry.declaration);
             return result;
         }
     }

src/services/goToDefinition.ts

@@ -281,12 +281,11 @@ namespace ts.GoToDefinition {
             containerKind: undefined!, // TODO: GH#18217
             containerName,
         };
-        const declarationNode = FindAllReferences.getDeclarationForDeclarationSpan(declaration);
-        if (declarationNode) {
-            result.declarationSpan = FindAllReferences.isDeclarationNodeWithStartAndEnd(declarationNode) ?
-                createTextSpanFromNode(declarationNode.start, sourceFile, declarationNode.end) :
-                createTextSpanFromNode(declarationNode, sourceFile);
-        }
+        FindAllReferences.setDeclarationSpan(
+            result,
+            sourceFile,
+            FindAllReferences.getDeclarationForDeclarationSpan(declaration)
+        );
         return result;
     }
 

tests/cases/fourslash/localGetReferences.ts

@@ -117,7 +117,7 @@
 ////array.forEach(
 ////
 ////
-////function([|{| "isWriteAccess": true, "isDefinition": true, "declarationRangeIndex": 43 |}str|]) {
+////function([|{| "isWriteAccess": true, "isDefinition": true |}str|]) {
 ////
 ////
 ////

tests/cases/fourslash/renameDestructuringFunctionParameter.ts

@@ -1,6 +1,6 @@
 /// <reference path='fourslash.ts' />
 
-////function f([|{[|{| "declarationRangeIndex": 0 |}a|]}: {[|{| "declarationRangeIndex": 2 |}a|]}|]) {
+////function f([|{[|{| "declarationRangeIndex": 0 |}a|]}: {[|a|]}|]) {
 ////    f({[|a|]});
 ////}
 

tests/cases/fourslash/renameJsSpecialAssignmentRhs1.ts

@@ -5,7 +5,7 @@
 ////    set: function (x) {
 ////        this._x = x;
 ////    },
-////    copy: function ([|{| "declarationRangeIndex": 0 |}x|]) {
+////    copy: function ([|x|]) {
 ////        this._x = [|x|].prop;
 ////    }
 ////};

tests/cases/fourslash/renameJsSpecialAssignmentRhs2.ts

@@ -5,7 +5,7 @@
 ////    set: function (x) {
 ////        this._x = x;
 ////    },
-////    copy: function ([|{| "declarationRangeIndex": 0 |}x|]) {
+////    copy: function ([|x|]) {
 ////        this._x = [|x|].prop;
 ////    }
 ////};

tests/cases/fourslash/renameThis.ts

@@ -1,6 +1,6 @@
 /// <reference path='fourslash.ts'/>
 
-////function f([|{| "declarationRangeIndex": 0 |}this|]) {
+////function f([|this|]) {
 ////    return [|this|];
 ////}
 ////this/**/;