@Override
public boolean pointsToTheSameElementAs(@NotNull final SmartPointerElementInfo other) {
if (other instanceof SelfElementInfo) {
SelfElementInfo otherInfo = (SelfElementInfo) other;
if (!getVirtualFile().equals(other.getVirtualFile()) || myType != otherInfo.myType)
return false;
Segment range1 = getPsiRange();
Segment range2 = otherInfo.getPsiRange();
return range1 != null
&& range2 != null
&& range1.getStartOffset() == range2.getStartOffset()
&& range1.getEndOffset() == range2.getEndOffset();
}
return areRestoredElementsEqual(other);
}
private void initGuardedBlocks(Place shreds) {
int origOffset = -1;
int curOffset = 0;
for (PsiLanguageInjectionHost.Shred shred : shreds) {
Segment hostRangeMarker = shred.getHostRangeMarker();
int start = shred.getRange().getStartOffset() + shred.getPrefix().length();
int end = shred.getRange().getEndOffset() - shred.getSuffix().length();
if (curOffset < start) {
RangeMarker guard = myNewDocument.createGuardedBlock(curOffset, start);
if (curOffset == 0 && shred == shreds.get(0)) guard.setGreedyToLeft(true);
String padding =
origOffset < 0
? ""
: myOrigDocument.getText().substring(origOffset, hostRangeMarker.getStartOffset());
guard.putUserData(REPLACEMENT_KEY, fixQuotes(padding));
}
curOffset = end;
origOffset = hostRangeMarker.getEndOffset();
}
if (curOffset < myNewDocument.getTextLength()) {
RangeMarker guard =
myNewDocument.createGuardedBlock(curOffset, myNewDocument.getTextLength());
guard.setGreedyToRight(true);
guard.putUserData(REPLACEMENT_KEY, "");
}
}
void setRange(@Nullable Segment range) {
if (range != null) {
myStartOffset = range.getStartOffset();
myEndOffset = range.getEndOffset();
} else {
myStartOffset = -1;
myEndOffset = -1;
}
}
@Override
public PsiElement restoreElement() {
Segment segment = getPsiRange();
if (segment == null) return null;
PsiFile file = restoreFile();
if (file == null || !file.isValid()) return null;
return findElementInside(file, segment.getStartOffset(), segment.getEndOffset(), myType);
}