Examples with TextEdit org.eclipse.text.edits.TextEdit used on opensource projects

Example 66 with TextEdit

use of org.eclipse.text.edits.TextEdit in project liferay-ide by liferay.

the class NewLiferayModuleProjectOpMethods method addProperties.

@SuppressWarnings("unchecked")
public static boolean addProperties(File dest, List<String> properties) throws Exception {
    if (ListUtil.isEmpty(properties)) {
        return false;
    }
    try {
        ASTParser parser = ASTParser.newParser(AST.JLS8);
        String readContents = FileUtil.readContents(dest, true);
        parser.setSource(readContents.toCharArray());
        parser.setKind(ASTParser.K_COMPILATION_UNIT);
        parser.setResolveBindings(true);
        CompilationUnit cu = (CompilationUnit) parser.createAST(new NullProgressMonitor());
        cu.recordModifications();
        Document document = new Document(new String(readContents));
        cu.accept(new ASTVisitor() {

            @Override
            public boolean visit(NormalAnnotation node) {
                String qualifiedName = node.getTypeName().getFullyQualifiedName();
                if (qualifiedName.equals("Component")) {
                    ASTRewrite rewrite = ASTRewrite.create(cu.getAST());
                    AST ast = cu.getAST();
                    List<ASTNode> values = node.values();
                    boolean hasProperty = false;
                    for (ASTNode astNode : values) {
                        if (astNode instanceof MemberValuePair) {
                            MemberValuePair pairNode = (MemberValuePair) astNode;
                            String fullQualifiedName = pairNode.getName().getFullyQualifiedName();
                            if (fullQualifiedName.equals("property")) {
                                Expression express = pairNode.getValue();
                                if (express instanceof ArrayInitializer) {
                                    ListRewrite lrw = rewrite.getListRewrite(express, ArrayInitializer.EXPRESSIONS_PROPERTY);
                                    ArrayInitializer initializer = (ArrayInitializer) express;
                                    List<ASTNode> expressions = initializer.expressions();
                                    ASTNode propertyNode = null;
                                    for (int i = properties.size() - 1; i >= 0; i--) {
                                        StringLiteral stringLiteral = ast.newStringLiteral();
                                        stringLiteral.setLiteralValue(properties.get(i));
                                        if (ListUtil.isNotEmpty(expressions)) {
                                            propertyNode = expressions.get(expressions.size() - 1);
                                            lrw.insertAfter(stringLiteral, propertyNode, null);
                                        } else {
                                            lrw.insertFirst(stringLiteral, null);
                                        }
                                    }
                                }
                                hasProperty = true;
                            }
                        }
                    }
                    if (hasProperty == false) {
                        ListRewrite clrw = rewrite.getListRewrite(node, NormalAnnotation.VALUES_PROPERTY);
                        ASTNode lastNode = values.get(values.size() - 1);
                        ArrayInitializer newArrayInitializer = ast.newArrayInitializer();
                        MemberValuePair propertyMemberValuePair = ast.newMemberValuePair();
                        propertyMemberValuePair.setName(ast.newSimpleName("property"));
                        propertyMemberValuePair.setValue(newArrayInitializer);
                        clrw.insertBefore(propertyMemberValuePair, lastNode, null);
                        ListRewrite newLrw = rewrite.getListRewrite(newArrayInitializer, ArrayInitializer.EXPRESSIONS_PROPERTY);
                        for (String property : properties) {
                            StringLiteral stringLiteral = ast.newStringLiteral();
                            stringLiteral.setLiteralValue(property);
                            newLrw.insertAt(stringLiteral, 0, null);
                        }
                    }
                    try (OutputStream fos = Files.newOutputStream(dest.toPath())) {
                        TextEdit edits = rewrite.rewriteAST(document, null);
                        edits.apply(document);
                        fos.write(document.get().getBytes());
                        fos.flush();
                    } catch (Exception e) {
                        ProjectCore.logError(e);
                    }
                }
                return super.visit(node);
            }
        });
        return true;
    } catch (Exception e) {
        ProjectCore.logError("error when adding properties to " + dest.getAbsolutePath(), e);
        return false;
    }
}

Example 67 with TextEdit

use of org.eclipse.text.edits.TextEdit in project eclipse.jdt.ls by eclipse.

the class OrganizeImportsCommand method organizeImportsInCompilationUnit.

public void organizeImportsInCompilationUnit(ICompilationUnit unit, WorkspaceEdit rootEdit) {
    try {
        InnovationContext context = new InnovationContext(unit, 0, unit.getBuffer().getLength() - 1);
        CUCorrectionProposal proposal = new CUCorrectionProposal("OrganizeImports", unit, IProposalRelevance.ORGANIZE_IMPORTS) {

            @Override
            protected void addEdits(IDocument document, TextEdit editRoot) throws CoreException {
                CompilationUnit astRoot = context.getASTRoot();
                OrganizeImportsOperation op = new OrganizeImportsOperation(unit, astRoot, true, false, true, null);
                editRoot.addChild(op.createTextEdit(null));
            }
        };
        addWorkspaceEdit(unit, proposal, rootEdit);
    } catch (CoreException e) {
        JavaLanguageServerPlugin.logException("Problem organize imports ", e);
    }
}

Example 68 with TextEdit

use of org.eclipse.text.edits.TextEdit in project eclipse.jdt.ls by eclipse.

the class CompletionProposalReplacementProvider method addImports.

/**
 * Adds imports collected by importRewrite to item
 * @param item
 */
private void addImports(List<org.eclipse.lsp4j.TextEdit> additionalEdits) {
    if (this.importRewrite != null) {
        try {
            TextEdit edit = this.importRewrite.rewriteImports(new NullProgressMonitor());
            TextEditConverter converter = new TextEditConverter(this.compilationUnit, edit);
            additionalEdits.addAll(converter.convert());
        } catch (CoreException e) {
            JavaLanguageServerPlugin.logException("Error adding imports", e);
        }
    }
}

Example 69 with TextEdit

use of org.eclipse.text.edits.TextEdit in project eclipse.jdt.ls by eclipse.

the class ImportOrganizeTest method testStaticImports1.

@Test
public void testStaticImports1() throws Exception {
    IPackageFragment pack1 = fSourceFolder.createPackageFragment("pack1", false, null);
    StringBuilder buf = new StringBuilder();
    buf.append("package pack1;\n");
    buf.append("\n");
    buf.append("import static java.lang.System.out;\n");
    buf.append("\n");
    buf.append("public class C {\n");
    buf.append("    public int foo() {\n");
    buf.append("        out.print(File.separator);\n");
    buf.append("    }\n");
    buf.append("}\n");
    ICompilationUnit cu = pack1.createCompilationUnit("C.java", buf.toString(), false, null);
    String[] order = new String[] { "java", "pack", "#java" };
    IChooseImportQuery query = createQuery("C", new String[] {}, new int[] {});
    OrganizeImportsOperation op = createOperation(cu, order, false, true, true, query);
    TextEdit edit = op.createTextEdit(new NullProgressMonitor());
    IDocument document = new Document(cu.getSource());
    edit.apply(document);
    try {
        cu.becomeWorkingCopy(new NullProgressMonitor());
        cu.getBuffer().setContents(document.get());
        cu.reconcile(ICompilationUnit.NO_AST, true, null, new NullProgressMonitor());
        buf = new StringBuilder();
        buf.append("package pack1;\n");
        buf.append("\n");
        buf.append("import java.io.File;\n");
        buf.append("\n");
        buf.append("import static java.lang.System.out;\n");
        buf.append("\n");
        buf.append("public class C {\n");
        buf.append("    public int foo() {\n");
        buf.append("        out.print(File.separator);\n");
        buf.append("    }\n");
        buf.append("}\n");
        assertTrue(cu.getSource().equals(buf.toString()));
    } finally {
        cu.discardWorkingCopy();
    }
}

Example 70 with TextEdit

use of org.eclipse.text.edits.TextEdit in project eclipse.jdt.ls by eclipse.

the class ImportOrganizeTest method testStaticImports2.

@Test
public void testStaticImports2() throws Exception {
    IPackageFragment pack1 = fSourceFolder.createPackageFragment("pack1", false, null);
    StringBuilder buf = new StringBuilder();
    buf.append("package pack1;\n");
    buf.append("\n");
    buf.append("import static java.io.File.*;\n");
    buf.append("\n");
    buf.append("public class C {\n");
    buf.append("    public String foo() {\n");
    buf.append("        return pathSeparator + separator + File.separator;\n");
    buf.append("    }\n");
    buf.append("}\n");
    ICompilationUnit cu = pack1.createCompilationUnit("C.java", buf.toString(), false, null);
    String[] order = new String[] { "#java.io.File", "java", "pack" };
    IChooseImportQuery query = createQuery("C", new String[] {}, new int[] {});
    OrganizeImportsOperation op = createOperation(cu, order, false, true, true, query);
    TextEdit edit = op.createTextEdit(new NullProgressMonitor());
    IDocument document = new Document(cu.getSource());
    edit.apply(document);
    try {
        cu.becomeWorkingCopy(new NullProgressMonitor());
        cu.getBuffer().setContents(document.get());
        cu.reconcile(ICompilationUnit.NO_AST, true, null, new NullProgressMonitor());
        buf = new StringBuilder();
        buf.append("package pack1;\n");
        buf.append("\n");
        buf.append("import static java.io.File.pathSeparator;\n");
        buf.append("import static java.io.File.separator;\n");
        buf.append("\n");
        buf.append("import java.io.File;\n");
        buf.append("\n");
        buf.append("public class C {\n");
        buf.append("    public String foo() {\n");
        buf.append("        return pathSeparator + separator + File.separator;\n");
        buf.append("    }\n");
        buf.append("}\n");
        assertTrue(cu.getSource().equals(buf.toString()));
    } finally {
        cu.discardWorkingCopy();
    }
}