Support pods

bin/slather

@@ -17,6 +17,7 @@ Clamp do
     option ["--simple-output", "-s"], :flag, "Post coverage results to coveralls"
 
     option ["--build-directory", "-b"], "BUILD_DIRECTORY", "The directory where gcno files will be written to. Defaults to derived data."
+    option ["--source-directory"], "SOURCE_DIRECTORY", "The directory where your source files are located."
     option ["--ignore", "-i"], "IGNORE", "ignore files conforming to a path", :multivalued => true
 
     def execute
@@ -25,6 +26,7 @@ Clamp do
       setup_service_name
       setup_ignore_list
       setup_build_directory
+      setup_source_directory
       setup_coverage_service
 
       post
@@ -36,6 +38,10 @@ Clamp do
       project.build_directory = build_directory if build_directory
     end
 
+    def setup_source_directory
+      project.source_directory = source_directory if source_directory
+    end
+
     def setup_ignore_list
       project.ignore_list = ignore_list if !ignore_list.empty?
     end

lib/slather.rb

@@ -9,4 +9,10 @@ module Slather
 
   Encoding.default_external = "utf-8"
 
+  def self.prepare_pods(pods)
+    pods.post_install do |installer|
+      installer.project.slather_setup_for_coverage
+    end
+  end
+
 end

lib/slather/coverage_file.rb

@@ -12,8 +12,15 @@ def source_file_pathname
       @source_file_pathname ||= begin
         base_filename = gcno_file_pathname.basename.sub_ext("")
         # TODO: Handle Swift
-        pbx_file = project.files.detect { |pbx_file| pbx_file.real_path.basename.to_s == "#{base_filename}.m" }        
-        pbx_file && pbx_file.real_path
+        path = nil
+        if project.source_directory
+          path = Dir["#{project.source_directory}/**/#{base_filename}.m"].first
+          path &&= Pathname(path)
+        else
+          pbx_file = project.files.detect { |pbx_file| pbx_file.real_path.basename.to_s == "#{base_filename}.m" }
+          path = pbx_file && pbx_file.real_path
+        end
+        path
       end
     end
 
@@ -26,7 +33,7 @@ def source_data
     end
 
     def source_file_pathname_relative_to_repo_root
-      source_file_pathname.relative_path_from(Pathname("./").realpath)
+      source_file_pathname.realpath.relative_path_from(Pathname("./").realpath)
     end
 
     def gcov_data

lib/slather/project.rb

@@ -3,10 +3,25 @@
 require 'json'
 require 'yaml'
 
+module Xcodeproj
+  class Project
+
+    def slather_setup_for_coverage
+      build_configurations.each do |build_configuration|
+        build_configuration.build_settings["GCC_INSTRUMENT_PROGRAM_FLOW_ARCS"] = "YES"
+        build_configuration.build_settings["GCC_GENERATE_TEST_COVERAGE_FILES"] = "YES"
+      end
+    end
+
+  end
+end
+
 module Slather
   class Project < Xcodeproj::Project
 
-    attr_accessor :build_directory, :ignore_list, :ci_service, :coverage_service
+    attr_accessor :build_directory, :ignore_list, :ci_service, :coverage_service, :source_directory
+
+    alias_method :setup_for_coverage, :slather_setup_for_coverage
 
     def self.open(xcodeproj)
       proj = super
@@ -51,12 +66,17 @@ def configure_from_yml
       configure_ignore_list_from_yml
       configure_ci_service_from_yml
       configure_coverage_service_from_yml
+      configure_source_directory_from_yml
     end
 
     def configure_build_directory_from_yml
       self.build_directory = self.class.yml["build_directory"] if self.class.yml["build_directory"] && !@build_directory
     end
 
+    def configure_source_directory_from_yml
+      self.source_directory ||= self.class.yml["source_directory"] if self.class.yml["source_directory"]
+    end
+
     def configure_ignore_list_from_yml
       self.ignore_list ||= [(self.class.yml["ignore"] || [])].flatten
     end
@@ -85,13 +105,6 @@ def coverage_service=(service)
       @coverage_service = service
     end
 
-    def setup_for_coverage
-      build_configurations.each do |build_configuration|
-        build_configuration.build_settings["GCC_INSTRUMENT_PROGRAM_FLOW_ARCS"] = "YES"
-        build_configuration.build_settings["GCC_GENERATE_TEST_COVERAGE_FILES"] = "YES"
-      end
-    end
-
   end
 end