protected void restoreState(@NotNull final V psiField) {
if (!ReadonlyStatusHandler.ensureDocumentWritable(
myProject, InjectedLanguageUtil.getTopLevelEditor(myEditor).getDocument())) return;
ApplicationManager.getApplication()
.runWriteAction(
() -> {
final PsiFile containingFile = psiField.getContainingFile();
final RangeMarker exprMarker = getExprMarker();
if (exprMarker != null) {
myExpr = restoreExpression(containingFile, psiField, exprMarker, myExprText);
}
if (myLocalMarker != null) {
final PsiElement refVariableElement =
containingFile.findElementAt(myLocalMarker.getStartOffset());
if (refVariableElement != null) {
final PsiElement parent = refVariableElement.getParent();
if (parent instanceof PsiNamedElement) {
((PsiNamedElement) parent).setName(myLocalName);
}
}
final V localVariable = getLocalVariable();
if (localVariable != null && localVariable.isPhysical()) {
myLocalVariable = localVariable;
final PsiElement nameIdentifier = localVariable.getNameIdentifier();
if (nameIdentifier != null) {
myLocalMarker = createMarker(nameIdentifier);
}
}
}
final List<RangeMarker> occurrenceMarkers = getOccurrenceMarkers();
for (int i = 0, occurrenceMarkersSize = occurrenceMarkers.size();
i < occurrenceMarkersSize;
i++) {
RangeMarker marker = occurrenceMarkers.get(i);
if (getExprMarker() != null
&& marker.getStartOffset() == getExprMarker().getStartOffset()
&& myExpr != null) {
myOccurrences[i] = myExpr;
continue;
}
final E psiExpression =
restoreExpression(
containingFile,
psiField,
marker,
getLocalVariable() != null ? myLocalName : myExprText);
if (psiExpression != null) {
myOccurrences[i] = psiExpression;
}
}
if (myExpr != null && myExpr.isPhysical()) {
myExprMarker = createMarker(myExpr);
}
myOccurrenceMarkers = null;
deleteTemplateField(psiField);
});
}
private static void update(
@NotNull PsiElement[] elements,
@NotNull PairFunction<PsiElement[], List<FileDescriptor>, Boolean> fun) {
List<PsiElement> candidates = new ArrayList<PsiElement>(elements.length);
List<FileDescriptor> files = new ArrayList<FileDescriptor>(elements.length);
final Set<String> names = new HashSet<String>();
for (PsiElement element : elements) {
if (element instanceof PsiNamedElement) {
names.add(((PsiNamedElement) element).getName());
}
}
for (PsiElement element : elements) {
PsiFile file = getContainingFile(element);
if (file == null) continue;
final PsiElement parent = element.getParent();
files.add(new FileDescriptor(file, names.size() > 1 || parent == file ? element : parent));
candidates.add(element);
}
fun.fun(PsiUtilCore.toPsiElementArray(candidates), files);
}
public void setData(
PsiElement[] psiElements,
String targetPackageName,
final PsiDirectory initialTargetDirectory,
boolean isTargetDirectoryFixed,
boolean searchInComments,
boolean searchForTextOccurences,
String helpID) {
myInitialTargetDirectory = initialTargetDirectory;
myTargetDirectoryFixed = isTargetDirectoryFixed;
if (targetPackageName.length() != 0) {
myWithBrowseButtonReference.prependItem(targetPackageName);
myClassPackageChooser.prependItem(targetPackageName);
}
String nameFromCallback =
myMoveCallback instanceof MoveClassesOrPackagesCallback
? ((MoveClassesOrPackagesCallback) myMoveCallback).getElementsToMoveName()
: null;
if (nameFromCallback != null) {
myNameLabel.setText(nameFromCallback);
} else if (psiElements.length == 1) {
PsiElement firstElement = psiElements[0];
if (firstElement instanceof PsiClass) {
LOG.assertTrue(!MoveClassesOrPackagesImpl.isClassInnerOrLocal((PsiClass) firstElement));
} else {
PsiElement parent = firstElement.getParent();
LOG.assertTrue(parent != null);
}
myNameLabel.setText(
RefactoringBundle.message(
"move.single.class.or.package.name.label",
UsageViewUtil.getType(firstElement),
UsageViewUtil.getLongName(firstElement)));
} else if (psiElements.length > 1) {
myNameLabel.setText(
psiElements[0] instanceof PsiClass
? RefactoringBundle.message("move.specified.classes")
: RefactoringBundle.message("move.specified.packages"));
}
selectInitialCard();
myCbSearchInComments.setSelected(searchInComments);
myCbSearchTextOccurences.setSelected(searchForTextOccurences);
((DestinationFolderComboBox) myDestinationFolderCB)
.setData(
myProject,
myInitialTargetDirectory,
new Pass<String>() {
@Override
public void pass(String s) {
setErrorText(s);
}
},
myClassPackageChooser.getChildComponent());
UIUtil.setEnabled(
myTargetPanel,
getSourceRoots().length > 0 && isMoveToPackage() && !isTargetDirectoryFixed,
true);
validateButtons();
myHelpID = helpID;
}
public void setData(
PsiElement[] psiElements,
String targetPackageName,
PsiDirectory initialTargetDirectory,
boolean isTargetDirectoryFixed,
boolean suggestToMoveToAnotherRoot,
boolean searchInComments,
boolean searchForTextOccurences,
String helpID) {
myTargetDirectoryFixed = isTargetDirectoryFixed;
mySuggestToMoveToAnotherRoot = suggestToMoveToAnotherRoot;
if (targetPackageName.length() != 0) {
myWithBrowseButtonReference.prependItem(targetPackageName);
myClassPackageChooser.prependItem(targetPackageName);
}
String nameFromCallback =
myMoveCallback instanceof MoveClassesOrPackagesCallback
? ((MoveClassesOrPackagesCallback) myMoveCallback).getElementsToMoveName()
: null;
if (nameFromCallback != null) {
myNameLabel.setText(nameFromCallback);
} else if (psiElements.length == 1) {
PsiElement firstElement = psiElements[0];
if (firstElement instanceof PsiClass) {
LOG.assertTrue(!MoveClassesOrPackagesImpl.isClassInnerOrLocal((PsiClass) firstElement));
} else {
PsiElement parent = firstElement.getParent();
LOG.assertTrue(parent != null);
}
myNameLabel.setText(
RefactoringBundle.message(
"move.single.class.or.package.name.label",
UsageViewUtil.getType(firstElement),
UsageViewUtil.getLongName(firstElement)));
} else if (psiElements.length > 1) {
myNameLabel.setText(
psiElements[0] instanceof PsiClass
? RefactoringBundle.message("move.specified.classes")
: RefactoringBundle.message("move.specified.packages"));
}
selectInitialCard();
myCbSearchInComments.setSelected(searchInComments);
myCbSearchTextOccurences.setSelected(searchForTextOccurences);
if (initialTargetDirectory != null
&& JavaMoveClassesOrPackagesHandler.packageHasMultipleDirectoriesInModule(
myProject, initialTargetDirectory)) {
final ProjectFileIndex fileIndex = ProjectRootManager.getInstance(myProject).getFileIndex();
final Set<VirtualFile> initialRoots = new HashSet<VirtualFile>();
collectSourceRoots(psiElements, fileIndex, initialRoots);
if (initialRoots.size() > 1) {
initialTargetDirectory = null;
}
}
((DestinationFolderComboBox) myDestinationFolderCB)
.setData(
myProject,
initialTargetDirectory,
new Pass<String>() {
@Override
public void pass(String s) {
setErrorText(s);
}
},
myHavePackages
? myWithBrowseButtonReference.getChildComponent()
: myClassPackageChooser.getChildComponent());
UIUtil.setEnabled(
myTargetPanel,
!getSourceRoots().isEmpty() && isMoveToPackage() && !isTargetDirectoryFixed,
true);
validateButtons();
myHelpID = helpID;
}
@Override
public JComponent getPreviewComponent(@NotNull PsiElement element) {
final PsiNewExpression psiNewExpression =
PsiTreeUtil.getParentOfType(element, PsiNewExpression.class);
if (psiNewExpression != null) {
final PsiJavaCodeReferenceElement referenceElement =
PsiTreeUtil.getChildOfType(psiNewExpression, PsiJavaCodeReferenceElement.class);
if (referenceElement != null) {
final PsiReference reference = referenceElement.getReference();
if (reference != null) {
final PsiElement psiElement = reference.resolve();
if (psiElement instanceof PsiClass
&& "java.awt.Color".equals(((PsiClass) psiElement).getQualifiedName())) {
final PsiExpressionList argumentList = psiNewExpression.getArgumentList();
if (argumentList != null) {
final PsiExpression[] expressions = argumentList.getExpressions();
int[] values = ArrayUtil.newIntArray(expressions.length);
float[] values2 = new float[expressions.length];
int i = 0;
int j = 0;
final PsiConstantEvaluationHelper helper =
JavaPsiFacade.getInstance(element.getProject()).getConstantEvaluationHelper();
for (final PsiExpression each : expressions) {
final Object o = helper.computeConstantExpression(each);
if (o instanceof Integer) {
values[i] = ((Integer) o).intValue();
if (expressions.length != 1) {
values[i] = values[i] > 255 ? 255 : values[i] < 0 ? 0 : values[i];
}
i++;
} else if (o instanceof Float) {
values2[j] = ((Float) o).floatValue();
values2[j] = values2[j] > 1 ? 1 : values2[j] < 0 ? 0 : values2[j];
j++;
}
}
Color c = null;
if (i == expressions.length) {
if (i == 1 && values[0] > 255) {
c = new Color(values[0]);
} else {
switch (values.length) {
case 1:
c = new Color(values[0]);
break;
case 3:
c = new Color(values[0], values[1], values[2]);
break;
case 4:
c = new Color(values[0], values[1], values[2], values[3]);
break;
default:
break;
}
}
} else if (j == expressions.length) {
switch (values2.length) {
case 3:
c = new Color(values2[0], values2[1], values2[2]);
break;
case 4:
c = new Color(values2[0], values2[1], values2[2], values2[3]);
break;
default:
break;
}
}
if (c != null) {
return new ColorPreviewComponent(c);
}
}
}
}
}
}
if (ColorChooserIntentionAction.isInsideDecodeOrGetColorMethod(element)) {
final String color = StringUtil.unquoteString(element.getText());
try {
return new ColorPreviewComponent(Color.decode(color));
} catch (NumberFormatException ignore) {
}
}
if (PlatformPatterns.psiElement(PsiIdentifier.class)
.withParent(PlatformPatterns.psiElement(PsiReferenceExpression.class))
.accepts(element)) {
final PsiReference reference = element.getParent().getReference();
if (reference != null) {
final PsiElement psiElement = reference.resolve();
if (psiElement instanceof PsiField) {
if ("java.awt.Color"
.equals(((PsiField) psiElement).getContainingClass().getQualifiedName())) {
final String colorName =
((PsiField) psiElement).getName().toLowerCase().replace("_", "");
final String hex = ColorSampleLookupValue.getHexCodeForColorName(colorName);
return new ColorPreviewComponent(Color.decode("0x" + hex.substring(1)));
}
}
}
}
if (PlatformPatterns.psiElement()
.withParent(PlatformPatterns.psiElement(PsiLiteralExpression.class))
.accepts(element)) {
final PsiLiteralExpression psiLiteralExpression = (PsiLiteralExpression) element.getParent();
if (psiLiteralExpression != null) {
return ImagePreviewComponent.getPreviewComponent(psiLiteralExpression);
}
}
return null;
}
