main.dev.zacsweers.moshix.ir.compiler.proguardgen.MoshiCompilationException.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of moshi-compiler-plugin Show documentation

Moshi Extensions

There is a newer version: 1.7.20-Beta-0.18.3

Show newest version

package dev.zacsweers.moshix.ir.compiler.proguardgen

import org.jetbrains.kotlin.codegen.CompilationException
import org.jetbrains.kotlin.com.intellij.psi.PsiElement
import org.jetbrains.kotlin.com.intellij.psi.PsiNameIdentifierOwner
import org.jetbrains.kotlin.descriptors.ClassDescriptor
import org.jetbrains.kotlin.descriptors.annotations.AnnotationDescriptor
import org.jetbrains.kotlin.ir.IrElement
import org.jetbrains.kotlin.ir.declarations.IrClass
import org.jetbrains.kotlin.ir.symbols.IrSymbol
import org.jetbrains.kotlin.ir.util.render
import org.jetbrains.kotlin.js.resolve.diagnostics.findPsi
import org.jetbrains.kotlin.resolve.source.KotlinSourceElement
import org.jetbrains.kotlin.resolve.source.getPsi
import org.jetbrains.kotlin.util.getExceptionMessage

/**
 * `AnvilCompilationException` is thrown when Anvil specific code can't be processed or an error
 * occurs while generating code.
 */
internal class MoshiCompilationException(
  message: String,
  cause: Throwable? = null,
  element: PsiElement? = null
) : CompilationException(message, cause, element) {
  companion object {
    operator fun invoke(
      annotationDescriptor: AnnotationDescriptor,
      message: String,
      cause: Throwable? = null
    ): MoshiCompilationException {
      return MoshiCompilationException(
        message = message,
        cause = cause,
        element = annotationDescriptor.identifier
      )
    }

    operator fun invoke(
      classDescriptor: ClassDescriptor,
      message: String,
      cause: Throwable? = null
    ): MoshiCompilationException {
      return MoshiCompilationException(
        message = message,
        cause = cause,
        element = classDescriptor.identifier
      )
    }

    operator fun invoke(
      element: IrElement? = null,
      message: String,
      cause: Throwable? = null
    ): MoshiCompilationException {
      return MoshiCompilationException(
          message =
            getExceptionMessage(
              subsystemName = "Anvil",
              message = message,
              cause = cause,
              location = element?.render()
            ),
          cause = cause,
          element = element?.psi
        )
        .apply {
          if (element != null) {
            withAttachment("element.kt", element.render())
          }
        }
    }

    operator fun invoke(
      element: IrSymbol? = null,
      message: String,
      cause: Throwable? = null,
    ): MoshiCompilationException {
      return MoshiCompilationException(message = message, cause = cause, element = element?.owner)
    }
  }
}

private val ClassDescriptor.identifier: PsiElement?
  get() = (findPsi() as? PsiNameIdentifierOwner)?.identifyingElement

private val AnnotationDescriptor.identifier: PsiElement?
  get() = (source as? KotlinSourceElement)?.psi

private val IrElement.psi: PsiElement?
  get() =
    when (this) {
      is IrClass -> (this.source.getPsi() as? PsiNameIdentifierOwner)?.identifyingElement
      else -> null
    }