import os, time

'''
Plugin handler for the Scream plugin.
'''
class SeedlinkPluginHandler:
  # Create defaults
  def __init__(self):
    self.instances = {}
    self.channelMap = {}
    self.idMap = {}


  def push(self, seedlink):
    # Check and set defaults
    try: host = seedlink.param('sources.scream.address')
    except:
      host = "127.0.0.1"
      seedlink.setParam('sources.scream.address',host)

    try: port = int(seedlink.param('sources.scream.port'))
    except:
      port = 1567
      seedlink.setParam('sources.scream.port',str(port))

    try:
      if seedlink.param('sources.scream.tcp').lower() in ("yes","true","1"):
        seedlink.setParam('sources.scream.tcpFlag',' -tcp')
      else:
        seedlink.setParam('sources.scream.tcpFlag','')
    except:
      seedlink.setParam('sources.scream.tcpFlag','')

    key = (host, port, seedlink.param('sources.scream.tcpFlag'))

    try:
      screamId = self.instances[key]

    except KeyError:
      screamId = len(self.instances)
      self.instances[key] = screamId
      self.channelMap[screamId] = []


    try:
      channelItems = [ x.strip() for x in seedlink.param('sources.scream.channels').split(',') ]
      map = os.path.join(seedlink.config_dir, "scream2sl%d.map" % screamId)
      seedlink.setParam('sources.scream.mapFlag',map)

    except KeyError:
      try:
        map = seedlink.param('sources.scream.map')
        if not os.path.isabs(map):
          map = os.path.join(seedlink.config_dir, map)
      except: map = os.path.join(seedlink.config_dir, 'scream2sl.map')

      seedlink.setParam('sources.scream.mapFlag',map)
      channelItems = []

    for item in channelItems:
      mapping = [x.strip() for x in item.split(':')]
      if len(mapping) != 2:
        raise Exception("Error: invalid channel mapping '%s' in %s" % (item, seedlink.station_config_file))
      if not mapping[0] or not mapping[1]:
        raise Exception("Error: invalid channel mapping '%s' in %s" % (item, seedlink.station_config_file))

      # Prepend current station code if not explicitely given
      if not " " in mapping[1]:
        mapping[1] = seedlink._get('seedlink.station.code') + " " + mapping[1]

      mapping[1] = [x.strip() for x in mapping[1].split()]

      # Add network code
      mapping.insert(1, seedlink._get('seedlink.station.network'))

      if not mapping in self.channelMap[screamId]:
        self.channelMap[screamId].append(mapping)

      try:
        if not mapping[1] in self.idMap[mapping[0]]:
          self.idMap[mapping[0]].append(mapping[1])
      except KeyError:
        self.idMap[mapping[0]] = [mapping[1]]

    return (key,map)


  # Flush generates the map file(s)
  def flush(self, seedlink):
    for x in self.idMap.keys():
      if len(self.idMap[x]) > 1:
        raise Exception("Error: Scream plugin has multiple mappings for id(s) %s" % ", ".join(self.idMap.keys()))

    cols = [0,0,0,0]

    for x in self.channelMap.keys():
      for c in self.channelMap[x]:
        vals = [c[0], c[1], c[2][0], c[2][1]]
        for i in range(len(vals)):
          if len(vals[i]) > cols[i]: cols[i] = len(vals[i])

    for x in self.channelMap.keys():
      mappings = self.channelMap[x]
      if len(mappings) == 0: continue

      scream2slmap = os.path.join(seedlink.config_dir, "scream2sl%d.map" % x)

      fd = open(scream2slmap, "w")

      line_id = 1
      fd.write("# Generated at %s - Do not edit!\n" % time.strftime("%Y-%m-%d %H:%M:%S UTC", time.gmtime()))
      for c in mappings:
        fd.write("ChanInfo    %s%s    %s%s    %s%s    %s%s    %d\n" % (c[0], " "*(cols[0]-len(c[0])), c[1], " "*(cols[1]-len(c[1])), c[2][0], " "*(cols[2]-len(c[2][0])), c[2][1], " "*(cols[3]-len(c[2][1])), line_id))
        line_id = line_id + 1

      fd.close()