#!/usr/bin/python3 import glob import http.server import mistune import os import pathlib import puzzles import socketserver HTTPStatus = http.server.HTTPStatus def page(title, body): return """ {}
{}
""".format(title, body) def mdpage(body): try: title, _ = body.split('\n', 1) except ValueError: title = "Result" title = title.lstrip("#") title = title.strip() return page(title, mistune.markdown(body)) class ThreadingServer(socketserver.ThreadingMixIn, http.server.HTTPServer): pass class MothHandler(http.server.CGIHTTPRequestHandler): def do_GET(self): if self.path == "/": self.serve_front() elif self.path.startswith("/puzzles"): self.serve_puzzles() elif self.path.startswith("/files"): self.serve_file() else: self.send_error(HTTPStatus.NOT_FOUND, "File not found") def translate_path(self, path): if path.startswith('/files'): path = path[7:] return super().translate_path(path) def serve_front(self): page = """ MOTH Development Server Front Page ==================== Yo, it's the front page. There's stuff you can do here: * [Available puzzles](/puzzles) * [Raw filesystem view](/files/) * [Documentation](/files/doc/) * [Instructions](/files/doc/devel-server.md) for using this server If you use this development server to run a contest, you are a fool. """ self.serve_md(page) def serve_puzzles(self): body = [] path = self.path.rstrip('/') parts = path.split("/") if len(parts) < 3: # List all categories body.append("# Puzzle Categories") for i in glob.glob(os.path.join("puzzles", "*", "")): body.append("* [{}](/{})".format(i, i)) elif len(parts) == 3: # List all point values in a category body.append("# Puzzles in category `{}`".format(parts[2])) puzz = [] for i in glob.glob(os.path.join("puzzles", parts[2], "*.moth")): base = os.path.basename(i) root, _ = os.path.splitext(base) points = int(root) puzz.append(points) for puzzle in sorted(puzz): body.append("* [puzzles/{cat}/{points}](/puzzles/{cat}/{points}/)".format(cat=parts[2], points=puzzle)) elif len(parts) == 4: body.append("# {} puzzle {}".format(parts[2], parts[3])) with open("puzzles/{}/{}.moth".format(parts[2], parts[3]), encoding="utf-8") as f: p = puzzles.Puzzle(f) body.append("* Author: `{}`".format(p.fields.get("author"))) body.append("* Summary: `{}`".format(p.fields.get("summary"))) body.append('') body.append("## Body") body.append(p.body) body.append("## Answers:") for a in p.answers: body.append("* `{}`".format(a)) body.append("") else: body.append("# Not Implemented Yet") self.serve_md('\n'.join(body)) def serve_file(self): if self.path.endswith(".md"): self.serve_md() else: super().do_GET() def serve_md(self, text=None): fspathstr = self.translate_path(self.path) fspath = pathlib.Path(fspathstr) if not text: try: text = fspath.read_text() except OSError: self.send_error(HTTPStatus.NOT_FOUND, "File not found") return None content = mdpage(text) self.send_response(http.server.HTTPStatus.OK) self.send_header("Content-type", "text/html; charset=utf-8") self.send_header("Content-Length", len(content)) try: fs = fspath.stat() self.send_header("Last-Modified", self.date_time_string(fs.st_mtime)) except: pass self.end_headers() self.wfile.write(content.encode('utf-8')) def run(address=('', 8080)): httpd = ThreadingServer(address, MothHandler) print("=== Listening on http://{}:{}/".format(address[0] or "localhost", address[1])) httpd.serve_forever() if __name__ == '__main__': run()