Java/C#: Share model coverage code.

shared/mad/codeql/mad/static/MaD.qll

@@ -121,4 +121,63 @@ module ModelsAsData<ExtensionsSig Extensions> {
           "; " + ext + "; " + input + "; " + output + "; " + kind + "; " + provenance
     )
   }
+
+  private predicate relevantNamespace(string namespace) {
+    Extensions::sourceModel(namespace, _, _, _, _, _, _, _, _, _) or
+    Extensions::sinkModel(namespace, _, _, _, _, _, _, _, _, _) or
+    Extensions::summaryModel(namespace, _, _, _, _, _, _, _, _, _, _)
+  }
+
+  private predicate namespaceLink(string shortns, string longns) {
+    relevantNamespace(shortns) and
+    relevantNamespace(longns) and
+    longns.prefix(longns.indexOf(".")) = shortns
+  }
+
+  private predicate canonicalNamespace(string namespace) {
+    relevantNamespace(namespace) and not namespaceLink(_, namespace)
+  }
+
+  private predicate canonicalNamespaceLink(string namespace, string subns) {
+    canonicalNamespace(namespace) and
+    (subns = namespace or namespaceLink(namespace, subns))
+  }
+
+  /**
+   * Holds if MaD framework coverage of `namespace` is `n` api endpoints of the
+   * kind `(kind, part)`, and `namespaces` is the number of subnamespaces of
+   * `namespace` which have MaD framework coverage (including `namespace`
+   * itself).
+   */
+  predicate modelCoverage(string namespace, int namespaces, string kind, string part, int n) {
+    namespaces = strictcount(string subns | canonicalNamespaceLink(namespace, subns)) and
+    (
+      part = "source" and
+      n =
+        strictcount(string subns, string type, boolean subtypes, string name, string signature,
+          string ext, string output, string provenance |
+          canonicalNamespaceLink(namespace, subns) and
+          Extensions::sourceModel(subns, type, subtypes, name, signature, ext, output, kind,
+            provenance, _)
+        )
+      or
+      part = "sink" and
+      n =
+        strictcount(string subns, string type, boolean subtypes, string name, string signature,
+          string ext, string input, string provenance |
+          canonicalNamespaceLink(namespace, subns) and
+          Extensions::sinkModel(subns, type, subtypes, name, signature, ext, input, kind,
+            provenance, _)
+        )
+      or
+      part = "summary" and
+      n =
+        strictcount(string subns, string type, boolean subtypes, string name, string signature,
+          string ext, string input, string output, string provenance |
+          canonicalNamespaceLink(namespace, subns) and
+          Extensions::summaryModel(subns, type, subtypes, name, signature, ext, input, output, kind,
+            provenance, _)
+        )
+    )
+  }
 }