module Aws::Api::ServiceCustomizations

def apply(client_class)
  apply_protocol_plugin(client_class)
  endpoint_prefix = client_class.api.metadata('endpointPrefix')
  @customizations[endpoint_prefix].each do |customization|
    Customizer.new(client_class).apply(&customization)
  end
end

def apply_protocol_plugin(client_class)
  protocol = client_class.api.metadata('protocol')
  plugin = case protocol
  when 'ec2'       then Aws::Plugins::Protocols::EC2
  when 'query'     then Aws::Plugins::Protocols::Query
  when 'json'      then Aws::Plugins::Protocols::JsonRpc
  when 'rest-json' then Aws::Plugins::Protocols::RestJson
  when 'rest-xml'  then Aws::Plugins::Protocols::RestXml
  end
  client_class.add_plugin(plugin) if plugin
end

def customize(endpoint_prefix, &block)
  @customizations[endpoint_prefix] << block
end