#! /usr/bin/env python

class DownloadHandler():
  
  def __init__(self, ListCtrl):
    self.ListCtrl = ListCtrl
    self.finished = False
    self.close = False
    self.error = False
    self.handlers = []
  
  def _has_closed(self):
    return self.close
 
  def _has_finished(self):
    return self.finished
    
  def _has_error(self):
    return self.error
 
  def handle(self, msg):
    ''' Handles msg base to Signals.txt '''
    data = msg.data
    index = self.get_index(data)
    self.check_for_error(data)
    if index == -1:
      if data[0] == 'finish':
	self.finished = True
      elif data[0] == 'close':
	self.close = True
    else:
      ''' Manage handlers for its index '''
      if index == len(self.handlers): 
	''' Create new IndexDownloadHandler and add it to handlers '''
	self.handlers.append(IndexDownloadHandler(self.ListCtrl, index))
      ''' Let IndexDownloadHandler handle message data for current index '''
      self.handlers[index].handle(data)
  
  def get_index(self, data):
    return data.pop()

  def check_for_error(self, data):
    if data[0] == 'error':
      self.error = True
    
class IndexDownloadHandler():
  
  def __init__(self, ListCtrl, index):
    self.ListCtrl = ListCtrl
    self.index = index
    self.info = ''
    
  def handle(self, data):
    ''' Handle its data message for current index '''
    if data[0] == 'finish':
      self.finish()
    elif data[0] == 'close':
      self.close()
    elif data[0] == 'error':
      self.error()
    elif data[0] == 'playlist':
      self.playlist(data)
    elif data[0] == '[youtube]':
      self.pre_proc()
    elif data[0] == '[download]':
      self.download(data)
    elif data[0] == '[ffmpeg]':
      self.post_proc()
    elif data[0] == 'remove':
      self.remove()
      
  def finish(self):
    self.ListCtrl._write_data(self.index, 4, '')
    self.ListCtrl._write_data(self.index, 5, 'Finished')
     
  def close(self):
    self.ListCtrl._write_data(self.index, 3, '')
    self.ListCtrl._write_data(self.index, 4, '')
    self.ListCtrl._write_data(self.index, 5, 'Stopped')
     
  def error(self):
    self.ListCtrl._write_data(self.index, 3, '')
    self.ListCtrl._write_data(self.index, 4, '')
    self.ListCtrl._write_data(self.index, 5, 'Error')
     
  def pre_proc(self):
    self.ListCtrl._write_data(self.index, 5, 'Pre-Processing %s' % self.info)
     
  def post_proc(self):
    self.ListCtrl._write_data(self.index, 4, '')
    self.ListCtrl._write_data(self.index, 5, 'Post-Processing %s' % self.info)
     
  def download(self, data):
    self.ListCtrl._write_data(self.index, 1, data[3])
    self.ListCtrl._write_data(self.index, 2, data[1])
    self.ListCtrl._write_data(self.index, 3, data[7])
    self.ListCtrl._write_data(self.index, 4, data[5])
    self.ListCtrl._write_data(self.index, 5, 'Downloading %s' % self.info)
     
  def playlist(self, data):
    self.ListCtrl._write_data(self.index, 1, '')
    self.ListCtrl._write_data(self.index, 2, '')
    self.ListCtrl._write_data(self.index, 3, '')
    self.ListCtrl._write_data(self.index, 4, '')
    self.info = '%s/%s' % (data[1], data[2])
  
  def remove(self):
    self.ListCtrl._write_data(self.index, 5, 'Removing DASH %s' % self.info)