org.jetbrains.uast.UElement Java Examples

private static void detectR2(JavaContext context, UElement node) {
  UFile sourceFile = context.getUastFile();
  List<UClass> classes = sourceFile.getClasses();
  if (!classes.isEmpty() && classes.get(0).getName() != null) {
    String qualifiedName = classes.get(0).getName();
    if (qualifiedName.contains("_ViewBinder")
        || qualifiedName.contains("_ViewBinding")
        || qualifiedName.equals(R2)) {
      // skip generated files and R2
      return;
    }
  }
  boolean isR2 = isR2Expression(node);
  if (isR2 && !context.isSuppressedWithComment(node, ISSUE)) {
    context.report(ISSUE, node, context.getLocation(node), LINT_ERROR_BODY);
  }
}
 

/**
 * Generate method identifier from method information.
 *
 * @param node UCallExpression
 * @return className + methodName + parametersType
 */
@Nullable
private static String methodIdentifier(@NotNull UCallExpression node) {
    UElement element = node.getUastParent();
    while (element != null) {
        if (element instanceof UClass) {
            break;
        }
        element = element.getUastParent();
    }
    UClass uClass = (UClass) element;
    if (uClass == null || node.getMethodName() == null) {
        return null;
    }
    return uClass.getName() + COLON + node.getMethodName();
}
 

private static boolean isR2Expression(UElement node) {
  UElement parentNode = node.getUastParent();
  if (parentNode == null) {
    return false;
  }
  String text = node.asSourceString();
  UElement parent = LintUtils.skipParentheses(parentNode);
  return (text.equals(R2) || text.contains(".R2"))
      && parent instanceof UExpression
      && endsWithAny(parent.asSourceString(), SUPPORTED_TYPES);
}
 

private boolean verifyConfigFieldReference(final UElement node, final PsiElement valueElement) {
	if (!ElementUtils.isFieldReference(valueElement)) {
		report(node);
		return false;
	}

	final PsiField configField = ElementUtils.getReferencedField(valueElement);
	if (!ConfigElementsUtils.isConfigField(configField)) {
		report(node);
		return false;
	}
	return true;
}
 

protected LintFix addAnnotationFix(UElement target, String targetName, Class<? extends Annotation> clazz, String annotationContent) {
	final String source = target.asSourceString();
	return replaceFix(target).name(String.format(ADD_ANNOTATION_FIX_FORMAT, clazz.getSimpleName(), targetName))
	                         .text(source)
	                         .with("@" + clazz.getCanonicalName() + "(" + annotationContent + ")" + source)
	                         .build();
}
 

public static UClass getContainingClass(UElement element) {
	if (element instanceof UClass) {
		return (UClass) element;
	}
	if (element == null) {
		return null;
	}

	return getContainingClass(element.getUastParent());
}
 

/**
 * Generate method identifier from method information.
 *
 * @param node UMethod
 * @return className + methodName + parametersType
 */
@Nullable
private static String methodIdentifier(@NotNull UMethod node) {
    UElement parent = node.getUastParent();
    if (!(parent instanceof UClass)) {
        return null;
    }
    UClass uClass = (UClass) parent;
    return uClass.getName() + COLON + node.getName();
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    final List<Class<? extends UElement>> ret = new ArrayList<>();
    ret.add(UClass.class);
    return ret;
}
 

@Override public List<Class<? extends UElement>> getApplicableUastTypes() {
  return Collections.singletonList(UClass.class);
}
 

@Override public List<Class<? extends UElement>> getApplicableUastTypes() {
  return Collections.singletonList(UClass.class);
}
 

@Nullable
@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    //noinspection unchecked
    return (List) Collections.singletonList(UCallExpression.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.singletonList(ULiteralExpression.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.<Class<? extends UElement>>singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.<Class<? extends UElement>>singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.<Class<? extends UElement>>singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.<Class<? extends UElement>>singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
  return Collections.singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.<Class<? extends UElement>>singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
    return Collections.<Class<? extends UElement>>singletonList(UClass.class);
}
 

@Override
public List<Class<? extends UElement>> getApplicableUastTypes() {
	return Collections.<Class<? extends UElement>> singletonList(UClass.class);
}
 

protected LintFix.ReplaceStringBuilder replaceFix(UElement target) {
	return fix().replace()
	            .range(mContext.getLocation(target))
	            .shortenNames()
	            .reformat(true);
}
 

protected void report(final UElement node, String desc, LintFix lintFix) {
	if (REPORTED_ISSUES.get(node) != mIssue) {
		REPORTED_ISSUES.put(node, mIssue);
		mContext.report(mIssue, node, mContext.getLocation(node), desc, lintFix);
	}
}
 

protected void report(final UElement node, String desc) {
	report(node, desc, null);
}
 

protected void report(final UElement node, LintFix lintFix) {
	report(node, mIssue.getExplanation(TextFormat.RAW), lintFix);
}
 

protected void report(final UElement node) {
	report(node, mIssue.getExplanation(TextFormat.RAW));
}
 

private LintFix addSourceAnnotationFix(UElement target, String targetName) {
	return addAnnotationFix(target, targetName, Source.class, "SomeConfigSource.class");
}