Enums can also be entry points

peripheryapp · Aug 20, 2024 · 5bf1c46 · 5bf1c46
1 parent 531b0aa
commit 5bf1c46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,7 +10,7 @@
 
 ##### Bug Fixes
 
-- None.
+- Enums with the `@main` attribute are now retained.
 
 ## 2.21.0 (2024-06-15)
 

diff --git a/Sources/SourceGraph/Mutators/EntryPointAttributeRetainer.swift b/Sources/SourceGraph/Mutators/EntryPointAttributeRetainer.swift
@@ -10,7 +10,7 @@ final class EntryPointAttributeRetainer: SourceGraphMutator {
 
     func mutate() {
         graph
-            .declarations(ofKinds: [.class, .struct])
+            .declarations(ofKinds: [.class, .struct, .enum])
             .lazy
             .filter {
                 $0.attributes.contains("NSApplicationMain") ||
-Original file line number
+Diff line change
@@ Expand Up / @@ -10,7 +10,7 @@ @@
     ##### Bug Fixes
-    - None.
+    - Enums with the `@main` attribute are now retained.
     ## 2.21.0 (2024-06-15)
@@ Expand Down @@