#!/usr/bin/env python

import subprocess

# Purge snap packages
snap_packages = [
    "cdk-addons",
    "kube-apiserver",
    "kube-controller-manager",
    "kube-proxy",
    "kube-scheduler",
    "kubelet",
    "kubectl",
    "etcd"
]

for package in snap_packages:
    subprocess.run(["snap", "remove", package, "--purge"])

# Purge Debian packages
packages_to_purge = [
    "nginx-full",
    "containerd",
    "keepalived"
]

subprocess.run(["apt", "purge", "-y"] + packages_to_purge + ["--allow-change-held-packages"])
subprocess.run(["apt", "autoremove", "-y", "--purge"])
subprocess.run(["/sbin/remove-juju-services"])

# Remove directories and files
directories = [
    "/var/log/juju",
    "/var/lib/juju",
    "/home/ubuntu/config*",
    "/root/cdk",
    "/etc/juju-proxy*",
    "/var/run/calico",
    "/var/lib/calico",
    "/var/log/calico",
    "/etc/containerd",
    "/var/lib/containerd",
    "/opt/calicoctl",
    "/opt/cni",
    "/opt/containerd",
    "/etc/keepalived"
]

for directory in directories:
    subprocess.run(["rm", "-rf", directory])