An exception null cannot be cast to non-null type org.jacodb.api.JcClassType occurs #200 (#201)

Author: lehvolk

jacodb-api/src/main/kotlin/org/jacodb/api/Exceptions.kt

@@ -21,6 +21,12 @@ package org.jacodb.api
  */
 class NoClassInClasspathException(val className: String) : Exception("Class $className not found in classpath")
 
+/**
+ * This exception should be thrown when classpath is incomplete
+ */
+class TypeNotFoundException(val typeName: String) : Exception("Type $typeName not found in classpath")
+
+class MethodNotFoundException(msg: String) : Exception(msg)
 
 fun String.throwClassNotFound(): Nothing {
     throw NoClassInClasspathException(this)

jacodb-api/src/main/kotlin/org/jacodb/api/ext/JcTypes.kt

@@ -193,4 +193,8 @@ val JcTypedMethod.humanReadableSignature: String get() {
         it.joinToString(prefix = "<", separator = ",", postfix = ">") { it.symbol }
     } ?: ""
     return "${enclosingType.typeName}#$generics$name($params):${returnType.typeName}"
+}
+
+fun JcClasspath.findType(name: String): JcType {
+    return findTypeOrNull(name) ?: throw TypeNotFoundException(name)
 }

jacodb-core/src/main/kotlin/org/jacodb/impl/cfg/TypedMethodRefImpl.kt

@@ -18,6 +18,7 @@ package org.jacodb.impl.cfg
 
 import org.jacodb.api.*
 import org.jacodb.api.cfg.*
+import org.jacodb.api.ext.findType
 import org.jacodb.api.ext.jvmName
 import org.jacodb.impl.cfg.util.typeName
 import org.jacodb.impl.softLazy
@@ -76,7 +77,7 @@ abstract class MethodSignatureRef(
         }
 
     fun JcType.throwNotFoundException(): Nothing {
-        throw IllegalStateException(this.methodNotFoundMessage)
+        throw MethodNotFoundException(this.methodNotFoundMessage)
     }
 
 }
@@ -89,14 +90,14 @@ class TypedStaticMethodRefImpl(
 ) : MethodSignatureRef(type, name, argTypes, returnType) {
 
     constructor(classpath: JcClasspath, raw: JcRawStaticCallExpr) : this(
-            classpath.findTypeOrNull(raw.declaringClass.typeName) as JcClassType,
+            classpath.findType(raw.declaringClass.typeName) as JcClassType,
             raw.methodName,
             raw.argumentTypes,
             raw.returnType
     )
 
     override val method: JcTypedMethod by weakLazy {
-        type.lookup.staticMethod(name, description) ?: throw IllegalStateException(methodNotFoundMessage)
+        type.lookup.staticMethod(name, description) ?: type.throwNotFoundException()
     }
 }
 
@@ -108,7 +109,7 @@ class TypedSpecialMethodRefImpl(
 ) : MethodSignatureRef(type, name, argTypes, returnType) {
 
     constructor(classpath: JcClasspath, raw: JcRawSpecialCallExpr) : this(
-            classpath.findTypeOrNull(raw.declaringClass.typeName) as JcClassType,
+            classpath.findType(raw.declaringClass.typeName) as JcClassType,
             raw.methodName,
             raw.argumentTypes,
             raw.returnType
@@ -130,7 +131,7 @@ class VirtualMethodRefImpl(
 
     companion object {
         private fun JcRawCallExpr.resolvedType(classpath: JcClasspath): Pair<JcClassType, JcClassType> {
-            val declared = classpath.findTypeOrNull(declaringClass.typeName) as JcClassType
+            val declared = classpath.findType(declaringClass.typeName) as JcClassType
             if (this is JcRawInstanceExpr) {
                 val instance = instance
                 if (instance is JcRawLocal) {
@@ -182,7 +183,7 @@ class TypedMethodRefImpl(
 ) : MethodSignatureRef(type, name, argTypes, returnType) {
 
     constructor(classpath: JcClasspath, raw: JcRawCallExpr) : this(
-            classpath.findTypeOrNull(raw.declaringClass.typeName) as JcClassType,
+            classpath.findType(raw.declaringClass.typeName) as JcClassType,
             raw.methodName,
             raw.argumentTypes,
             raw.returnType