# python-mpd2: Python MPD client library
#
# Copyright (C) 2008-2010  J. Alexander Treuman <jat@spatialrift.net>
# Copyright (C) 2012  J. Thalheim <jthalheim@gmail.com>
# Copyright (C) 2016  Robert Niederreiter <rnix@squarewave.at>
#
# python-mpd2 is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# python-mpd2 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with python-mpd2.  If not, see <http://www.gnu.org/licenses/>.

import gi
import re, socket, sys, time
from enum import Enum

gi.require_version('Gtk', '3.0')
from gi.repository import GObject

VERSION = (3, 1, 0)
HELLO_PREFIX = "OK MPD "
ERROR_PREFIX = "ACK "
ERROR_PATTERN = re.compile(r"\[(?P<errno>\d+)@(?P<offset>\d+)\]\s+{(?P<command>\w+)}\s+(?P<msg>.*)")
SUCCESS = "OK"
NEXT = "list_OK"



def escape(text):
  return text.replace("\\", "\\\\").replace('"', '\\"')



# MPD Protocol errors as found in CommandError exceptions
# https://github.com/MusicPlayerDaemon/MPD/blob/master/src/protocol/Ack.hxx
class FailureResponseCode(Enum):
    NOT_LIST = 1
    ARG = 2
    PASSWORD = 3
    PERMISSION = 4
    UNKNOWN = 5

    NO_EXIST = 50
    PLAYLIST_MAX = 51
    SYSTEM = 52
    PLAYLIST_LOAD = 53
    UPDATE_ALREADY = 54
    PLAYER_SYNC = 55
    EXIST = 56


class ProtocolError(Exception):
    pass


class CommandError(Exception):
    def __init__(self, error):
        self.errno = None
        self.offset = None
        self.command = None
        self.msg = None

        match = ERROR_PATTERN.match(error)
        if match:
            self.errno = FailureResponseCode(int(match.group("errno")))
            self.offset = int(match.group("offset"))
            self.command = match.group("command")
            self.msg = match.group("msg")


class CommandListError(Exception):
  pass


class IteratingError(Exception):
  pass


class mpd_commands(object):
    """Decorator for registering MPD commands with it's corresponding result
    callback.
    """

    def __init__(self, *commands, **kwargs):
        self.commands = commands
        self.is_direct = kwargs.pop("is_direct", False)
        self.is_binary = kwargs.pop("is_binary", False)
        if kwargs:
            raise AttributeError(
                "mpd_commands() got unexpected keyword"
                " arguments %s" % ",".join(kwargs)
            )

    def __call__(self, ob):
        ob.mpd_commands = self.commands
        ob.mpd_commands_direct = self.is_direct
        ob.mpd_commands_binary = self.is_binary
        return ob


def mpd_command_provider(cls):
    """Decorator hooking up registered MPD commands to concrete client
    implementation.

    A class using this decorator must inherit from ``MPDClientBase`` and
    implement it's ``add_command`` function.
    """

    def collect(cls, callbacks=dict()):
        """Collect MPD command callbacks from given class.

        Searches class __dict__ on given class and all it's bases for functions
        which have been decorated with @mpd_commands and returns a dict
        containing callback name as keys and
        (callback, callback implementing class) tuples as values.
        """
        for name, ob in cls.__dict__.items():
            if hasattr(ob, "mpd_commands") and name not in callbacks:
                callbacks[name] = (ob, cls)
        for base in cls.__bases__:
            callbacks = collect(base, callbacks)
        return callbacks

    for name, value in collect(cls).items():
        callback, from_ = value
        for command in callback.mpd_commands:
            cls.add_command(command, callback)
    return cls


class Noop(object):
    """An instance of this class represents a MPD command callback which
    does nothing.
    """

    mpd_commands = None


class MPDClientBase():
  """Abstract MPD client.

  This class defines a general client contract, provides MPD protocol parsers
  and defines all available MPD commands and it's corresponding result
  parsing callbacks. There might be the need of overriding some callbacks on
  subclasses.
  """
  
  def __init__(self):
    self.iterate = False
    self._reset()


  @classmethod
  def add_command(cls, name, callback):
      raise NotImplementedError(
          "Abstract ``MPDClientBase`` does not implement ``add_command``"
      )

  def noidle(self):
      raise NotImplementedError(
          "Abstract ``MPDClientBase`` does not implement ``noidle``"
      )

  def command_list_ok_begin(self):
      raise NotImplementedError(
          "Abstract ``MPDClientBase`` does not implement " "``command_list_ok_begin``"
      )

  def command_list_end(self):
      raise NotImplementedError(
          "Abstract ``MPDClientBase`` does not implement " "``command_list_end``"
      )

  def _reset(self):
      self.mpd_version = None
      self._command_list = None

  def _parse_pair(self, line, separator):
      if line is None:
          return
      pair = line.split(separator, 1)
      if len(pair) < 2:
          raise ProtocolError("Could not parse pair: '{}'".format(line))
      return pair

  def _parse_pairs(self, lines, separator=": "):
      for line in lines:
          yield self._parse_pair(line, separator)

  def _parse_objects(self, lines, delimiters=[], lookup_delimiter=False):
      obj = {}
      for key, value in self._parse_pairs(lines):
          key = key.lower()
          if lookup_delimiter and key not in delimiters:
              delimiters = delimiters + [key]
          if obj:
              if key in delimiters:
                  if lookup_delimiter:
                      if key in obj:
                          yield obj
                          obj = obj.copy()
                          while delimiters[-1] != key:
                              obj.pop(delimiters[-1], None)
                              delimiters.pop()
                  else:
                      yield obj
                      obj = {}
              elif key in obj:
                  if not isinstance(obj[key], list):
                      obj[key] = [obj[key], value]
                  else:
                      obj[key].append(value)
                  continue
          obj[key] = value
      if obj:
          yield obj

  # Use this instead of _parse_objects whenever the result is returned
  # immediately in a command implementation
  _parse_objects_direct = _parse_objects

  def _parse_raw_stickers(self, lines):
      for key, sticker in self._parse_pairs(lines):
          value = sticker.split("=", 1)
          if len(value) < 2:
              raise ProtocolError("Could not parse sticker: {}".format(repr(sticker)))
          yield tuple(value)

  NOOP = mpd_commands("close", "kill")(Noop())

  @mpd_commands("plchangesposid", is_direct=True)
  def _parse_changes(self, lines):
      return self._parse_objects_direct(lines, ["cpos"])

  @mpd_commands("listall", "listallinfo", "listfiles", "lsinfo", is_direct=True)
  def _parse_database(self, lines):
      return self._parse_objects_direct(lines, ["file", "directory", "playlist"])

  @mpd_commands("idle")
  def _parse_idle(self, lines):
      return self._parse_list(lines)

  @mpd_commands("addid", "config", "replay_gain_status", "rescan", "update")
  def _parse_item(self, lines):
      pairs = list(self._parse_pairs(lines))
      if len(pairs) != 1:
          return
      return pairs[0][1]

  @mpd_commands(
      "channels", "commands", "listplaylist", "notcommands", "tagtypes", "urlhandlers"
  )
  def _parse_list(self, lines):
      seen = None
      for key, value in self._parse_pairs(lines):
          if key != seen:
              if seen is not None:
                  raise ProtocolError("Expected key '{}', got '{}'".format(seen, key))
              seen = key
          yield value

  @mpd_commands("list", is_direct=True)
  def _parse_list_groups(self, lines):
      return self._parse_objects_direct(lines, lookup_delimiter=True)

  @mpd_commands("readmessages", is_direct=True)
  def _parse_messages(self, lines):
      return self._parse_objects_direct(lines, ["channel"])

  @mpd_commands("listmounts", is_direct=True)
  def _parse_mounts(self, lines):
      return self._parse_objects_direct(lines, ["mount"])

  @mpd_commands("listneighbors", is_direct=True)
  def _parse_neighbors(self, lines):
      return self._parse_objects_direct(lines, ["neighbor"])

  @mpd_commands("listpartitions", is_direct=True)
  def _parse_partitions(self, lines):
      return self._parse_objects_direct(lines, ["partition"])

  @mpd_commands(
      "add",
      "addtagid",
      "binarylimit",
      "clear",
      "clearerror",
      "cleartagid",
      "consume",
      "crossfade",
      "delete",
      "deleteid",
      "delpartition",
      "disableoutput",
      "enableoutput",
      "findadd",
      "load",
      "mixrampdb",
      "mixrampdelay",
      "mount",
      "move",
      "moveid",
      "moveoutput",
      "newpartition",
      "next",
      "outputvolume",
      "partition",
      "password",
      "pause",
      "ping",
      "play",
      "playid",
      "playlistadd",
      "playlistclear",
      "playlistdelete",
      "playlistmove",
      "previous",
      "prio",
      "prioid",
      "random",
      "rangeid",
      "rename",
      "repeat",
      "replay_gain_mode",
      "rm",
      "save",
      "searchadd",
      "searchaddpl",
      "seek",
      "seekcur",
      "seekid",
      "sendmessage",
      "setvol",
      "shuffle",
      "single",
      "sticker delete",
      "sticker set",
      "stop",
      "subscribe",
      "swap",
      "swapid",
      "toggleoutput",
      "unmount",
      "unsubscribe",
      "volume",
  )
  def _parse_nothing(self, lines):
      for line in lines:
          raise ProtocolError(
              "Got unexpected return value: '{}'".format(", ".join(lines))
          )

  @mpd_commands("count", "currentsong", "readcomments", "stats", "status")
  def _parse_object(self, lines):
      objs = list(self._parse_objects(lines))
      if not objs:
          return {}
      return objs[0]

  @mpd_commands("outputs", is_direct=True)
  def _parse_outputs(self, lines):
      return self._parse_objects_direct(lines, ["outputid"])

  @mpd_commands("playlist")
  def _parse_playlist(self, lines):
      for key, value in self._parse_pairs(lines, ":"):
          yield value

  @mpd_commands("listplaylists", is_direct=True)
  def _parse_playlists(self, lines):
      return self._parse_objects_direct(lines, ["playlist"])

  @mpd_commands("decoders", is_direct=True)
  def _parse_plugins(self, lines):
      return self._parse_objects_direct(lines, ["plugin"])

  @mpd_commands(
      "find",
      "listplaylistinfo",
      "playlistfind",
      "playlistid",
      "playlistinfo",
      "playlistsearch",
      "plchanges",
      "search",
      "sticker find",
      is_direct=True,
  )
  def _parse_songs(self, lines):
      return self._parse_objects_direct(lines, ["file"])

  @mpd_commands("sticker get")
  def _parse_sticker(self, lines):
      key, value = list(self._parse_raw_stickers(lines))[0]
      return value

  @mpd_commands("sticker list")
  def _parse_stickers(self, lines):
      return dict(self._parse_raw_stickers(lines))

  @mpd_commands("albumart", "readpicture", is_binary=True)
  def _parse_plain_binary(self, structure):
      return structure


def _create_callback(self, function, wrap_result):
    """Create MPD command related response callback.
    """
    if not callable(function):
        return None

    def command_callback():
        # command result callback expects response from MPD as iterable lines,
        # thus read available lines from socket
        res = function(self, self._read_lines())
        # wrap result in iterator helper if desired
        if wrap_result:
            res = self._wrap_iterator(res)
        return res

    return command_callback


def _create_command(wrapper, name, return_value, wrap_result):
    """Create MPD command related function.
    """

    def mpd_command(self, *args):
        callback = _create_callback(self, return_value, wrap_result)
        return wrapper(self, name, args, callback)

    return mpd_command


@mpd_command_provider
class MPDClient(MPDClientBase, GObject.Object):
    
  __gsignals__ = {
    'online': (GObject.SignalFlags.RUN_FIRST, None, ()),
    'offline': (GObject.SignalFlags.RUN_FIRST, None, ()),
    'playing': (GObject.SignalFlags.RUN_FIRST, None, (object, object)),
    'song_changed': (GObject.SignalFlags.RUN_FIRST, None, (object, object)),
    'state_changed': (GObject.SignalFlags.RUN_FIRST, None, (object,)),
    'error': (GObject.SignalFlags.RUN_FIRST, None, (object,))
  }


  # __events__ = {}

  connected = False

  try_conn_timer = None
  heart_beat_timer = None

  current_song_id = None
  current_state = 'stop'
  
  idletimeout = None
  _timeout = None
  _wrap_iterator_parsers = [
      MPDClientBase._parse_list,
      MPDClientBase._parse_list_groups,
      MPDClientBase._parse_playlist,
      MPDClientBase._parse_changes,
      MPDClientBase._parse_songs,
      MPDClientBase._parse_mounts,
      MPDClientBase._parse_neighbors,
      MPDClientBase._parse_partitions,
      MPDClientBase._parse_playlists,
      MPDClientBase._parse_database,
      MPDClientBase._parse_messages,
      MPDClientBase._parse_outputs,
      MPDClientBase._parse_plugins,
  ]

  def __init__(self, host = '127.0.0.1', port = 6600):
    super(MPDClient, self).__init__()
    super(GObject.Object, self).__init__()
    self.host = host
    self.port = port

  def _reset(self):
    super(MPDClient, self)._reset()
    self.connected = False
    self._iterating = False
    self._sock = None
    self._rbfile = None
    self._wfile = None

  def _execute(self, command, args, retval):
      if self._iterating:
          raise IteratingError("Cannot execute '{}' while iterating".format(command))
      if self._command_list is not None:
          if not callable(retval):
              raise CommandListError(
                  "'{}' not allowed in command list".format(command)
              )
          self._write_command(command, args)
          self._command_list.append(retval)
      else:
          self._write_command(command, args)
          if callable(retval):
              return retval()
          return retval

  def _write_line(self, line):
    try:
      if self._wfile == None:
        print('MPD server is not connected!!!')
      else:
        self._wfile.write("{}\n".format(line))
        self._wfile.flush()

    except socket.error as e:
      error_message = "Connection to server was reset"
      self._reset()
      e = ConnectionError(error_message)
      raise e.with_traceback(sys.exc_info()[2])

  def _write_command(self, command, args=[]):
      parts = [command]
      for arg in args:
          if type(arg) is tuple:
              if len(arg) == 0:
                  parts.append('":"')
              elif len(arg) == 1:
                  parts.append('"{}:"'.format(int(arg[0])))
              else:
                  parts.append('"{}:{}"'.format(int(arg[0]), int(arg[1])))
          else:
              parts.append('"{}"'.format(escape(str(arg))))
      # Minimize logging cost if the logging is not activated.
  
      cmd = " ".join(parts)
      self._write_line(cmd)

  def _read_line(self):
      line = self._rbfile.readline().decode("utf-8")
      if not line.endswith("\n"):
          self.destroy()
          raise ConnectionError("Connection lost while reading line")
      line = line.rstrip("\n")
      if line.startswith(ERROR_PREFIX):
          error = line[len(ERROR_PREFIX) :].strip()
          raise CommandError(error)
      if self._command_list is not None:
          if line == NEXT:
              return
          if line == SUCCESS:
              raise ProtocolError("Got unexpected '{}'".format(SUCCESS))
      elif line == SUCCESS:
          return
      return line

  def _read_lines(self):
      line = self._read_line()
      while line is not None:
          yield line
          line = self._read_line()

  def _read_chunk(self, amount):
      chunk = bytearray()
      while amount > 0:
          result = self._rbfile.read(amount)
          if len(result) == 0:
              break
          chunk.extend(result)
          amount -= len(result)
      return bytes(chunk)

  def _read_binary(self):
      """From the data stream, read Unicode lines until one says "binary:
      <number>\\n"; at that point, read binary data of the given length.

      This behaves like _parse_objects (with empty set of delimiters; even
      returning only a single result), but rather than feeding from a lines
      iterable (which would be preprocessed too far), it reads directly off
      the stream."""

      obj = {}

      while True:
          line = self._read_line()
          if line is None:
              break

          key, value = self._parse_pair(line, ": ")

          if key == "binary":
              chunk_size = int(value)
              value = self._read_chunk(chunk_size)

              if len(value) != chunk_size:
                  self.destroy()
                  raise ConnectionError(
                      "Connection lost while reading binary data: "
                      "expected %d bytes, got %d" % (chunk_size, len(value))
                  )

              if self._rbfile.read(1) != b"\n":
                  # newline after binary content
                  self.destroy()
                  raise ConnectionError("Connection lost while reading line")

          obj[key] = value
      return obj

  def _execute_binary(self, command, args):
      """Execute a command repeatedly with an additional offset argument,
      keeping all the identical returned dictionary items and concatenating
      the binary chunks following the binary item into one of exactly size.

      This differs from _execute in that rather than passing the lines to the
      callback which'd then call on something like _parse_objects, it builds
      a parsed object on its own (as a prerequisite to the chunk driving
      process) and then joins together the chunks into a single big response."""
      if self._iterating or self._command_list is not None:
          raise IteratingError("Cannot execute '{}' with command lists".format(command))
      data = None
      args = list(args)
      assert len(args) == 1
      args.append(0)
      final_metadata = None
      while True:
          self._write_command(command, args)
          metadata = self._read_binary()
          chunk = metadata.pop('binary', None)

          if final_metadata is None:
              data = chunk
              final_metadata = metadata
              if not data:
                  break
              try:
                  size = int(final_metadata['size'])
              except KeyError:
                  size = len(chunk)
              except ValueError:
                  raise CommandError("Size data unsuitable for binary transfer")
          else:
              if metadata != final_metadata:
                  raise CommandError("Metadata of binary data changed during transfer")
              if chunk is None:
                  raise CommandError("Binary field vanished changed during transfer")
              data += chunk
          args[-1] = len(data)
          if len(data) > size:
              raise CommandListError("Binary data announced size exceeded")
          elif len(data) == size:
              break

      if data is not None:
          final_metadata['binary'] = data

      final_metadata.pop('size', None)

      return final_metadata

  def _read_command_list(self):
      try:
          for retval in self._command_list:
              yield retval()
      finally:
          self._command_list = None
      self._parse_nothing(self._read_lines())

  def _iterator_wrapper(self, iterator):
      try:
          for item in iterator:
              yield item
      finally:
          self._iterating = False

  def _wrap_iterator(self, iterator):
      if not self.iterate:
          return list(iterator)
      self._iterating = True
      return self._iterator_wrapper(iterator)

  def _check_is_mpd_server(self, line):
    if not line.endswith("\n"):
      self.destroy()
      return
    
    line = line.rstrip("\n")
    if not line.startswith(HELLO_PREFIX):
      return
    
    self.connected = True
    self.mpd_version = line[len(HELLO_PREFIX) :].strip()

  def _connect_unix(self):
      if not hasattr(socket, "AF_UNIX"):
          raise ConnectionError("Unix domain sockets not supported on this platform")
      sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
      sock.settimeout(self.timeout)
      sock.connect(self.host)
      return sock

  def _connect_tcp(self):

      try:
          flags = socket.AI_ADDRCONFIG
      except AttributeError:
          flags = 0
    
      for res in socket.getaddrinfo(
          self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM, socket.IPPROTO_TCP, flags
      ):
          af, socktype, proto, canonname, sa = res
          sock = None
          try:
              sock = socket.socket(af, socktype, proto)
              sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
              sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
              sock.settimeout(self.timeout)
              sock.connect(sa)
              return sock
          except Exception as e:
              
              if e.strerror == 'Connection refused':
                  self.emit('offline')
              else:
                  self.emit('error', e)

              if sock is not None:
                  sock.close()


  @mpd_commands("idle")
  def _parse_idle(self, lines):
      self._sock.settimeout(self.idletimeout)
      ret = self._wrap_iterator(self._parse_list(lines))
      self._sock.settimeout(self._timeout)
      return ret

  @property
  def timeout(self):
      return self._timeout

  @timeout.setter
  def timeout(self, timeout):
      self._timeout = timeout
      if self._sock is not None:
          self._sock.settimeout(timeout)


  def _try_connect(self):

    while True:
      if self._sock is not None:
        break

      if self.host.startswith(("/", "\0")):
        self._sock = self._connect_unix()
      else:
        self._sock = self._connect_tcp()
          
    
      if self._sock is not None:
        self._rbfile = self._sock.makefile("rb", newline="\n")
        self._wfile = self._sock.makefile("w", encoding="utf-8", newline="\n")

        try:
          helloline = self._rbfile.readline().decode("utf-8")
          self._check_is_mpd_server(helloline)
          if self.connected:
            break
          
        except Exception as e:
          self.connected = False
          self.destroy()

      
      time.sleep(2)


  def heart_beat(self):

    while True:
      if self.connected:
        try:
          status = self.status()
          song = self.currentsong()
          state = status.get('state')

          if state != self.current_state:
            self.emit('state_changed', state)
            self.current_state = state
          

          if song.get('id') != self.current_song_id:
            if self.current_song_id is not None:
              self.emit('song_changed', status, song)
            self.current_song_id = song.get('id')
          
          
          if state == 'play':
            self.emit('playing', status, song)

          time.sleep(0.2)

        except Exception as err:
          print(err)
  
      else:
         break

    self.destroy()
    self.start()
  

  def start(self):
    self._try_connect()
    if self.connected:
      self.emit('online')
      self.heart_beat()


  def destroy(self):
    if self._rbfile is not None:
      self._rbfile.close()
    if self._wfile is not None:
      self._wfile.close()
    if self._sock is not None:
      self._sock.close()
    self._reset()

  def fileno(self):
    if self._sock is None:
      raise ConnectionError("Not connected")
    return self._sock.fileno()

  def command_list_ok_begin(self):
    if self._command_list is not None:
      raise CommandListError("Already in command list")
    if self._iterating:
      raise IteratingError("Cannot begin command list while iterating")
    self._write_command("command_list_ok_begin")
    self._command_list = []

  def command_list_end(self):
    if self._command_list is None:
      raise CommandListError("Not in command list")
    if self._iterating:
      raise IteratingError("Already iterating over a command list")
    self._write_command("command_list_end")
    return self._wrap_iterator(self._read_command_list())

  @classmethod
  def add_command(cls, name, callback):
      wrap_result = callback in cls._wrap_iterator_parsers
      if callback.mpd_commands_binary:
          method = lambda self, *args: callback(self, cls._execute_binary(self, name, args))
      else:
          method = _create_command(cls._execute, name, callback, wrap_result)
      # create new mpd commands as function:
      escaped_name = name.replace(" ", "_")
      setattr(cls, escaped_name, method)

  @classmethod
  def remove_command(cls, name):
      if not hasattr(cls, name):
          raise ValueError("Can't remove not existent '{}' command".format(name))
      name = name.replace(" ", "_")
      delattr(cls, str(name))


# vim: set expandtab shiftwidth=4 softtabstop=4 textwidth=79: