tagan.processor-ksp.1.4.0.source-code.KspYataganProcessor.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2022 Yandex LLC
 *
 * 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 com.yandex.yatagan.processor.ksp

import com.google.devtools.ksp.processing.Dependencies
import com.google.devtools.ksp.processing.Resolver
import com.google.devtools.ksp.processing.SymbolProcessor
import com.google.devtools.ksp.processing.SymbolProcessorEnvironment
import com.google.devtools.ksp.symbol.KSAnnotated
import com.google.devtools.ksp.symbol.KSClassDeclaration
import com.yandex.yatagan.Component
import com.yandex.yatagan.lang.LangModelFactory
import com.yandex.yatagan.lang.TypeDeclaration
import com.yandex.yatagan.lang.ksp.KspModelFactoryImpl
import com.yandex.yatagan.lang.ksp.ProcessingUtils
import com.yandex.yatagan.lang.ksp.TypeDeclaration
import com.yandex.yatagan.lang.use
import com.yandex.yatagan.processor.common.Logger
import com.yandex.yatagan.processor.common.Options
import com.yandex.yatagan.processor.common.ProcessorDelegate
import com.yandex.yatagan.processor.common.process
import java.io.Writer

internal class KspYataganProcessor(
    private val environment: SymbolProcessorEnvironment,
) : SymbolProcessor, ProcessorDelegate {
    override val logger: Logger = KspLogger(environment.logger)
    override val options: Options = Options(environment.options)

    override fun process(resolver: Resolver): List {
        ProcessingUtils(resolver).use {
            LangModelFactory.use(KspModelFactoryImpl()) {
                process(
                    sources = resolver.getSymbolsWithAnnotation(Component::class.java.canonicalName)
                        .filterIsInstance(),
                    delegate = this,
                )
                return emptyList()
            }
        }
    }

    override fun createDeclaration(source: KSClassDeclaration) = TypeDeclaration(source)

    override fun getSourceFor(declaration: TypeDeclaration): KSClassDeclaration {
        return declaration.platformModel as KSClassDeclaration
    }

    override fun openFileForGenerating(
        sources: Sequence,
        packageName: String,
        className: String,
    ): Writer {
        val newFile = environment.codeGenerator.createNewFile(
            Dependencies(
                aggregating = false,
            ),
            packageName = packageName,
            fileName = className,
            extensionName = "java",
        )
        environment.codeGenerator.associateWithClasses(
            classes = sources.toList(),
            packageName = packageName,
            fileName = className,
            extensionName = "java",
        )
        return newFile.bufferedWriter()
    }
}