import urllib.request
from invoke import task


def updateConfigmapCmd(path):
    return f'kubectl create configmap victoriametrics-config --from-file {path}=config/{path} -o yaml --dry-run=client | kubectl apply -f -'


def reload(ctx, svc):
    host = ctx.run(f'khost {svc}').stdout
    path = {'victoriametrics': '/m', 'vmalert': '/vmalert'}[svc]
    print(' -> status',
          urllib.request.urlopen(f'http://{host}{path}/-/reload').status)


@task
def sync_config(ctx):
    ctx.run(updateConfigmapCmd("scrape_config.yaml"))
    ctx.run(updateConfigmapCmd("scrape_ssl.yaml"))
    reload(ctx, 'victoriametrics')

    ctx.run(updateConfigmapCmd("rules_expected_hosts.yaml"))
    ctx.run(updateConfigmapCmd("rules_expected_nodes.yaml"))
    ctx.run(updateConfigmapCmd("rules_k8s.yaml"))
    ctx.run(updateConfigmapCmd("rules_main.yaml"))
    reload(ctx, 'vmalert')