1 files changed, 71 insertions, 4 deletions

diff --git a/roka.py b/roka.py
index 4c4da57..8d81641 100755
--- a/roka.py
+++ b/roka.py
@@ -2,13 +2,15 @@
 
 import argparse
 import os
-from flask import Flask, request, Response, render_template, send_file
+import shutil
+from flask import Flask, request, Response, render_template, send_file, templating
 from lib.books import Books
 from lib.util import check_auth, escape, generate_rss, read_cache
 
 abs_path = os.path.dirname(os.path.abspath(__file__))
 app = Flask(__name__)
-app.config.from_pyfile(os.path.join(abs_path, 'app.cfg'))
+if os.path.exists(os.path.join(abs_path, 'app.cfg')):
+    app.config.from_pyfile(os.path.join(abs_path, 'app.cfg'))
 cache_path = os.path.join(abs_path, 'cache')
 json_path = os.path.join(cache_path, 'audiobooks.json')
 
@@ -40,7 +42,7 @@ def list_books():
         if not books.get(a):
             return 'book not found', 404
 
-        rss = generate_rss(request, books)
+        rss = generate_rss(request.base_url, a, books)
         return Response(rss, mimetype='text/xml')
 
     else:
@@ -51,17 +53,82 @@ def list_books():
 
         return render_template('index.html', books=books)
 
+# TODO does this really need to be here?
+def my_render_template(
+    app, template_name_or_list, **context
+) -> str:
+    """Renders a template from the template folder with the given
+    context.
+
+    :param template_name_or_list: the name of the template to be
+                                  rendered, or an iterable with template names
+                                  the first one existing will be rendered
+    :param context: the variables that should be available in the
+                    context of the template.
+    """
+    app.update_template_context(context)
+    return templating._render(
+        app.jinja_env.get_or_select_template(template_name_or_list),
+        context,
+        app,
+    )
+
+def generate(static_path, base_url, audiobook_dirs):
+    static_index_path = os.path.join(static_path, 'index.html')
+
+    books = Books()
+    books.scan_books(audiobook_dirs)
+    # TODO avoid writing and reading cache
+    books.write_cache()
+    books = read_cache(json_path)
+    index = my_render_template(app, 'index.html', books=books, static=True)
+
+    os.makedirs(static_path, exist_ok=True)
+
+    indexfile = open(static_index_path, 'w')
+    indexfile.write(index)
+    indexfile.close()
+
+    for b_key, book in books.items():
+        rss = generate_rss(base_url, b_key, books, static=True)
+        rss_path = os.path.join(static_path, b_key + '.xml')
+        rssfile = open(rss_path, 'w')
+        rssfile.write(rss.decode('utf-8'))
+        rssfile.close()
+
+        book_dir = os.path.join(static_path, b_key)
+        os.makedirs(book_dir, exist_ok=True)
+
+        for f_key, file in book['files'].items():
+            f_path = file['path']
+            copy_path = os.path.join(book_dir, f_key + '.mp3')
+            shutil.copyfile(f_path, copy_path)
+
 if __name__ == '__main__':
     desc = 'roka: listen to audiobooks with podcast apps via RSS'
     parser = argparse.ArgumentParser(description=desc)
     parser.add_argument('--scan', dest='scan', action='store_true',
                         help='scan audiobooks directory for new books',
                         required=False)
+    parser.add_argument('--generate', dest='static_path', type=str, action='store',
+                        help='Output directory to generate static files',
+                        required=False)
+    parser.add_argument('--base_url', dest='base_url', type=str, action='store',
+                        help='Base URL to use in static files',
+                        required=False)
+    parser.add_argument('--scan_dir', dest='scan_dir', type=str, action='store',
+                        help='Directory to scan for audiobooks',
+                        required=False)
     args = parser.parse_args()
 
+    if args.static_path and not args.base_url or args.base_url and not args.static_path:
+        parser.error('--generate and --base_url must be included together')
+
     if args.scan:
         books = Books()
-        books.scan_books()
+        books.scan_books(args.scan_dir)
         books.write_cache()
+    elif args.static_path:
+        generate(args.static_path, args.base_url, args.scan_dir)
     else:
         app.run(host='127.0.0.1', port='8085', threaded=True)