-
Notifications
You must be signed in to change notification settings - Fork 408
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add Code Action: Generate Constructor
Signed-off-by: Jinbo Wang <[email protected]>
- Loading branch information
1 parent
b75859a
commit 7a70139
Showing
12 changed files
with
608 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
201 changes: 201 additions & 0 deletions
201
...jdt.ls.core/src/org/eclipse/jdt/ls/core/internal/handlers/GenerateConstructorHandler.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,201 @@ | ||
| /******************************************************************************* | ||
| * Copyright (c) 2019 Microsoft Corporation and others. | ||
| * All rights reserved. This program and the accompanying materials | ||
| * are made available under the terms of the Eclipse Public License v1.0 | ||
| * which accompanies this distribution, and is available at | ||
| * http://www.eclipse.org/legal/epl-v10.html | ||
| * | ||
| * Contributors: | ||
| * Microsoft Corporation - initial API and implementation | ||
| *******************************************************************************/ | ||
|
|
||
| package org.eclipse.jdt.ls.core.internal.handlers; | ||
|
|
||
| import java.util.ArrayList; | ||
| import java.util.Arrays; | ||
| import java.util.HashMap; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.Optional; | ||
|
|
||
| import org.eclipse.core.runtime.CoreException; | ||
| import org.eclipse.jdt.core.IField; | ||
| import org.eclipse.jdt.core.IJavaElement; | ||
| import org.eclipse.jdt.core.IType; | ||
| import org.eclipse.jdt.core.JavaModelException; | ||
| import org.eclipse.jdt.core.dom.AST; | ||
| import org.eclipse.jdt.core.dom.ASTNode; | ||
| import org.eclipse.jdt.core.dom.CompilationUnit; | ||
| import org.eclipse.jdt.core.dom.IMethodBinding; | ||
| import org.eclipse.jdt.core.dom.ITypeBinding; | ||
| import org.eclipse.jdt.core.dom.IVariableBinding; | ||
| import org.eclipse.jdt.core.dom.Modifier; | ||
| import org.eclipse.jdt.core.dom.VariableDeclarationFragment; | ||
| import org.eclipse.jdt.internal.corext.codemanipulation.AddCustomConstructorOperation; | ||
| import org.eclipse.jdt.internal.corext.codemanipulation.CodeGenerationSettings; | ||
| import org.eclipse.jdt.internal.corext.codemanipulation.StubUtility2Core; | ||
| import org.eclipse.jdt.internal.corext.dom.ASTNodes; | ||
| import org.eclipse.jdt.internal.corext.dom.Bindings; | ||
| import org.eclipse.jdt.internal.corext.dom.IASTSharedValues; | ||
| import org.eclipse.jdt.internal.corext.refactoring.util.RefactoringASTParser; | ||
| import org.eclipse.jdt.ls.core.internal.JavaLanguageServerPlugin; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.JdtDomModels.LspMethodBinding; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.JdtDomModels.LspVariableBinding; | ||
| import org.eclipse.jdt.ls.core.internal.preferences.Preferences; | ||
| import org.eclipse.jdt.ls.core.internal.text.correction.SourceAssistProcessor; | ||
| import org.eclipse.lsp4j.CodeActionParams; | ||
| import org.eclipse.lsp4j.WorkspaceEdit; | ||
| import org.eclipse.text.edits.TextEdit; | ||
|
|
||
| public class GenerateConstructorHandler { | ||
|
|
||
| public static CheckConstructorResponse checkConstructorStatus(CodeActionParams params) { | ||
| IType type = SourceAssistProcessor.getSelectionType(params); | ||
| return checkConstructorStatus(type); | ||
| } | ||
|
|
||
| public static CheckConstructorResponse checkConstructorStatus(IType type) { | ||
| if (type == null || type.getCompilationUnit() == null) { | ||
| return new CheckConstructorResponse(); | ||
| } | ||
|
|
||
| try { | ||
| RefactoringASTParser astParser = new RefactoringASTParser(IASTSharedValues.SHARED_AST_LEVEL); | ||
| CompilationUnit astRoot = astParser.parse(type.getCompilationUnit(), true); | ||
| ITypeBinding typeBinding = ASTNodes.getTypeBinding(astRoot, type); | ||
| if (typeBinding == null) { | ||
| return new CheckConstructorResponse(); | ||
| } | ||
|
|
||
| IMethodBinding[] superConstructors = getVisibleConstructors(astRoot, typeBinding); | ||
| Map<IJavaElement, IVariableBinding> fieldsToBindings = new HashMap<>(); | ||
| for (IVariableBinding field : typeBinding.getDeclaredFields()) { | ||
| if (field.isSynthetic() || Modifier.isStatic(field.getModifiers())) { | ||
| continue; | ||
| } | ||
|
|
||
| if (Modifier.isFinal(field.getModifiers())) { | ||
| ASTNode declaringNode = astRoot.findDeclaringNode(field); | ||
| // Do not add final fields which have been set in the <clinit> | ||
| if (declaringNode instanceof VariableDeclarationFragment && ((VariableDeclarationFragment) declaringNode).getInitializer() != null) { | ||
| continue; | ||
| } | ||
| } | ||
|
|
||
| fieldsToBindings.put(field.getJavaElement(), field); | ||
| } | ||
|
|
||
| List<IVariableBinding> fields = new ArrayList<>(); | ||
| // Sort the fields by the order in which they appear in the source or class file. | ||
| for (IField field : type.getFields()) { | ||
| IVariableBinding fieldBinding = fieldsToBindings.remove(field); | ||
| if (fieldBinding != null) { | ||
| fields.add(fieldBinding); | ||
| } | ||
| } | ||
|
|
||
| //@formatter:off | ||
| return new CheckConstructorResponse( | ||
| Arrays.stream(superConstructors).map(binding -> new LspMethodBinding(binding)).toArray(LspMethodBinding[]::new), | ||
| fields.stream().map(binding -> new LspVariableBinding(binding)).toArray(LspVariableBinding[]::new) | ||
| ); | ||
| //@formatter:on | ||
| } catch (JavaModelException e) { | ||
| JavaLanguageServerPlugin.logException("Failed to check Constructor status", e); | ||
| } | ||
|
|
||
| return new CheckConstructorResponse(); | ||
| } | ||
|
|
||
| private static IMethodBinding getObjectConstructor(AST ast) { | ||
| final ITypeBinding binding = ast.resolveWellKnownType("java.lang.Object"); | ||
| return Bindings.findMethodInType(binding, "Object", new ITypeBinding[0]); | ||
| } | ||
|
|
||
| private static IMethodBinding[] getVisibleConstructors(CompilationUnit astRoot, ITypeBinding typeBinding) { | ||
| if (typeBinding.isEnum()) { | ||
| return new IMethodBinding[] { getObjectConstructor(astRoot.getAST()) }; | ||
| } else { | ||
| return StubUtility2Core.getVisibleConstructors(typeBinding, false, true); | ||
| } | ||
| } | ||
|
|
||
| public static WorkspaceEdit generateConstructor(GenerateConstructorParams params) { | ||
| IType type = SourceAssistProcessor.getSelectionType(params.context); | ||
| TextEdit edit = generateConstructor(type, params.constructor, params.fields); | ||
| return (edit == null) ? null : SourceAssistProcessor.convertToWorkspaceEdit(type.getCompilationUnit(), edit); | ||
| } | ||
|
|
||
| public static TextEdit generateConstructor(IType type, LspMethodBinding constructor, LspVariableBinding[] fields) { | ||
| Preferences preferences = JavaLanguageServerPlugin.getPreferencesManager().getPreferences(); | ||
| CodeGenerationSettings settings = new CodeGenerationSettings(); | ||
| settings.createComments = preferences.isCodeGenerationTemplateGenerateComments(); | ||
| return generateConstructor(type, constructor, fields, settings); | ||
| } | ||
|
|
||
| public static TextEdit generateConstructor(IType type, LspMethodBinding constructor, LspVariableBinding[] fields, CodeGenerationSettings settings) { | ||
| if (type == null || type.getCompilationUnit() == null || constructor == null) { | ||
| return null; | ||
| } | ||
|
|
||
| try { | ||
| RefactoringASTParser astParser = new RefactoringASTParser(IASTSharedValues.SHARED_AST_LEVEL); | ||
| CompilationUnit astRoot = astParser.parse(type.getCompilationUnit(), true); | ||
| ITypeBinding typeBinding = ASTNodes.getTypeBinding(astRoot, type); | ||
| if (typeBinding != null) { | ||
| Map<String, IVariableBinding> fieldBindings = new HashMap<>(); | ||
| for (IVariableBinding binding : typeBinding.getDeclaredFields()) { | ||
| fieldBindings.put(binding.getKey(), binding); | ||
| } | ||
|
|
||
| IVariableBinding[] selectedFields = Arrays.stream(fields).map(field -> fieldBindings.get(field.bindingKey)).filter(binding -> binding != null).toArray(IVariableBinding[]::new); | ||
| IMethodBinding[] superConstructors = getVisibleConstructors(astRoot, typeBinding); | ||
| Optional<IMethodBinding> selectedSuperConstructor = Arrays.stream(superConstructors).filter(superConstructor -> compareConstructor(superConstructor, constructor)).findAny(); | ||
| if (selectedSuperConstructor.isPresent()) { | ||
| IMethodBinding superConstructor = selectedSuperConstructor.get(); | ||
| AddCustomConstructorOperation constructorOperation = new AddCustomConstructorOperation(astRoot, typeBinding, selectedFields, superConstructor, null, settings, false, false); | ||
| constructorOperation.setOmitSuper(superConstructor.getParameterTypes().length == 0); | ||
| constructorOperation.setVisibility(typeBinding.isEnum() ? Modifier.PRIVATE : Modifier.PUBLIC); | ||
| constructorOperation.run(null); | ||
| return constructorOperation.getResultingEdit(); | ||
| } | ||
| } | ||
| } catch (CoreException e) { | ||
| JavaLanguageServerPlugin.logException("Failed to generate constructor", e); | ||
| } | ||
|
|
||
| return null; | ||
| } | ||
|
|
||
| private static boolean compareConstructor(IMethodBinding binding, LspMethodBinding lspBinding) { | ||
| if (lspBinding == null) { | ||
| return binding == null; | ||
| } | ||
|
|
||
| if (binding == null) { | ||
| return lspBinding == null; | ||
| } | ||
|
|
||
| String[] parameters = Arrays.stream(binding.getParameterTypes()).map(type -> type.getName()).toArray(String[]::new); | ||
| return Arrays.equals(parameters, lspBinding.parameters); | ||
| } | ||
|
|
||
| public static class CheckConstructorResponse { | ||
| public LspMethodBinding[] constructors = new LspMethodBinding[0]; | ||
| public LspVariableBinding[] fields = new LspVariableBinding[0]; | ||
|
|
||
| public CheckConstructorResponse() { | ||
| } | ||
|
|
||
| public CheckConstructorResponse(LspMethodBinding[] constructors, LspVariableBinding[] fields) { | ||
| this.constructors = constructors; | ||
| this.fields = fields; | ||
| } | ||
| } | ||
|
|
||
| public static class GenerateConstructorParams { | ||
| public CodeActionParams context; | ||
| public LspMethodBinding constructor; | ||
| public LspVariableBinding[] fields; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.