from pyinfra import host
from pyinfra.operations import server, files

from multikube_config import nodes, server_node

if host.name in nodes + [server_node]:
    files.put(dest='/usr/local/bin/k3s-killall.sh', src='files/kube/k3s-killall.sh', mode='a+rx')
    files.put(dest='/usr/local/bin/k3s-uninstall.sh', src='files/kube/k3s-uninstall.sh', mode='a+rx')
    server.shell(commands=['k3s-uninstall.sh'])
    for n in ['k3s-agent.service', 'k3s-agent.service.env', 'k3s-node.service']:
        files.file(path=f'/etc/systemd/system/{n}', present=False)