#!/usr/bin/env python
'''ovhtool - OVH IP address management tool
usage: ovhtool <command>

Available commands are:
    list    Lists IP addresses
    sync    Creates vmac records where missing
'''
import logging

logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.DEBUG)

import ovh
import json
from collections import namedtuple
from docopt import docopt

try:
    import libvirt
except:
    libvirt = None
    logger.warning('No libvirt - not using IP availability data')

from xml.dom import minidom

current_server = None

class OVHIP(object):
    def __init__(self, ip, mac, domain):
        self.ip = ip
        self.mac = mac
        self.domain = domain

    def __repr__(self):
        return '<OVHIP {0.ip} {0.mac} {0.domain}>'.format(self)

def list_ips(server=None, client=None):
    domains = {}

    if libvirt:
        c = libvirt.openReadOnly(None)

        for domid in c.listDomainsID():
            domain = c.lookupByID(domid)
            domains[domid] = []

            dom = minidom.parseString(domain.XMLDesc(0))
            for interface in dom.getElementsByTagName('interface'):
                mac = interface.getElementsByTagName('mac')[0].getAttribute('address')
                domains[mac.lower()] = domain

    client = client or ovh.Client()
    if not server:
        server = client.get('/dedicated/server')[0]

    ips = client.get('/ip?routedTo.serviceName=%s&type=failover' % server)

    # TODO: IP ranges
    ips = {v.split('/')[0]: OVHIP(v.split('/')[0], None, None) for v in ips}

    macs = client.get('/dedicated/server/%s/virtualMac' % server)
    for mac in macs:
        resp = client.get('/dedicated/server/%s/virtualMac/%s/virtualAddress' %
                          (server, mac))[0]
        ips[resp].mac = mac
        ips[resp].domain = domains.get(mac.lower())

    return ips

if __name__ == "__main__":
    args = docopt(__doc__, version='ovhtool 0.1')
    if args['<command>'] == 'list':
        ips = list_ips()
        for _, ip in ips.items():
            hostname = ip.domain.name() if ip.domain else ''
            print '%15s | %17s | %s' % (ip.ip, ip.mac, hostname)
    elif args['<command>'] == 'sync':
        client = ovh.Client()
        server = client.get('/dedicated/server')[0]
        ips = list_ips(client=client)
        for ip in ips.values():
            if ip.mac is None:
                try:
                    logger.info('Missing MAC for %s, creating', ip.ip)
                    resp = client.post('/dedicated/server/%s/virtualMac' %
                                       (server,),
                                       ipAddress=ip.ip, type='ovh',
                                       virtualMachineName='vm')
                    logger.info('Result: %r', resp)
                except:
                    logger.exception('Failed!')