earlgrey 0.1.6 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 11cbb8397c02fad28c1a9540138d60fec164964c
-  data.tar.gz: 56cdee3bd95121a638c3da2c68b1ed2a4959fb17
+  metadata.gz: 119eb6c49d0f9ec0a629f2e46b073ea0b27b7098
+  data.tar.gz: f28b6a153bd07b865c043667a9d327811f4eb62f
 SHA512:
-  metadata.gz: bde9b7f311fc7355be93dd48e27dfdd7e5ec13954a6815ab30f55c56bbad8e436742f8c2490ca126daf81d40bc63c296330401e38dc788d09346064a2d0ec279
-  data.tar.gz: ff4b8ae1225ae473f0cf4c0a6c85e9b90cb56298fd1c29bfec5ec8172c66d4fcceb20d26eb38bff42ec966e00bc06ebaa20beef1fb706e070818517c08f46d0a
+  metadata.gz: 322cb875b7ddce6157ff0205e7c105c98a3a00ad52e40340e5aff81473915e5e504d9ee5a31672c8f93a9149586019cf9f2b6e2b33b3a431d2943b91091084b6
+  data.tar.gz: 027b47b75110a4901670b6e87136575dec90600d950ba1df5209ae3e3e37aaaec53be49f4bb52070af3a73dfa4a41ce0abcc3a11e65961d7d8d77df02cde59c2

data/bin/earlgrey

@@ -15,6 +15,6 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-require_relative '../lib/earlgrey'
+require 'earlgrey/cli'
 
 EarlGrey::CLI.start

data/lib/{earlgrey/earlgrey.rb → cocoapods_plugin.rb}

@@ -1,5 +1,5 @@
 #
-#  Copyright 2016 Google Inc.
+#  Copyright 2017 Google Inc.
 #
 #  Licensed under the Apache License, Version 2.0 (the "License");
 #  you may not use this file except in compliance with the License.
@@ -13,13 +13,4 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-require 'rubygems'
-require 'xcodeproj'
-require 'colored'
-require 'thor'
-
-require 'fileutils'
-
-require_relative 'version'
-require_relative 'configure_earlgrey'
-require_relative 'cli'
+require_relative 'earlgrey/extensions'

data/lib/earlgrey.rb

@@ -13,4 +13,4 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-require_relative 'earlgrey/earlgrey'
+require 'earlgrey/configure_earlgrey'

data/lib/earlgrey/cli.rb

@@ -12,6 +12,8 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
+require 'earlgrey/configure_earlgrey'
+require 'thor'
 
 # rubocop:disable Metrics/LineLength
 module EarlGrey
@@ -36,8 +38,6 @@ module EarlGrey
     def install
       o = options.dup
 
-      # CLI will never use Cocoapod's `post_install do |installer|`
-      podfile_installer = nil
       opts = { swift: o[SWIFT], carthage: o[CARTHAGE], swift_version: o[SWIFT_VERSION] }
 
       # Use target as the default Scheme name.
@@ -46,7 +46,7 @@ module EarlGrey
       o[PROJECT] ||= Dir.glob(File.join(Dir.pwd, '*.xcodeproj')).first
       raise 'No project found' unless o[PROJECT]
 
-      EarlGrey.configure_for_earlgrey podfile_installer, o[PROJECT], o[TARGET], o[SCHEME], opts
+      EarlGrey.configure_for_earlgrey o[PROJECT], o[TARGET], o[SCHEME], opts
     end
   end
 end

data/lib/earlgrey/configure_earlgrey.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 #
 #  Copyright 2016 Google Inc.
 #
@@ -13,19 +15,50 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-# global export for unqualified use in Pods file
-def configure_for_earlgrey(*args)
-  EarlGrey.configure_for_earlgrey(*args)
+require 'colored'
+require 'fileutils'
+require 'xcodeproj'
+
+def configure_for_earlgrey(*_)
+  EarlGrey.puts_yellow <<-S
+    configure_for_earlgrey in post_install is now deprecated, you should safely remove it now.
+    EarlGrey pod will automatically set up your project and schemes.
+
+    for more information, please visit our installation guideline:
+    https://github.com/google/EarlGrey/blob/master/docs/install-and-run.md
+    S
 end
 
 module EarlGrey
+  XCScheme = Xcodeproj::XCScheme
+  EnvironmentVariable = XCScheme::EnvironmentVariable
+
+  XCSCHEME_EXT = '*.xcscheme'.freeze
+  ENVIRONMENT_KEY = 'DYLD_INSERT_LIBRARIES'.freeze
+  ENVIRONMENT_VALUE = '@executable_path/EarlGrey.framework/EarlGrey'.freeze
+
+  FRAMEWORK_SEARCH_PATHS = 'FRAMEWORK_SEARCH_PATHS'.freeze
+  HEADER_SEARCH_PATHS = 'HEADER_SEARCH_PATHS'.freeze
+
+  SWIFT_FILETYPE = 'sourcecode.swift'.freeze
+  UNITTEST_PRODUCTTYPE = 'com.apple.product-type.bundle.unit-test'.freeze
+
+  CARTHAGE_BUILD_IOS = '$(SRCROOT)/Carthage/Build/iOS'.freeze
+  CARTHAGE_HEADERS_IOS = '$(SRCROOT)/Carthage/Build/iOS/**'.freeze
+
+  EARLGREY_FRAMEWORK = 'EarlGrey.framework'.freeze
+  CARTHAGE_FRAMEWORK_PATH =
+    '${SRCROOT}/Carthage/Build/iOS/EarlGrey.framework'.freeze
+  POD_FRAMEWORK_PATH =
+    '${SRCROOT}/Pods/EarlGrey/EarlGrey/EarlGrey.framework'.freeze
+
   class << self
-    attr_reader :project_name, :installer, :test_target, :test_target_name, :scheme_file,
-                :user_project, :carthage, :swift, :swift_version
+    attr_reader :project_name, :installer, :test_target, :test_target_name,
+                :scheme_file, :user_project, :carthage, :swift, :swift_version
 
     # Returns path to Xcode file, prepending current working dir if necessary.
-    # @param project_name [String] name of the .xcodeproj file
-    # @param ext [String] xcode file extension
+    # @param [String] project_name name of the .xcodeproj file
+    # @param [String] ext xcode file extension
     # @return [String] path to Xcode file
     def path_for(project_name, ext)
       ext_match = File.extname(project_name) == ext
@@ -34,64 +67,59 @@ module EarlGrey
       path ? path : nil
     end
 
-    # Returns the project's directory. If CocoaPods hasn't had it passed in, then the current
-    # directory is chosen.
+    # Returns the project's directory. If CocoaPods hasn't had it passed in,
+    # then the current directory is chosen.
     # @return [String] directory path for the Xcode project
     def dir_path
       installer ? installer.config.installation_root : Dir.pwd
     end
 
     # Strips each line in a string
-    # @param string [String] the string to process
+    # @param [String] string the string to process
     # @return [String] the modified string
     def strip(string)
       string.split("\n").map(&:strip).join("\n")
     end
 
     # Raise error message after removing excessive spaces.
-    # @param message [String] the message to raise
+    # @param [String] message the message to raise
     # @return [nil]
     def error(message)
       raise strip(message)
     end
 
-    # Prints string as blue after stripping excess spacing
-    # @param string [String] the string to print
+    # Prints string as magenta after stripping excess spacing
+    # @param [String] string the string to print
     # @return [nil]
-    def puts_blue(string)
-      puts strip(string).blue
+    def puts_magenta(string)
+      puts strip(string).magenta
     end
 
     # Prints string as yellow after stripping excess spacing
-    # @param string [String] the string to print
+    # @param [String] string the string to print
     # @return [nil]
     def puts_yellow(string)
       puts strip(string).yellow
     end
 
-    def set_defaults(installer, project_name, test_target_name, scheme_file, opts = {})
+    def set_defaults(project_name, test_target_name, scheme_file, opts = {})
       @swift = opts.fetch(:swift, false)
       @carthage = opts.fetch(:carthage, false)
       @swift_version = opts.fetch(:swift_version, '3.0')
-      @installer = installer
 
-      puts_blue "Checking and Updating #{project_name} for EarlGrey."
-      pods_project = installer ? installer.pods_project : true
+      puts_magenta "Checking and Updating #{project_name} for EarlGrey."
       project_file = path_for project_name, '.xcodeproj'
 
       raise 'No test target provided' unless test_target_name
 
-      if pods_project.nil? || project_file.nil?
+      if project_file.nil?
         error <<-E
-          The target's xcodeproj file could not be found.
-          Please check if the correct PROJECT_NAME is being passed in the Podfile.
+          The target's xcodeproj file could not be found. Please check if the
+          correct PROJECT_NAME is being passed in the Podfile.
           Current PROJECT_NAME is: #{project_name}
         E
       end
 
-      # must reset state on every configure
-      @add_earlgrey_product = nil
-
       @project_name = project_name
       @test_target_name = test_target_name
       @scheme_file = File.basename(scheme_file, '.*') + '.xcscheme'
@@ -108,243 +136,244 @@ module EarlGrey
 
     # Main entry point. Configures An Xcode project for use with EarlGrey.
     #
-    # @param installer [Installer] Cococapods installer object or nil.
-    # @param project_name [String] the xcodeproj file name
-    # @param test_target_name [String] the test target name contained in xcodeproj
-    # @param scheme_file [String] the scheme file name. defaults to project name when nil.
+    # @param [String] project_name
+    #        the xcodeproj file name
+    # @param [String] test_target_name
+    #        the test target name contained in xcodeproj
+    # @param [String] scheme_file
+    #        the scheme file name. defaults to project name when nil.
     # @return [nil]
-    def configure_for_earlgrey(installer, project_name, test_target_name, scheme_file, opts = {})
-      set_defaults(installer, project_name, test_target_name, scheme_file, opts)
+    def configure_for_earlgrey(project_name, test_target_name,
+                               scheme_file, opts = {})
+      set_defaults(project_name, test_target_name, scheme_file, opts)
 
-      # Add a Test Action to the User Project Scheme.
-      scheme = modify_scheme_for_actions
+      # Add DYLD_INSERT_LIBRARIES to the schemes
+      modify_scheme_for_actions(user_project, [test_target]).each do |_, scheme|
+        scheme.save!
+      end
 
-      # Add a Copy Files Build Phase for EarlGrey.framework to embed it into the app under test.
-      add_earlgrey_copy_files_script
+      # Add a Copy Files Build Phase for EarlGrey.framework to embed it into
+      # the app under test.
+      framework_ref = add_earlgrey_product user_project, carthage
+      add_earlgrey_framework test_target, framework_ref
+      add_earlgrey_copy_files_script test_target, framework_ref
 
       # Add header/framework search paths for carthage
-      add_carthage_search_paths
+      add_carthage_search_paths test_target if carthage
 
       # Adds EarlGrey.swift
-      copy_swift_files
-
-      save_earlgrey_scheme_changes(scheme) unless scheme.nil?
+      copy_swift_files(user_project, test_target, swift_version) if swift
 
-      puts_blue <<-S
+      user_project.save
+      puts_magenta <<-S
         EarlGrey setup complete.
         You can use the Test Target: #{test_target_name} for EarlGrey testing.
       S
     end
 
-    # Scheme changes to ensure that EarlGrey is correctly loaded before main() is called.
-    def modify_scheme_for_actions
-      scheme_filename = scheme_file
-      # If you do not pass on a scheme name, we set it to the project name itself.
-      scheme_filename = project_name if scheme_filename.to_s == ''
-
-      xcdata_dir = Xcodeproj::XCScheme.user_data_dir(user_project.path)
-      unless File.exist?(File.join(xcdata_dir, scheme_filename).to_s)
-        xcdata_dir = Xcodeproj::XCScheme.shared_data_dir(user_project.path)
+    # Returns the schemes that contain the given targets
+    #
+    # @param [Xcodeproj::Project] project
+    # @param [Array<Xcodeproj::PBXNativeTarget>] targets
+    # @return [Array<Xcodeproj::XCScheme>]
+    def schemes_for_native_targets(project, targets)
+      schemes = Dir[File.join(XCScheme.shared_data_dir(project.path), XCSCHEME_EXT)] +
+                Dir[File.join(XCScheme.user_data_dir(project.path), XCSCHEME_EXT)]
+
+      schemes = schemes.map { |scheme| [scheme, Xcodeproj::XCScheme.new(scheme)] }
+
+      targets_names = targets.map(&:name)
+      schemes.select do |scheme|
+        scheme[1].test_action.testables.any? do |testable|
+          testable.buildable_references.any? do |buildable|
+            targets_names.include? buildable.target_name
+          end
+        end
       end
+    end
 
-      unless File.exist?(File.join(xcdata_dir, scheme_filename).to_s)
-        error <<-E
-          The required scheme "#{scheme_filename}" could not be found.
-          Please ensure that the required scheme file exists within your project directory.
-        E
+    # Add DYLD_INSERT_LIBRARIES to the launching environments for the test
+    # schemes to ensure that EarlGrey is correctly loaded before main() is
+    # called.
+    #
+    # @param [Xcodeproj::Project] project
+    # @param [Array<Xcodeproj::PBXNativeTarget>] targets
+    # @return [Array<String, Xcodeproj::XCScheme>]
+    def modify_scheme_for_actions(project, targets)
+      schemes = schemes_for_native_targets(project, targets).uniq do |name, _|
+        name
       end
-      scheme = Xcodeproj::XCScheme.new File.join(xcdata_dir, scheme_filename)
-      test_action_key = 'DYLD_INSERT_LIBRARIES'
-      test_action_value = '@executable_path/EarlGrey.framework/EarlGrey'
-      unless scheme.test_action.xml_element.to_s.include? test_action_value
-        scheme = add_environment_variables_to_test_action_scheme(scheme_filename,
-                                                                 scheme,
-                                                                 test_action_key,
-                                                                 test_action_value)
+      schemes.each do |name, scheme|
+        add_environment_variables_to_test_scheme(name, scheme)
       end
-
-      scheme
     end
 
     # Load the EarlGrey framework when the app binary is loaded by
     # the dynamic loader, before the main() method is called.
-    def add_environment_variables_to_test_action_scheme(scheme_filename,
-                                                        scheme,
-                                                        test_action_key,
-                                                        test_action_value)
+    #
+    # @param [String] name
+    # @param [Xcodeproj::XCScheme] scheme
+    def add_environment_variables_to_test_scheme(name, scheme)
+      name = File.basename(name, '.xcscheme')
       test_action = scheme.test_action
-      if (scheme.test_action.xml_element.to_s.include? test_action_key) ||
-         (scheme.launch_action.xml_element.to_s.include? test_action_key)
-        puts_yellow <<-S
-          //////////////////// EARLGREY SCHEME ISSUE ////////////////////
-          EarlGrey failed to modify the Test Action part of the scheme: #{scheme_filename}
-          for one of following reasons:
-
-          1) DYLD_INSERT_LIBRARIES is already defined under Environment Variables of
-          the Test Action.
-          2) Run Action's environment variables are used for Test Action.
-
-          To ensure correct functioning of EarlGrey, please manually add the
-          following under Test Action's Environment Variables of the scheme: #{scheme_filename}
-          Environment Variables or EarlGrey's location will not be found.
-          Name: DYLD_INSERT_LIBRARIES
-          Value: @executable_path/EarlGrey.framework/EarlGrey
-          ///////////////////////////////////////////////////////////////
+      test_variables = test_action.environment_variables
+
+      # If any environment variables or arguments were being used in the test
+      # action by being copied from the launch (run) action then copy them over
+      # to the test action along with the EarlGrey environment variable.
+      if test_action.should_use_launch_scheme_args_env?
+        scheme.launch_action.environment_variables.all_variables.each do |var|
+          test_variables.assign_variable var
+        end
+      end
+
+      env_variable = test_variables[ENVIRONMENT_KEY] ||
+                     EnvironmentVariable.new(key: ENVIRONMENT_KEY, value: '')
+      if env_variable.value.include? ENVIRONMENT_VALUE
+        puts_magenta <<-S
+          DYLD_INSERT_LIBRARIES is already set up for #{name}, ignored.
         S
-        return
+        return scheme
       end
-      puts strip(<<-S)
+      puts_magenta <<-S
         Adding EarlGrey Framework Location as an Environment Variable
-        in the App Project's Test Target's Scheme Test Action.
+        in the App Project's Test Target's Scheme Test Action #{name}.
       S
 
-      # Check if the test action uses the run action's environment variables and arguments.
-      launch_action_env_args_present = false
-      if (scheme.test_action.xml_element.to_s.include? 'shouldUseLaunchSchemeArgsEnv') &&
-         ((scheme.launch_action.xml_element.to_s.include? '<EnvironmentVariables>') ||
-            (scheme.launch_action.xml_element.to_s.include? '<CommandLineArguments>'))
-        launch_action_env_args_present = true
-      end
-
-      test_action_is_enabled = 'YES'
       test_action.should_use_launch_scheme_args_env = false
+      env_variable.value += env_variable.value.empty? ? '' : ':'
+      env_variable.value += ENVIRONMENT_VALUE
+      env_variable.enabled = true
+      test_variables.assign_variable env_variable
+      test_action.environment_variables = test_variables
 
-      # If no environment variables are set, then create the element itself.
-      unless scheme.test_action.xml_element.to_s.include? '<EnvironmentVariables>'
-        scheme.test_action.xml_element.add_element('EnvironmentVariables')
-      end
-
-      # If Launch Action Arguments are present and none are present in the test
-      # action, then please add them in.
-      if (scheme.launch_action.xml_element.to_s.include? '<CommandLineArguments>') &&
-         !(scheme.test_action.xml_element.to_s.include? '<CommandLineArguments>')
-        scheme.test_action.xml_element.add_element('CommandLineArguments')
-      end
-
-      # Create a new environment variable and add it to the Environment Variables.
-      test_action_env_vars = scheme.test_action.xml_element.elements['EnvironmentVariables']
-      test_action_args = scheme.test_action.xml_element.elements['CommandLineArguments']
-
-      earl_grey_environment_variable = REXML::Element.new 'EnvironmentVariable'
-      earl_grey_environment_variable.attributes['key'] = test_action_key
-      earl_grey_environment_variable.attributes['value'] = test_action_value
-      earl_grey_environment_variable.attributes['isEnabled'] = test_action_is_enabled
-      test_action_env_vars.add_element(earl_grey_environment_variable)
-
-      # If any environment variables or arguments were being used in the test action by
-      # being copied from the launch (run) action then copy them over to the test action
-      # along with the EarlGrey environment variable.
-      if launch_action_env_args_present
-        launch_action_env_vars = scheme.launch_action.xml_element.elements['EnvironmentVariables']
-        launch_action_args = scheme.launch_action.xml_element.elements['CommandLineArguments']
-
-        # Add in the Environment Variables
-        unless launch_action_env_vars.nil?
-          launch_action_env_vars.elements.each('EnvironmentVariable') do |launch_action_env_var|
-            environment_variable = REXML::Element.new 'EnvironmentVariable'
-            launch_attributes = launch_action_env_var.attributes
-            environment_variable.attributes['key'] = launch_attributes['key']
-            environment_variable.attributes['value'] = launch_attributes['value']
-            environment_variable.attributes['isEnabled'] = launch_attributes['isEnabled']
-            test_action_env_vars.add_element(environment_variable)
-          end
-        end
-
-        # Add in the Arguments
-        unless launch_action_args.nil?
-          launch_action_args.elements.each('CommandLineArgument') do |launch_action_arg|
-            argument = REXML::Element.new 'CommandLineArgument'
-            argument.attributes['argument'] = launch_action_arg.attributes['argument']
-            argument.attributes['isEnabled'] = launch_action_arg.attributes['isEnabled']
-            test_action_args.add_element(argument)
-          end
-        end
-
-      end
-      scheme.test_action = test_action
-      scheme
+      scheme.save!
     end
 
     # Adds EarlGrey.framework to products group. Returns file ref.
-    def add_earlgrey_product
-      return @add_earlgrey_product if @add_earlgrey_product
+    #
+    # @param [Xcodeproj::Project] project
+    #        the xcodeproject that the app is in, and EarlGrey.framework will
+    #        be added to.
+    # @param [Boolean] carthage
+    #        if the project is carthage
+    def add_earlgrey_product(project, carthage)
       framework_path = if carthage
-                         '${SRCROOT}/Carthage/Build/iOS/EarlGrey.framework'
+                         CARTHAGE_FRAMEWORK_PATH
                        else
-                         '${SRCROOT}/Pods/EarlGrey/EarlGrey/EarlGrey.framework'
+                         POD_FRAMEWORK_PATH
                        end
 
-      framework_ref = user_project.products_group.files.find { |f| f.path == framework_path }
-      return @add_earlgrey_product = framework_ref if framework_ref
-
-      framework_ref = user_project.products_group.new_file(framework_path)
-      framework_ref.source_tree = 'SRCROOT'
-
-      @add_earlgrey_product = framework_ref
+      framework_ref = project.products_group.files.find do |f|
+        # TODO: should have some md5 check on the actual binary
+        f.path == framework_path
+      end
+      unless framework_ref
+        framework_ref = project.products_group.new_file(framework_path)
+        framework_ref.source_tree = 'SRCROOT'
+      end
+      framework_ref
     end
 
     # Generates a copy files build phase to embed the EarlGrey framework into
     # the app under test.
-    def add_earlgrey_copy_files_script
+    #
+    # @param [PBXNativeTarget] target
+    #        the native target to add a copy script that copies the earlgrey
+    #        framework into its host app
+    # @param [PBXFileReference] framework_ref
+    #        the framework reference pointing to the EarlGrey.framework
+    def add_earlgrey_copy_files_script(target, framework_ref)
       earlgrey_copy_files_phase_name = 'EarlGrey Copy Files'
-      earlgrey_copy_files_exists = false
-      test_target.copy_files_build_phases.each do |copy_files_phase|
-        if copy_files_phase.name == earlgrey_copy_files_phase_name
-          earlgrey_copy_files_exists = true
-        end
+      return true if target.copy_files_build_phases.any? do |copy_files_phase|
+        copy_files_phase.name == earlgrey_copy_files_phase_name
       end
 
-      unless earlgrey_copy_files_exists
-        name = earlgrey_copy_files_phase_name
-        new_copy_files_phase = test_target.new_copy_files_build_phase(name)
-        new_copy_files_phase.dst_path = '$(TEST_HOST)/../'
-        new_copy_files_phase.dst_subfolder_spec = '0'
+      return false unless target.product_type.eql? UNITTEST_PRODUCTTYPE
+      new_copy_files_phase = target.new_copy_files_build_phase(earlgrey_copy_files_phase_name)
+      new_copy_files_phase.dst_path = '$(TEST_HOST)/../'
+      new_copy_files_phase.dst_subfolder_spec = '0'
 
-        file_ref = add_earlgrey_product
-        build_file = new_copy_files_phase.add_file_reference(file_ref, true)
-        build_file.settings = { 'ATTRIBUTES' => ['CodeSignOnCopy'] }
-        user_project.save
-      end
+      build_file = new_copy_files_phase.add_file_reference framework_ref, true
+      build_file.settings = { 'ATTRIBUTES' => ['CodeSignOnCopy'] }
+      build_file
     end
 
-    FRAMEWORK_SEARCH_PATHS = 'FRAMEWORK_SEARCH_PATHS'.freeze
-    HEADER_SEARCH_PATHS = 'HEADER_SEARCH_PATHS'.freeze
-
-    # Updates test target's build configuration framework and header search paths for carthage.
-    # No-op if carthage isn't true.
-    def add_carthage_search_paths
-      return unless carthage
-      carthage_build_ios = '$(SRCROOT)/Carthage/Build/iOS'
-      carthage_headers_ios = '$(SRCROOT)/Carthage/Build/iOS/**'
-
-      test_target.build_configurations.each do |config|
+    # Updates test target's build configuration framework and header search
+    # paths for carthage. Generates a copy files build phase to embed the
+    # EarlGrey framework into the app under test.
+    #
+    # @param [PBXNativeTarget] target
+    # @return [PBXNativeTarget] target
+    def add_carthage_search_paths(target)
+      target.build_configurations.each do |config|
         settings = config.build_settings
         settings[FRAMEWORK_SEARCH_PATHS] = Array(settings[FRAMEWORK_SEARCH_PATHS])
-        unless settings[FRAMEWORK_SEARCH_PATHS].include?(carthage_build_ios)
-          settings[FRAMEWORK_SEARCH_PATHS] << carthage_build_ios
+        unless settings[FRAMEWORK_SEARCH_PATHS].include?(CARTHAGE_BUILD_IOS)
+          settings[FRAMEWORK_SEARCH_PATHS] << CARTHAGE_BUILD_IOS
         end
 
         settings[HEADER_SEARCH_PATHS] = Array(settings[HEADER_SEARCH_PATHS])
-        unless settings[HEADER_SEARCH_PATHS].include?(carthage_headers_ios)
-          settings[HEADER_SEARCH_PATHS] << carthage_headers_ios
+        unless settings[HEADER_SEARCH_PATHS].include?(CARTHAGE_HEADERS_IOS)
+          settings[HEADER_SEARCH_PATHS] << CARTHAGE_HEADERS_IOS
         end
       end
+      target
+    end
 
-      user_project.save
+    # Add Carthage copy phase
+    #
+    # @param [PBXNativeTarget] target
+    def add_carthage_copy_phase(target)
+      shell_script_name = 'Carthage copy-frameworks Run Script'
+      target_names = target.shell_script_build_phases.map(&:name)
+      unless target_names.include?(shell_script_name)
+        shell_script = target.new_shell_script_build_phase shell_script_name
+        shell_script.shell_path = '/bin/bash'
+        shell_script.shell_script = '/usr/local/bin/carthage copy-frameworks'
+        shell_script.input_paths = [CARTHAGE_FRAMEWORK_PATH]
+      end
     end
 
+    # Add EarlGrey.framework into the build phase "Link Binary With Libraries"
+    #
+    # @param [PBXNativeTarget] target
+    # @param [PBXFileReference] framework_ref
+    #        the framework reference pointing to the EarlGrey.framework
+    def add_earlgrey_framework(target, framework_ref)
+      linked_frameworks = target.frameworks_build_phase.files.map(&:display_name)
+      unless linked_frameworks.include? EARLGREY_FRAMEWORK
+        target.frameworks_build_phase.add_file_reference framework_ref, true
+      end
+    end
+
+    # Check if the target contains a swift source file
+    # @param [PBXNativeTarget] target
+    # @return [Boolean]
+    # rubocop:disable Style/PredicateName
+    def has_swift?(target)
+      target.source_build_phase.files_references.any? do |ref|
+        SWIFT_FILETYPE == (ref.last_known_file_type || ref.explicit_file_type)
+      end
+    end
+    # rubocop:enable Style/PredicateName
+
     # Copies EarlGrey.swift and adds it to the project.
-    # Adds EarlGrey.swift to project's test target group.
-    # Adds EarlGrey.swift to testing target's source build phase
-    # Adds Link Binary With Libraries phase to test target for EarlGrey.framework
-    # Adds shell script phase that runs Carthage copy-frameworks
+    # No op if the target doesn't contain swift.
     #
-    # No op if swift isn't true
-    def copy_swift_files
-      return unless swift
+    # @param [Xcodeproj::Project] project
+    # @param [PBXNativeTarget] target
+    def copy_swift_files(project, target, swift_version = nil)
+      return unless has_swift?(target) || !swift_version.nil?
 
+      project_test_targets = project.main_group.children
+      test_target_group = project_test_targets.find { |g| g.display_name == target.name }
+
+      raise "Test target group not found! #{test_target_group}" unless test_target_group
+
+      swift_version ||= '3.0'
       src_root = File.join(__dir__, 'files')
-      dst_root = File.join(dir_path, test_target_name)
+      dst_root = test_target_group.real_path
       raise "Missing target folder #{dst_root}" unless File.exist? dst_root
 
       src_swift_name = 'EarlGrey.swift'
@@ -355,10 +384,6 @@ module EarlGrey
 
       FileUtils.copy src_swift, dst_swift
 
-      project_test_targets = user_project.main_group.children
-      test_target_group = project_test_targets.find { |g| g.display_name == test_target_name }
-      raise "Test target group not found! #{test_target_group}" unless test_target_group
-
       # Add files to testing target group otherwise Xcode can't read them.
       new_files = [src_swift_name]
       existing_files = test_target_group.children.map(&:display_name)
@@ -369,30 +394,15 @@ module EarlGrey
       end
 
       # Add EarlGrey.swift to sources build phase
-      existing_sources = test_target.source_build_phase.files.map(&:display_name)
+      existing_sources = target.source_build_phase.files.map(&:display_name)
       unless existing_sources.include? src_swift_name
         target_files = test_target_group.files
         earlgrey_swift_file_ref = target_files.find { |f| f.display_name == src_swift_name }
-        raise 'EarlGrey.swift not found in testing target' unless earlgrey_swift_file_ref
-        test_target.source_build_phase.add_file_reference(earlgrey_swift_file_ref, true)
-      end
-
-      # Link Binary With Libraries - frameworks_build_phase - Add EarlGrey.framework
-      earlgrey_framework = 'EarlGrey.framework'
-      existing_frameworks = test_target.frameworks_build_phase.files.map(&:display_name)
-      unless existing_frameworks.include? earlgrey_framework
-        framework_ref = add_earlgrey_product
-        # must pass 'true' to avoid duplicates
-        test_target.frameworks_build_phase.add_file_reference(framework_ref, true)
+        unless earlgrey_swift_file_ref
+          raise 'EarlGrey.swift not found in testing target'
+        end
+        target.source_build_phase.add_file_reference earlgrey_swift_file_ref, true
       end
-
-      user_project.save
-    end
-
-    # Save the scheme changes. This is done here to prevent any irreversible changes in case
-    # of an exception being thrown.
-    def save_earlgrey_scheme_changes(scheme)
-      scheme.save!
     end
   end
 end

data/lib/earlgrey/extensions.rb

@@ -0,0 +1,17 @@
+#
+#  Copyright 2017 Google Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'extensions/aggregate_target_extensions'
+require_relative 'extensions/analyzer_extensions'

data/lib/earlgrey/extensions/aggregate_target_extensions.rb

@@ -0,0 +1,36 @@
+#
+#  Copyright 2017 Google Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative '../configure_earlgrey'
+
+module EarlGrey
+  module AggregateTargetExtensions
+    # rubocop:disable Style/PredicateName
+    def is_earlgrey?
+      @is_earlgrey ||= specs.any? { |spec| 'EarlGrey'.eql? spec.name }
+    end
+    # rubocop:enable Style/PredicateName
+
+    def schemes_for_native_targets
+      EarlGrey.schemes_for_native_targets user_project, user_targets
+    end
+  end
+end
+
+module Pod
+  class AggregateTarget
+    include EarlGrey::AggregateTargetExtensions
+  end
+end

data/lib/earlgrey/extensions/analyzer_extensions.rb

@@ -0,0 +1,51 @@
+#
+#  Copyright 2017 Google Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative '../configure_earlgrey'
+require_relative 'aggregate_target_extensions'
+
+module EarlGrey
+  module AnalyzerExtension
+    def analyze(*_)
+      result = super
+      eg_targets = result.targets.select(&:is_earlgrey?).each do |target|
+        target.user_targets.each do |native_target|
+          EarlGrey.copy_swift_files(target.user_project, native_target,
+                                    target.target_definition.swift_version)
+
+          framework_ref = EarlGrey.add_earlgrey_product target.user_project, false
+          EarlGrey.add_earlgrey_copy_files_script native_target, framework_ref
+          EarlGrey.add_earlgrey_framework native_target, framework_ref
+        end
+      end
+
+      schemes = eg_targets.map(&:schemes_for_native_targets).flatten(1).uniq do |name, _|
+        name
+      end
+      schemes.each do |name, scheme|
+        EarlGrey.add_environment_variables_to_test_scheme(name, scheme)
+      end
+      result
+    end
+  end
+end
+
+module Pod
+  class Installer
+    class Analyzer
+      prepend EarlGrey::AnalyzerExtension
+    end
+  end
+end

data/lib/earlgrey/version.rb

@@ -14,5 +14,5 @@
 #  limitations under the License.
 
 module EarlGrey
-  VERSION = '0.1.6'.freeze unless defined? ::EarlGrey::VERSION
+  VERSION = '0.2.0'.freeze unless defined? ::EarlGrey::VERSION
 end

metadata

@@ -1,16 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: earlgrey
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.2.0
 platform: ruby
 authors:
 - khandpur
 - tirodkar
 - bootstraponline
+- wuhao5
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-03-18 00:00:00.000000000 Z
+date: 2017-03-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colored
@@ -82,20 +83,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.39.0
-- !ruby/object:Gem::Dependency
-  name: coveralls
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.8.13
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.8.13
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -132,10 +119,13 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/earlgrey
+- lib/cocoapods_plugin.rb
 - lib/earlgrey.rb
 - lib/earlgrey/cli.rb
 - lib/earlgrey/configure_earlgrey.rb
-- lib/earlgrey/earlgrey.rb
+- lib/earlgrey/extensions.rb
+- lib/earlgrey/extensions/aggregate_target_extensions.rb
+- lib/earlgrey/extensions/analyzer_extensions.rb
 - lib/earlgrey/files/Swift-2.3/EarlGrey.swift
 - lib/earlgrey/files/Swift-3.0/EarlGrey.swift
 - lib/earlgrey/version.rb
@@ -159,7 +149,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.8
+rubygems_version: 2.6.11
 signing_key: 
 specification_version: 4
 summary: EarlGrey installer gem