# Copyright 2019 Grabtaxi Holdings PTE LTE (GRAB), All rights reserved.
# Use of this source code is governed by an MIT-style license that can be found in the LICENSE file

require 'rexml/document'

class SchemeEditor
  def self.edit_to_support_code_coverage(sandbox)
    pod_proj_path = sandbox.project_path
    puts "Modify schemes of pod project to support code coverage of prebuilt local pod: #{pod_proj_path}"
    scheme_files = Dir["#{pod_proj_path}/**/*.xcscheme"]
    scheme_files.each do |file_path|
      scheme_name = File.basename(file_path, '.*')
      next unless sandbox.local?(scheme_name)

      puts "Modify scheme to enable coverage symbol when prebuild: #{scheme_name}"

      doc = File.open(file_path, 'r') { |f| REXML::Document.new(f) }
      scheme = doc.elements['Scheme']
      test_action = scheme.elements['TestAction']
      next if test_action.attributes['codeCoverageEnabled'] == 'YES'

      test_action.add_attribute('codeCoverageEnabled', 'YES')
      test_action.add_attribute('onlyGenerateCoverageForSpecifiedTargets', 'YES')
      coverage_targets = REXML::Element.new('CodeCoverageTargets')
      buildable_ref = scheme.elements['BuildAction'] \
                            .elements['BuildActionEntries'] \
                            .elements['BuildActionEntry'] \
                            .elements['BuildableReference']
      new_buildable_ref = buildable_ref.clone # Need to clone, otherwise the original one will be move to new place
      coverage_targets.add_element(new_buildable_ref)
      test_action.add_element(coverage_targets)
      File.open(file_path, 'w') { |f| doc.write(f) }
    end
  end
end