ugsig.actions-dispatcher-processor.1.0.7.source-code.ActionType.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of actions-dispatcher-processor Show documentation

A simple annotation processor for generating actions dispatcher

There is a newer version: 1.0.9

Show newest version

package com.github.rougsig.actionsdispatcher.processor

import com.github.rougsig.actionsdispatcher.processor.utils.Logger
import com.squareup.kotlinpoet.ClassName
import com.squareup.kotlinpoet.ParameterizedTypeName
import com.squareup.kotlinpoet.asTypeName
import me.eugeniomarletti.kotlin.metadata.KotlinClassMetadata
import me.eugeniomarletti.kotlin.metadata.classKind
import me.eugeniomarletti.kotlin.metadata.kotlinMetadata
import me.eugeniomarletti.kotlin.metadata.shadow.metadata.ProtoBuf
import me.eugeniomarletti.kotlin.metadata.shadow.metadata.ProtoBuf.Class
import me.eugeniomarletti.kotlin.metadata.visibility
import javax.lang.model.element.Element
import javax.lang.model.element.TypeElement
import javax.lang.model.util.Elements
import javax.lang.model.util.Types

internal data class ActionType(
  val proto: Class,
  val element: TypeElement,
  val actions: List,
  val isInternal: Boolean
) {
  val name = element.className
  val packageName = name.packageName()

  companion object {
    fun get(logger: Logger, elements: Elements, types: Types, element: Element): ActionType? {
      val typeMetadata = element.kotlinMetadata
      if (element !is TypeElement || typeMetadata !is KotlinClassMetadata) {
        logger.error("@ActionElement can't be applied to $element: must be kotlin class", element)
        return null
      }

      val proto = typeMetadata.data.classProto
      when {
        proto.classKind != Class.Kind.CLASS -> {
          logger.error("@ActionReducer can't be applied to $element: must be a class", element)
          return null
        }
      }

      val actions = getActions(element)
      val isInternal = proto.visibility!! == ProtoBuf.Visibility.INTERNAL

      return ActionType(proto, element, actions, isInternal)
    }

    private fun getActions(actionTypeElement: TypeElement): List {
      val actions = mutableListOf()

      for (element in actionTypeElement.enclosingElement.enclosedElements) {
        val typeMetadata = element.kotlinMetadata

        if (element !is TypeElement || typeMetadata !is KotlinClassMetadata) continue
        if (element.superclass.toString() != actionTypeElement.asType().toString()) continue

        actions.add(element)
      }

      return actions
    }
  }

  private val Element.className: ClassName
    get() {
      val typeName = asType().asTypeName()
      return when (typeName) {
        is ClassName -> typeName
        is ParameterizedTypeName -> typeName.rawType
        else -> throw IllegalStateException("unexpected TypeName: ${typeName::class}")
      }
    }
}