Yan Zhulanow
101 lines
4.9 KiB
Kotlin
101 lines
4.9 KiB
Kotlin
/*
|
|
* Copyright 2010-2016 JetBrains s.r.o.
|
|
*
|
|
* 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.jetbrains.kotlin.noarg
|
|
|
|
import com.intellij.mock.MockProject
|
|
import org.jetbrains.kotlin.codegen.extensions.ExpressionCodegenExtension
|
|
import org.jetbrains.kotlin.compiler.plugin.CliOption
|
|
import org.jetbrains.kotlin.compiler.plugin.CliOptionProcessingException
|
|
import org.jetbrains.kotlin.compiler.plugin.CommandLineProcessor
|
|
import org.jetbrains.kotlin.compiler.plugin.ComponentRegistrar
|
|
import org.jetbrains.kotlin.config.CompilerConfiguration
|
|
import org.jetbrains.kotlin.config.CompilerConfigurationKey
|
|
import org.jetbrains.kotlin.container.StorageComponentContainer
|
|
import org.jetbrains.kotlin.container.useInstance
|
|
import org.jetbrains.kotlin.descriptors.ModuleDescriptor
|
|
import org.jetbrains.kotlin.extensions.StorageComponentContainerContributor
|
|
import org.jetbrains.kotlin.noarg.NoArgCommandLineProcessor.Companion.SUPPORTED_PRESETS
|
|
import org.jetbrains.kotlin.noarg.NoArgConfigurationKeys.ANNOTATION
|
|
import org.jetbrains.kotlin.noarg.NoArgConfigurationKeys.INVOKE_INITIALIZERS
|
|
import org.jetbrains.kotlin.noarg.NoArgConfigurationKeys.PRESET
|
|
import org.jetbrains.kotlin.noarg.diagnostic.CliNoArgDeclarationChecker
|
|
import org.jetbrains.kotlin.resolve.TargetPlatform
|
|
import org.jetbrains.kotlin.resolve.jvm.platform.JvmPlatform
|
|
|
|
object NoArgConfigurationKeys {
|
|
val ANNOTATION: CompilerConfigurationKey<List<String>> =
|
|
CompilerConfigurationKey.create("annotation qualified name")
|
|
|
|
val PRESET: CompilerConfigurationKey<List<String>> = CompilerConfigurationKey.create("annotation preset")
|
|
|
|
val INVOKE_INITIALIZERS: CompilerConfigurationKey<Boolean> = CompilerConfigurationKey.create(
|
|
"invoke instance initializers in a no-arg constructor")
|
|
}
|
|
|
|
class NoArgCommandLineProcessor : CommandLineProcessor {
|
|
companion object {
|
|
val SUPPORTED_PRESETS = mapOf(
|
|
"jpa" to listOf("javax.persistence.Entity", "javax.persistence.Embeddable", "javax.persistence.MappedSuperclass"))
|
|
|
|
val ANNOTATION_OPTION = CliOption("annotation", "<fqname>", "Annotation qualified names",
|
|
required = false, allowMultipleOccurrences = true)
|
|
|
|
val PRESET_OPTION = CliOption("preset", "<name>", "Preset name (${SUPPORTED_PRESETS.keys.joinToString()})",
|
|
required = false, allowMultipleOccurrences = true)
|
|
|
|
val INVOKE_INITIALIZERS_OPTION = CliOption("invokeInitializers", "true/false",
|
|
"Invoke instance initializers in a no-arg constructor",
|
|
required = false, allowMultipleOccurrences = false)
|
|
|
|
val PLUGIN_ID = "org.jetbrains.kotlin.noarg"
|
|
}
|
|
|
|
override val pluginId = PLUGIN_ID
|
|
override val pluginOptions = listOf(ANNOTATION_OPTION, PRESET_OPTION, INVOKE_INITIALIZERS_OPTION)
|
|
|
|
override fun processOption(option: CliOption, value: String, configuration: CompilerConfiguration) = when (option) {
|
|
ANNOTATION_OPTION -> configuration.appendList(ANNOTATION, value)
|
|
PRESET_OPTION -> configuration.appendList(PRESET, value)
|
|
INVOKE_INITIALIZERS_OPTION -> configuration.put(INVOKE_INITIALIZERS, value == "true")
|
|
else -> throw CliOptionProcessingException("Unknown option: ${option.name}")
|
|
}
|
|
}
|
|
|
|
class NoArgComponentRegistrar : ComponentRegistrar {
|
|
override fun registerProjectComponents(project: MockProject, configuration: CompilerConfiguration) {
|
|
val annotations = configuration.get(ANNOTATION)?.toMutableList() ?: mutableListOf()
|
|
configuration.get(PRESET)?.forEach { preset ->
|
|
SUPPORTED_PRESETS[preset]?.let { annotations += it }
|
|
}
|
|
if (annotations.isEmpty()) return
|
|
|
|
StorageComponentContainerContributor.registerExtension(project, CliNoArgComponentContainerContributor(annotations))
|
|
|
|
val invokeInitializers = configuration[INVOKE_INITIALIZERS] ?: false
|
|
ExpressionCodegenExtension.registerExtension(project, NoArgExpressionCodegenExtension(invokeInitializers))
|
|
}
|
|
}
|
|
|
|
class CliNoArgComponentContainerContributor(val annotations: List<String>) : StorageComponentContainerContributor {
|
|
override fun registerModuleComponents(
|
|
container: StorageComponentContainer, platform: TargetPlatform, moduleDescriptor: ModuleDescriptor
|
|
) {
|
|
if (platform != JvmPlatform) return
|
|
|
|
container.useInstance(CliNoArgDeclarationChecker(annotations))
|
|
}
|
|
} |