| /* |
| * Copyright (C) 2020 The Android Open Source Project |
| * |
| * 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 androidx.room.compiler.processing.javac |
| |
| import androidx.room.compiler.processing.XElement |
| import androidx.room.compiler.processing.XMessager |
| import java.io.StringWriter |
| import javax.annotation.processing.ProcessingEnvironment |
| import javax.lang.model.element.Element |
| import javax.lang.model.element.ElementKind |
| import javax.lang.model.util.Elements |
| import javax.tools.Diagnostic |
| |
| internal class JavacProcessingEnvMessager( |
| private val processingEnv: ProcessingEnvironment |
| ) : XMessager { |
| override fun printMessage(kind: Diagnostic.Kind, msg: String, element: XElement?) { |
| val javacElement = (element as? JavacElement)?.element |
| processingEnv.messager.printMessage( |
| kind, |
| if (javacElement != null && javacElement.isFromCompiledClass()) { |
| msg.appendElement(processingEnv.elementUtils, javacElement) |
| } else { |
| msg |
| }, |
| javacElement |
| ) |
| } |
| |
| companion object { |
| /** |
| * Indicates whether an element comes from a compiled class. |
| * |
| * If this method fails to identify if the element comes from a compiled class it will |
| * default to returning false. Note that this is a poor-man's method of identifying if |
| * the java source of the element is available without depending on compiler tools. |
| */ |
| private fun Element.isFromCompiledClass(): Boolean { |
| fun getClassFileString(symbol: Any): String = |
| try { |
| symbol.javaClass.getDeclaredField("classfile").get(symbol).toString() |
| } catch (ex: NoSuchFieldException) { |
| getClassFileString( |
| symbol.javaClass.superclass.getDeclaredField("owner").get(symbol) |
| ) |
| } |
| |
| return try { |
| getClassFileString(this).let { |
| it.contains(".jar") || it.contains(".class") |
| } |
| } catch (ex: Throwable) { |
| false |
| } |
| } |
| |
| private fun String.appendElement(elementUtils: Elements, element: Element): String { |
| return StringBuilder(this).apply { |
| append(" - ") |
| when (element.kind) { |
| ElementKind.CLASS, ElementKind.INTERFACE, ElementKind.CONSTRUCTOR -> |
| append(element) |
| ElementKind.FIELD, ElementKind.METHOD, ElementKind.PARAMETER -> |
| append("$element in ${element.enclosingElement}") |
| else -> { |
| // Not sure how to nicely print the element, delegate to utils then. |
| append("In:\n") |
| append(StringWriter().apply { |
| elementUtils.printElements(this, element) |
| }.toString()) |
| } |
| } |
| }.toString() |
| } |
| } |
| } |