#!/bin/bash
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

set -euo pipefail

ENDPOINT_URL=""
ENDPOINT_URL_OPTION=""
REGION="us-east-1"
VEHICLE_NAME=""
TIMESTAMP=""
ACCOUNT_ID=`aws sts get-caller-identity --query "Account" --output text`
SERVICE_ROLE="IoTFleetWiseServiceRole"
SERVICE_ROLE_POLICY_ARN="arn:aws:iam::${ACCOUNT_ID}:policy/"
FLEET_SIZE=1
BATCH_SIZE=$((`nproc`*4))

parse_args() {
    while [ "$#" -gt 0 ]; do
        case $1 in
        --vehicle-name)
            VEHICLE_NAME=$2
            shift
            ;;
        --fleet-size)
            FLEET_SIZE=$2
            shift
            ;;
        --timestamp)
            TIMESTAMP=$2
            shift
            ;;
        --endpoint-url)
            ENDPOINT_URL=$2
            shift
            ;;
        --region)
            REGION=$2
            shift
            ;;
        --help)
            echo "Usage: $0 [OPTION]"
            echo "  --vehicle-name <NAME>  Vehicle name"
            echo "  --fleet-size <SIZE>    Size of fleet, default: ${FLEET_SIZE}. When greater than 1,"
            echo "                         the instance number will be appended to each"
            echo "                         Vehicle name after a '-', e.g. fwdemo-42"
            echo "  --timestamp <TS>       Timestamp of demo.sh script"
            echo "  --endpoint-url <URL>   The endpoint URL used for AWS CLI calls"
            echo "  --region <REGION>      The region used for AWS CLI calls, default: ${REGION}"
            exit 0
            ;;
        esac
        shift
    done

    if [ "${ENDPOINT_URL}" != "" ]; then
        ENDPOINT_URL_OPTION="--endpoint-url ${ENDPOINT_URL}"
    fi
    if [ "${VEHICLE_NAME}" == "" ]; then
        echo "Error: Vehicle name not provided"
        exit -1
    fi
    if [ "${TIMESTAMP}" == "" ]; then
        echo "Error: Timestamp not provided"
        exit -1
    fi
}

echo "======================================="
echo "AWS IoT FleetWise Cloud Clean-up Script"
echo "======================================="

parse_args "$@"

NAME="${VEHICLE_NAME}-${TIMESTAMP}"
SERVICE_ROLE="${SERVICE_ROLE}-${REGION}-${TIMESTAMP}"
SERVICE_ROLE_POLICY_ARN="${SERVICE_ROLE_POLICY_ARN}${SERVICE_ROLE}-policy"

echo -n "Date: "
date --rfc-3339=seconds

echo "Timestamp: ${TIMESTAMP}"
echo "Vehicle name: ${VEHICLE_NAME}"
echo "Fleet size: ${FLEET_SIZE}"

# $1 is the campaign name
suspend_and_delete_campaign(){
    echo "Suspending campaign..."
    aws iotfleetwise update-campaign \
        ${ENDPOINT_URL_OPTION} --region ${REGION} \
        --name $1 \
        --action SUSPEND 2> /dev/null | jq -r .arn || true

    echo "Deleting campaign..."
    aws iotfleetwise delete-campaign \
        ${ENDPOINT_URL_OPTION} --region ${REGION} \
        --name $1 2> /dev/null | jq -r .arn || true
}

suspend_and_delete_campaign ${NAME}-campaign
suspend_and_delete_campaign ${NAME}-campaign-s3-json
suspend_and_delete_campaign ${NAME}-campaign-s3-parquet

if ((FLEET_SIZE==1)); then
    echo "Disassociating vehicle ${VEHICLE_NAME}..."
    aws iotfleetwise disassociate-vehicle-fleet \
        ${ENDPOINT_URL_OPTION} --region ${REGION} \
        --fleet-id ${NAME}-fleet \
        --vehicle-name "${VEHICLE_NAME}" 2> /dev/null || true
else
    echo "Disassociating vehicle ${VEHICLE_NAME}-0..$((FLEET_SIZE-1))..."
    for ((i=0; i<${FLEET_SIZE}; i+=${BATCH_SIZE})); do
        for ((j=0; j<${BATCH_SIZE} && i+j<${FLEET_SIZE}; j++)); do
            # This output group is run in a background process. Note that stderr is redirected to stream 3 and back,
            # to print stderr from the output group, but not info about the background process.
            { \
                aws iotfleetwise disassociate-vehicle-fleet \
                    ${ENDPOINT_URL_OPTION} --region ${REGION} \
                    --fleet-id ${NAME}-fleet \
                    --vehicle-name "${VEHICLE_NAME}-$((i+j))" 2> /dev/null || true \
            2>&3 &} 3>&2 2>/dev/null
        done
        # Wait for all background processes to finish
        wait
    done
fi

if ((FLEET_SIZE==1)); then
    echo "Deleting vehicle ${VEHICLE_NAME}..."
    aws iotfleetwise delete-vehicle \
        ${ENDPOINT_URL_OPTION} --region ${REGION} \
        --vehicle-name "${VEHICLE_NAME}" 2> /dev/null || true
else
    echo "Deleting vehicle ${VEHICLE_NAME}-0..$((FLEET_SIZE-1))..."
    for ((i=0; i<${FLEET_SIZE}; i+=${BATCH_SIZE})); do
        for ((j=0; j<${BATCH_SIZE} && i+j<${FLEET_SIZE}; j++)); do
            # This output group is run in a background process. Note that stderr is redirected to stream 3 and back,
            # to print stderr from the output group, but not info about the background process.
            { \
                aws iotfleetwise delete-vehicle \
                    ${ENDPOINT_URL_OPTION} --region ${REGION} \
                    --vehicle-name "${VEHICLE_NAME}-$((i+j))" 2> /dev/null || true \
            2>&3 &} 3>&2 2>/dev/null
        done
        # Wait for all background processes to finish
        wait
    done
fi

echo "Deleting fleet..."
aws iotfleetwise delete-fleet \
    ${ENDPOINT_URL_OPTION} --region ${REGION} \
    --fleet-id ${NAME}-fleet 2> /dev/null || true

echo "Deleting decoder manifest..."
aws iotfleetwise delete-decoder-manifest \
    ${ENDPOINT_URL_OPTION} --region ${REGION} \
    --name ${NAME}-decoder-manifest 2> /dev/null || true

echo "Deleting model manifest..."
aws iotfleetwise delete-model-manifest \
    ${ENDPOINT_URL_OPTION} --region ${REGION} \
    --name ${NAME}-model-manifest 2> /dev/null || true

echo "Deleting service role and policy..."
aws iam detach-role-policy --role-name ${SERVICE_ROLE} --policy-arn ${SERVICE_ROLE_POLICY_ARN} --region ${REGION} || true
aws iam delete-policy --policy-arn ${SERVICE_ROLE_POLICY_ARN} --region ${REGION} || true
aws iam delete-role --role-name ${SERVICE_ROLE} --region ${REGION} || true

# Note: As the service currently only supports one signal catalog, do not delete it
# echo "Deleting signal catalog..."
# aws iotfleetwise delete-signal-catalog \
#     ${ENDPOINT_URL_OPTION} --region ${REGION} \
#     --name ${NAME}-signal-catalog 2> /dev/null || true