From 4deff601b0a407fd5c980027291b33a5bb0fd0c3 Mon Sep 17 00:00:00 2001
From: 1024jp <1024jp@wolfrosch.com>
Date: Fri, 18 Mar 2022 18:16:35 +0900
Subject: [PATCH] Refactor SyntaxMapBuilder

---
 SyntaxMapBuilder/Sources/SyntaxMapBuilder/main.swift | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/SyntaxMapBuilder/Sources/SyntaxMapBuilder/main.swift b/SyntaxMapBuilder/Sources/SyntaxMapBuilder/main.swift
index 19377071f..dbd1f2739 100644
--- a/SyntaxMapBuilder/Sources/SyntaxMapBuilder/main.swift
+++ b/SyntaxMapBuilder/Sources/SyntaxMapBuilder/main.swift
@@ -24,6 +24,7 @@
 //
 
 import Foundation
+import UniformTypeIdentifiers
 import Yams
 
 private struct SyntaxStyle: Codable {
@@ -43,8 +44,8 @@ private func buildSyntaxMap(directoryPath: String) throws -> String {
     
     // find syntax style files
     let directoryURL = URL(fileURLWithPath: directoryPath, isDirectory: true)
-    let urls = try FileManager.default.contentsOfDirectory(at: directoryURL, includingPropertiesForKeys: nil)
-        .filter { $0.pathExtension == "yml" }
+    let urls = try FileManager.default.contentsOfDirectory(at: directoryURL, includingPropertiesForKeys: [.contentTypeKey])
+        .filter { try $0.resourceValues(forKeys: [.contentTypeKey]).contentType?.conforms(to: .yaml) == true }
     
     // build syntaxMap from syntax style files
     let decoder = YAMLDecoder()