[CLI] Notify JS IR perf manager when generation is started and finished

#KT-67473

Author: ivandev0

compiler/cli/cli-js/src/org/jetbrains/kotlin/cli/js/K2JsIrCompiler.kt

@@ -111,6 +111,8 @@ class K2JsIrCompiler : CLICompiler<K2JSCompilerArguments>() {
         val messageCollector: MessageCollector,
         val mainCallArguments: List<String>?
     ) {
+        private val performanceManager = module.compilerConfiguration[CLIConfigurationKeys.PERF_MANAGER]
+
         private fun lowerIr(): LoweredIr {
             return compile(
                 mainCallArguments,
@@ -139,11 +141,18 @@ class K2JsIrCompiler : CLICompiler<K2JSCompilerArguments>() {
             val transformer = IrModuleToJsTransformer(ir.context, ir.moduleFragmentToUniqueName, mainCallArguments != null)
 
             val mode = TranslationMode.fromFlags(arguments.irDce, arguments.granularity, arguments.irMinimizedMemberNames)
-            return transformer.makeJsCodeGenerator(ir.allModules, mode)
+            return transformer
+                .also { performanceManager?.notifyIRGenerationStarted() }
+                .makeJsCodeGenerator(ir.allModules, mode)
         }
 
         fun compileAndTransformIrNew(): CompilationOutputsBuilt {
-            return makeJsCodeGenerator().generateJsCode(relativeRequirePath = true, outJsProgram = false)
+            return makeJsCodeGenerator()
+                .generateJsCode(relativeRequirePath = true, outJsProgram = false)
+                .also {
+                    performanceManager?.notifyIRGenerationFinished()
+                    performanceManager?.notifyGenerationFinished()
+                }
         }
     }
 

compiler/ir/backend.js/build.gradle.kts

@@ -4,6 +4,7 @@ plugins {
 }
 
 dependencies {
+    api(project(":compiler:cli-base"))
     api(project(":compiler:util"))
     api(project(":compiler:frontend"))
     api(project(":compiler:backend-common"))

compiler/ir/backend.js/src/org/jetbrains/kotlin/ir/backend/js/compiler.kt

@@ -8,6 +8,7 @@ package org.jetbrains.kotlin.ir.backend.js
 import org.jetbrains.kotlin.backend.common.linkage.issues.checkNoUnboundSymbols
 import org.jetbrains.kotlin.backend.common.phaser.PhaseConfig
 import org.jetbrains.kotlin.backend.common.phaser.PhaserState
+import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
 import org.jetbrains.kotlin.config.CompilerConfiguration
 import org.jetbrains.kotlin.ir.IrBuiltIns
 import org.jetbrains.kotlin.ir.backend.js.lower.*
@@ -48,6 +49,8 @@ fun compile(
     filesToLower: Set<String>? = null,
     granularity: JsGenerationGranularity = JsGenerationGranularity.WHOLE_PROGRAM,
 ): LoweredIr {
+    val performanceManager = depsDescriptors.compilerConfiguration[CLIConfigurationKeys.PERF_MANAGER]
+    performanceManager?.notifyIRTranslationStarted()
 
     val (moduleFragment: IrModuleFragment, dependencyModules, irBuiltIns, symbolTable, deserializer, moduleToName) =
         loadIr(depsDescriptors, irFactory, verifySignatures, filesToLower, loadFunctionInterfacesIntoStdlib = true)
@@ -93,6 +96,7 @@ fun compileIr(
     val moduleDescriptor = moduleFragment.descriptor
     val irFactory = symbolTable.irFactory
     val shouldGeneratePolyfills = configuration.getBoolean(JSConfigurationKeys.GENERATE_POLYFILLS)
+    val performanceManager = configuration[CLIConfigurationKeys.PERF_MANAGER]
 
     val allModules = when (mainModule) {
         is MainModule.SourceFiles -> dependencyModules + listOf(moduleFragment)
@@ -132,7 +136,10 @@ fun compileIr(
 
     // TODO should be done incrementally
     generateJsTests(context, allModules.last(), groupByPackage = false)
+    performanceManager?.notifyIRTranslationFinished()
 
+    performanceManager?.notifyGenerationStarted()
+    performanceManager?.notifyIRLoweringStarted()
     (irFactory.stageController as? WholeWorldStageController)?.let {
         lowerPreservingTags(allModules, context, phaseConfig, it)
     } ?: run {
@@ -143,6 +150,7 @@ fun compileIr(
             }
         }
     }
+    performanceManager?.notifyIRLoweringFinished()
 
     return LoweredIr(context, moduleFragment, allModules, moduleToName)
 }