Only attempt to remove file headers from swift files

stackotter · Jan 12, 2022 · 333f456 · 333f456
1 parent 2f84f51
commit 333f456
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/swift-bundler/Subcommands/RemoveFileHeaders.swift b/Sources/swift-bundler/Subcommands/RemoveFileHeaders.swift
@@ -18,7 +18,7 @@ extension Bundler {
       if let enumerator = FileManager.default.enumerator(at: sourcesDir, includingPropertiesForKeys: [.isRegularFileKey], options: [.skipsHiddenFiles, .skipsPackageDescendants]) {
         for case let fileURL as URL in enumerator {
           let fileAttributes = try fileURL.resourceValues(forKeys: [.isRegularFileKey])
-          if fileAttributes.isRegularFile! {
+          if fileAttributes.isRegularFile! && fileURL.pathExtension == "swift" {
             contents.append(fileURL)
           }
         }
@@ -39,4 +39,4 @@ extension Bundler {
       }
     }
   }
-}
+}