from enum import Enum
from .client import aws_client


class ParameterType(Enum):
    STRING = 'String'
    LIST = 'StringList'
    SECURE = 'SecureString'


def update_parameter(session, name, description, value, type: ParameterType, encrypt=False, kms_key_id=None, overwrite=False, region: str='us-east-1'):
    ssm_client = aws_client(client_type='ssm', session=session, region=region)
    if encrypt is True:
        response = ssm_client.put_parameter(
            Name=name,
            Description=description,
            Value=value,
            Type=type.value,
            KeyId=kms_key_id,
            Overwrite=overwrite
        )
    else:
        response = ssm_client.put_parameter(
            Name=name,
            Description=description,
            Value=value,
            Type=type.value,
            Overwrite=overwrite
        )
    return response