Kotlin: Add WhenBranch.getWhenExpr()

igfoo · igfoo · commit 7baefbb9e580 · 2022-05-10T19:51:06.000+01:00
diff --git a/java/ql/lib/semmle/code/java/Expr.qll b/java/ql/lib/semmle/code/java/Expr.qll
@@ -2201,6 +2201,11 @@ class WhenBranch extends Top, @whenbranch {
   /** Holds if this is an `else` branch. */
   predicate isElseBranch() { when_branch_else(this) }
 
+  /** Gets the `when` expression this is a branch of. */
+  WhenExpr getWhenExpr() {
+    this = result.getBranch(_)
+  }
+
   override string toString() { result = "... -> ..." }
 
   override string getAPrimaryQlClass() { result = "WhenBranch" }
