Swift: define ImportDecl:imported_module optional

Author: redsun82

swift/codegen/schema.yml

@@ -311,7 +311,7 @@ IfConfigClause:
 ImportDecl:
   _extends: Decl
   is_exported: predicate
-  imported_module: ModuleDecl
+  imported_module: ModuleDecl?  # may be none in inactive #if clauses
   declarations: ValueDecl*
 
 MissingMemberDecl:

swift/ql/lib/codeql/swift/generated/Raw.qll

@@ -607,7 +607,7 @@ module Raw {
 
     predicate isExported() { import_decl_is_exported(this) }
 
-    ModuleDecl getImportedModule() { import_decls(this, result) }
+    ModuleDecl getImportedModule() { import_decl_imported_modules(this, result) }
 
     ValueDecl getDeclaration(int index) { import_decl_declarations(this, index, result) }
   }

swift/ql/lib/codeql/swift/generated/decl/ImportDecl.qll

@@ -19,6 +19,8 @@ class ImportDeclBase extends Synth::TImportDecl, Decl {
 
   final ModuleDecl getImportedModule() { result = getImmediateImportedModule().resolve() }
 
+  final predicate hasImportedModule() { exists(getImportedModule()) }
+
   ValueDecl getImmediateDeclaration(int index) {
     result =
       Synth::convertValueDeclFromRaw(Synth::convertImportDeclToRaw(this)

swift/ql/lib/swift.dbscheme

@@ -703,15 +703,20 @@ if_config_clause_is_active(  //dir=decl
 );
 
 import_decls(  //dir=decl
-  unique int id: @import_decl,
-  int imported_module: @module_decl ref
+  unique int id: @import_decl
 );
 
 #keyset[id]
 import_decl_is_exported(  //dir=decl
   int id: @import_decl ref
 );
 
+#keyset[id]
+import_decl_imported_modules(  //dir=decl
+  int id: @import_decl ref,
+  int imported_module: @module_decl ref
+);
+
 #keyset[id, index]
 import_decl_declarations(  //dir=decl
   int id: @import_decl ref,

swift/ql/test/extractor-tests/generated/decl/ImportDecl/ImportDecl.ql

@@ -2,12 +2,10 @@
 import codeql.swift.elements
 import TestUtils
 
-from ImportDecl x, ModuleDecl getModule, string isExported, ModuleDecl getImportedModule
+from ImportDecl x, ModuleDecl getModule, string isExported
 where
   toBeTested(x) and
   not x.isUnknown() and
   getModule = x.getModule() and
-  (if x.isExported() then isExported = "yes" else isExported = "no") and
-  getImportedModule = x.getImportedModule()
-select x, "getModule:", getModule, "isExported:", isExported, "getImportedModule:",
-  getImportedModule
+  if x.isExported() then isExported = "yes" else isExported = "no"
+select x, "getModule:", getModule, "isExported:", isExported

swift/ql/test/extractor-tests/generated/decl/ImportDecl/ImportDecl_getImportedModule.ql

@@ -0,0 +1,7 @@
+// generated by codegen/codegen.py
+import codeql.swift.elements
+import TestUtils
+
+from ImportDecl x
+where toBeTested(x) and not x.isUnknown()
+select x, x.getImportedModule()