fixes sourceplusplus/sourceplusplus#421

Author: BFergerson

marker/jvm-marker/src/main/kotlin/spp/jetbrains/marker/jvm/JVMLineMarkerProvider.kt

@@ -28,6 +28,7 @@ import org.jetbrains.plugins.groovy.lang.psi.GroovyFile
 import org.jetbrains.uast.UClass
 import org.jetbrains.uast.UMethod
 import org.jetbrains.uast.toUElement
+import org.jetbrains.uast.toUElementOfType
 import org.slf4j.LoggerFactory
 import spp.jetbrains.marker.SourceMarker
 import spp.jetbrains.marker.impl.ArtifactCreationService
@@ -95,7 +96,7 @@ abstract class JVMLineMarkerProvider : SourceLineMarkerProvider() {
 
     private fun getMethodGutterMark(element: PsiElement): LineMarkerInfo<PsiElement>? {
         val fileMarker = SourceMarker.getSourceFileMarker(element.containingFile) ?: return null
-        val uMethod = element.parent.toUElement() as UMethod?
+        val uMethod = element.parent.toUElementOfType<UMethod>()
         if (uMethod == null) {
             log.warn("Unable to transform to UMethod: {}", element.parent)
             return null

marker/jvm-marker/src/main/kotlin/spp/jetbrains/marker/jvm/psi/LoggerDetector.kt

@@ -32,7 +32,7 @@ import kotlinx.coroutines.GlobalScope
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.runBlocking
 import org.jetbrains.uast.UMethod
-import org.jetbrains.uast.toUElement
+import org.jetbrains.uast.toUElementOfType
 import org.slf4j.LoggerFactory
 import spp.jetbrains.marker.source.SourceFileMarker
 import spp.jetbrains.marker.source.mark.api.MethodSourceMark
@@ -91,7 +91,7 @@ class LoggerDetector(val vertx: Vertx) {
             log.trace("Found logger statements: $loggerStatements")
             loggerStatements
         } else {
-            val uMethod = sourceMark.getPsiMethod().toUElement() as UMethod?
+            val uMethod = sourceMark.getPsiMethod().toUElementOfType<UMethod>()
             if (uMethod != null) {
                 val foundLoggerStatements = getOrFindLoggerStatements(uMethod).await()
                 sourceMark.putUserData(LOGGER_STATEMENTS, foundLoggerStatements)