#!/usr/bin/python3

# Collects all the assigned roles of all the minions, and checks if the

# declared {salt,pillar}/role/*.sls files actually match a minion-assigned role

import yaml

import os

import sys

all_roles = ['base']

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

    with open("pillar/id/%s" % sls) as f:

        try:

            _roles = yaml.load(f)['grains']['roles']

        except KeyError:

            continue

        for item in _roles:

            all_roles.append(item)

all_roles = sorted(set(all_roles))

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

    for sls in os.listdir('%s/role' % directory):

        if sls.endswith('.sls'):

            with open('salt/role/%s' % sls) as f:

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

                    print ('%s/role/%s not in roles' % (directory, sls))

                    status = 1

sys.exit(status)