WIP: extract annotations on classes, with constant values

java/ql/lib/semmle/code/java/Annotation.qll

@@ -249,7 +249,7 @@ private predicate filteredAnnotValue(Annotation a, Method m, Expr val) {
 
 private predicate sourceAnnotValue(Annotation a, Method m, Expr val) {
   annotValue(a, m, val) and
-  val.getFile().getExtension() = "java"
+  val.getFile().isSourceFile()
 }
 
 /** An abstract representation of language elements that can be annotated. */

java/ql/lib/semmle/code/java/DependencyCounts.qll

@@ -91,7 +91,7 @@ predicate numDepends(RefType t, RefType dep, int value) {
         elem = a and usesType(a.getType(), dep)
         or
         elem = [a.getValue(_), a.getAnArrayValue(_)] and
-        elem.getFile().getExtension() = "java" and
+        elem.getFile().isSourceFile() and
         usesType(elem.(Expr).getType(), dep)
       )
       or