Implement given type to sort, support .h and .m now

marcuswu0814 · May 15, 2018 · a852495 · a852495
1 parent 3f5b3bd
commit a852495
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Sources/ObjcImportSorter/main.swift b/Sources/ObjcImportSorter/main.swift
@@ -1,10 +1,16 @@
 import Commander
 import ObjcImportSorterCore
+import Rainbow
 
 Group {
 
-    $0.command("sort") {
-        ImportSorter().sort(by: .h)
+    $0.command("sort") { (fileType: String) in
+        guard let fileType = FileType(rawValue: "*." + fileType) else {
+            print("Wrong given type".red)
+            return
+        }
+
+        ImportSorter().sort(by: fileType)
     }
 
 }.run()