#!/usr/bin/python
# doteast; base from skvidal's freezelist
# doteast porting to ansible 2.0
# list hosts with ansible var[=value], Or
# list all hosts with their corresponding vars
# Note that the script will attempt to "match" the supplied value of the var against the values if it the var is multivalued

from ansible import constants as C
from ansible.parsing.dataloader import DataLoader
from ansible.vars.manager import VariableManager
from ansible.inventory.manager import InventoryManager
import sys
from optparse import OptionParser


parser = OptionParser(version="1.0")
parser.add_option('-i', dest='inventory', default=C.DEFAULT_HOST_LIST,
                  help="Path to inventory file/dir")
parser.add_option('-o', dest='variable', default=None,
                  help="variable name to check")
parser.add_option('-a', action="store_true", dest='all_vars', default=None,
                  help="get all vars")
opts, args = parser.parse_args(sys.argv[1:])

if ((opts.variable == None and opts.all_vars == None) or (opts.variable != None and opts.all_vars != None)):
    print("Usage: hosts_with_var_set -o varname[=value] | -a")
    sys.exit(-1)

loader = DataLoader()
inv = InventoryManager(loader=loader, sources=opts.inventory)
variable_manager = VariableManager(loader=loader, inventory=inv)
matching = True

if opts.variable != None:
    if opts.variable.find("=") == -1:
        matching = False
        var_name = opts.variable
    else:
        var_name, value = opts.variable.split('=')
        if value == "":
            value = "None"

var_set = []
for host in inv.get_hosts():
    vars = variable_manager.get_vars(host=host)
    if opts.variable == None:
        # remove expanded 'all' groups
        vars.pop('groups')
        vars['groups'] = host.get_groups()
        print("%s\n%s\n" % (host.get_name(), vars))
    else:
        if var_name in vars:
            if not matching:
                var_set.append(host.get_name())
            else:
                if str(vars.get(var_name)).find(value) != -1:
                    var_set.append(host.get_name())

if opts.variable != None:
    if not matching:
        print('hosts with variable %s:' % var_name)
    else:
        print('hosts with variable %s matching %s value' % (var_name, value))
    for host in sorted(var_set):
        print(host)