public static boolean processImplicitImports(
@NotNull PsiScopeProcessor processor,
ResolveState state,
PsiElement lastParent,
PsiElement place,
@NotNull GroovyFile file) {
JavaPsiFacade facade = JavaPsiFacade.getInstance(file.getProject());
final DelegatingScopeProcessor packageSkipper =
new DelegatingScopeProcessor(processor) {
@Override
public boolean execute(@NotNull PsiElement element, ResolveState state) {
if (element instanceof PsiPackage) return true;
return super.execute(element, state);
}
};
for (final String implicitlyImported : getImplicitlyImportedPackages(file)) {
PsiPackage aPackage = facade.findPackage(implicitlyImported);
if (aPackage == null) continue;
if (!aPackage.processDeclarations(packageSkipper, state, lastParent, place)) {
return false;
}
}
GroovyPsiManager groovyPsiManager = GroovyPsiManager.getInstance(file.getProject());
for (String implicitlyImportedClass : GroovyFileBase.IMPLICITLY_IMPORTED_CLASSES) {
PsiClass clazz =
groovyPsiManager.findClassWithCache(implicitlyImportedClass, file.getResolveScope());
if (clazz != null && !ResolveUtil.processElement(processor, clazz, state)) return false;
}
return true;
}
@Override
public void tuneConfiguration(
@NotNull GroovyFile file,
@NotNull GroovyScriptRunConfiguration configuration,
Location location) {
List<String> tasks = getTasksTarget(location);
if (tasks != null) {
String s = StringUtil.join(tasks, " ");
configuration.setScriptParameters(s);
configuration.setName("gradle:" + s);
}
RunManagerEx.disableTasks(
file.getProject(),
configuration,
CompileStepBeforeRun.ID,
CompileStepBeforeRunNoErrorCheck.ID);
}
@Override
public GlobalSearchScope patchResolveScope(
@NotNull GroovyFile file, @NotNull GlobalSearchScope baseScope) {
final Module module = ModuleUtilCore.findModuleForPsiElement(file);
final GradleInstallationManager libraryManager =
ServiceManager.getService(GradleInstallationManager.class);
if (module != null) {
if (libraryManager.getGradleHome(module) != null) {
return baseScope;
}
}
final Collection<VirtualFile> files = libraryManager.getClassRoots(file.getProject());
if (files == null || files.isEmpty()) {
return baseScope;
}
GlobalSearchScope result = baseScope;
for (final VirtualFile root : files) {
result = result.uniteWith(new NonClasspathDirectoryScope(root));
}
return result;
}
public void doCollectInformation(ProgressIndicator progress) {
if (!(myFile instanceof GroovyFile)) return;
GroovyFile groovyFile = (GroovyFile) myFile;
GroovyImportsTracker importsTracker = GroovyImportsTracker.getInstance(groovyFile.getProject());
myUnusedImports = importsTracker.getUnusedImportStatements(groovyFile);
}
protected GrKeywordAndDeclarationHighlighter(GroovyFile file, Document document) {
super(file.getProject(), document);
myFile = file;
}
public JavaCodeFragment createCodeFragment(
TextWithImports textWithImports, PsiElement context, Project project) {
String text = textWithImports.getText();
String imports = textWithImports.getImports();
final GroovyPsiElementFactory factory = GroovyPsiElementFactory.getInstance(project);
final GroovyFile toEval = factory.createGroovyFile(text, false, context);
final Set<String> namesList = new HashSet<String>();
final GrClosableBlock closure = PsiTreeUtil.getParentOfType(context, GrClosableBlock.class);
final Set<String> valList = new HashSet<String>();
toEval.accept(
new GroovyRecursiveElementVisitor() {
public void visitReferenceExpression(GrReferenceExpression referenceExpression) {
super.visitReferenceExpression(referenceExpression);
PsiElement resolved = referenceExpression.resolve();
if (resolved instanceof PsiMethod
&& "getDelegate".equals(((PsiMethod) resolved).getName())
&& closure != null) {
replaceWithReference(referenceExpression, "owner");
return;
}
if (resolved instanceof GrField && !referenceExpression.isQualified()) {
replaceWithReference(
referenceExpression,
(closure == null ? "delegate" : "owner")
+ "."
+ referenceExpression.getReferenceName());
return;
}
if (resolved instanceof GrVariableBase
&& !(resolved instanceof GrField)
&& !PsiTreeUtil.isAncestor(toEval, resolved, false)) {
final String name = ((GrVariableBase) resolved).getName();
if (resolved instanceof ClosureSyntheticParameter
&& PsiTreeUtil.isAncestor(
toEval, ((ClosureSyntheticParameter) resolved).getClosure(), false)) {
return;
}
namesList.add(name);
if (closure != null
&& PsiTreeUtil.findCommonParent(resolved, closure) != closure
&& !(resolved instanceof ClosureSyntheticParameter)) {
// Evaluating inside closure for outer variable definitions
// All non-local variables are accessed by references
valList.add("this." + name);
} else {
valList.add(name);
}
}
}
@Override
public void visitThisExpression(final GrThisReferenceExpression thisExpression) {
super.visitThisExpression(thisExpression);
replaceWithReference(thisExpression, closure == null ? "delegate" : "owner");
}
@Override
public void visitSuperExpression(final GrSuperReferenceExpression superExpression) {
super.visitSuperExpression(superExpression);
replaceWithReference(superExpression, closure == null ? "delegate" : "owner");
}
private void replaceWithReference(GrExpression expr, final String exprText) {
final GroovyPsiElementFactory factory =
GroovyPsiElementFactory.getInstance(expr.getProject());
visitReferenceExpression(
(GrReferenceExpression)
expr.replaceWithExpression(factory.createExpressionFromText(exprText), false));
}
public void visitCodeReferenceElement(GrCodeReferenceElement refElement) {
if (refElement.getQualifier() != null) {
super.visitCodeReferenceElement(refElement);
} else {
PsiElement resolved = refElement.resolve();
if (resolved instanceof PsiClass) {
String qName = ((PsiClass) resolved).getQualifiedName();
if (qName != null) {
int dotIndex = qName.lastIndexOf(".");
if (dotIndex < 0) return;
String packageName = qName.substring(0, dotIndex);
refElement.setQualifier(factory.createReferenceElementFromText(packageName));
}
}
}
}
});
text = toEval.getText();
String[] names = namesList.toArray(new String[namesList.size()]);
String[] vals = valList.toArray(new String[valList.size()]);
PsiClass contextClass = PsiUtil.getContextClass(context);
boolean isStatic = isStaticContext(context);
StringBuffer javaText = new StringBuffer();
javaText.append("groovy.lang.MetaClass mc;\n");
javaText.append("java.lang.Class clazz;\n");
if (!isStatic) {
javaText.append("clazz = ((java.lang.Object)this).getClass();\n");
javaText.append("mc = ((groovy.lang.GroovyObject)this).getMetaClass();\n");
} else {
javaText
.append("clazz = java.lang.Class.forName(\"")
.append(contextClass.getQualifiedName())
.append("\");\n");
javaText.append(
"mc = groovy.lang.GroovySystem.getMetaClassRegistry().getMetaClass(clazz);\n");
}
javaText.append(createProperty(stripImports(text, toEval), imports, names));
javaText.append(
"groovy.lang.ExpandoMetaClass emc = new groovy.lang.ExpandoMetaClass(clazz);\n");
if (!isStatic) {
javaText.append("emc.setProperty(\"").append(EVAL_NAME).append("\", closure);\n");
javaText.append("((groovy.lang.GroovyObject)this).setMetaClass(emc);\n");
} else {
javaText
.append("((groovy.lang.GroovyObject)emc.getProperty(\"static\")).setProperty(\"")
.append(EVAL_NAME)
.append("\", closure);\n");
javaText.append(
"groovy.lang.GroovySystem.getMetaClassRegistry().setMetaClass(clazz, emc);\n");
}
javaText.append("emc.initialize();\n");
javaText.append(unwrapVals(vals));
if (!isStatic) {
javaText
.append("java.lang.Object res = ((groovy.lang.MetaClassImpl)emc).invokeMethod(this, \"")
.append(EVAL_NAME)
.append("\", ")
.append("resVals")
.append(");\n");
javaText.append(
"((groovy.lang.GroovyObject)this).setMetaClass(mc);"); // try/finally is not supported
} else {
javaText
.append(
"java.lang.Object res = ((groovy.lang.MetaClassImpl)emc).invokeStaticMethod(clazz, \"")
.append(EVAL_NAME)
.append("\", ")
.append("resVals")
.append(");\n");
javaText.append("groovy.lang.GroovySystem.getMetaClassRegistry().setMetaClass(clazz, mc);\n");
}
javaText.append("res");
PsiElementFactory elementFactory =
JavaPsiFacade.getInstance(toEval.getProject()).getElementFactory();
JavaCodeFragment result =
elementFactory.createCodeBlockCodeFragment(javaText.toString(), null, true);
if (contextClass != null) {
result.setThisType(elementFactory.createType(contextClass));
}
return result;
}
