import pygame
import time
import json
from mutagen.mp3 import MP3
import copy

from state_manager import stateManager

class bapsicle():
  state = None

  __default_state = {
    "filename": "",
    "channel": -1,
    "playing": False,
    "pos": 0,
    "remaining": 0,
    "length": 0,
    "loop": False,
    "output": None
  }

  def isInit(self):
    try:
      pygame.mixer.music.get_busy()
    except:
      return False
    else:
      return True

  def isPlaying(self):
    return bool(pygame.mixer.music.get_busy())

  def play(self):


    pygame.mixer.music.play(0)

  def pause(self):
    pygame.mixer.music.pause()

  def unpause(self):
    pygame.mixer.music.play(0, self.state.state["pos"])

  def stop(self):
    pygame.mixer.music.stop()

  def seek(self, pos):
    if self.isPlaying():
      pygame.mixer.music.play(0, pos)
    else:
      self.updateState(pos)

  def load(self, filename):
    if not self.isPlaying():
      self.state.update("filename",filename)
      pygame.mixer.music.load(filename)
      if ".mp3" in filename:
        song = MP3(filename)
        self.state.update("length",song.info.length)
      else:
        self.state.update("length",pygame.mixer.Sound(filename).get_length()/1000)

  def output(self, name = None):
    pygame.mixer.quit()
    try:
      if name:
        pygame.mixer.init(44100, -16, 1, 1024, devicename=name)
      else:
        pygame.mixer.init(44100, -16, 1, 1024)
    except:
      return "FAIL:Failed to init mixer, check sound devices."
    else:
      self.state.update("output",name)
    
    return "OK"


  def updateState(self, pos = None):
    self.state.update("playing",self.isPlaying())
    if (pos):
      self.state.update("pos",max(0,pos))
    else:
      self.state.update("pos",max(0,pygame.mixer.music.get_pos()/1000))
    self.state.update("remaining",self.state.state["length"] - self.state.state["pos"])

  def getDetails(self):
    res = "RESP:DETAILS: " + json.dumps(self.state.state)
    return res


  def __init__(self, channel, in_q, out_q):

    self.state = stateManager("channel" + str(channel), self.__default_state)

    self.state.update("channel", channel)

    loaded_state = copy.copy(self.state.state)

    if loaded_state["output"]:
      print("Setting output to: " + loaded_state["output"])
      self.output(loaded_state["output"])
    else:
      self.output()

    if loaded_state["filename"]:
      print("Loading filename: " + loaded_state["filename"])
      self.load(loaded_state["filename"])

    if loaded_state["pos"] != 0:
      print("Seeking to pos: " + str(loaded_state["pos"]))
      self.seek(loaded_state["pos"])

    if loaded_state["playing"] == True:
      print("Resuming.")
      self.unpause()




    while True:
      time.sleep(0.01)
      incoming_msg = in_q.get()
      if (not incoming_msg):
        continue
      if self.isInit():
        self.updateState()
        if (incoming_msg == 'PLAY'):
          self.play()
        if (incoming_msg == 'PAUSE'):
          self.pause()
        if (incoming_msg == 'UNPAUSE'):
          self.unpause()
        if (incoming_msg == 'STOP'):
          self.stop()
        if (incoming_msg.startswith("SEEK")):
          split = incoming_msg.split(":")
          self.seek(float(split[1]))
        if (incoming_msg.startswith("LOAD")):
          split = incoming_msg.split(":")
          self.load(split[1])
        if (incoming_msg == 'DETAILS'):
          out_q.put(self.getDetails())


      if (incoming_msg.startswith("OUTPUT")):
        split = incoming_msg.split(":")
        out_q.put(self.output(split[1]))