void (@NotNull PsiReferenceRegistrar registrar) { UastReferenceRegistrar .registerUastReferenceProvider(registrar, injectionHostUExpression() .sourcePsiFilter(psi -> PsiUtil.isPluginProject(psi.getProject())) .methodCallParameter(0, psiMethod() .withName(string().oneOf("isFeatureEnabled", "setFeatureEnabled")) .definedInClass(Experiments.class.getName())), UastReferenceRegistrar.uastInjectionHostReferenceProvider( (expression, host) -> new PsiReference[]{new ExperimentalFeatureIdReference(host)}), PsiReferenceRegistrar.DEFAULT_PRIORITY); }