def _get_contract_token_from_cloud_identity(cfg: config.UAConfig) -> str:
"""Detect cloud_type and request a contract token from identity info.
:param cfg: a ``config.UAConfig`` instance
:raise NonAutoAttachImageError: When not on an auto-attach image type.
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
:return: contract token obtained from identity doc
"""
cloud_type = identity.get_cloud_type()
if cloud_type not in ("aws", ): # TODO(avoid hard-coding supported types)
raise exceptions.NonAutoAttachImageError(
ua_status.MESSAGE_UNSUPPORTED_AUTO_ATTACH_CLOUD_TYPE.format(
cloud_type=cloud_type))
instance = identity.cloud_instance_factory()
contract_client = contract.UAContractClient(cfg)
pkcs7 = instance.identity_doc
try:
# TODO(make this logic cloud-agnostic if possible)
tokenResponse = contract_client.request_aws_contract_token(pkcs7)
except contract.ContractAPIError as e:
if contract.API_ERROR_MISSING_INSTANCE_INFORMATION in e:
raise exceptions.NonAutoAttachImageError(
ua_status.MESSAGE_UNSUPPORTED_AUTO_ATTACH)
raise e
return tokenResponse["contractToken"]
def auto_attach(
cfg: config.UAConfig, cloud: clouds.AutoAttachCloudInstance
) -> None:
"""
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
:raise NonAutoAttachImageError: If this cloud type does not have
auto-attach support.
"""
contract_client = contract.UAContractClient(cfg)
try:
tokenResponse = contract_client.request_auto_attach_contract_token(
instance=cloud
)
except exceptions.ContractAPIError as e:
if e.code and 400 <= e.code < 500:
raise exceptions.NonAutoAttachImageError(
messages.UNSUPPORTED_AUTO_ATTACH
)
raise e
token = tokenResponse["contractToken"]
attach_with_token(cfg, token=token, allow_enable=True)
def cloud_instance_factory() -> clouds.AutoAttachCloudInstance:
from uaclient.clouds import aws
from uaclient.clouds import azure
cloud_instance_map = {
"aws": aws.UAAutoAttachAWSInstance,
"aws-china": aws.UAAutoAttachAWSInstance,
"aws-gov": aws.UAAutoAttachAWSInstance,
"azure": azure.UAAutoAttachAzureInstance,
}
cloud_type = get_cloud_type()
if not cloud_type:
raise exceptions.UserFacingError(
status.MESSAGE_UNABLE_TO_DETERMINE_CLOUD_TYPE)
cls = cloud_instance_map.get(cloud_type)
if not cls:
raise exceptions.NonAutoAttachImageError(
status.MESSAGE_UNSUPPORTED_AUTO_ATTACH_CLOUD_TYPE.format(
cloud_type=cloud_type))
instance = cls()
if not instance.is_viable:
raise exceptions.UserFacingError(
status.MESSAGE_UNSUPPORTED_AUTO_ATTACH)
return instance
def _get_contract_token_from_cloud_identity(cfg: config.UAConfig) -> str:
"""Detect cloud_type and request a contract token from identity info.
:param cfg: a ``config.UAConfig`` instance
:raise NonAutoAttachImageError: When not on an auto-attach image type.
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
:raise NonAutoAttachImageError: If this cloud type does not have
auto-attach support.
:return: contract token obtained from identity doc
"""
instance = identity.cloud_instance_factory()
contract_client = contract.UAContractClient(cfg)
try:
tokenResponse = contract_client.request_auto_attach_contract_token(
instance=instance)
except contract.ContractAPIError as e:
if contract.API_ERROR_MISSING_INSTANCE_INFORMATION in e:
raise exceptions.NonAutoAttachImageError(
ua_status.MESSAGE_UNSUPPORTED_AUTO_ATTACH)
raise e
return tokenResponse["contractToken"]
def _get_contract_token_from_cloud_identity(cfg: config.UAConfig) -> str:
"""Detect cloud_type and request a contract token from identity info.
:param cfg: a ``config.UAConfig`` instance
:raise NonAutoAttachImageError: When not on an auto-attach image type.
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
:raise NonAutoAttachImageError: If this cloud type does not have
auto-attach support.
:return: contract token obtained from identity doc
"""
try:
instance = identity.cloud_instance_factory()
except exceptions.UserFacingError as e:
if cfg.is_attached:
# We are attached on non-Pro Image, just report already attached
raise exceptions.AlreadyAttachedError(cfg)
# Unattached on non-Pro return UserFacing error msg details
raise e
current_iid = identity.get_instance_id()
if cfg.is_attached:
prev_iid = cfg.read_cache("instance-id")
if current_iid == prev_iid:
raise exceptions.AlreadyAttachedError(cfg)
print("Re-attaching Ubuntu Advantage subscription on new instance")
if _detach(cfg, assume_yes=True) != 0:
raise exceptions.UserFacingError(
ua_status.MESSAGE_DETACH_AUTOMATION_FAILURE
)
contract_client = contract.UAContractClient(cfg)
try:
tokenResponse = contract_client.request_auto_attach_contract_token(
instance=instance
)
except contract.ContractAPIError as e:
if e.code and 400 <= e.code < 500:
raise exceptions.NonAutoAttachImageError(
ua_status.MESSAGE_UNSUPPORTED_AUTO_ATTACH
)
raise e
if current_iid:
cfg.write_cache("instance-id", current_iid)
return tokenResponse["contractToken"]