Current Path : /opt/puppetlabs/puppet/lib/ruby/gems/2.5.0/gems/facter-ng-4.2.5/lib/facter/resolvers/

Upload File :

Current File : //opt/puppetlabs/puppet/lib/ruby/gems/2.5.0/gems/facter-ng-4.2.5/lib/facter/resolvers/az.rb

# frozen_string_literal: true

module Facter
  module Resolvers
    class Az < BaseResolver
      init_resolver

      AZ_METADATA_URL = 'http://169.254.169.254/metadata/instance?api-version=2020-09-01'
      AZ_SESSION_TIMEOUT = 5

      class << self
        private

        def post_resolve(fact_name, _options)
          log.debug('Querying Az metadata')
          @fact_list.fetch(fact_name) { read_facts(fact_name) }
        end

        def read_facts(fact_name)
          @fact_list[:metadata] = {}
          data = get_data_from(AZ_METADATA_URL)
          @fact_list[:metadata] = JSON.parse(data) unless data.empty?

          @fact_list[fact_name]
        end

        def get_data_from(url)
          headers = { Metadata: 'true' }
          Facter::Util::Resolvers::Http.get_request(url, headers, { session: determine_session_timeout })
        end

        def determine_session_timeout
          session_env = ENV['AZ_SESSION_TIMEOUT']
          session_env ? session_env.to_i : AZ_SESSION_TIMEOUT
        end
      end
    end
  end
end

Copyright 2K16 - 2K18 Indonesian Hacker Rulez