Java ElementManipulators类代码示例

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Nullable
protected Pair<PsiElement, TextRange> getRangePair(final PsiFile file, final Editor editor) {
  final int offset = editor.getCaretModel().getOffset();
  final PsiLanguageInjectionHost host =
    PsiTreeUtil.getParentOfType(file.findElementAt(offset), PsiLanguageInjectionHost.class, false);
  if (host == null || ElementManipulators.getManipulator(host) == null) return null;
  final List<Pair<PsiElement, TextRange>> injections = InjectedLanguageManager.getInstance(host.getProject()).getInjectedPsiFiles(host);
  if (injections == null || injections.isEmpty()) return null;
  final int offsetInElement = offset - host.getTextRange().getStartOffset();
  final Pair<PsiElement, TextRange> rangePair = ContainerUtil.find(injections, new Condition<Pair<PsiElement, TextRange>>() {
    @Override
    public boolean value(final Pair<PsiElement, TextRange> pair) {
      return pair.second.containsRange(offsetInElement, offsetInElement);
    }
  });
  if (rangePair != null) {
    final Language language = rangePair.first.getContainingFile().getLanguage();
    final Object action = language.getUserData(EDIT_ACTION_AVAILABLE);
    if (action != null && action.equals(false)) return null;

    myLastLanguageName = language.getDisplayName();
  }
  return rangePair;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@NotNull
public List<TextRange> getInjectedArea(final PsiElement element) {
  final TextRange textRange = ElementManipulators.getValueTextRange(element);
  if (myCompiledValuePattern == null) {
    return Collections.singletonList(textRange);
  }
  else {
    final LiteralTextEscaper<? extends PsiLanguageInjectionHost> textEscaper =
            ((PsiLanguageInjectionHost)element).createLiteralTextEscaper();
    final StringBuilder sb = new StringBuilder();
    textEscaper.decode(textRange, sb);
    final List<TextRange> ranges = getMatchingRanges(myCompiledValuePattern.matcher(StringPattern.newBombedCharSequence(sb)), sb.length());
    return !ranges.isEmpty() ? ContainerUtil.map(ranges, new Function<TextRange, TextRange>() {
      public TextRange fun(TextRange s) {
        return new TextRange(textEscaper.getOffsetInHost(s.getStartOffset(), textRange), textEscaper.getOffsetInHost(s.getEndOffset(), textRange));
      }
    }) : Collections.<TextRange>emptyList();
  }
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
protected void checkValue( XmlAttributeValue value, XmlFile file, XmlRefCountHolder refHolder,
    XmlTag tag, ProblemsHolder holder )
{
    if( refHolder.isValidatable( tag.getParent() )
        && ((XmlAttribute)(value).getParent()).getName().equals( "xmi:id" ) )
    {

        int matches =
            StringUtils.countMatches( file.getText(), ((XmlAttribute)(value).getParent()).getText() );

        if( matches > 1 )
        {
            holder.registerProblem( value, "XMI Duplicate id", ProblemHighlightType.GENERIC_ERROR,
                ElementManipulators.getValueTextRange( value ), new DuplicateIdQuickFix() );
        }
    }
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@NotNull
@Override
public PsiReference[] createReferences(GenericDomValue value, final PsiElement element, final ConvertContext context) {
  String text = value.getStringValue();
  TextRange range = ElementManipulators.getValueTextRange(element);
  return new PsiReference[]{new MavenUrlPsiReference(element, text, range) {

    @NotNull
    @Override
    public Object[] getVariants() {
      Module module = context.getModule();

      if (module != null) {
        String name = MavenRepositoriesProvider.getInstance().getRepositoryUrl(getRepositoryId(context));
        if (!StringUtil.isEmptyOrSpaces(name)) return new Object[]{name};
      }
      return super.getVariants();
    }
  }};
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Nullable
protected Pair<PsiElement, TextRange> getRangePair(final PsiFile file, final Editor editor) {
  final int offset = editor.getCaretModel().getOffset();
  final PsiLanguageInjectionHost host =
          PsiTreeUtil.getParentOfType(file.findElementAt(offset), PsiLanguageInjectionHost.class, false);
  if (host == null || ElementManipulators.getManipulator(host) == null) return null;
  final List<Pair<PsiElement, TextRange>> injections = InjectedLanguageManager.getInstance(host.getProject()).getInjectedPsiFiles(host);
  if (injections == null || injections.isEmpty()) return null;
  final int offsetInElement = offset - host.getTextRange().getStartOffset();
  final Pair<PsiElement, TextRange> rangePair = ContainerUtil.find(injections, new Condition<Pair<PsiElement, TextRange>>() {
    @Override
    public boolean value(final Pair<PsiElement, TextRange> pair) {
      return pair.second.containsRange(offsetInElement, offsetInElement);
    }
  });
  if (rangePair != null) {
    final Language language = rangePair.first.getContainingFile().getLanguage();
    final Object action = language.getUserData(EDIT_ACTION_AVAILABLE);
    if (action != null && action.equals(false)) return null;

    myLastLanguageName = language.getDisplayName();
  }
  return rangePair;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@NotNull
public PsiReference[] createReferences(final GenericDomValue genericDomValue, final PsiElement element, final ConvertContext context) {
  final String s = genericDomValue.getStringValue();
  if (s != null) {
    final int offset = ElementManipulators.getOffsetInElement(element);
    final ArrayList<PsiReference> list = new ArrayList<PsiReference>();
    int pos = -1;
    while (true) {
      while (pos + 1 < s.length()) {
        if (!Character.isWhitespace(s.charAt(pos + 1))) {
          break;
        }
        pos++;
      }
      int nextPos = s.indexOf(',', pos + 1);
      if (nextPos == -1) {
        createReference(element, s.substring(pos + 1), pos + 1 + offset, list);
        break;
      }
      else {
        createReference(element, s.substring(pos + 1, nextPos), pos + 1 + offset, list);
        pos = nextPos;
      }
    }
    return list.toArray(new PsiReference[list.size()]);
  }
  return PsiReference.EMPTY_ARRAY;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
public void testManipulatorRegistered() {
  LanguageExtensionPoint[] extensions =
    Extensions.getExtensions(new ExtensionPointName<LanguageExtensionPoint>("com.intellij.lang.parserDefinition"));
  Set<String> classes = new HashSet<String>();
  List<String> failures = new ArrayList<String>();
  int total = 0;
  for (LanguageExtensionPoint e : extensions) {
    ParserDefinition definition = (ParserDefinition)e.getInstance();

    for (IElementType type : IElementType.enumerate(IElementType.TRUE)) {
      if (type instanceof ILeafElementType) continue;
      try {
        CompositeElement treeElement = ASTFactory.composite(type);
        total++;
        PsiElement element = treeElement instanceof PsiElement? (PsiElement)treeElement : definition.createElement(treeElement);
        if (element instanceof PsiLanguageInjectionHost && classes.add(element.getClass().getName())) {
          boolean ok = ElementManipulators.getManipulator(element) != null;
          System.out.println((ok ? "OK  " : "FAIL") + " " + element.getClass().getSimpleName() + " [" + definition.getClass().getSimpleName() + "]");
          if (!ok) failures.add(element.getClass().getName());
        }
      }
      catch (Throwable ignored) {
      }
    }
  }
  System.out.println("count: " + classes.size() + ", total: " + total);
  assertEmpty("PsiLanguageInjectionHost requires " + ElementManipulators.EP_NAME, failures);
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@NotNull
public static <T extends PsiElement> ElementManipulator<T> getManipulator(T currentElement){
  ElementManipulator<T> manipulator = ElementManipulators.getManipulator(currentElement);
  if (manipulator == null) {
    throw new IncorrectOperationException("Manipulator for this element is not defined: " + currentElement + "; " + currentElement.getClass());
  }
  return manipulator;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Override
public boolean createReferences(@NotNull final PsiElement psiElement, final @NotNull List<PsiReference> references, final boolean soft) {

  final TextRange range = ElementManipulators.getValueTextRange(psiElement);
  int offset = range.getStartOffset();
  int endOffset = range.getEndOffset();
  final String elementText = psiElement.getText();
  for (DynamicContextProvider provider: Extensions.getExtensions(DynamicContextProvider.EP_NAME)) {
    final int dynamicOffset = provider.getOffset(psiElement, offset, elementText);
    if (dynamicOffset == -1) {
      return false;
    } else if (dynamicOffset != offset) {
      offset = dynamicOffset;
    }
  }

  final int pos = getLastPosOfURL(offset, elementText);
  if (pos != -1 && pos < endOffset) {
    endOffset = pos;
  }
  try {
    final String text = elementText.substring(offset, endOffset);
    return createReferences(psiElement, offset, text, references, soft);
  } catch (StringIndexOutOfBoundsException e) {
    LOG.error("Cannot process string: '" + psiElement.getParent().getParent().getText() + "'", e);
    return false;
  }
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Override
public List<TextRange> select(PsiElement e, CharSequence editorText, int cursorOffset, Editor editor) {
  final IElementType type = e.getNode().getElementType();
  final StringLiteralLexer lexer = new StringLiteralLexer(type == SINGLE_QUOTED_STRING ? '\'' : '"', type, false, "/", false, false);
  final List<TextRange> result = new ArrayList<TextRange>();
  SelectWordUtil.addWordHonoringEscapeSequences(editorText, e.getTextRange(), cursorOffset, lexer, result);

  final PsiElement parent = e.getParent();
  result.add(ElementManipulators.getValueTextRange(parent).shiftRight(parent.getTextOffset()));
  return result;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Override
public PsiElement handleElementRename(String newElementName) throws IncorrectOperationException {
  final String canonicalText = getCanonicalText();

  final PsiElement element = ElementManipulators.getManipulator(myElement)
    .handleContentChange(myElement, getRangeInElement(), newElementName);
  myRange = new TextRange(myRange.getStartOffset(),myRange.getEndOffset() - (canonicalText.length() - newElementName.length()));
  return element;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
public PsiElement handleElementRename(String newElementName) throws IncorrectOperationException {
  return ElementManipulators.getManipulator(myElement).handleContentChange(
    myElement,
    getRangeInElement(),
    newElementName
  );
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
public static SchemaPrefix createJspPrefix(XmlAttributeValue element, String prefix) {
  TextRange range = ElementManipulators.getValueTextRange(element).shiftRight(element.getStartOffsetInParent());
  return new SchemaPrefix((XmlAttribute)element.getParent(), range, prefix) {
    @Override
    protected String getNamespace() {
      return ((XmlAttribute)getParent()).getParent().getAttributeValue("uri");
    }
  };
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Override
public PsiElement setName(@NotNull String name) throws IncorrectOperationException {
  final PsiElement nameElement = getNameElement();

  if (nameElement != null) {
    return ElementManipulators.getManipulator(nameElement).handleContentChange(
      nameElement,
      new TextRange(0,nameElement.getTextLength()),
      name
    );
  }
  return null;
}
 

import com.intellij.psi.ElementManipulators; //导入依赖的package包/类
@Override
public boolean createReferences(@NotNull final PsiElement psiElement, @NotNull final List<PsiReference> references, final boolean soft) {

  final TextRange range = ElementManipulators.getValueTextRange(psiElement);
  final String elementText = psiElement.getText();
  final int anchorOffset = elementText.indexOf('#');
  if (anchorOffset == -1) {
    return false;
  }
  final boolean dynamic = isDynamic(psiElement, anchorOffset + 1, elementText);
  if (dynamic) {
    return false;
  }

  FileReference fileReference = null;
  if (range.getStartOffset() != anchorOffset) {
    fileReference = findFileReference(references);
    if (fileReference == null || fileReference.resolve() == null) {
      return false;
    }
  }
  final int pos = elementText.indexOf('?', anchorOffset);
  final String anchor;
  try {
    int endIndex = pos != -1 ? pos : range.getEndOffset();
    if (endIndex <= anchorOffset) {
      endIndex = anchorOffset + 1;
    }
    anchor = elementText.substring(anchorOffset + 1, endIndex);
  }
  catch (StringIndexOutOfBoundsException e) {      
    throw new RuntimeException(elementText, e);
  }
  final AnchorReferenceImpl anchorReference = new AnchorReferenceImpl(anchor, fileReference, psiElement, anchorOffset + 1, soft);
  references.add(anchorReference);
  return false;
}