Fui: Add support for .mm files.

CHANGELOG.md

@@ -1,6 +1,7 @@
 ### 0.4.1 (Next)
 
 * Your contribution here.
+* [#24](https://github.com/dblock/fui/pull/24): Support .mm files - [@shachlan](https://github.com/Shachlan).
 
 ### 0.4.0 (5/14/2016)
 

lib/fui/finder.rb

@@ -20,7 +20,7 @@ def references(&block)
         end
         Find.find(path) do |path|
           next unless File.ftype(path) == 'file'
-          if ['.m', '.h', '.pch'].include?(File.extname(path))
+          if ['.m', '.mm', '.h', '.pch'].include?(File.extname(path))
             process_code references, path, &block
           elsif ['.storyboard', '.xib'].include?(File.extname(path))
             process_xml references, path, &block

spec/fixtures/mm/main.mm

@@ -0,0 +1,6 @@
+#import "used_class.h"
+
+- void main()
+{
+
+}

spec/fixtures/mm/unused_class.h

@@ -0,0 +1,8 @@
+//
+//  UnusedClass.h
+//  FUI
+//
+
+@interface UnusedClass
+
+@end

spec/fixtures/mm/unused_class.mm

@@ -0,0 +1,10 @@
+//
+//  UnusedClass.mm
+//  FUI
+//
+
+#import "unused_class.h"
+
+@implementation UnusedClass
+
+@end

spec/fixtures/mm/used_class.h

@@ -0,0 +1,10 @@
+//
+//  ImageView.h
+//  FUI
+//
+
+#import <UIKit/UIKit.h>
+
+@interface ImageView
+- (NSString *)fooForBar;
+@end

spec/fixtures/mm/used_class.mm

@@ -0,0 +1,14 @@
+
+//
+//  UsedClass.mm
+//  FUI
+//
+
+#import "used_class.h"
+
+@implementation UsedClass
+- (NSString *)fooForBar
+{
+
+}
+@end

spec/fui/finder_spec.rb

@@ -34,6 +34,39 @@
       end
     end
   end
+  context 'included from a .mm file' do
+    before :each do
+      @fixtures_dir = File.expand_path(File.join(__FILE__, '../../fixtures/mm'))
+    end
+    describe '#find' do
+      it 'finds all files for which the block yields true' do
+        files = Fui::Finder.send(:find, @fixtures_dir) do |file|
+          File.extname(file) == '.h'
+        end
+        expect(files.sort).to eq Dir["#{@fixtures_dir}/*.h"].sort
+      end
+    end
+    describe '#headers' do
+      it 'finds all headers' do
+        finder = Fui::Finder.new(@fixtures_dir)
+        expect(finder.headers.map(&:filename).sort).to eq(['unused_class.h', 'used_class.h'])
+      end
+    end
+    describe '#references' do
+      it 'maps references' do
+        finder = Fui::Finder.new(@fixtures_dir)
+        expect(finder.references.size).to eq(2)
+        expect(Hash[finder.references.map { |k, v| [k.filename, v.count] }]).to eq('unused_class.h' => 0,
+                                                                                   'used_class.h' => 1)
+      end
+    end
+    describe '#unsed_references' do
+      it 'finds unused references' do
+        finder = Fui::Finder.new(@fixtures_dir)
+        expect(Hash[finder.unused_references.map { |k, v| [k.filename, v.count] }]).to eq('unused_class.h' => 0)
+      end
+    end
+  end
   context 'included from a .pch file' do
     before :each do
       @fixtures_dir = File.expand_path(File.join(__FILE__, '../../fixtures/pch'))