#!/usr/bin/env python
# vim:ts=2:et:sw=2:ai
#
# Build topological network graph
# Rick van der Zwet <info@rickvanderzwet.nl>

import cgi
import gformat
import re
import sys
import urllib
import yaml
import math
import rdnap

def get_yaml(gfile):
  """ Get configuration yaml for 'item'"""
  f = open(gfile, 'r')
  datadump = yaml.load(f)
  return datadump

def write_yaml(gfile, datadump):
  """ Write configuration yaml for 'item'"""
  f = open(gfile, 'w')
  f.write(yaml.dump(datadump, default_flow_style=False))
  f.close()


def make_graph():
  status = None
  try:
    stream = file('/tmp/nodemap_status.yaml','r')
    status = yaml.load(stream)
  except IOError,e:
    print "# Error loading status '%s'" % e

  f = open('kmlfile.kml', 'w')
  f.write("""<?xml version="1.0" encoding="UTF-8"?>
<kml xmlns="http://earth.google.com/kml/2.0">
  <Document>
    <name>WirelessLeiden Nodemap</name>
    <open>1</open>
    <description>Generated realtime status of all Wireless Leiden AccessPoints</description>
    <Style id="node_status_up">
      <IconStyle>
        <scale>0.5</scale>
        <Icon><href>http://www.google.com/mapfiles/kml/paddle/grn-stars-lv.png</href></Icon>  
      </IconStyle>
    </Style>
    <Style id="node_status_down">
      <IconStyle>
        <scale>0.5</scale>
        <Icon><href>http://www.google.com/mapfiles/kml/paddle/red-stars-lv.png</href></Icon>  
      </IconStyle>
    </Style>
    <Style id="node_status_planned">
      <IconStyle>
        <scale>0.5</scale>
        <Icon><href>http://www.google.com/mapfiles/kml/paddle/wht-stars-lv.png</href></Icon>  
      </IconStyle>
    </Style>
    <Style id="node_status_unknown">
      <IconStyle>
        <scale>0.5</scale>
        <Icon><href>http://www.google.com/mapfiles/kml/paddle/wht-stars-lv.png</href></Icon>  
      </IconStyle>
    </Style>
    <Folder>
      <name>Nodes</name>
      <visibility>0</visibility>
      <description>All active nodes and links</description>
    """)

  poel = {}
  link_type = {}
  node = {}
  
  nodes = []
  links = []
  try:
    for host in gformat.get_hostlist():
      fqdn = host + ".wLeiden.NET"
      print "## Processing host", host
      datadump = gformat.get_yaml(host)
      try:
        lam, phi = rdnap.rd2etrs(datadump['rdnap_x'], datadump['rdnap_y'])
      except ValueError:
        raise ValueError("Host '%s' yaml broken. Invalid rdnap_x or rdnap_y" %
        host)
      print "### RDNAP:%s,%s -> %s,%s" % (datadump['rdnap_x'], datadump['rdnap_y'], lam, phi)
      iface_keys = [elem for elem in datadump.keys() if (elem.startswith('iface_') and not "lo0" in elem)]
      for iface_key in iface_keys:
        l = datadump[iface_key]['ip']
        addr, mask = l.split('/')

        addr = gformat.parseaddr(addr)
        mask = int(mask)
        addr = addr & ~((1 << (32 - mask)) - 1)
        if poel.has_key(addr): 
          poel[addr] += [host]
        else: 
          poel[addr] = [host]
          # Assume all eth2wifibridge to be 11a for a moment
          iface_parent = '_'.join(iface_key.split('_')[0:2])
          if datadump[iface_parent].has_key('extra_type') and datadump[iface_parent]['extra_type'] == 'eth2wifibridge':
            link_type[addr] = '11a'
          else:
            link_type[addr] = datadump[iface_parent]['type']
          print "### %s [%s] is of type %s" % (gformat.showaddr(addr), iface_key, link_type[addr])
      node[host] = (lam, phi)
      if not status:
        # By default assume up
        node_status = "up"
      elif not status['node'].has_key(fqdn):
        node_status = "unknown"
      elif status['node'][fqdn] == gformat.OK:
        node_status = "up"
      elif status['node'][fqdn] == gformat.DOWN:
        node_status = "down"
      elif status['node'][fqdn] == gformat.UNKNOWN:
        node_status = "unknown"
      else:
        assert False, "Status cannot be generated"

      print "### Node status is '%s'" % node_status
      f.write("""
      <description>All active nodes</description>
      <Placemark>
        <name>%(nodetype)s %(nodename)s</name>
        <description>%(desc)s</description>
        <styleUrl>%(style)s</styleUrl>
        <Point><coordinates>%(lam)s,%(phi)s,0</coordinates></Point>
      </Placemark>
   """ % {'nodetype' : datadump['nodetype'], 'nodename' : datadump['nodename'], 'desc' : cgi.escape(datadump['location']), 'style' : '#node_status_' + node_status, 'lam' : lam, 'phi' : phi})
      nodes += [("POINT(%s, %s)" % (lam, phi))]
  except (KeyError, ValueError), e:
    print "[FOUT] in '%s' interface '%s'" % (host,iface_key) 
    raise
    sys.exit(1)

  f.write("""
    </Folder>
    <Folder>
      <name>Links</name>
      <visibility>0</visibility>
      <description>All links</description>
  """)
  for addr,leden in poel.iteritems():
    if not gformat.valid_addr(gformat.showaddr(addr)):
      print "## Adres '%s' met leden '%s' ignored'" % (gformat.showaddr(addr), ','.join(leden))
      continue
    if link_type[addr] == '11a':
      weight = 2
    elif link_type[addr] == 'eth':
      weight = 4
    else:
      weight = 1

    leden = sorted(set(leden))
    for index,lid in enumerate(leden[:-1]):
      for buur in leden[index + 1:]:
        key = (lid + ".wLeiden.NET", buur + ".wLeiden.NET")
        rev_key = (key[1],key[0])
        link_status = None
        if not status:
          # Default assume OK
          link_status = gformat.OK
        elif status['link'].has_key(key):
          link_status = status['link'][key]
        elif status['link'].has_key(rev_key):
          link_status = status['link'][rev_key]
        else:
          # If link is not known assume nothing
          link_status = gformat.UNKNOWN


        if link_status == gformat.OK:
          color = '#ffff0000' # green
        elif link_status == gformat.DOWN:
          color = '#ff0000ff' # red
        elif link_status == gformat.UNKNOWN:
          color = '#55000000' # black
        else:
          assert False, "Link status not mapped properly"

        f.write("""
      <Placemark>
        <name>%(name)s</name>
        <visibility>0</visibility>
        <description>%(desc)s</description>
        <LineString>
          <tessellate>0</tessellate>
          <coordinates> %(lam1)s, %(phi1)s, 0
            %(lam2)s , %(phi2)s, 0 </coordinates>
        </LineString>
        <Style>%(style)s</Style>
      </Placemark>
      """ % { 'lam1' : node[lid][0], 
              'phi1' : node[lid][1], 
              'lam2' : node[buur][0], 
              'phi2' : node[buur][1],
              'name' : "Interlink: %s --- %s" % (lid, buur), 
              'desc' : "%s [%s]" % (gformat.showaddr(addr), link_type[addr]),
              'style' : "<LineStyle><color>%s</color><width>%s</width></LineStyle>" % (color, weight),
              })
  f.write("""
    </Folder>
  </Document>
</kml>
  """)
  f.close()


if __name__ == "__main__":
  make_graph()