private boolean getStringToReplace(
int textOffset,
int textEndOffset,
Document document,
FindModel findModel,
Ref<String> stringToReplace)
throws FindManager.MalformedReplacementStringException {
if (textOffset < 0 || textOffset >= document.getTextLength()) {
return false;
}
if (textEndOffset < 0 || textOffset > document.getTextLength()) {
return false;
}
FindManager findManager = FindManager.getInstance(myProject);
final CharSequence foundString =
document.getCharsSequence().subSequence(textOffset, textEndOffset);
PsiFile file = PsiDocumentManager.getInstance(myProject).getPsiFile(document);
FindResult findResult =
findManager.findString(
document.getCharsSequence(),
textOffset,
findModel,
file != null ? file.getVirtualFile() : null);
if (!findResult.isStringFound()) {
return false;
}
stringToReplace.set(
FindManager.getInstance(myProject)
.getStringToReplace(foundString.toString(), findModel, textOffset, document.getText()));
return true;
}
private void updateResults(final boolean allowedToChangedEditorSelection) {
final String text = myFindModel.getStringToFind();
if (text.length() == 0) {
nothingToSearchFor();
} else {
if (myFindModel.isRegularExpressions()) {
try {
Pattern.compile(text);
} catch (Exception e) {
setNotFoundBackground();
myClickToHighlightLabel.setVisible(false);
mySearchResults.clear();
myMatchInfoLabel.setText("Incorrect regular expression");
return;
}
}
final FindManager findManager = FindManager.getInstance(myProject);
if (allowedToChangedEditorSelection) {
findManager.setFindWasPerformed();
FindModel copy = new FindModel();
copy.copyFrom(myFindModel);
copy.setReplaceState(false);
findManager.setFindNextModel(copy);
}
if (myLivePreviewController != null) {
myLivePreviewController.updateInBackground(myFindModel, allowedToChangedEditorSelection);
}
}
}
private static int addToUsages(
@NotNull Document document,
@NotNull Processor<UsageInfo> consumer,
@NotNull FindModel findModel,
@NotNull final PsiFile psiFile,
@NotNull int[] offsetRef,
int maxUsages) {
int count = 0;
CharSequence text = document.getCharsSequence();
int textLength = document.getTextLength();
int offset = offsetRef[0];
Project project = psiFile.getProject();
FindManager findManager = FindManager.getInstance(project);
while (offset < textLength) {
FindResult result = findManager.findString(text, offset, findModel, psiFile.getVirtualFile());
if (!result.isStringFound()) break;
final SearchScope customScope = findModel.getCustomScope();
if (customScope instanceof LocalSearchScope) {
final TextRange range = new TextRange(result.getStartOffset(), result.getEndOffset());
if (!((LocalSearchScope) customScope).containsRange(psiFile, range)) break;
}
UsageInfo info = new FindResultUsageInfo(findManager, psiFile, offset, findModel, result);
if (!consumer.process(info)) {
throw new ProcessCanceledException();
}
count++;
final int prevOffset = offset;
offset = result.getEndOffset();
if (prevOffset == offset) {
// for regular expr the size of the match could be zero -> could be infinite loop in finding
// usages!
++offset;
}
if (maxUsages > 0 && count >= maxUsages) {
break;
}
}
offsetRef[0] = offset;
return count;
}
public void replaceInProject(@NotNull DataContext dataContext) {
final FindManager findManager = FindManager.getInstance(myProject);
final FindModel findModel = (FindModel) findManager.getFindInProjectModel().clone();
findModel.setReplaceState(true);
FindInProjectUtil.setDirectoryName(findModel, dataContext);
Editor editor = CommonDataKeys.EDITOR.getData(dataContext);
FindUtil.initStringToFindWithSelection(findModel, editor);
findManager.showFindDialog(
findModel,
new Runnable() {
@Override
public void run() {
final PsiDirectory psiDirectory =
FindInProjectUtil.getPsiDirectory(findModel, myProject);
if (!findModel.isProjectScope()
&& psiDirectory == null
&& findModel.getModuleName() == null
&& findModel.getCustomScope() == null) {
return;
}
UsageViewManager manager = UsageViewManager.getInstance(myProject);
if (manager == null) return;
findManager.getFindInProjectModel().copyFrom(findModel);
final FindModel findModelCopy = (FindModel) findModel.clone();
final UsageViewPresentation presentation =
FindInProjectUtil.setupViewPresentation(true, findModelCopy);
final FindUsagesProcessPresentation processPresentation =
FindInProjectUtil.setupProcessPresentation(myProject, true, presentation);
UsageSearcherFactory factory =
new UsageSearcherFactory(findModelCopy, psiDirectory, processPresentation);
searchAndShowUsages(
manager, factory, findModelCopy, presentation, processPresentation, findManager);
}
});
}
private static FindModel createDefaultFindModel(Project project, Editor editor) {
FindModel findModel = new FindModel();
findModel.copyFrom(FindManager.getInstance(project).getFindInFileModel());
if (editor.getSelectionModel().hasSelection()) {
String selectedText = editor.getSelectionModel().getSelectedText();
if (selectedText != null) {
findModel.setStringToFind(selectedText);
}
}
findModel.setPromptOnReplace(false);
return findModel;
}
public void showHistory(final boolean byClickingToolbarButton, JTextComponent textField) {
FeatureUsageTracker.getInstance().triggerFeatureUsed("find.recent.search");
FindSettings settings = FindSettings.getInstance();
String[] recent =
textField == mySearchTextComponent
? settings.getRecentFindStrings()
: settings.getRecentReplaceStrings();
final boolean toShowAd =
textField == mySearchTextComponent
&& textField.getText().isEmpty()
&& FindManager.getInstance(myProject).getPreviousFindModel() != null;
Utils.showCompletionPopup(
byClickingToolbarButton ? mySearchActionsToolbar1 : null,
new JBList((Object[]) ArrayUtil.reverseArray(recent)),
"Recent " + (textField == mySearchTextComponent ? "Searches" : "Replaces"),
textField,
toShowAd ? RestorePreviousSettingsAction.getAd() : null);
}
private void replaceWithPrompt(final ReplaceContext replaceContext) {
final List<Usage> _usages = replaceContext.getUsageView().getSortedUsages();
if (hasReadOnlyUsages(_usages)) {
WindowManager.getInstance()
.getStatusBar(myProject)
.setInfo(FindBundle.message("find.replace.occurrences.found.in.read.only.files.status"));
return;
}
final Usage[] usages = _usages.toArray(new Usage[_usages.size()]);
// usageView.expandAll();
for (int i = 0; i < usages.length; ++i) {
final Usage usage = usages[i];
final UsageInfo usageInfo = ((UsageInfo2UsageAdapter) usage).getUsageInfo();
final PsiElement elt = usageInfo.getElement();
if (elt == null) continue;
final PsiFile psiFile = elt.getContainingFile();
if (!psiFile.isWritable()) continue;
Runnable selectOnEditorRunnable =
new Runnable() {
@Override
public void run() {
final VirtualFile virtualFile = psiFile.getVirtualFile();
if (virtualFile != null
&& ApplicationManager.getApplication()
.runReadAction(
new Computable<Boolean>() {
@Override
public Boolean compute() {
return virtualFile.isValid() ? Boolean.TRUE : Boolean.FALSE;
}
})
.booleanValue()) {
if (usage.isValid()) {
usage.highlightInEditor();
replaceContext.getUsageView().selectUsages(new Usage[] {usage});
}
}
}
};
CommandProcessor.getInstance()
.executeCommand(
myProject,
selectOnEditorRunnable,
FindBundle.message("find.replace.select.on.editor.command"),
null);
String title = FindBundle.message("find.replace.found.usage.title", i + 1, usages.length);
int result;
try {
replaceUsage(
usage, replaceContext.getFindModel(), replaceContext.getExcludedSetCached(), true);
result =
FindManager.getInstance(myProject)
.showPromptDialog(replaceContext.getFindModel(), title);
} catch (FindManager.MalformedReplacementStringException e) {
markAsMalformedReplacement(replaceContext, usage);
result =
FindManager.getInstance(myProject)
.showMalformedReplacementPrompt(replaceContext.getFindModel(), title, e);
}
if (result == FindManager.PromptResult.CANCEL) {
return;
}
if (result == FindManager.PromptResult.SKIP) {
continue;
}
final int currentNumber = i;
if (result == FindManager.PromptResult.OK) {
final Ref<Boolean> success = Ref.create();
Runnable runnable =
new Runnable() {
@Override
public void run() {
success.set(replaceUsageAndRemoveFromView(usage, replaceContext));
}
};
CommandProcessor.getInstance()
.executeCommand(myProject, runnable, FindBundle.message("find.replace.command"), null);
if (closeUsageViewIfEmpty(replaceContext.getUsageView(), success.get())) {
return;
}
}
if (result == FindManager.PromptResult.ALL_IN_THIS_FILE) {
final int[] nextNumber = new int[1];
Runnable runnable =
new Runnable() {
@Override
public void run() {
int j = currentNumber;
boolean success = true;
for (; j < usages.length; j++) {
final Usage usage = usages[j];
final UsageInfo usageInfo = ((UsageInfo2UsageAdapter) usage).getUsageInfo();
final PsiElement elt = usageInfo.getElement();
if (elt == null) continue;
PsiFile otherPsiFile = elt.getContainingFile();
if (!otherPsiFile.equals(psiFile)) {
break;
}
if (!replaceUsageAndRemoveFromView(usage, replaceContext)) {
success = false;
}
}
closeUsageViewIfEmpty(replaceContext.getUsageView(), success);
nextNumber[0] = j;
}
};
CommandProcessor.getInstance()
.executeCommand(myProject, runnable, FindBundle.message("find.replace.command"), null);
//noinspection AssignmentToForLoopParameter
i = nextNumber[0] - 1;
}
if (result == FindManager.PromptResult.ALL_FILES) {
CommandProcessor.getInstance()
.executeCommand(
myProject,
new Runnable() {
@Override
public void run() {
final boolean success = replaceUsages(replaceContext, _usages);
closeUsageViewIfEmpty(replaceContext.getUsageView(), success);
}
},
FindBundle.message("find.replace.command"),
null);
break;
}
}
}