#!/bin/bash

set -e

# Here don't need to remove/purge the tedge_mapper, tedge_agent, and tedge_watchdog packages explicitly,
# as they will be removed by removing the tedge package.
packages=("tedge" "tedge_apt_plugin" "tedge_apama_plugin" "c8y_log_plugin" "c8y_configuration_plugin")

extension_services=("tedge-watchdog.service" "tedge-mapper-collectd.service" "c8y-log-plugin.service" "c8y-configuration-plugin.service")

clouds=("c8y" "az")

usage() {
    cat <<EOF
USAGE:
    uninstall-thin-edge_io.sh [COMMAND]
    
COMMANDS:
    remove     Uninstall thin-edge.io with keeping configuration files
    purge      Uninstall thin-edge.io and also remove configuration files  

EOF
}

disconnect_from_cloud() {
    for cloud in "${clouds[@]}"; do
        if [ -f "/etc/tedge/mosquitto-conf/$cloud-bridge.conf" ]; then
            sudo tedge disconnect "$cloud"
        fi
    done
}

stop_extension_services() {
    for service in "${extension_services[@]}"; do
        status=$(sudo systemctl is-active "$service") || true
        if [ "$status" = "active" ]; then
            sudo systemctl stop "$service"
        fi
    done
}

remove_or_purge_package_if_exists() {
    disconnect_from_cloud
    stop_extension_services
    for package in "${packages[@]}"; do
        status=$(dpkg -s "$package" | grep -w installed) || true
        if [ "$status" = "Status: install ok installed" ]; then
            sudo apt --assume-yes "$1" "$package"
        fi
    done
}

if [ $# -eq 1 ]; then
    DELETE_OR_PURGE=$1
    if [ "$DELETE_OR_PURGE" = 'remove' ]; then
        remove_or_purge_package_if_exists "remove"
    elif [ "$DELETE_OR_PURGE" = 'purge' ]; then
        remove_or_purge_package_if_exists "purge"
    fi
else
    usage
fi