1 files changed, 179 insertions, 0 deletions
diff --git a/pathod/libpathod/app.py b/pathod/libpathod/app.py
new file mode 100644
index 00000000..debebaf2
--- /dev/null
+++ b/pathod/libpathod/app.py
@@ -0,0 +1,179 @@
+import logging
+import pprint
+import cStringIO
+import copy
+from flask import Flask, jsonify, render_template, request, abort, make_response
+from . import version, language, utils
+from netlib.http import user_agents
+
+logging.basicConfig(level="DEBUG")
+EXAMPLE_HOST = "example.com"
+EXAMPLE_WEBSOCKET_KEY = "examplekey"
+
+# pylint: disable=unused-variable
+
+
+def make_app(noapi, debug):
+    app = Flask(__name__)
+    app.debug = debug
+
+    if not noapi:
+        @app.route('/api/info')
+        def api_info():
+            return jsonify(
+                version=version.IVERSION
+            )
+
+    @app.route('/api/log')
+    def api_log():
+        return jsonify(
+            log=app.config["pathod"].get_log()
+        )
+
+    @app.route('/api/clear_log')
+    def api_clear_log():
+        app.config["pathod"].clear_log()
+        return "OK"
+
+    def render(s, cacheable, **kwargs):
+        kwargs["noapi"] = app.config["pathod"].noapi
+        kwargs["nocraft"] = app.config["pathod"].nocraft
+        kwargs["craftanchor"] = app.config["pathod"].craftanchor
+        resp = make_response(render_template(s, **kwargs), 200)
+        if cacheable:
+            resp.headers["Cache-control"] = "public, max-age=4320"
+        return resp
+
+    @app.route('/')
+    @app.route('/index.html')
+    def index():
+        return render(
+            "index.html",
+            True,
+            section="main",
+            version=version.VERSION
+        )
+
+    @app.route('/download')
+    @app.route('/download.html')
+    def download():
+        return render(
+            "download.html", True, section="download", version=version.VERSION
+        )
+
+    @app.route('/about')
+    @app.route('/about.html')
+    def about():
+        return render("about.html", True, section="about")
+
+    @app.route('/docs/pathod')
+    def docs_pathod():
+        return render(
+            "docs_pathod.html", True, section="docs", subsection="pathod"
+        )
+
+    @app.route('/docs/language')
+    def docs_language():
+        return render(
+            "docs_lang.html", True,
+            section="docs", uastrings=user_agents.UASTRINGS,
+            subsection="lang"
+        )
+
+    @app.route('/docs/pathoc')
+    def docs_pathoc():
+        return render(
+            "docs_pathoc.html", True, section="docs", subsection="pathoc"
+        )
+
+    @app.route('/docs/libpathod')
+    def docs_libpathod():
+        return render(
+            "docs_libpathod.html", True, section="docs", subsection="libpathod"
+        )
+
+    @app.route('/docs/test')
+    def docs_test():
+        return render(
+            "docs_test.html", True, section="docs", subsection="test"
+        )
+
+    @app.route('/log')
+    def log():
+        if app.config["pathod"].noapi:
+            abort(404)
+        return render(
+            "log.html",
+            False,
+            section="log",
+            log=app.config["pathod"].get_log()
+        )
+
+    @app.route('/log/<int:lid>')
+    def onelog(lid):
+        item = app.config["pathod"].log_by_id(int(lid))
+        if not item:
+            abort(404)
+        l = pprint.pformat(item)
+        return render("onelog.html", False, section="log", alog=l, lid=lid)
+
+    def _preview(is_request):
+        if is_request:
+            template = "request_preview.html"
+        else:
+            template = "response_preview.html"
+
+        spec = request.args["spec"]
+
+        args = dict(
+            spec=spec,
+            section="main",
+            syntaxerror=None,
+            error=None,
+        )
+        if not spec.strip():
+            args["error"] = "Can't parse an empty spec."
+            return render(template, False, **args)
+
+        try:
+            if is_request:
+                r = language.parse_pathoc(spec).next()
+            else:
+                r = language.parse_pathod(spec).next()
+        except language.ParseException as v:
+            args["syntaxerror"] = str(v)
+            args["marked"] = v.marked()
+            return render(template, False, **args)
+
+        s = cStringIO.StringIO()
+
+        settings = copy.copy(app.config["pathod"].settings)
+        settings.request_host = EXAMPLE_HOST
+        settings.websocket_key = EXAMPLE_WEBSOCKET_KEY
+
+        safe = r.preview_safe()
+        err, safe = app.config["pathod"].check_policy(
+            safe,
+            settings
+        )
+        if err:
+            args["error"] = err
+            return render(template, False, **args)
+        if is_request:
+            settings.request_host = EXAMPLE_HOST
+            language.serve(safe, s, settings)
+        else:
+            settings.websocket_key = EXAMPLE_WEBSOCKET_KEY
+            language.serve(safe, s, settings)
+
+        args["output"] = utils.escape_unprintables(s.getvalue())
+        return render(template, False, **args)
+
+    @app.route('/response_preview')
+    def response_preview():
+        return _preview(False)
+
+    @app.route('/request_preview')
+    def request_preview():
+        return _preview(True)
+    return app