From e1e920e1e678293cae5ad555ab5c029ad73c1587 Mon Sep 17 00:00:00 2001
From: Edwin Jakobs <edwin@rndr.studio>
Date: Fri, 22 May 2020 21:19:57 +0200
Subject: [PATCH] Add null checks to LambdaExtractor.enterAnnotatedLambda

---
 orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt b/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt
index b4e37d8b..7cf88a25 100644
--- a/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt
+++ b/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt
@@ -43,11 +43,13 @@ class LambdaExtractor(val ruleNames: List<String>, val lambdaName: String) : Kot
 
     var result: String? = null
     override fun enterAnnotatedLambda(ctx: KotlinParser.AnnotatedLambdaContext?) {
-        val puec = ctx!!.parent!!.parent!!.parent!! as KotlinParser.PostfixUnaryExpressionContext
-        val identifier = puec.primaryExpression().simpleIdentifier().Identifier().text
-        if (identifier == lambdaName) {
-            if (result == null) {
-                result = ctx.verbatimText(1, 1)
+        val puec = ctx?.parent?.parent?.parent as? KotlinParser.PostfixUnaryExpressionContext
+        if (puec != null) {
+            val identifier = puec.primaryExpression()?.simpleIdentifier()?.Identifier()?.text
+            if (identifier == lambdaName) {
+                if (result == null) {
+                    result = ctx.verbatimText(1, 1)
+                }
             }
         }
     }