diff --git a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/GroovyCoreEnvironment.java b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/GroovyCoreEnvironment.java index fd6ee23487b3..978319a5023d 100644 --- a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/GroovyCoreEnvironment.java +++ b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/GroovyCoreEnvironment.java @@ -85,8 +85,10 @@ import org.jetbrains.plugins.groovy.lang.folding.GroovyFoldingBuilder; import org.jetbrains.plugins.groovy.lang.parser.GroovyElementTypes; import org.jetbrains.plugins.groovy.lang.parser.GroovyParserDefinition; import org.jetbrains.plugins.groovy.lang.psi.GroovyPsiElementFactory; +import org.jetbrains.plugins.groovy.lang.psi.api.auxiliary.GrListOrMap; import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrMethodCall; import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrReferenceExpression; +import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.path.GrIndexProperty; import org.jetbrains.plugins.groovy.lang.psi.expectedTypes.GroovyExpectedTypesContributor; import org.jetbrains.plugins.groovy.lang.psi.impl.*; import org.jetbrains.plugins.groovy.lang.psi.impl.javaView.GroovyClassFinder; @@ -111,10 +113,7 @@ import org.jetbrains.plugins.groovy.lang.resolve.ast.builder.strategy.SimpleBuil import org.jetbrains.plugins.groovy.lang.resolve.noncode.GrCollectionTypeMembersProvider; import org.jetbrains.plugins.groovy.lang.resolve.noncode.MixinMemberContributor; import org.jetbrains.plugins.groovy.lang.stubs.GroovyShortNamesCache; -import org.jetbrains.plugins.groovy.lang.typing.GrClosureDelegateTypeCalculator; -import org.jetbrains.plugins.groovy.lang.typing.GrTypeCalculator; -import org.jetbrains.plugins.groovy.lang.typing.MethodCallTypeCalculator; -import org.jetbrains.plugins.groovy.lang.typing.ReferenceExpressionTypeCalculator; +import org.jetbrains.plugins.groovy.lang.typing.*; import org.jetbrains.plugins.groovy.structure.GroovyStructureViewFactory; import org.jetbrains.plugins.groovy.swingBuilder.SwingBuilderNamedArgumentProvider; import org.jetbrains.plugins.groovy.swingBuilder.SwingBuilderNonCodeMemberContributor; @@ -353,6 +352,8 @@ public class GroovyCoreEnvironment { appEnvironment.addExtension(ReadWriteAccessDetector.EP_NAME, new GroovyReadWriteAccessDetector()); appEnvironment.addExplicitExtension(GrTypeCalculator.EP, GrReferenceExpression.class, new GrClosureDelegateTypeCalculator()); + appEnvironment.addExplicitExtension(GrTypeCalculator.EP, GrListOrMap.class, new DefaultListOrMapTypeCalculator()); + appEnvironment.addExplicitExtension(GrTypeCalculator.EP, GrIndexProperty.class, new DefaultIndexAccessTypeCalculator()); appEnvironment.addExplicitExtension(GrTypeCalculator.EP, GrReferenceExpression.class, new ReferenceExpressionTypeCalculator()); appEnvironment.addExplicitExtension(GrTypeCalculator.EP, GrMethodCall.class, new MethodCallTypeCalculator());