class Aws::SSO::Plugins::Endpoints::Handler

def apply_endpoint_headers(context, headers)
  headers.each do |key, values|
    value = values
      .compact
      .map { |s| Seahorse::Util.escape_header_list_string(s.to_s) }
      .join(',')
    context.http_request.headers[key] = value
  end
end

def call(context)
  # If endpoint was discovered, do not resolve or apply the endpoint.
  unless context[:discovered_endpoint]
    params = parameters_for_operation(context)
    endpoint = context.config.endpoint_provider.resolve_endpoint(params)
    context.http_request.endpoint = endpoint.url
    apply_endpoint_headers(context, endpoint.headers)
  end
  context[:endpoint_params] = params
  context[:auth_scheme] =
    Aws::Endpoints.resolve_auth_scheme(context, endpoint)
  @handler.call(context)
end

def parameters_for_operation(context)
  case context.operation_name
  when :get_role_credentials
    Aws::SSO::Endpoints::GetRoleCredentials.build(context)
  when :list_account_roles
    Aws::SSO::Endpoints::ListAccountRoles.build(context)
  when :list_accounts
    Aws::SSO::Endpoints::ListAccounts.build(context)
  when :logout
    Aws::SSO::Endpoints::Logout.build(context)
  end
end