|
14 | 14 | * limitations under the License.
|
15 | 15 | */
|
16 | 16 |
|
| 17 | +@file:OptIn(ExperimentalCompilerApi::class) |
| 18 | + |
17 | 19 | package com.bnorm.template
|
18 | 20 |
|
19 | 21 | import com.google.auto.service.AutoService
|
20 | 22 | import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
|
21 | 23 | import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
|
22 | 24 | import org.jetbrains.kotlin.cli.common.messages.MessageCollector
|
23 |
| -import org.jetbrains.kotlin.com.intellij.mock.MockProject |
24 |
| -import org.jetbrains.kotlin.compiler.plugin.ComponentRegistrar |
| 25 | +import org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar |
| 26 | +import org.jetbrains.kotlin.compiler.plugin.ExperimentalCompilerApi |
25 | 27 | import org.jetbrains.kotlin.config.CompilerConfiguration
|
26 | 28 |
|
27 |
| -@AutoService(ComponentRegistrar::class) |
28 |
| -class TemplateComponentRegistrar( |
| 29 | +@AutoService(CompilerPluginRegistrar::class) |
| 30 | +class TemplateCompilerRegistrar( |
29 | 31 | private val defaultString: String,
|
30 | 32 | private val defaultFile: String,
|
31 |
| -) : ComponentRegistrar { |
| 33 | +) : CompilerPluginRegistrar() { |
| 34 | + override val supportsK2 = true |
32 | 35 |
|
33 | 36 | @Suppress("unused") // Used by service loader
|
34 | 37 | constructor() : this(
|
35 | 38 | defaultString = "Hello, World!",
|
36 | 39 | defaultFile = "file.txt"
|
37 | 40 | )
|
38 | 41 |
|
39 |
| - override fun registerProjectComponents( |
40 |
| - project: MockProject, |
41 |
| - configuration: CompilerConfiguration |
42 |
| - ) { |
| 42 | + override fun ExtensionStorage.registerExtensions(configuration: CompilerConfiguration) { |
43 | 43 | val messageCollector = configuration.get(CLIConfigurationKeys.MESSAGE_COLLECTOR_KEY, MessageCollector.NONE)
|
44 | 44 | val string = configuration.get(TemplateCommandLineProcessor.ARG_STRING, defaultString)
|
45 | 45 | val file = configuration.get(TemplateCommandLineProcessor.ARG_FILE, defaultFile)
|
46 | 46 |
|
47 |
| - IrGenerationExtension.registerExtension(project, TemplateIrGenerationExtension(messageCollector, string, file)) |
| 47 | + IrGenerationExtension.registerExtension(TemplateIrGenerationExtension(messageCollector, string, file)) |
48 | 48 | }
|
49 | 49 | }
|
0 commit comments