type inference for closures in groovy1.8
[idea/community.git] / plugins / groovy / src / org / jetbrains / plugins / groovy / lang / resolve / GroovyStringLiteralManipulator.java
1 /*
2  * Copyright 2000-2009 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.jetbrains.plugins.groovy.lang.resolve;
17
18 import com.intellij.openapi.util.TextRange;
19 import com.intellij.openapi.util.text.StringUtil;
20 import com.intellij.psi.AbstractElementManipulator;
21 import com.intellij.psi.PsiElement;
22 import com.intellij.util.IncorrectOperationException;
23 import org.jetbrains.plugins.groovy.lang.psi.GroovyPsiElementFactory;
24 import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrExpression;
25 import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.literals.GrLiteral;
26
27 public class GroovyStringLiteralManipulator extends AbstractElementManipulator<GrLiteral> {
28   public GrLiteral handleContentChange(GrLiteral expr, TextRange range, String newContent) throws IncorrectOperationException {
29     if (!(expr.getValue() instanceof String)) throw new IncorrectOperationException("cannot handle content change");
30     String oldText = expr.getText();
31     newContent = StringUtil.escapeStringCharacters(newContent);
32     String newText = oldText.substring(0, range.getStartOffset()) + newContent + oldText.substring(range.getEndOffset());
33     final GrExpression newExpr = GroovyPsiElementFactory.getInstance(expr.getProject()).createExpressionFromText(newText);
34
35     PsiElement firstChild = expr.getFirstChild();
36     assert firstChild != null && firstChild.getNextSibling() == null;
37
38     PsiElement newElement = newExpr.getFirstChild();
39     assert newElement != null;
40     firstChild.replace(newElement);
41
42     return expr;
43   }
44
45   public TextRange getRangeInElement(final GrLiteral element) {
46     final String text = element.getText();
47     return getLiteralRange(text);
48   }
49
50   public static TextRange getLiteralRange(String text) {
51     if (text.length() > 6 && text.startsWith("\"\"\"") && text.endsWith("\"\"\"")) {
52       return new TextRange(3, text.length() - 3);
53     }
54     return new TextRange(1, Math.max(1, text.length() - 1));
55   }
56 }