awscli/customizations/overridesslcommonname.py (116 lines of code) (raw):

# Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"). You # may not use this file except in compliance with the License. A copy of # the License is located at # # http://aws.amazon.com/apache2.0/ # # or in the "license" file accompanying this file. This file is # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF # ANY KIND, either express or implied. See the License for the specific # language governing permissions and limitations under the License. SSL_COMMON_NAMES = { "sqs": { "af-south-1": "af-south-1.queue.amazonaws.com", "ap-east-1": "ap-east-1.queue.amazonaws.com", "ap-northeast-1": "ap-northeast-1.queue.amazonaws.com", "ap-northeast-2": "ap-northeast-2.queue.amazonaws.com", "ap-northeast-3": "ap-northeast-3.queue.amazonaws.com", "ap-south-1": "ap-south-1.queue.amazonaws.com", "ap-southeast-1": "ap-southeast-1.queue.amazonaws.com", "ap-southeast-2": "ap-southeast-2.queue.amazonaws.com", "ap-southeast-3": "ap-southeast-3.queue.amazonaws.com", "ca-central-1": "ca-central-1.queue.amazonaws.com", "eu-central-1": "eu-central-1.queue.amazonaws.com", "eu-north-1": "eu-north-1.queue.amazonaws.com", "eu-south-1": "eu-south-1.queue.amazonaws.com", "eu-west-1": "eu-west-1.queue.amazonaws.com", "eu-west-2": "eu-west-2.queue.amazonaws.com", "eu-west-3": "eu-west-3.queue.amazonaws.com", "me-south-1": "me-south-1.queue.amazonaws.com", "sa-east-1": "sa-east-1.queue.amazonaws.com", "us-east-1": "queue.amazonaws.com", "us-east-2": "us-east-2.queue.amazonaws.com", "us-west-1": "us-west-1.queue.amazonaws.com", "us-west-2": "us-west-2.queue.amazonaws.com", "cn-north-1": "cn-north-1.queue.amazonaws.com.cn", "cn-northwest-1": "cn-northwest-1.queue.amazonaws.com.cn", "us-gov-west-1": "us-gov-west-1.queue.amazonaws.com", "us-isob-east-1": "us-isob-east-1.queue.sc2s.sgov.gov", }, "emr": { "af-south-1": "af-south-1.elasticmapreduce.amazonaws.com", "ap-east-1": "ap-east-1.elasticmapreduce.amazonaws.com", "ap-northeast-1": "ap-northeast-1.elasticmapreduce.amazonaws.com", "ap-northeast-2": "ap-northeast-2.elasticmapreduce.amazonaws.com", "ap-northeast-3": "ap-northeast-3.elasticmapreduce.amazonaws.com", "ap-south-1": "ap-south-1.elasticmapreduce.amazonaws.com", "ap-southeast-1": "ap-southeast-1.elasticmapreduce.amazonaws.com", "ap-southeast-2": "ap-southeast-2.elasticmapreduce.amazonaws.com", "ap-southeast-3": "ap-southeast-3.elasticmapreduce.amazonaws.com", "ca-central-1": "ca-central-1.elasticmapreduce.amazonaws.com", "eu-north-1": "eu-north-1.elasticmapreduce.amazonaws.com", "eu-south-1": "eu-south-1.elasticmapreduce.amazonaws.com", "eu-west-1": "eu-west-1.elasticmapreduce.amazonaws.com", "eu-west-2": "eu-west-2.elasticmapreduce.amazonaws.com", "eu-west-3": "eu-west-3.elasticmapreduce.amazonaws.com", "me-south-1": "me-south-1.elasticmapreduce.amazonaws.com", "sa-east-1": "sa-east-1.elasticmapreduce.amazonaws.com", "us-east-2": "us-east-2.elasticmapreduce.amazonaws.com", "us-west-1": "us-west-1.elasticmapreduce.amazonaws.com", "us-west-2": "us-west-2.elasticmapreduce.amazonaws.com", }, "rds": { "us-east-1": "rds.amazonaws.com", }, "docdb": { "us-east-1": "rds.amazonaws.com", }, "neptune": { "us-east-1": "rds.amazonaws.com", }, "health": { "aws-global": "health.us-east-1.amazonaws.com", "af-south-1": "health.us-east-1.amazonaws.com", "ap-east-1": "health.us-east-1.amazonaws.com", "ap-northeast-1": "health.us-east-1.amazonaws.com", "ap-northeast-2": "health.us-east-1.amazonaws.com", "ap-northeast-3": "health.us-east-1.amazonaws.com", "ap-south-1": "health.us-east-1.amazonaws.com", "ap-southeast-1": "health.us-east-1.amazonaws.com", "ap-southeast-2": "health.us-east-1.amazonaws.com", "ap-southeast-3": "health.us-east-1.amazonaws.com", "ca-central-1": "health.us-east-1.amazonaws.com", "eu-central-1": "health.us-east-1.amazonaws.com", "eu-north-1": "health.us-east-1.amazonaws.com", "eu-south-1": "health.us-east-1.amazonaws.com", "eu-west-1": "health.us-east-1.amazonaws.com", "eu-west-2": "health.us-east-1.amazonaws.com", "eu-west-3": "health.us-east-1.amazonaws.com", "me-south-1": "health.us-east-1.amazonaws.com", "sa-east-1": "health.us-east-1.amazonaws.com", "us-east-1": "health.us-east-1.amazonaws.com", "us-east-2": "health.us-east-1.amazonaws.com", "us-west-1": "health.us-east-1.amazonaws.com", "us-west-2": "health.us-east-1.amazonaws.com", "cn-north-1": "health.cn-northwest-1.amazonaws.com.cn", "cn-northwest-1": "health.cn-northwest-1.amazonaws.com.cn", "aws-cn-global": "health.cn-northwest-1.amazonaws.com.cn", }, } REGION_TO_PARTITION_OVERRIDE = { "aws-global": "aws", "aws-cn-global": "aws-cn", } def register_override_ssl_common_name(cli): cli.register_last( "before-building-argument-table-parser", update_endpoint_url ) def update_endpoint_url(session, parsed_globals, **kwargs): service = parsed_globals.command endpoints = SSL_COMMON_NAMES.get(service) # only change url if user has not overridden already themselves if endpoints is not None and parsed_globals.endpoint_url is None: region = session.get_config_variable("region") endpoint_url = endpoints.get(region) if endpoint_url is not None: parsed_globals.endpoint_url = f"https://{endpoint_url}" if service == "health": _override_health_region(region, session, parsed_globals) def _override_health_region(region, session, parsed_globals): if region in REGION_TO_PARTITION_OVERRIDE: partition = REGION_TO_PARTITION_OVERRIDE[region] else: partition = session.get_partition_for_region(region) if partition == "aws-cn": parsed_globals.region = "cn-northwest-1" else: parsed_globals.region = "us-east-1"