7 files changed, 135 insertions, 52 deletions

diff --git a/youtube_podcaster/__init__.py b/youtube_podcaster/__init__.py
index 32aee7c..77ab9cc 100644
--- a/youtube_podcaster/__init__.py
+++ b/youtube_podcaster/__init__.py
@@ -1,27 +1,47 @@
 #!/usr/bin/env python3
 
-import json
+import argparse
 
-from http.server import (
-    HTTPServer,
-)
+from http.server import HTTPServer
 
-from . import (
-    youtube,
+from .podcastfeeder import create_feeder
+from .config import (
+        Config,
+        ConfigException
 )
 
-from .podcastfeeder import (
-    create_feeder
-)
+"""
+Start the program
+"""
 
 
 def main():
-    config = json.load(open("youtube-podcaster.json"))
+    arg_parser = argparse.ArgumentParser(prog="youtube-podcaster",
+                                         description="Converts youtube \
+                                         playlists to RSS-feeds")
+    arg_parser.add_argument("-c", "--config",
+                            dest="config",
+                            help="Use CONFIG as the config file")
+    arg_parser.add_argument("-i", "--interface",
+                            dest="interface",
+                            help="The interface the http server will listen on")
+    arg_parser.add_argument("-p", "--port",
+                            dest="port",
+                            help="The port the http server will listen on")
+    arg_parser.add_argument("--api-key",
+                            dest="apikey",
+                            help="The YouTube API v3 key")
+    args = arg_parser.parse_args()
 
     try:
-        PodcastFeeder = create_feeder(config["youtube"], config["podcasts"])
-        server = HTTPServer(("", 8888), PodcastFeeder)
+        config = Config.parse_config(args)
+
+        PodcastFeeder = create_feeder(config)
+
+        server = HTTPServer(config.get_server_address(), PodcastFeeder)
         server.serve_forever()
+    except ConfigException as e:
+        print(e)
     except KeyboardInterrupt:
         server.socket.close()
 
diff --git a/youtube_podcaster/__main__.py b/youtube_podcaster/__main__.py
deleted file mode 100644
index 941a79a..0000000
--- a/youtube_podcaster/__main__.py
+++ /dev/null
@@ -1,8 +0,0 @@
-#!/usr/bin/env python3
-
-import youtube_podcaster
-
-if __name__ == "__main__":
-    youtube_podcaster.main()
-
-#  vim: set ts=8 sw=4 tw=0 et :
diff --git a/youtube_podcaster/config.py b/youtube_podcaster/config.py
new file mode 100644
index 0000000..bfce902
--- /dev/null
+++ b/youtube_podcaster/config.py
@@ -0,0 +1,64 @@
+import sys
+import os
+import json
+
+
+class ConfigException(Exception):
+    def __init__(self, msg):
+        super(ConfigException, self).__init__("Config exception: %s" % (msg))
+
+
+class Config:
+    instance = None
+
+    def parse_config(config_file=None):
+        if not Config.instance:
+            Config.instance = Config(config_file)
+
+        return Config.instance
+
+    def __init__(self, args):
+        if args.config:
+            config = args.config
+        else:
+            config_file = "youtube-podcaster.json"
+
+            if sys.platform == "linux" and not hasattr(sys, "real_prefix"):
+                config = "/etc/%s" % (config_file)
+            else:
+                config = "%s/etc/%s" % (sys.prefix, config_file)
+
+        if not os.path.isfile(config):
+            raise ConfigException("%s not found" % (config))
+
+        try:
+            config = json.load(open(config))
+        except json.decoder.JSONDecodeError as e:
+            raise ConfigException("%s is not valid json: %s" % (
+                                  config, str(e)))
+
+        try:
+            self.server = config["server"]
+            self.youtube = config["youtube"]
+            self.podcasts = config["podcasts"]
+        except KeyError as e:
+            raise ConfigException("Missing %s-section in %s" % (
+                                  str(e), config))
+
+        for arg, value in vars(args).items():
+            if not value:
+                continue
+
+            if arg == "interface":
+                self.server["interface"] = value
+            elif arg == "port":
+                self.server["port"] = value
+            elif arg == "apikey":
+                self.youtube["api-key"] = value
+
+    def get_server_address(self):
+        interface = str(self.server["interface"])
+        port = int(self.server["port"])
+        return interface, port
+
+#  vim: set ts=8 sw=4 tw=0 et :
diff --git a/youtube_podcaster/podcastfeeder.py b/youtube_podcaster/podcastfeeder.py
index 3f752d2..4181d6c 100644
--- a/youtube_podcaster/podcastfeeder.py
+++ b/youtube_podcaster/podcastfeeder.py
@@ -5,10 +5,10 @@ from .podcastupdater import (
 )
 
 
-def create_feeder(youtube_config, podcast_config):
+def create_feeder(config):
     class PodcastFeeder(BaseHTTPRequestHandler):
         def __init__(self, request, client_address, server):
-            self.updater = PodcastUpdater(youtube_config, podcast_config)
+            self.updater = PodcastUpdater(config)
             super(PodcastFeeder, self).__init__(request, client_address, server)
 
         def do_GET(self):
diff --git a/youtube_podcaster/podcastupdater.py b/youtube_podcaster/podcastupdater.py
index 4a0a017..75b8b10 100644
--- a/youtube_podcaster/podcastupdater.py
+++ b/youtube_podcaster/podcastupdater.py
@@ -2,6 +2,7 @@ import pickle
 import os
 import time
 import hashlib
+import sys
 
 from feedgen.feed import FeedGenerator
 
@@ -11,11 +12,19 @@ from . import (
 
 
 class PodcastUpdater:
-    def __init__(self, youtube_config, podcast_config):
-        self.podcasts = podcast_config
-        self.youtube = youtube.Youtube(youtube_config["api-key"])
+    def __init__(self, config):
+        self.podcasts = config.podcasts
+        self.youtube = youtube.Youtube(config.youtube["api-key"])
+
+        if sys.platform == "linux" and not hasattr(sys, "real_prefix"):
+            self.data_dir = "/var/lib/youtube-podcaster"
+        else:
+            self.data_dir = "%s/var/lib/youtube-podcaster" % (sys.prefix)
+
+        os.makedirs(self.data_dir, 0o755, True)
+
+        self.feeds_file = "%s/feeds.dump" % (self.data_dir)
 
-        self.feeds_file = "feeds.save"
         if os.path.isfile(self.feeds_file):
             with open(self.feeds_file, "rb") as feeds:
                 self.feeds = pickle.load(feeds)
@@ -42,6 +51,8 @@ class PodcastUpdater:
 
     def update_podcast(self, channel, playlist):
         feed_id = hashlib.sha1(bytes("%s %s" % (channel, playlist), "UTF-8")).hexdigest()
+        feed_file = "%s/%s.xml" % (self.data_dir, feed_id)
+
         yt_channel = self.youtube.get_channel(channel)[0]
         yt_playlists = self.youtube.get_playlists(yt_channel, 50)
 
@@ -58,17 +69,16 @@ class PodcastUpdater:
 
         if feed.last_updated < time.time() - 600:
             self.populate_feed(feed, feed_id, yt_playlist)
+            feed.rss_file(feed_file)
 
-            feed_file = "%s.xml" % (feed_id)
-            self.feeds[feed_id].rss_file(feed_file)
-
+            with open(self.feeds_file, "wb") as feeds:
+                pickle.dump(self.feeds, feeds)