Factor player code out of pandora package

3 files changed, 235 insertions, 2 deletions

diff --git a/pydora/mpg123.py b/pydora/mpg123.py
new file mode 100644
index 0000000..bb2e76a
--- /dev/null
+++ b/pydora/mpg123.py
@@ -0,0 +1,154 @@
+import os
+import select
+import subprocess
+
+
+def iterate_forever(func, *args, **kwargs):
+    """Iterate over a finite iterator forever
+
+    When the iterator is exhausted will call the function again to generate a
+    new iterator and keep iterating.
+    """
+    output = func(*args, **kwargs)
+
+    while True:
+        try:
+            yield next(output)
+        except StopIteration:
+            output = func(*args, **kwargs)
+
+
+class SilentPopen(subprocess.Popen):
+    """A Popen varient that dumps it's output and error
+    """
+
+    def __init__(self, *args, **kwargs):
+        self._dev_null = open(os.devnull, 'w')
+        kwargs['stdin'] = subprocess.PIPE
+        kwargs['stdout'] = subprocess.PIPE
+        kwargs['stderr'] = self._dev_null
+        super(SilentPopen, self).__init__(*args, **kwargs)
+
+    def __del__(self):
+        self._dev_null.close()
+        super(SilentPopen, self).__del__()
+
+
+class PlayerCallbacks(object):
+
+    def play(self, song):
+        """Called once when a song starts playing
+        """
+        pass
+
+    def pre_poll(self):
+        """Called before polling for process status
+        """
+        pass
+
+    def post_poll(self):
+        """Called after polling for process status
+        """
+        pass
+
+    def input(self, value, song):
+        """Called after user input during song playback
+        """
+        pass
+
+
+class Player(object):
+    """Remote control for an mpg123 process
+
+    Starts and owns a handle to an mpg123 process then feeds commands to it to
+    play pandora audio
+    """
+
+    def __init__(self, callbacks, control_channel):
+        self._control_channel = control_channel
+        self._control_fd = control_channel.fileno()
+        self._callbacks = callbacks
+        self._process = None
+        self._ensure_started()
+
+    def __del__(self):
+        self._process.kill()
+
+    def _ensure_started(self):
+        """Ensure mpg123 is started
+        """
+        if self._process and self._process.poll() is None:
+            return
+
+        self._process = SilentPopen(
+            ['mpg123', '-q', '-R', '--preload', '0.1'])
+
+        # Only output play status in the player stdout
+        self._send_cmd('silence')
+
+    def _send_cmd(self, cmd):
+        """Write command to remote mpg123 process
+        """
+        self._process.stdin.write("{}\n".format(cmd))
+        self._process.stdin.flush()
+
+    def stop(self):
+        """Stop the currently playing song
+        """
+        self._send_cmd('stop')
+
+    def pause(self):
+        """Pause the player
+        """
+        self._send_cmd('pause')
+
+    def _player_stopped(self, value):
+        """Determine if player has stopped
+        """
+        return value.startswith("@P") and value[3] == "0"
+
+    def play(self, song):
+        """Play a new song from a Pandora model
+
+        Returns once the stream starts but does not shut down the remote mpg123
+        process. Calls the input callback when the user has input.
+        """
+        self._callbacks.play(song)
+        self._send_cmd('load {}'.format(song.audio_url))
+
+        while True:
+            try:
+                self._callbacks.pre_poll()
+                self._ensure_started()
+
+                readers, _, _ = select.select(
+                    [self._control_channel, self._process.stdout], [], [], 1)
+
+                for fd in readers:
+                    value = fd.readline().strip()
+
+                    if fd.fileno() == self._control_fd:
+                        self._callbacks.input(value, song)
+                    else:
+                        if self._player_stopped(value):
+                            return
+            finally:
+                self._callbacks.post_poll()
+
+    def end_station(self):
+        """Stop playing the station
+        """
+        raise StopIteration
+
+    def play_station(self, station):
+        """Play the station until something ends it
+
+        This function will run forever until termintated by calling
+        end_station.
+        """
+        for song in iterate_forever(station.get_playlist):
+            try:
+                self.play(song)
+            except StopIteration:
+                self.stop()
+                return
diff --git a/pydora/player.py b/pydora/player.py
index 2b00ad3..6a9c345 100755
--- a/pydora/player.py
+++ b/pydora/player.py
@@ -10,8 +10,8 @@ import os
 import sys
 
 from pandora import APIClient
-from pandora.player import Player
-from pandora.utils import Colors, Screen
+from pydora.mpg123 import Player
+from .utils import Colors, Screen
 
 
 class PlayerApp:
diff --git a/pydora/utils.py b/pydora/utils.py
new file mode 100644
index 0000000..7a86d7a
--- /dev/null
+++ b/pydora/utils.py
@@ -0,0 +1,79 @@
+from __future__ import print_function
+
+import sys
+import termios
+
+
+class Colors:
+
+    def __wrap_with(code):
+        @staticmethod
+        def inner(text, bold=False):
+            c = code
+            if bold:
+                c = u"1;{}".format(c)
+            return u"\033[{}m{}\033[0m".format(c, text)
+        return inner
+
+    red = __wrap_with('31')
+    green = __wrap_with('32')
+    yellow = __wrap_with('33')
+    blue = __wrap_with('34')
+    magenta = __wrap_with('35')
+    cyan = __wrap_with('36')
+    white = __wrap_with('37')
+
+
+class Screen:
+
+    @staticmethod
+    def set_echo(enabled):
+        fd = sys.stdin.fileno()
+        (iflag, oflag, cflag, lflag, ispeed, ospeed, cc) = \
+            termios.tcgetattr(fd)
+
+        if enabled:
+            lflag |= termios.ECHO
+        else:
+            lflag &= ~termios.ECHO
+
+        termios.tcsetattr(fd, termios.TCSANOW,
+            [iflag, oflag, cflag, lflag, ispeed, ospeed, cc])
+
+    @staticmethod
+    def clear():
+        sys.stdout.write('\x1b[2J\x1b[H')
+        sys.stdout.flush()
+
+    @staticmethod
+    def print_error(msg):
+        print(Colors.red(msg))
+
+    @staticmethod
+    def print_success(msg):
+        print(Colors.green(msg))
+
+    @staticmethod
+    def get_integer(prompt):
+        """Gather user input and convert it to an integer
+
+        Will keep trying till the user enters an interger or until they ^C the
+        program.
+        """
+        try:
+            raw_input
+        except NameError:
+            raw_input = input
+
+        while True:
+            try:
+                return int(raw_input(prompt).strip())
+            except ValueError:
+                print(Colors.red('Invaid Input!'))
+
+
+def clear_screen():
+    """Clear the terminal
+    """
+    sys.stdout.write('\x1b[2J\x1b[H')
+    sys.stdout.flush()