115 lines
4.0 KiB
Python
Executable File
115 lines
4.0 KiB
Python
Executable File
#!/usr/bin/python -tt
|
|
# vim: fileencoding=utf8 foldmethod=marker
|
|
#{{{ License header: MIT
|
|
"""Copyright (c) 2013 Till Maas <opensource@till.name>
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
this software and associated documentation files (the "Software"), to deal in
|
|
the Software without restriction, including without limitation the rights to
|
|
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
of the Software, and to permit persons to whom the Software is furnished to do
|
|
so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE."""
|
|
#}}}
|
|
""" :author: Till Maas
|
|
:contact: opensource@till.name
|
|
:license: MIT
|
|
"""
|
|
""" :changes: Ali AlKhalidi
|
|
:contact: doteast@fedoraproject.org
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import copy
|
|
import itertools
|
|
|
|
from ansible import plugins
|
|
from ansible.errors import AnsibleOptionsError
|
|
from ansible.plugins.callback import CallbackBase
|
|
from ansible.plugins.callback import default
|
|
from ansible.cli.adhoc import AdHocCLI
|
|
ALIAS_PATH = '/srv/web/infra/hosts/{hostname}/host_aliases'
|
|
|
|
|
|
class ResultAccumulator(CallbackBase):
|
|
def __init__(self, *args, **kwargs):
|
|
super(ResultAccumulator, self).__init__(*args, **kwargs)
|
|
self.unreachable = set()
|
|
self.host_status = {}
|
|
self.sshhostkeys = {}
|
|
|
|
def v2_runner_on_unreachable(self, result):
|
|
self.unreachable.add(result._host.get_name())
|
|
|
|
def v2_runner_on_ok(self, result, *args, **kwargs):
|
|
facts = result._result['ansible_facts']
|
|
key = "ssh-rsa {0}".format(facts['ansible_ssh_host_key_rsa_public'])
|
|
|
|
names = [result._host.get_name()]
|
|
ansible_fqdn = facts['ansible_fqdn']
|
|
if ansible_fqdn not in names:
|
|
names.append(ansible_fqdn)
|
|
|
|
ansible_hostname = facts['ansible_hostname']
|
|
if ansible_hostname not in names:
|
|
if ansible_fqdn.find('.stg.') != -1 or result._host.get_name().find('.stg.') != -1:
|
|
names.append(ansible_hostname+".stg")
|
|
else:
|
|
names.append(ansible_hostname)
|
|
|
|
try:
|
|
with open(ALIAS_PATH.format(hostname=result._host.get_name()),
|
|
"rb") as alias_file:
|
|
aliases = [a.strip() for a in alias_file.readlines()]
|
|
for alias in aliases:
|
|
if alias not in names:
|
|
names.append(alias)
|
|
except IOError:
|
|
pass
|
|
|
|
ipv4_addresses = facts["ansible_default_ipv4"]
|
|
if ipv4_addresses:
|
|
names.append(ipv4_addresses["address"])
|
|
|
|
# ignore link local addresses
|
|
non_link_local = facts["ansible_default_ipv6"]
|
|
if non_link_local:
|
|
names.append(non_link_local["address"])
|
|
|
|
#get tunnel addresses; hardcoded to tun0
|
|
if facts.has_key('ansible_tun0'):
|
|
tunnel_addresses=facts["ansible_tun0"]
|
|
names.append(tunnel_addresses['ipv4']['address'])
|
|
|
|
self.sshhostkeys[result._host.get_name()] = {"key": key,
|
|
"names": ",".join(names)}
|
|
|
|
if __name__ == '__main__':
|
|
args = copy.copy(sys.argv)
|
|
args.extend(['-m', 'setup'])
|
|
cb = ResultAccumulator()
|
|
cli = AdHocCLI(copy.copy(args), callback=cb)
|
|
try:
|
|
cli.parse()
|
|
except AnsibleOptionsError:
|
|
if len(cli.args) != 1:
|
|
cli.args = copy.copy(args)
|
|
cli.args.append('all')
|
|
cli.parse()
|
|
|
|
cli.run()
|
|
|
|
for host in sorted(cb.sshhostkeys.items()):
|
|
print "{names} {key} {comment}".format(comment=host[0],**host[1])
|