J2K JavaElementCollectionFromPsiArrayUtil
This commit is contained in:
Alexander Udalov
-202
@@ -1,202 +0,0 @@
|
||||
/*
|
||||
* Copyright 2010-2016 JetBrains s.r.o.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.jetbrains.kotlin.load.java.structure.impl;
|
||||
|
||||
import com.intellij.psi.*;
|
||||
import com.intellij.psi.search.GlobalSearchScope;
|
||||
import kotlin.collections.CollectionsKt;
|
||||
import kotlin.jvm.functions.Function1;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.kotlin.load.java.structure.*;
|
||||
import org.jetbrains.kotlin.name.Name;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
public class JavaElementCollectionFromPsiArrayUtil {
|
||||
private JavaElementCollectionFromPsiArrayUtil() {
|
||||
}
|
||||
|
||||
private interface Factory<Psi, Java> {
|
||||
@NotNull
|
||||
Java create(@NotNull Psi psi);
|
||||
}
|
||||
|
||||
private static class Factories {
|
||||
private static final Factory<PsiClass, JavaClass> CLASSES = new Factory<PsiClass, JavaClass>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaClass create(@NotNull PsiClass psiClass) {
|
||||
return new JavaClassImpl(psiClass);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiMethod, JavaMethod> METHODS = new Factory<PsiMethod, JavaMethod>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaMethod create(@NotNull PsiMethod psiMethod) {
|
||||
return new JavaMethodImpl(psiMethod);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiMethod, JavaConstructor> CONSTRUCTORS = new Factory<PsiMethod, JavaConstructor>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaConstructor create(@NotNull PsiMethod psiMethod) {
|
||||
return new JavaConstructorImpl(psiMethod);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiField, JavaField> FIELDS = new Factory<PsiField, JavaField>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaField create(@NotNull PsiField psiField) {
|
||||
return new JavaFieldImpl(psiField);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiParameter, JavaValueParameter> VALUE_PARAMETERS = new Factory<PsiParameter, JavaValueParameter>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaValueParameter create(@NotNull PsiParameter psiParameter) {
|
||||
return new JavaValueParameterImpl(psiParameter);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiTypeParameter, JavaTypeParameter> TYPE_PARAMETERS =
|
||||
new Factory<PsiTypeParameter, JavaTypeParameter>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaTypeParameter create(@NotNull PsiTypeParameter psiTypeParameter) {
|
||||
return new JavaTypeParameterImpl(psiTypeParameter);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiClassType, JavaClassifierType> CLASSIFIER_TYPES = new Factory<PsiClassType, JavaClassifierType>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaClassifierType create(@NotNull PsiClassType psiClassType) {
|
||||
return new JavaClassifierTypeImpl(psiClassType);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiAnnotation, JavaAnnotation> ANNOTATIONS = new Factory<PsiAnnotation, JavaAnnotation>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaAnnotation create(@NotNull PsiAnnotation psiAnnotation) {
|
||||
return new JavaAnnotationImpl(psiAnnotation);
|
||||
}
|
||||
};
|
||||
|
||||
private static final Factory<PsiNameValuePair, JavaAnnotationArgument> NAMED_ANNOTATION_ARGUMENTS =
|
||||
new Factory<PsiNameValuePair, JavaAnnotationArgument>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaAnnotationArgument create(@NotNull PsiNameValuePair psiNameValuePair) {
|
||||
String name = psiNameValuePair.getName();
|
||||
PsiAnnotationMemberValue value = psiNameValuePair.getValue();
|
||||
assert value != null : "Annotation argument value cannot be null: " + name;
|
||||
return JavaAnnotationArgumentImpl.Factory.create(value, name == null ? null : Name.identifier(name));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private static <Psi, Java> List<Java> convert(@NotNull Psi[] elements, @NotNull Factory<Psi, Java> factory) {
|
||||
if (elements.length == 0) return Collections.emptyList();
|
||||
List<Java> result = new ArrayList<Java>(elements.length);
|
||||
for (Psi element : elements) {
|
||||
result.add(factory.create(element));
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private static <Psi, Java> List<Java> convert(@NotNull Iterable<Psi> elements, @NotNull final Factory<Psi, Java> factory) {
|
||||
if (!elements.iterator().hasNext()) return Collections.emptyList();
|
||||
|
||||
return CollectionsKt.map(elements, new Function1<Psi, Java>() {
|
||||
@Override
|
||||
public Java invoke(Psi psi) {
|
||||
return factory.create(psi);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaClass> classes(@NotNull PsiClass[] classes) {
|
||||
return convert(classes, Factories.CLASSES);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaClass> classes(@NotNull Iterable<PsiClass> classes) {
|
||||
return convert(classes, Factories.CLASSES);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaPackage> packages(@NotNull PsiPackage[] packages, @NotNull final GlobalSearchScope scope) {
|
||||
return convert(packages, new Factory<PsiPackage, JavaPackage>() {
|
||||
@NotNull
|
||||
@Override
|
||||
public JavaPackage create(@NotNull PsiPackage aPackage) {
|
||||
return new JavaPackageImpl(aPackage, scope);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaMethod> methods(@NotNull Iterable<PsiMethod> methods) {
|
||||
return convert(methods, Factories.METHODS);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaConstructor> constructors(@NotNull Iterable<PsiMethod> methods) {
|
||||
return convert(methods, Factories.CONSTRUCTORS);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaField> fields(@NotNull Iterable<PsiField> fields) {
|
||||
return convert(fields, Factories.FIELDS);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static List<JavaValueParameter> valueParameters(@NotNull PsiParameter[] parameters) {
|
||||
return convert(parameters, Factories.VALUE_PARAMETERS);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static List<JavaTypeParameter> typeParameters(@NotNull PsiTypeParameter[] typeParameters) {
|
||||
return convert(typeParameters, Factories.TYPE_PARAMETERS);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaClassifierType> classifierTypes(@NotNull PsiClassType[] classTypes) {
|
||||
return convert(classTypes, Factories.CLASSIFIER_TYPES);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaAnnotation> annotations(@NotNull PsiAnnotation[] annotations) {
|
||||
return convert(annotations, Factories.ANNOTATIONS);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Collection<JavaAnnotationArgument> namedAnnotationArguments(@NotNull PsiNameValuePair[] nameValuePairs) {
|
||||
return convert(nameValuePairs, Factories.NAMED_ANNOTATION_ARGUMENTS);
|
||||
}
|
||||
}
|
||||
+75
@@ -0,0 +1,75 @@
|
||||
/*
|
||||
* Copyright 2010-2016 JetBrains s.r.o.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
@file:JvmName("JavaElementCollectionFromPsiArrayUtil")
|
||||
|
||||
package org.jetbrains.kotlin.load.java.structure.impl
|
||||
|
||||
import com.intellij.psi.*
|
||||
import com.intellij.psi.search.GlobalSearchScope
|
||||
import org.jetbrains.kotlin.load.java.structure.*
|
||||
import org.jetbrains.kotlin.name.Name
|
||||
|
||||
private inline fun <Psi, Java> Array<Psi>.convert(factory: (Psi) -> Java): List<Java> =
|
||||
when (size) {
|
||||
0 -> emptyList()
|
||||
1 -> listOf(factory(first()))
|
||||
else -> map(factory)
|
||||
}
|
||||
|
||||
private fun <Psi, Java> Collection<Psi>.convert(factory: (Psi) -> Java): List<Java> =
|
||||
when (size) {
|
||||
0 -> emptyList()
|
||||
1 -> listOf(factory(first()))
|
||||
else -> map(factory)
|
||||
}
|
||||
|
||||
internal fun classes(classes: Array<PsiClass>): Collection<JavaClass> =
|
||||
classes.convert(::JavaClassImpl)
|
||||
|
||||
internal fun classes(classes: Collection<PsiClass>): Collection<JavaClass> =
|
||||
classes.convert(::JavaClassImpl)
|
||||
|
||||
internal fun packages(packages: Array<PsiPackage>, scope: GlobalSearchScope): Collection<JavaPackage> =
|
||||
packages.convert { psi -> JavaPackageImpl(psi, scope) }
|
||||
|
||||
internal fun methods(methods: Collection<PsiMethod>): Collection<JavaMethod> =
|
||||
methods.convert(::JavaMethodImpl)
|
||||
|
||||
internal fun constructors(methods: Collection<PsiMethod>): Collection<JavaConstructor> =
|
||||
methods.convert(::JavaConstructorImpl)
|
||||
|
||||
internal fun fields(fields: Collection<PsiField>): Collection<JavaField> =
|
||||
fields.convert(::JavaFieldImpl)
|
||||
|
||||
internal fun valueParameters(parameters: Array<PsiParameter>): List<JavaValueParameter> =
|
||||
parameters.convert(::JavaValueParameterImpl)
|
||||
|
||||
internal fun typeParameters(typeParameters: Array<PsiTypeParameter>): List<JavaTypeParameter> =
|
||||
typeParameters.convert(::JavaTypeParameterImpl)
|
||||
|
||||
internal fun classifierTypes(classTypes: Array<PsiClassType>): Collection<JavaClassifierType> =
|
||||
classTypes.convert(::JavaClassifierTypeImpl)
|
||||
|
||||
internal fun annotations(annotations: Array<PsiAnnotation>): Collection<JavaAnnotation> =
|
||||
annotations.convert(::JavaAnnotationImpl)
|
||||
|
||||
internal fun namedAnnotationArguments(nameValuePairs: Array<PsiNameValuePair>): Collection<JavaAnnotationArgument> =
|
||||
nameValuePairs.convert { psi ->
|
||||
val name = psi.name?.let(Name::identifier)
|
||||
val value = psi.value ?: error("Annotation argument value cannot be null: $name")
|
||||
JavaAnnotationArgumentImpl.create(value, name)
|
||||
}
|
||||
-2
@@ -20,8 +20,6 @@ import com.intellij.psi.PsiPackage
|
||||
import com.intellij.psi.search.GlobalSearchScope
|
||||
import org.jetbrains.kotlin.load.java.structure.JavaClass
|
||||
import org.jetbrains.kotlin.load.java.structure.JavaPackage
|
||||
import org.jetbrains.kotlin.load.java.structure.impl.JavaElementCollectionFromPsiArrayUtil.classes
|
||||
import org.jetbrains.kotlin.load.java.structure.impl.JavaElementCollectionFromPsiArrayUtil.packages
|
||||
import org.jetbrains.kotlin.name.FqName
|
||||
import org.jetbrains.kotlin.name.Name
|
||||
|
||||
|
||||
Reference in New Issue
Block a user