---
# yaml-language-server: $schema=https://taskfile.dev/schema.json
version: "3"

vars:
  PYTHON_BIN: python3
  ANSIBLE_LINT_FILE: "{{.ANSIBLE_DIR}}/.ansible-lint"
  ANSIBLE_INVENTORY_FILE: "{{.ANSIBLE_DIR}}/inventory/hosts.yaml"
  ANSIBLE_REQUIREMENTS_FILE: "{{.ROOT_DIR}}/requirements.yaml"
  PIP_REQUIREMENTS_FILE: "{{.ROOT_DIR}}/requirements.txt"

env:
  PATH: "{{.ROOT_DIR}}/.venv/bin:$PATH"
  VIRTUAL_ENV: "{{.ROOT_DIR}}/.venv"
  ANSIBLE_COLLECTIONS_PATH: "{{.ROOT_DIR}}/.venv/galaxy"
  ANSIBLE_ROLES_PATH: "{{.ROOT_DIR}}/.venv/galaxy/ansible_roles"
  ANSIBLE_VARS_ENABLED: "host_group_vars"
  ANSIBLE_LOCALHOST_WARNING: "False"
  ANSIBLE_INVENTORY_UNPARSED_WARNING: "False"

tasks:

  deps:
    desc: Set up Ansible dependencies for the environment
    cmds:
      - task: .venv
        vars:
          force: '{{.force | default "true"}}'

  run:
    desc: Run an Ansible playbook for configuring a cluster
    summary: |
      Args:
        playbook: Playbook to run (required)
    prompt: Run Ansible playbook '{{.playbook}}'... continue?
    deps: ["deps"]
    cmd: .venv/bin/ansible-playbook --inventory {{.ANSIBLE_INVENTORY_FILE}} {{.ANSIBLE_DIR}}/playbooks/{{.playbook}}.yaml {{.CLI_ARGS}}
    preconditions:
      - sh: test -n {{.playbook}}
      - sh: test -d {{.ROOT_DIR}}/.venv
      - sh: test -f {{.ANSIBLE_INVENTORY_FILE}}
      - sh: test -f {{.ANSIBLE_DIR}}/playbooks/{{.playbook}}.yaml

  poweroff:
    desc: Shutdown all the k8s nodes
    deps: ["deps"]
    cmd: .venv/bin/ansible kubernetes --inventory {{.ANSIBLE_INVENTORY_FILE}} -a '/usr/bin/systemctl poweroff' --become
    preconditions:
      - sh: test -d {{.ROOT_DIR}}/.venv
      - sh: test -f {{.ANSIBLE_INVENTORY_FILE}}

  list:
    desc: List all the hosts
    deps: ["deps"]
    cmd: .venv/bin/ansible kubernetes --inventory {{.ANSIBLE_INVENTORY_FILE}} --list-hosts
    preconditions:
      - sh: test -d {{.ROOT_DIR}}/.venv
      - sh: test -f {{.ANSIBLE_INVENTORY_FILE}}

  ping:
    desc: Ping all the hosts
    deps: ["deps"]
    cmd: .venv/bin/ansible kubernetes --inventory {{.ANSIBLE_INVENTORY_FILE}} --one-line -m 'ping'
    preconditions:
      - sh: test -d {{.ROOT_DIR}}/.venv
      - sh: test -f {{.ANSIBLE_INVENTORY_FILE}}

  uptime:
    desc: Uptime of all the hosts
    deps: ["deps"]
    cmd: .venv/bin/ansible kubernetes --inventory {{.ANSIBLE_INVENTORY_FILE}} --one-line -a 'uptime'
    preconditions:
      - sh: test -d {{.ROOT_DIR}}/.venv
      - sh: test -f {{.ANSIBLE_INVENTORY_FILE}}

  lint:
    desc: Lint Ansible
    deps: ["deps"]
    cmd: ./.venv/bin/ansible-lint --config-file {{.ANSIBLE_LINT_FILE}} {{.ANSIBLE_DIR}}/**/*.yaml
    preconditions:
      - sh: test -d {{.ANSIBLE_DIR}}
      - sh: test -f {{.ANSIBLE_LINT_FILE}}

  .venv:
    internal: true
    cmds:
      - "{{.PYTHON_BIN}} -m venv {{.ROOT_DIR}}/.venv"
      - .venv/bin/python3 -m pip install --upgrade pip setuptools wheel
      - .venv/bin/python3 -m pip install --upgrade --requirement "{{.PIP_REQUIREMENTS_FILE}}"
      - .venv/bin/ansible-galaxy install --role-file "{{.ANSIBLE_REQUIREMENTS_FILE}}" {{if eq .force "true"}}--force{{end}}
    sources:
      - "{{.PIP_REQUIREMENTS_FILE}}"
      - "{{.ANSIBLE_REQUIREMENTS_FILE}}"
    generates:
      - "{{.ROOT_DIR}}/.venv/pyvenv.cfg"
    preconditions:
      - sh: command -v {{.PYTHON_BIN}}
    vars:
      force: '{{.force | default "true"}}'