/*
 * Copyright 2024 the original author or authors.
 *
 * 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 org.gradle.kotlin.dsl.provider.plugins

import org.gradle.api.internal.initialization.ClassLoaderScope
import org.gradle.api.reflect.TypeOf
import org.gradle.cache.internal.CrossBuildInMemoryCache
import org.gradle.cache.internal.CrossBuildInMemoryCacheFactory
import org.gradle.declarative.dsl.evaluation.InterpretationSequence
import org.gradle.kotlin.dsl.accessors.ContainerElementFactoryEntry
import org.gradle.kotlin.dsl.accessors.NestedModelEntry
import org.gradle.kotlin.dsl.accessors.ProjectFeatureEntry
import org.gradle.plugin.software.internal.ProjectFeatureDeclarations

typealias ContainerElementFactories = List<ContainerElementFactoryEntry<TypeOf<*>>>
typealias NestedModels = List<NestedModelEntry<TypeOf<*>>>
typealias ProjectTypeEntries = List<ProjectFeatureEntry<TypeOf<*>>>

interface KotlinDslDclSchemaCache {
    fun getOrPutContainerElementFactories(
        forInterpretationSequence: InterpretationSequence,
        classLoaderScope: ClassLoaderScope,
        produceIfAbsent: () -> ContainerElementFactories
    ): ContainerElementFactories

    fun getOrPutNestedModels(
        forInterpretationSequence: InterpretationSequence,
        classLoaderScope: ClassLoaderScope,
        produceIfAbsent: () -> NestedModels
    ): NestedModels

    fun getOrPutProjectTypes(
        forRegistry: ProjectFeatureDeclarations,
        produceIfAbsent: () -> ProjectTypeEntries
    ): ProjectTypeEntries
}

class CrossBuildInMemoryKotlinDslDclSchemaCache(
    crossBuildInMemoryCacheFactory: CrossBuildInMemoryCacheFactory
) : KotlinDslDclSchemaCache {

    private val containerElementFactoriesCache: CrossBuildInMemoryCache<InterpretationSequenceKey, ContainerElementFactories> =
        crossBuildInMemoryCacheFactory.newCache()

    private val nestedModelsCache: CrossBuildInMemoryCache<InterpretationSequenceKey, NestedModels> =
        crossBuildInMemoryCacheFactory.newCache()

    private val projectTypeEntriesCache: CrossBuildInMemoryCache<ProjectFeatureDeclarations, ProjectTypeEntries> =
        crossBuildInMemoryCacheFactory.newCache()

    private data class InterpretationSequenceKey(
        val interpretationSequence: InterpretationSequence,
        val classLoaderScope: ClassLoaderScope
    )

    override fun getOrPutNestedModels(forInterpretationSequence: InterpretationSequence, classLoaderScope: ClassLoaderScope, produceIfAbsent: () -> NestedModels): NestedModels =
        nestedModelsCache.get(InterpretationSequenceKey(forInterpretationSequence, classLoaderScope)) { _ ->
            produceIfAbsent().ifEmpty { emptyList() }
        }

    override fun getOrPutContainerElementFactories(
        forInterpretationSequence: InterpretationSequence,
        classLoaderScope: ClassLoaderScope,
        produceIfAbsent: () -> ContainerElementFactories
    ): ContainerElementFactories = containerElementFactoriesCache
        .get(InterpretationSequenceKey(forInterpretationSequence, classLoaderScope)) { _ ->
            produceIfAbsent().ifEmpty { emptyList() } // avoid referencing lots of empty lists, store a reference to the singleton instead
        }

    override fun getOrPutProjectTypes(
        forRegistry: ProjectFeatureDeclarations,
        produceIfAbsent: () -> ProjectTypeEntries
    ): ProjectTypeEntries = projectTypeEntriesCache.get(forRegistry) { _ ->
        produceIfAbsent()
    }
}
