Register Fir checkers to compiler processor

Summary:
$title

Fir compiler uses the FirExtensionRegistrar for all Fir extensions, which helps simplify the installation site in the compiler processor

Reviewed By: fbcbl

Differential Revision: D65447524

fbshipit-source-id: 4d777831f1f4b7a375b4e8b511140449b90ef019

litho-compiler-plugin/compiler/src/main/kotlin/com/facebook/litho/LithoComponentRegistrar.kt

@@ -19,13 +19,15 @@ package com.facebook.litho
 import com.facebook.litho.common.LithoCompilerConfig
 import com.facebook.litho.common.get
 import com.facebook.litho.k1.LithoFeCheckersContributor
+import com.facebook.litho.k2.LithoFirExtensionRegistrar
 import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
 import org.jetbrains.kotlin.cli.common.messages.CompilerMessageSeverity
 import org.jetbrains.kotlin.cli.common.messages.MessageCollector
 import org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar
 import org.jetbrains.kotlin.compiler.plugin.ExperimentalCompilerApi
 import org.jetbrains.kotlin.config.CompilerConfiguration
 import org.jetbrains.kotlin.extensions.StorageComponentContainerContributor
+import org.jetbrains.kotlin.fir.extensions.FirExtensionRegistrarAdapter
 import org.jetbrains.kotlin.resolve.extensions.AnalysisHandlerExtension
 
 /**
@@ -50,6 +52,7 @@ class LithoComponentRegistrar : CompilerPluginRegistrar() {
           })
     }
     // K2 extensions
+    FirExtensionRegistrarAdapter.registerExtension(LithoFirExtensionRegistrar())
     // Backend extensions
   }
 

litho-compiler-plugin/compiler/src/main/kotlin/com/facebook/litho/k2/LithoFirCheckersExtension.kt

@@ -0,0 +1,55 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * 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.facebook.litho.k2
+
+import com.facebook.litho.common.LithoNames
+import com.facebook.litho.k2.diagnostics.LithoFirHookUsageChecker
+import org.jetbrains.kotlin.fir.FirSession
+import org.jetbrains.kotlin.fir.analysis.checkers.expression.ExpressionCheckers
+import org.jetbrains.kotlin.fir.analysis.extensions.FirAdditionalCheckersExtension
+import org.jetbrains.kotlin.fir.extensions.FirDeclarationPredicateRegistrar
+import org.jetbrains.kotlin.fir.extensions.predicate.DeclarationPredicate
+
+/**
+ * Responsible for supplying additional checkers that will be called when the compiler analyses
+ * code. It can also report additional errors or warnings that can be visualized by the IDE.
+ *
+ * There are different kinds of checkers available, and can be as fine-grained as possible.
+ *
+ * @see [FirAdditionalCheckersExtension]
+ * @see [LithoFirHookUsageChecker]
+ */
+class LithoFirCheckersExtension(session: FirSession) : FirAdditionalCheckersExtension(session) {
+
+  override val expressionCheckers: ExpressionCheckers =
+      object : ExpressionCheckers() {
+        override val functionCallCheckers = setOf(LithoFirHookUsageChecker())
+      }
+
+  override fun FirDeclarationPredicateRegistrar.registerPredicates() {
+    register(predicate)
+  }
+
+  /**
+   * Defined a set of predicates that will be used to filter the declarations that should trigger
+   * the checkers defined in this extension.
+   */
+  private val predicate =
+      DeclarationPredicate.create {
+        metaAnnotated(LithoNames.Unconditional.asSingleFqName(), includeItself = false)
+      }
+}

litho-compiler-plugin/compiler/src/main/kotlin/com/facebook/litho/k2/LithoFirExtensionRegistrar.kt

@@ -0,0 +1,34 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * 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.facebook.litho.k2
+
+import org.jetbrains.kotlin.fir.extensions.FirExtensionRegistrar
+
+/**
+ * Responsible for registering all FIR specific extensions for Litho to the compiler.
+ *
+ * It exposes a shorthand API that makes it easy to register extensions correctly. The API
+ * implicitly passes an [FirSession] object to the extension constructor, and also provides
+ * mechanisms to pass explicit dependencies as necessary.
+ *
+ * @see [LithoFirCheckersExtension]
+ */
+class LithoFirExtensionRegistrar : FirExtensionRegistrar() {
+  override fun ExtensionRegistrarContext.configurePlugin() {
+    +::LithoFirCheckersExtension
+  }
+}