From 68a47d6efb614e77cba0fdee7c8c82d34ad4516c Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Wed, 23 Sep 2020 18:27:42 +0300
Subject: [PATCH] [FIR] Add helpers for creating diagnostics on nullable source

---
 .../diagnostics/FirDiagnosticFactory.kt       | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/FirDiagnosticFactory.kt b/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/FirDiagnosticFactory.kt
index 6184b81708b..57728ae4ac3 100644
--- a/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/FirDiagnosticFactory.kt
+++ b/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/FirDiagnosticFactory.kt
@@ -83,3 +83,28 @@ class FirDiagnosticFactory3<E : FirSourceElement, P : PsiElement, A : Any, B : A
 private fun incorrectElement(element: FirSourceElement): Nothing {
     throw IllegalArgumentException("Unknown element type: ${element::class}")
 }
+
+fun <E : FirSourceElement, P : PsiElement> FirDiagnosticFactory0<E, P>.on(element: E?): FirSimpleDiagnostic<E>? {
+    return element?.let { on(it) }
+}
+
+fun <E : FirSourceElement, P : PsiElement, A : Any> FirDiagnosticFactory1<E, P, A>.on(element: E?, a: A): FirDiagnosticWithParameters1<E, A>? {
+    return element?.let { on(it, a) }
+}
+
+fun <E : FirSourceElement, P : PsiElement, A : Any, B : Any> FirDiagnosticFactory2<E, P, A, B>.on(
+    element: E?,
+    a: A,
+    b: B
+): FirDiagnosticWithParameters2<E, A, B>? {
+    return element?.let { on(it, a, b) }
+}
+
+fun <E : FirSourceElement, P : PsiElement, A : Any, B : Any, C : Any> FirDiagnosticFactory3<E, P, A, B, C>.on(
+    element: E?,
+    a: A,
+    b: B,
+    c: C
+): FirDiagnosticWithParameters3<E, A, B, C>? {
+    return element?.let { on(it, a, b, c) }
+}