main.dev.zacsweers.moshix.ir.compiler.sealed.MoshiSealedSymbols.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
/*
 * Copyright (C) 2021 Zac Sweers
 *
 * 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
 *
 *    https://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 dev.zacsweers.moshix.ir.compiler.sealed

import dev.zacsweers.moshix.ir.compiler.BaseSymbols
import dev.zacsweers.moshix.ir.compiler.MoshiSymbols
import org.jetbrains.kotlin.descriptors.ClassKind
import org.jetbrains.kotlin.descriptors.Modality
import org.jetbrains.kotlin.ir.builders.declarations.addFunction
import org.jetbrains.kotlin.ir.builders.declarations.addTypeParameter
import org.jetbrains.kotlin.ir.builders.declarations.addValueParameter
import org.jetbrains.kotlin.ir.declarations.IrPackageFragment
import org.jetbrains.kotlin.ir.types.defaultType
import org.jetbrains.kotlin.ir.types.makeNullable
import org.jetbrains.kotlin.ir.types.typeWith
import org.jetbrains.kotlin.ir.types.typeWithParameters
import org.jetbrains.kotlin.ir.util.constructors
import org.jetbrains.kotlin.ir.util.defaultType
import org.jetbrains.kotlin.ir.util.getSimpleFunction
import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.types.Variance

internal class MoshiSealedSymbols(
  private val moshiSymbols: MoshiSymbols,
) : BaseSymbols(moshiSymbols) {

  private val moshiAdaptersPackage: IrPackageFragment by lazy {
    createPackage("com.squareup.moshi.adapters")
  }

  val pjaf by lazy {
    // TODO https://youtrack.jetbrains.com/issue/KT-55269
    //    pluginContext.referenceClass(
    //      ClassId.fromString("com/squareup/moshi/adapters/PolymorphicJsonAdapterFactory")
    //    )!!
    createClass(
      moshiAdaptersPackage,
      "PolymorphicJsonAdapterFactory",
      ClassKind.CLASS,
      Modality.FINAL
    ) {
      val classTypeParameter = addTypeParameter("T", irBuiltIns.anyNType, Variance.INVARIANT)
      //  val pjafWithDefaultValue by lazy { pjaf.getSimpleFunction("withDefaultValue")!! }
      //  val pjafWithFallbackJsonAdapter by lazy {
      // pjaf.getSimpleFunction("withFallbackJsonAdapter")!! }
      // public static  PolymorphicJsonAdapterFactory of(Class baseType, String labelKey)
      addFunction(
          name = Name.identifier("of").identifier,
          returnType = defaultType,
          isStatic = true,
        )
        .apply {
          val typeParameter = addTypeParameter("T", irBuiltIns.anyNType, Variance.INVARIANT)
          addValueParameter("baseType", javaLangClass.typeWithParameters(listOf(typeParameter)))
          addValueParameter("labelKey", irBuiltIns.stringType)
        }

      // public PolymorphicJsonAdapterFactory withSubtype(Class subtype, String
      // label) {
      addFunction(
          name = Name.identifier("withSubtype").identifier,
          returnType = typeWith(classTypeParameter.defaultType),
        )
        .apply {
          // TODO need ? extends T
          addValueParameter("subtype", javaLangClass.typeWithParameters(listOf(classTypeParameter)))
          addValueParameter("label", irBuiltIns.stringType)
        }

      // public PolymorphicJsonAdapterFactory withDefaultValue(@Nullable T defaultValue)
      addFunction(
          name = Name.identifier("withDefaultValue").identifier,
          returnType = typeWith(classTypeParameter.defaultType),
        )
        .apply { addValueParameter("defaultValue", classTypeParameter.defaultType.makeNullable()) }

      // public PolymorphicJsonAdapterFactory withFallbackJsonAdapter(@Nullable
      // JsonAdapter