# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
---
schemaVersion: '0.3'
description: |
  ### Document Name - ASR-AFSBP_1.0.0_Redshift.6

  ## What does this document do?
  This document enables automatic version upgrade on a Redshift cluster by calling another SSM document

  ## Input Parameters
  * Finding: (Required) Security Hub finding details JSON
  * AutomationAssumeRole: (Optional) The ARN of the role that allows Automation to perform the actions on your behalf.
  * RemediationRoleName: (Optional) The name of the role that allows Automation to remediate the finding on your behalf.

  ## Documentation Links
  * [AFSBP Redshift.6](https://docs.aws.amazon.com/securityhub/latest/userguide/securityhub-standards-fsbp-controls.html#fsbp-redshift-6)
assumeRole: '{{AutomationAssumeRole}}'
outputs:
- 'Remediation.Output'
- 'ParseInput.AffectedObject'
parameters:
  Finding:
    type: 'StringMap'
    description: 'The input from the Orchestrator Step function for the Redshift.6 finding'
  AutomationAssumeRole:
    type: 'String'
    description: '(Required) The ARN of the role that allows Automation to perform the actions on your behalf.'
    allowedPattern: '^arn:(?:aws|aws-us-gov|aws-cn):iam::\d{12}:role/[\w+=,.@-]+$'
  RemediationRoleName:
    type: 'String'
    default: 'SO0111-EnableAutomaticVersionUpgradeOnRedshiftCluster'
    allowedPattern: '^[\w+=,.@-]+$'
mainSteps:
- name: 'ParseInput'
  action: 'aws:executeScript'
  inputs:
    InputPayload:
      Finding: '{{Finding}}'
      parse_id_pattern: '^arn:(?:aws|aws-cn|aws-us-gov):redshift:(?:[a-z]{2}(?:-gov)?-[a-z]+-\d):\d{12}:cluster:(?!.*--)([a-z][a-z0-9-]{0,62})(?<!-)$'
      expected_control_id:
      - 'Redshift.6'
    Runtime: 'python3.8'
    Handler: 'parse_event'
    Script: |-
      %%SCRIPT=common/parse_input.py%%
  outputs:
  - Name: 'ClusterIdentifier'
    Selector: '$.Payload.resource_id'
    Type: 'String'
  - Name: AllowVersionUpgradeSerialized
    Selector: $.Payload.aws_config_rule.InputParameters
    Type: String
  - Name: 'AffectedObject'
    Selector: '$.Payload.object'
    Type: 'StringMap'
  - Name: 'FindingId'
    Selector: '$.Payload.finding.Id'
    Type: 'String'
  - Name: 'ProductArn'
    Selector: '$.Payload.finding.ProductArn'
    Type: 'String'
  - Name: 'RemediationRegion'
    Selector: '$.Payload.resource_region'
    Type: 'String'
  - Name: 'RemediationAccount'
    Selector: '$.Payload.account_id'
    Type: 'String'
- name: 'ExtractConfigRuleParameters'
  action: 'aws:executeScript'
  inputs:
    InputPayload:
      SerializedJson: '{{ParseInput.AllowVersionUpgradeSerialized}}'
    Runtime: 'python3.8'
    Handler: 'event_handler'
    Script: |-
      %%SCRIPT=common/deserialize_json.py%%
  outputs:
  - Name: 'AllowVersionUpgrade'
    Selector: '$.Payload.allowVersionUpgrade'
    Type: 'String'
- name: 'Remediation'
  action: 'aws:executeAutomation'
  inputs:
    DocumentName: 'ASR-EnableAutomaticVersionUpgradeOnRedshiftCluster'
    TargetLocations:
      - Accounts:
        - '{{ParseInput.RemediationAccount}}'
        Regions:
        - '{{ParseInput.RemediationRegion}}'
        ExecutionRoleName: '{{RemediationRoleName}}'
    RuntimeParameters:
      ClusterIdentifier: '{{ParseInput.ClusterIdentifier}}'
      AllowVersionUpgrade: '{{ExtractConfigRuleParameters.AllowVersionUpgrade}}'
      AutomationAssumeRole: 'arn:{{global:AWS_PARTITION}}:iam::{{global:ACCOUNT_ID}}:role/{{RemediationRoleName}}'
- name: 'UpdateFinding'
  action: 'aws:executeAwsApi'
  inputs:
    Service: 'securityhub'
    Api: 'BatchUpdateFindings'
    FindingIdentifiers:
    - Id: '{{ParseInput.FindingId}}'
      ProductArn: '{{ParseInput.ProductArn}}'
    Note:
      Text: 'Enabled automatic version upgrade on Redshift cluster'
      UpdatedBy: 'ASR-AFSBP_1.0.0_Redshift.6'
    Workflow:
      Status: 'RESOLVED'
  description: 'Update finding'
  isEnd: true