Merge pull request #1 from insanoid/swift-support

Swift support added.
insanoid · Oct 15, 2015 · 1948c2f · 1948c2f
2 parents e093a9f + 70c9b5b
commit 1948c2f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/CleanHeaders/CleanHeaders.m b/CleanHeaders/CleanHeaders.m
@@ -68,10 +68,12 @@ - (void)cleanHeaderAction {
 }
 
 - (NSString *)formatSelection:(NSString *)content {
-  // Let's assume all header files start with an #import, #include, or @import.
+  // Let's assume all header files start with an #import, #include, @import,
+  // import.
   NSString *traditionalImportPrefix = @"#import";
   NSString *frameworkImportPrefix = @"@import";
   NSString *traditionalIncludePrefix = @"#include";
+  NSString *swiftPrefix = @"import";
 
   // Convert the entire source into an array based on new lines.
   // Hence the imports have to be alteast in new line to work.
@@ -95,7 +97,8 @@ - (NSString *)formatSelection:(NSString *)content {
 
     BOOL isLineHeader = [cleansedLine hasPrefix:traditionalImportPrefix] ||
                         [cleansedLine hasPrefix:traditionalIncludePrefix] ||
-                        [cleansedLine hasPrefix:frameworkImportPrefix];
+                        [cleansedLine hasPrefix:frameworkImportPrefix] ||
+                        [cleansedLine hasPrefix:swiftPrefix];
 
     // If the line is a header and no header element has been detected so far,
     // mark this as the start of the header segment.
@@ -169,7 +172,7 @@ - (void)formatRanges:(NSArray *)ranges
 
 - (BOOL)shouldFormat:(IDESourceCodeDocument *)document {
   return [[NSSet setWithObjects:@"c", @"h", @"cpp", @"cc", @"cxx", @"hh",
-                                @"hpp", @"ipp", @"m", @"mm", nil]
+                                @"hpp", @"ipp", @"m", @"mm", @"swift", nil]
       containsObject:[[[document fileURL] pathExtension] lowercaseString]];
 }
 

diff --git a/README.md b/README.md
@@ -2,6 +2,7 @@
 
 An Xcode plug-in to format your import headers in a systematic manner. It simply removes duplicates, spaces and sorts them alphabetically making it much more easier to read and avoid duplicate imports. Works with `@imports` and `#include` along with `#import`.
 
+
 ![Preview](https://raw.githubusercontent.com/insanoid/CleanHeaders-Xcode/master/diff_image.png)
 
 ## Installation
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,6 +2,7 @@

		An Xcode plug-in to format your import headers in a systematic manner. It simply removes duplicates, spaces and sorts them alphabetically making it much more easier to read and avoid duplicate imports. Works with `@imports` and `#include` along with `#import`.


		![Preview](https://raw.githubusercontent.com/insanoid/CleanHeaders-Xcode/master/diff_image.png)

		## Installation
Expand Down