Allow fact file to be set per branch

Author: kpaulisse

lib/octocatalog-diff/cli/options.rb

@@ -112,14 +112,17 @@ def self.option_globally_or_per_branch_string(opts)
           translated = translate_option(opts[:translator], x)
           options[to_option] ||= translated
           options[from_option] ||= translated
+          post_process(opts[:post_process], options)
         end
         parser.on("--to-#{flag}", "#{desc} for the to branch") do |x|
           validate_option(opts[:validator], x) if opts[:validator]
           options[to_option] = translate_option(opts[:translator], x)
+          post_process(opts[:post_process], options)
         end
         parser.on("--from-#{flag}", "#{desc} for the from branch") do |x|
           validate_option(opts[:validator], x) if opts[:validator]
           options[from_option] = translate_option(opts[:translator], x)
+          post_process(opts[:post_process], options)
         end
       end
 
@@ -178,6 +181,15 @@ def self.translate_option(translator, value)
         return value if translator.nil?
         translator.call(value)
       end
+
+      # Code that can run after a translation and operate upon all options. This returns nothing but may
+      # modify options that were input.
+      # @param processor [Code] Processor function
+      # @param options [Hash] Options hash
+      def self.post_process(processor, options)
+        return if processor.nil?
+        processor.call(options)
+      end
     end
   end
 end

lib/octocatalog-diff/cli/options/fact_file.rb

@@ -7,19 +7,38 @@
   has_weight 150
 
   def parse(parser, options)
-    parser.on('--fact-file FILENAME', 'Fact file to use instead of node lookup') do |fact_file|
-      raise Errno::ENOENT, 'Invalid fact file provided' unless File.file?(fact_file)
-      facts = nil
-      local_opts = { fact_file_string: File.read(fact_file) }
-      if fact_file =~ /\.ya?ml$/
-        facts = OctocatalogDiff::Facts.new(local_opts.merge(backend: :yaml))
-      elsif fact_file =~ /\.json$/
-        facts = OctocatalogDiff::Facts.new(local_opts.merge(backend: :json))
-      else
-        raise ArgumentError, 'I do not know how to parse the provided fact file. Needs .yaml or .json extension.'
+    OctocatalogDiff::Cli::Options.option_globally_or_per_branch(
+      parser: parser,
+      options: options,
+      cli_name: 'fact-file',
+      option_name: 'facts',
+      desc: 'Override fact',
+      datatype: '',
+      validator: ->(fact_file) { File.file?(fact_file) && (fact_file =~ /\.ya?ml$/ || fact_file =~ /\.json$/) },
+      translator: lambda do |fact_file|
+        local_opts = { fact_file_string: File.read(fact_file) }
+        if fact_file =~ /\.ya?ml$/
+          OctocatalogDiff::Facts.new(local_opts.merge(backend: :yaml))
+        elsif fact_file =~ /\.json$/
+          OctocatalogDiff::Facts.new(local_opts.merge(backend: :json))
+        else
+          # :nocov:
+          # Believed to be a bug condition since the validator should kick this out before it ever gets here.
+          raise ArgumentError, 'I do not know how to parse the provided fact file. Needs .yaml or .json extension.'
+          # :nocov:
+        end
+      end,
+      post_process: lambda do |opts|
+        unless options[:node]
+          %w[to_facts from_facts facts].each do |opt|
+            next unless opts[opt.to_sym] && opts[opt.to_sym].node
+            opts[:node] = opts[opt.to_sym].node
+            break
+          end
+        end
+
+        options[:facts] ||= options[:to_facts]
       end
-      options[:facts] = facts
-      options[:node] ||= facts.facts['name']
-    end
+    )
   end
 end

spec/octocatalog-diff/tests/cli/options/fact_file_spec.rb

@@ -48,4 +48,28 @@
       end.to raise_error(Errno::ENOENT)
     end
   end
+
+  describe '#opt_to_fact_file' do
+    it 'should distinguish between the to-facts and from-facts' do
+      fact_file_1 = OctocatalogDiff::Spec.fixture_path('facts/facts.yaml')
+      fact_file_2 = OctocatalogDiff::Spec.fixture_path('facts/fact-overrides-datatypes.yaml')
+      result = run_optparse(['--fact-file', fact_file_1, '--to-fact-file', fact_file_2])
+
+      result_facts_1 = result[:facts].facts
+      expect(result_facts_1).to be_a_kind_of(Hash)
+      expect(result_facts_1['name']).to eq('rspec-node.xyz.github.net')
+      expect(result_facts_1['values']).to be_a_kind_of(Hash)
+      expect(result_facts_1['values']['fqdn']).to eq('rspec-node.xyz.github.net')
+      expect(result_facts_1['values']['ipaddress']).to be_nil
+      expect(result_facts_1['values'].keys).not_to include('expiration')
+
+      result_facts_2 = result[:to_facts].facts
+      expect(result_facts_2).to be_a_kind_of(Hash)
+      expect(result_facts_2['name']).to eq('rspec-node.xyz.github.net')
+      expect(result_facts_2['values']).to be_a_kind_of(Hash)
+      expect(result_facts_2['values']['fqdn']).to eq('rspec-node.xyz.github.net')
+      expect(result_facts_2['values']['ipaddress']).to eq('10.20.30.40')
+      expect(result_facts_2['values'].keys).not_to include('expiration')
+    end
+  end
 end