lib/prx_auth/rails/ext/controller.rb in prx_auth-rails-1.3.0 vs lib/prx_auth/rails/ext/controller.rb in prx_auth-rails-1.4.0

@@ -1,10 +1,14 @@
 require 'prx_auth/rails/token'
+require 'open-uri'
 
 module PrxAuth
   module Rails
     module Controller
+
+      PRX_ACCOUNT_NAME_MAPPING_KEY = 'prx.account.name.mapping'.freeze
+
       def prx_auth_token
         rack_auth_token = env_prx_auth_token
         return rack_auth_token if rack_auth_token.present?
 
         session['prx.auth'] && Rack::PrxAuth::TokenData.new(session['prx.auth'])
@@ -24,14 +28,55 @@
         return if prx_auth_token.nil?
 
         PrxAuth::Rails::Token.new(prx_auth_token)
       end
 
+      def lookup_and_register_accounts_names
+        session[PRX_ACCOUNT_NAME_MAPPING_KEY] =
+          lookup_account_names_mapping
+      end
+
+      def account_name_for(id)
+        id = id.to_i
+
+        name =
+          if session[PRX_ACCOUNT_NAME_MAPPING_KEY].has_key?(id)
+            session[PRX_ACCOUNT_NAME_MAPPING_KEY][id]
+          else
+            session[PRX_ACCOUNT_NAME_MAPPING_KEY][id] = lookup_account_name_for(id)
+          end
+
+        name = "[#{id}] Unknown Account Name" unless name.present?
+
+        name
+      end
+
       def sign_in_user(token)
         session['prx.auth'] = token
       end
 
       private
+
+      def lookup_account_name_for(id)
+        id = id.to_i
+
+        res = lookup_account_names_mapping([id])
+        res[id]
+      end
+
+      def lookup_account_names_mapping(ids=current_user.resources)
+        id_host = PrxAuth::Rails.configuration.id_host
+        ids_param = ids.map(&:to_s).join(',')
+
+        options = {}
+        options[:ssl_verify_mode] = OpenSSL::SSL::VERIFY_NONE if ::Rails.env.development?
+
+        accounts = URI.open("https://#{id_host}/api/v1/accounts?account_ids=#{ids_param}", options).read
+
+        mapping = JSON.parse(accounts)['accounts'].map { |acct| [acct['id'], acct['display_name']] }.to_h
+
+        mapping
+      end
 
       def env_prx_auth_token
         if !defined? @_prx_auth_token
           @_prx_auth_token = request.env['prx.auth'] && PrxAuth::Rails::Token.new(request.env['prx.auth'])
         else