feat(57028): support inlining variables that are used in object literal shorthand

Author: a-tarasyuk

src/services/refactors/inlineVariable.ts

@@ -23,6 +23,7 @@ import {
     isNumericLiteral,
     isObjectLiteralExpression,
     isPropertyAccessExpression,
+    isShorthandPropertyAssignment,
     isTypeQueryNode,
     isVariableDeclarationInVariableStatement,
     isVariableStatement,
@@ -102,7 +103,6 @@ registerRefactor(refactorName, {
 
     getEditsForAction(context, actionName) {
         Debug.assert(actionName === refactorName, "Unexpected refactor invoked");
-
         const { file, program, startPosition } = context;
 
         // tryWithReferenceToken is true below since here we're already performing the refactor.
@@ -152,7 +152,7 @@ function getInliningInfo(file: SourceFile, startPosition: number, tryWithReferen
     }
 
     // Try finding the declaration and nodes to replace via the reference token.
-    if (tryWithReferenceToken) {
+    if (tryWithReferenceToken || isShorthandPropertyAssignment(parent)) {
         let definition = checker.resolveName(token.text, token, SymbolFlags.Value, /*excludeGlobals*/ false);
         definition = definition && checker.getMergedSymbol(definition);
 
@@ -191,7 +191,7 @@ function getReferenceNodes(declaration: InitializedVariableDeclaration, checker:
     const cannotInline = FindAllReferences.Core.eachSymbolReferenceInFile(declaration.name as Identifier, checker, file, ref => {
         // Only inline if all references are reads. Else we might end up with an invalid scenario like:
         // const y = x++ + 1 -> const y = 2++ + 1
-        if (FindAllReferences.isWriteAccessForReference(ref)) {
+        if (FindAllReferences.isWriteAccessForReference(ref) && !isShorthandPropertyAssignment(ref.parent)) {
             return true;
         }
 
@@ -216,7 +216,7 @@ function getReferenceNodes(declaration: InitializedVariableDeclaration, checker:
     return references.length === 0 || cannotInline ? undefined : references;
 }
 
-function getReplacementExpression(reference: Node, replacement: Expression): Expression {
+function getReplacementExpression(reference: Node, replacement: Expression) {
     // Make sure each reference site gets its own copy of the replacement node.
     replacement = getSynthesizedDeepClone(replacement);
     const { parent } = reference;
@@ -244,5 +244,9 @@ function getReplacementExpression(reference: Node, replacement: Expression): Exp
         return factory.createParenthesizedExpression(replacement);
     }
 
+    if (isIdentifier(reference) && isShorthandPropertyAssignment(parent)) {
+        return factory.createPropertyAssignment(reference, replacement);
+    }
+
     return replacement;
 }

tests/cases/fourslash/inlineVariableShorthandPropertyAssignment.ts

@@ -0,0 +1,13 @@
+/// <reference path="fourslash.ts" />
+
+////const foo = 1;
+////const bar = { /*a*/foo/*b*/ };
+
+goTo.select("a", "b");
+verify.refactorAvailable("Inline variable");
+edit.applyRefactor({
+    refactorName: "Inline variable",
+    actionName: "Inline variable",
+    actionDescription: "Inline variable",
+    newContent: "const bar = { foo: 1 };",
+});