Swift: extract extension declarations

AlexDenisov · AlexDenisov · commit cf0fc5829fdc · 2022-06-14T08:22:43.000+02:00
diff --git a/swift/extractor/visitors/DeclVisitor.h b/swift/extractor/visitors/DeclVisitor.h
@@ -193,6 +193,13 @@ class DeclVisitor : public AstVisitorBase<DeclVisitor> {
     emitAbstractStorageDecl(decl, label);
   }
 
+  void visitExtensionDecl(swift::ExtensionDecl* decl) {
+    auto label = dispatcher_.assignNewLabel(decl);
+    dispatcher_.emit(ExtensionDeclsTrap{label});
+    emitGenericContext(decl, label);
+    emitIterableDeclContext(decl, label);
+  }
+
  private:
   void emitConstructorDecl(swift::ConstructorDecl* decl, TrapLabel<ConstructorDeclTag> label) {
     emitAbstractFunctionDecl(decl, label);
diff --git a/swift/ql/test/extractor-tests/declarations/all.expected b/swift/ql/test/extractor-tests/declarations/all.expected
@@ -205,3 +205,7 @@
 | declarations.swift:132:5:132:5 | newValue |
 | declarations.swift:132:5:132:15 | willSet |
 | declarations.swift:134:5:134:14 | didSet |
+| declarations.swift:138:1:142:1 | extension |
+| declarations.swift:139:3:141:3 | id |
+| declarations.swift:139:8:139:8 | self |
+| declarations.swift:144:1:144:7 | { ... } |
diff --git a/swift/ql/test/extractor-tests/declarations/declarations.swift b/swift/ql/test/extractor-tests/declarations/declarations.swift
@@ -133,4 +133,13 @@ struct HasPropertyAndObserver {
 
     didSet { }
   }
-}
+}
+
+extension Int {
+  func id() -> Int {
+    return self
+  }
+}
+
+42.id()
+
diff --git a/swift/ql/test/extractor-tests/declarations/func.expected b/swift/ql/test/extractor-tests/declarations/func.expected
@@ -55,3 +55,4 @@
 | declarations.swift:131:7:131:7 | set | (unnamed function decl) |
 | declarations.swift:132:5:132:15 | willSet | (unnamed function decl) |
 | declarations.swift:134:5:134:14 | didSet | (unnamed function decl) |
+| declarations.swift:139:3:141:3 | id | id |
diff --git a/swift/ql/test/extractor-tests/declarations/func_body.expected b/swift/ql/test/extractor-tests/declarations/func_body.expected
@@ -50,3 +50,4 @@
 | declarations.swift:131:7:131:7 | set | declarations.swift:131:7:131:7 | { ... } |
 | declarations.swift:132:5:132:15 | willSet | declarations.swift:132:13:132:15 | { ... } |
 | declarations.swift:134:5:134:14 | didSet | declarations.swift:134:12:134:14 | { ... } |
+| declarations.swift:139:3:141:3 | id | declarations.swift:139:20:141:3 | { ... } |
diff --git a/swift/ql/test/library-tests/parent/declarations.swift b/swift/ql/test/library-tests/parent/declarations.swift
@@ -133,4 +133,13 @@ struct HasPropertyAndObserver {
 
     didSet { }
   }
-}
+}
+
+extension Int {
+  func id() -> Int {
+    return self
+  }
+}
+
+42.id()
+
diff --git a/swift/ql/test/library-tests/parent/parent.expected b/swift/ql/test/library-tests/parent/parent.expected
@@ -326,6 +326,13 @@
 | declarations.swift:132:5:132:15 | willSet | AccessorDecl | declarations.swift:132:5:132:5 | newValue | ParamDecl |
 | declarations.swift:132:5:132:15 | willSet | AccessorDecl | declarations.swift:132:13:132:15 | { ... } | BraceStmt |
 | declarations.swift:134:5:134:14 | didSet | AccessorDecl | declarations.swift:134:12:134:14 | { ... } | BraceStmt |
+| declarations.swift:139:3:141:3 | id | ConcreteFuncDecl | declarations.swift:139:20:141:3 | { ... } | BraceStmt |
+| declarations.swift:139:20:141:3 | { ... } | BraceStmt | declarations.swift:140:5:140:12 | return ... | ReturnStmt |
+| declarations.swift:140:5:140:12 | return ... | ReturnStmt | declarations.swift:140:12:140:12 | self | DeclRefExpr |
+| declarations.swift:144:1:144:4 | call to id | DotSyntaxCallExpr | declarations.swift:144:4:144:4 | id | DeclRefExpr |
+| declarations.swift:144:1:144:7 | call to ... | CallExpr | declarations.swift:144:1:144:4 | call to id | DotSyntaxCallExpr |
+| declarations.swift:144:1:144:7 | { ... } | BraceStmt | declarations.swift:144:1:144:7 | call to ... | CallExpr |
+| declarations.swift:144:1:144:7 | { ... } | TopLevelCodeDecl | declarations.swift:144:1:144:7 | { ... } | BraceStmt |
 | expressions.swift:1:1:1:9 | var ... = ... | PatternBindingDecl | expressions.swift:1:5:1:5 | a | NamedPattern |
 | expressions.swift:1:1:1:9 | var ... = ... | PatternBindingDecl | expressions.swift:1:9:1:9 | 15 | IntegerLiteralExpr |
 | expressions.swift:1:1:1:9 | { ... } | BraceStmt | expressions.swift:1:1:1:9 | var ... = ... | PatternBindingDecl |
