#!/usr/bin/python3

# Various roles related tests:

# - Checks if the declared {salt,pillar}/role/*.sls files actually match an

#   assigned role to a minion

# - Checks if the assigned roles to minions have equivalent

#   {salt,pillar}/role/*.sls files

# - Checks that the special roles are not included in any pillar/id/*.sls

import os

import sys

from get_roles import get_roles, get_roles_of_one_minion

status = 0

special_roles = ['base']

roles = get_roles() + special_roles

for directory in ['salt', 'pillar']:

    for sls in os.listdir(f'{directory}/role'):

        if sls.endswith('.sls'):

            if sls.split('.sls')[0] not in roles:

                print(f'Unused file {directory}/role/{sls} - not in roles')

                status = 1

for role in roles:

    role = role.replace('.', '/')

    if not os.path.isfile(f'salt/role/{role}.sls') and not os.path.isfile(f'salt/role/{role}/init.sls'):

        print(f'{role} role is missing the salt/role/{role}.sls or salt/role/{role}/init.sls file')

        status = 1

roles = get_roles()

for special_role in special_roles:

    if special_role in roles:

        for sls in os.listdir('pillar/id/'):

            _roles = get_roles_of_one_minion(sls)

            if special_role in _roles:

                print(f'{special_role} role should not be included in pillar/id/{sls} file')

                status = 1

sys.exit(status)