Support inferSeletion when extract variable (#1717)

Signed-off-by: Shi Chen <[email protected]>

Author: CsCherrYY

src/extension.ts

@@ -174,7 +174,7 @@ export function activate(context: ExtensionContext): Promise<ExtensionAPI> {
 						generateConstructorsPromptSupport: true,
 						generateDelegateMethodsPromptSupport: true,
 						advancedExtractRefactoringSupport: true,
-						inferSelectionSupport: ["extractMethod"],
+						inferSelectionSupport: ["extractMethod", "extractVariable"],
 						moveRefactoringSupport: true,
 						clientHoverProvider: true,
 						clientDocumentSymbolProvider: true,

src/refactorAction.ts

@@ -70,7 +70,10 @@ function registerApplyRefactorCommand(languageClient: LanguageClient, context: E
 
                     commandArguments.push(initializeIn);
                 }
-            } else if (command === 'extractMethod') {
+            } else if (command === 'extractMethod'
+                || command === 'extractVariableAllOccurrence'
+                || command === 'extractVariable'
+                || command === 'extractConstant') {
                 if (!params || !params.range) {
                     return;
                 }
@@ -81,12 +84,12 @@ function registerApplyRefactorCommand(languageClient: LanguageClient, context: E
                     });
                     const options: IExpressionItem[] = [];
                     for (const expression of expressions) {
-                        const extractMethodItem: IExpressionItem = {
+                        const extractItem: IExpressionItem = {
                             label: expression.name,
                             length: expression.length,
                             offset: expression.offset,
                         };
-                        options.push(extractMethodItem);
+                        options.push(extractItem);
                     }
                     let resultItem: IExpressionItem;
                     if (options.length === 1) {