io.github.mattshoe.shoebox.kernl.KernlSymbolProcessor.kt Maven / Gradle / Ivy

Go to download
package io.github.mattshoe.shoebox.kernl

import com.google.devtools.ksp.processing.SymbolProcessorEnvironment
import io.github.mattshoe.shoebox.kernl.annotations.Kernl
import io.github.mattshoe.shoebox.kernl.processors.impl.AssociativeMemoryCacheProcessor
import io.github.mattshoe.shoebox.kernl.processors.impl.MemoryCacheCodeGenerator
import io.github.mattshoe.shoebox.kernl.processors.impl.NoCacheProcessor
import io.github.mattshoe.shoebox.kernl.processors.impl.SingleMemoryCacheProcessor
import io.github.mattshoe.shoebox.stratify.StratifySymbolProcessor
import io.github.mattshoe.shoebox.stratify.ksp.StratifyResolver
import io.github.mattshoe.shoebox.stratify.strategy.AnnotationStrategy

class KernlSymbolProcessor: StratifySymbolProcessor() {

    override suspend fun buildStrategies(resolver: StratifyResolver) = listOf(
        noCacheStrategy(environment),
        singleMemoryCacheStrategy(environment),
        associativeMemoryCacheStrategy(environment)
    )

    private fun noCacheStrategy(environment: SymbolProcessorEnvironment): AnnotationStrategy {
        return AnnotationStrategy(
            annotation = Kernl.NoCache::class,
            processors = listOf (
                NoCacheProcessor(environment.logger)
            )
        )
    }

    private fun singleMemoryCacheStrategy(environment: SymbolProcessorEnvironment): AnnotationStrategy {
        return AnnotationStrategy(
            annotation = Kernl.SingleMemoryCache::class,
            processors = listOf(
                SingleMemoryCacheProcessor(
                    environment.logger,
                    MemoryCacheCodeGenerator()
                )
            )
        )
    }

    private fun associativeMemoryCacheStrategy(environment: SymbolProcessorEnvironment): AnnotationStrategy {
        return AnnotationStrategy(
            annotation = Kernl.AssociativeMemoryCache::class,
            processors = listOf(
                AssociativeMemoryCacheProcessor(
                    environment.logger,
                    MemoryCacheCodeGenerator()
                )
            )
        )
    }
}