arrow.meta.phases.CompilerContext.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arrow-meta Show documentation

Functional companion to Kotlin's Standard Library

There is a newer version: 1.6.3-alpha.2

package arrow.meta.phases

import java.util.concurrent.ConcurrentHashMap
import java.util.concurrent.atomic.AtomicBoolean
import org.jetbrains.kotlin.cli.common.messages.MessageCollector
import org.jetbrains.kotlin.com.intellij.openapi.project.Project
import org.jetbrains.kotlin.config.CompilerConfiguration
import org.jetbrains.kotlin.container.ComponentProvider
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
import org.jetbrains.kotlin.descriptors.ModuleDescriptor
import org.jetbrains.kotlin.js.resolve.diagnostics.findPsi
import org.jetbrains.kotlin.psi.KtElement
import org.jetbrains.kotlin.psi.KtPsiFactory
import org.jetbrains.kotlin.resolve.BindingTrace
import org.jetbrains.kotlin.script.jsr223.KotlinJsr223JvmLocalScriptEngineFactory

/**
 * The Compiler Context represents the environment received by all plugins. The Compiler Context
 * will get more services as they become relevant overtime to the development of compiler plugins.
 */
open class CompilerContext(
  val configuration: CompilerConfiguration?,
  open val project: Project,
  val messageCollector: MessageCollector? = null,
  val ktPsiElementFactory: KtPsiFactory = KtPsiFactory(project, false),
  val eval: (String) -> Any? = { KotlinJsr223JvmLocalScriptEngineFactory().scriptEngine.eval(it) }
) {
  private var md: ModuleDescriptor? = null
  private var cp: ComponentProvider? = null
  var bindingTrace: BindingTrace? = null

  val analysedDescriptors: MutableList = mutableListOf()

  val analysisPhaseWasRewind: AtomicBoolean = AtomicBoolean(false)

  val analysisPhaseCanBeRewind: AtomicBoolean = AtomicBoolean(false)

  var module: ModuleDescriptor?
    get() = md
    set(value) {
      md = value
    }

  var componentProvider: ComponentProvider?
    get() = cp
    set(value) {
      cp = value
    }

  val ctx: CompilerContext = this

  @PublishedApi internal val sessionData: ConcurrentHashMap = ConcurrentHashMap()

  fun  set(key: String, value: T): Unit {
    sessionData[key] = value
  }

  inline fun  get(key: String): T? = sessionData[key] as? T
}

fun  CompilerContext.evaluateDependsOn(
  noRewindablePhase: () -> T?,
  rewindablePhase: (Boolean) -> T?
): T? {
  if (!analysisPhaseCanBeRewind.get()) return noRewindablePhase()
  return rewindablePhase(analysisPhaseWasRewind.get())
}

fun  CompilerContext.evaluateDependsOnRewindableAnalysisPhase(evaluation: () -> T?): T? =
  evaluateDependsOn(
    noRewindablePhase = evaluation,
    rewindablePhase = { wasRewind -> if (wasRewind) evaluation() else null }
  )

inline fun  KtElement.findInAnalysedDescriptors(
  compilerContext: CompilerContext
): D? =
  compilerContext.analysedDescriptors.filterIsInstance().firstOrNull { it.findPsi() == this }