class Aws::BedrockAgent::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)
  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)
    context[:endpoint_params] = params
    context[:endpoint_properties] = endpoint.properties
  end
  context[:auth_scheme] =
    Aws::Endpoints.resolve_auth_scheme(context, endpoint)
  @handler.call(context)
end

def parameters_for_operation(context)
  case context.operation_name
  when :associate_agent_knowledge_base
    Aws::BedrockAgent::Endpoints::AssociateAgentKnowledgeBase.build(context)
  when :create_agent
    Aws::BedrockAgent::Endpoints::CreateAgent.build(context)
  when :create_agent_action_group
    Aws::BedrockAgent::Endpoints::CreateAgentActionGroup.build(context)
  when :create_agent_alias
    Aws::BedrockAgent::Endpoints::CreateAgentAlias.build(context)
  when :create_data_source
    Aws::BedrockAgent::Endpoints::CreateDataSource.build(context)
  when :create_knowledge_base
    Aws::BedrockAgent::Endpoints::CreateKnowledgeBase.build(context)
  when :delete_agent
    Aws::BedrockAgent::Endpoints::DeleteAgent.build(context)
  when :delete_agent_action_group
    Aws::BedrockAgent::Endpoints::DeleteAgentActionGroup.build(context)
  when :delete_agent_alias
    Aws::BedrockAgent::Endpoints::DeleteAgentAlias.build(context)
  when :delete_agent_version
    Aws::BedrockAgent::Endpoints::DeleteAgentVersion.build(context)
  when :delete_data_source
    Aws::BedrockAgent::Endpoints::DeleteDataSource.build(context)
  when :delete_knowledge_base
    Aws::BedrockAgent::Endpoints::DeleteKnowledgeBase.build(context)
  when :disassociate_agent_knowledge_base
    Aws::BedrockAgent::Endpoints::DisassociateAgentKnowledgeBase.build(context)
  when :get_agent
    Aws::BedrockAgent::Endpoints::GetAgent.build(context)
  when :get_agent_action_group
    Aws::BedrockAgent::Endpoints::GetAgentActionGroup.build(context)
  when :get_agent_alias
    Aws::BedrockAgent::Endpoints::GetAgentAlias.build(context)
  when :get_agent_knowledge_base
    Aws::BedrockAgent::Endpoints::GetAgentKnowledgeBase.build(context)
  when :get_agent_version
    Aws::BedrockAgent::Endpoints::GetAgentVersion.build(context)
  when :get_data_source
    Aws::BedrockAgent::Endpoints::GetDataSource.build(context)
  when :get_ingestion_job
    Aws::BedrockAgent::Endpoints::GetIngestionJob.build(context)
  when :get_knowledge_base
    Aws::BedrockAgent::Endpoints::GetKnowledgeBase.build(context)
  when :list_agent_action_groups
    Aws::BedrockAgent::Endpoints::ListAgentActionGroups.build(context)
  when :list_agent_aliases
    Aws::BedrockAgent::Endpoints::ListAgentAliases.build(context)
  when :list_agent_knowledge_bases
    Aws::BedrockAgent::Endpoints::ListAgentKnowledgeBases.build(context)
  when :list_agent_versions
    Aws::BedrockAgent::Endpoints::ListAgentVersions.build(context)
  when :list_agents
    Aws::BedrockAgent::Endpoints::ListAgents.build(context)
  when :list_data_sources
    Aws::BedrockAgent::Endpoints::ListDataSources.build(context)
  when :list_ingestion_jobs
    Aws::BedrockAgent::Endpoints::ListIngestionJobs.build(context)
  when :list_knowledge_bases
    Aws::BedrockAgent::Endpoints::ListKnowledgeBases.build(context)
  when :list_tags_for_resource
    Aws::BedrockAgent::Endpoints::ListTagsForResource.build(context)
  when :prepare_agent
    Aws::BedrockAgent::Endpoints::PrepareAgent.build(context)
  when :start_ingestion_job
    Aws::BedrockAgent::Endpoints::StartIngestionJob.build(context)
  when :tag_resource
    Aws::BedrockAgent::Endpoints::TagResource.build(context)
  when :untag_resource
    Aws::BedrockAgent::Endpoints::UntagResource.build(context)
  when :update_agent
    Aws::BedrockAgent::Endpoints::UpdateAgent.build(context)
  when :update_agent_action_group
    Aws::BedrockAgent::Endpoints::UpdateAgentActionGroup.build(context)
  when :update_agent_alias
    Aws::BedrockAgent::Endpoints::UpdateAgentAlias.build(context)
  when :update_agent_knowledge_base
    Aws::BedrockAgent::Endpoints::UpdateAgentKnowledgeBase.build(context)
  when :update_data_source
    Aws::BedrockAgent::Endpoints::UpdateDataSource.build(context)
  when :update_knowledge_base
    Aws::BedrockAgent::Endpoints::UpdateKnowledgeBase.build(context)
  end
end