added Ast.astInfoOrNull helper refs #21

kotlinx · Dec 17, 2020 · 8cef2d8 · 8cef2d8
1 parent fe958fe
commit 8cef2d8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/common/src/commonMain/kotlin/kotlinx/ast/common/ast/AstWithAstInfo.kt b/common/src/commonMain/kotlin/kotlinx/ast/common/ast/AstWithAstInfo.kt
@@ -11,6 +11,15 @@ interface AstWithAstInfo : AstWithAttachments {
     }
 }
 
+val Ast.astInfoOrNull: AstInfo?
+    get() {
+        return if (this is AstWithAstInfo) {
+            info
+        } else {
+            null
+        }
+    }
+
 interface AstSelfTypedWithAstInfo<Self> : AstSelfTypedWithAttachments<Self>, AstWithAstInfo
         where Self : AstSelfTypedWithAttachments<Self>, Self : AstWithAstInfo {
     override fun withAstInfo(info: AstInfo?): Self {

diff --git a/common/src/commonMain/kotlin/kotlinx/ast/common/ast/AstWithRawAst.kt b/common/src/commonMain/kotlin/kotlinx/ast/common/ast/AstWithRawAst.kt
@@ -17,7 +17,7 @@ interface AstWithRawAst : AstWithAttachments {
     }
 }
 
-fun Ast.rawOrNull(): RawAst? {
+fun Ast.rawAstOrNull(): RawAst? {
     return if(this is AstWithRawAst) {
         raw
     } else {