Add interactive configuration

2 files changed, 188 insertions, 1 deletions

diff --git a/pydora/configure.py b/pydora/configure.py
new file mode 100644
index 0000000..ed0d862
--- /dev/null
+++ b/pydora/configure.py
@@ -0,0 +1,166 @@
+import os
+import re
+import sys
+import requests
+
+try:
+    from configparser import SafeConfigParser
+except ImportError:
+    from ConfigParser import SafeConfigParser
+
+from pandora.client import APIClient
+from pandora.clientbuilder import PydoraConfigFileBuilder
+
+from .utils import Screen, Colors
+
+
+class umask(object):
+    """Set/Restore Umask Context Manager
+    """
+
+    def __init__(self, umask):
+        self.umask = umask
+        self.old_umask = None
+
+    def __enter__(self):
+        self.old_umask = os.umask(self.umask)
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        os.umask(self.old_umask)
+
+
+class PandoraKeysConfigParser(object):
+    """Parser for Pandora Keys Source Page
+
+    This is an extremely naive restructured text parser designed only to parse
+    the pandora API docs keys source file.
+    """
+
+    KEYS_URL = ("http://6xq.net/git/lars/pandora-apidoc.git/"
+                "plain/json/partners.rst")
+
+    FIELD_RE = re.compile(
+            ":(?P<key>[^:]+): (?:`{2})?(?P<value>[^`\n]+)(?:`{2})?$")
+
+    def _fixup_key(self, key):
+        key = key.lower()
+
+        if key.startswith("dec") and "password" in key:
+            return "decryption_key"
+        elif key.startswith("encrypt"):
+            return "encryption_key"
+        elif key == "deviceid":
+            return "device"
+        else:
+            return key
+
+    def _format_api_host(self, host):
+        return "{}/services/json/".format(host)
+
+    def _clean_device_name(self, name):
+        return re.sub("[^a-z]+", "_", name, flags=re.I)
+
+    def _fetch_config(self):
+        return requests.get(self.KEYS_URL).text.split("\n")
+
+    def _match_key(self, line):
+        key_match = self.FIELD_RE.match(line)
+        if key_match:
+            match = key_match.groupdict()
+            match["key"] = self._fixup_key(match["key"])
+            return match
+        else:
+            return None
+
+    def _is_host_terminator(self, line):
+        return line.startswith("--")
+
+    def _is_device_terminator(self, line):
+        return line.startswith("^^")
+
+    def load(self):
+        buffer = []
+        current_partner = {}
+        api_host = None
+        partners = {}
+
+        for line in self._fetch_config():
+            key_match = self._match_key(line)
+            if key_match:
+                current_partner[key_match["key"]] = key_match["value"]
+            elif self._is_host_terminator(line):
+                api_host = buffer.pop()
+            elif self._is_device_terminator(line):
+                key = self._clean_device_name(buffer.pop())
+                current_partner = partners[key] = {
+                        "api_host": self._format_api_host(api_host) }
+
+            buffer.append(line.strip().lower())
+
+        return partners
+
+
+class Configurator(object):
+    """Interactive Configuration Builder
+
+    Allows a user to configure pydora interactively. Ultimately writes the
+    pydora config file.
+    """
+
+    def __init__(self):
+        self.builder = PydoraConfigFileBuilder()
+
+        self.cfg = SafeConfigParser()
+        self.cfg.add_section("user")
+        self.cfg.add_section("api")
+
+    def fail(self, message):
+        print(Screen.print_error(message))
+        sys.exit(1)
+
+    def finished(self, message):
+        Screen.print_success(message)
+        sys.exit(0)
+
+    def print_message(self, message):
+        print(Colors.cyan(message))
+
+    def get_partner_config(self):
+        try:
+            return PandoraKeysConfigParser().load()["ios"]
+        except:
+            self.fail("Error loading config file. Unable to continue.")
+
+    def get_value(self, section, key, prompt):
+        self.cfg.set(section, key, Screen.get_string(prompt))
+
+    def get_password(self, section, key, prompt):
+        self.cfg.set(section, key, Screen.get_password(prompt))
+
+    def set_static_value(self, section, key, value):
+        self.cfg.set(section, key, value)
+
+    def add_partner_config(self, config):
+        for key, value in config.items():
+            self.cfg.set("api", key, value)
+
+    def write_config(self):
+        with umask(0o077), open(self.builder.path, "w") as fp:
+            self.cfg.write(fp)
+
+    def configure(self):
+        if self.builder.file_exists:
+            self.finished("You already have a pydora config!")
+
+        self.print_message("Welcome to Pydora, let's configure a few things")
+        self.add_partner_config(self.get_partner_config())
+        self.get_value("user", "username", "Pandora Username: ")
+        self.get_password("user", "password", "Pandora Password: ")
+        self.set_static_value("api", "default_audio_quality",
+                APIClient.HIGH_AUDIO_QUALITY)
+
+        self.write_config()
+
+
+def main():
+    Configurator().configure()
diff --git a/pydora/utils.py b/pydora/utils.py
index 76d12f3..92a15a3 100644
--- a/pydora/utils.py
+++ b/pydora/utils.py
@@ -2,6 +2,7 @@ from __future__ import print_function
 
 import sys
 import termios
+import getpass
 
 
 def input(prompt):
@@ -62,6 +63,26 @@ class Screen(object):
         print(Colors.green(msg))
 
     @staticmethod
+    def get_string(prompt):
+        while True:
+            value = input(prompt).strip()
+
+            if not value:
+                print(Colors.red("Value Required!"))
+            else:
+                return value
+
+    @staticmethod
+    def get_password(prompt="Password: "):
+        while True:
+            value = getpass.getpass(prompt)
+
+            if not value:
+                print(Colors.red("Value Required!"))
+            else:
+                return value
+
+    @staticmethod
     def get_integer(prompt):
         """Gather user input and convert it to an integer
 
@@ -72,7 +93,7 @@ class Screen(object):
             try:
                 return int(input(prompt).strip())
             except ValueError:
-                print(Colors.red("Invaid Input!"))
+                print(Colors.red("Invalid Input!"))
 
 
 def clear_screen():