/silverlining/commands/query.py

"""Query a server for its active instances"""
import re
import fnmatch
from silversupport.shell import ssh


def command_query(config):
    stdout, stderr, returncode = ssh(
        'www-mgr', config.node_hostname,
        'cat /var/www/appdata.map; echo "END"; '
        'cat /var/www/disabledapps.txt; echo END; '
        'ls /var/www',
        capture_stdout=True)
    hosts = {}
    lines = [line.strip()
             for line in stdout.splitlines()
             if line.strip() and not line.strip().startswith('#')]
    hosts = {}
    site_instances = {}
    instance_site = {}
    sites = set()
    disabled = set()

    # parse appdata.map
    while 1:
        if not lines:
            break
        line = lines.pop(0)
        if line == 'END':
            break
        hostname, path, data = line.split(None, 2)
        instance_name = data.split('|')[0]
        hosts[hostname + path] = instance_name

    # parse disabledsites.txt
    while 1:
        if not lines:
            break
        line = lines.pop(0)
        if line == 'END':
            break
        disabled.add(line)

    # parse directory listing
    for line in lines:
        if line == 'appdata.map' or line == 'disabledapps.txt':
            continue
        match = re.match(r'^(?:([a-z0-9_.-]+)\.(.*)|default-[a-z]+)$',
                         line)
        if not match:
            continue
        if not match.group(1):
            site_name = line
            release = ''
        else:
            site_name = match.group(1)
            release = match.group(2)
        site_instances.setdefault(site_name, {})[release] = line
        sites.add(site_name)
        instance_site[line] = site_name
    site_names = getattr(config.args, 'app-name')
    if site_names:
        matcher = re.compile('|'.join(fnmatch.translate(s) for s in site_names))
        new_hosts = {}
        new_site_instances = {}
        new_instance_site = {}
        new_sites = set()
        for site in sites:
            if matcher.match(site):
                new_sites.add(site)
        for hostname, instance in hosts.iteritems():
            if matcher.match(hostname):
                new_sites.add(instance.split('.')[0])
        for site in new_sites:
            new_site_instances[site] = site_instances[site]
            for n, v in instance_site.iteritems():
                if v == site:
                    new_instance_site[n] = v
            for n, v in hosts.iteritems():
                if v.startswith(site):
                    new_hosts[n] = v
        hosts = new_hosts
        site_instances = new_site_instances
        instance_site = new_instance_site
        sites = new_sites
    notify = config.logger.notify
    special_hosts = {}
    for hostname, site in hosts.items():
        if site in ('disabled', 'notfound'):
            special_hosts.setdefault(site, []).append(hostname)
        elif site in disabled:
            special_hosts.setdefault('disabled', []).append(hostname)
    for site in sorted(sites):
        if len(sites) > 1:
            notify('Site: %s' % site)
            config.logger.indent += 2
        try:
            for release, instance_name in sorted(site_instances[site].items()):
                hostnames = []
                for hostname, inst in hosts.items():
                    if ':' in hostname:
                        # boring
                        continue
                    if (hostname.startswith('www.')
                        or hostname.startswith('prev.www')):
                        continue
                    if inst == instance_name:
                        hostnames.append(hostname)
                if not hostnames:
                    notify('%s (defunct instance)' % instance_name)
                elif len(hostnames) == 1:
                    notify('%s: %s' % (instance_name, hostnames[0]))
                else:
                    notify('%s:' % instance_name)
                    for hostname in sorted(hostnames):
                        notify('  %s' % hostname)
        finally:
            if len(sites) > 1:
                config.logger.indent -= 2
    if special_hosts.get('disabled'):
        notify('Disabled hosts:')
        for hostname in sorted(special_hosts['disabled']):
            if (':' in hostname or hostname.startswith('www.')
                or hostname.startswith('prev.www.')):
                continue
            notify('  %s' % hostname)
    if special_hosts.get('notfound'):
        notify('Hosts marked as not-found:')
        for hostname in sorted(special_hosts['disabled']):
            if (':' in hostname or hostname.startswith('www.')
                or hostname.startswith('prev.www.')):
                continue
            notify('  %s' % hostname)