mirror of https://github.com/dirtbags/moth.git
152 lines
4.5 KiB
Python
Executable File
152 lines
4.5 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import glob
|
|
import http.server
|
|
import mistune
|
|
import os
|
|
import pathlib
|
|
import puzzles
|
|
import socketserver
|
|
|
|
if hasattr(http.server, 'HTTPStatus'):
|
|
HTTPStatus = http.HTTPStatus
|
|
else:
|
|
class HTTPStatus:
|
|
NOT_FOUND = 404
|
|
OK = 200
|
|
|
|
|
|
def page(title, body):
|
|
return """<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<title>{}</title>
|
|
<link rel="stylesheet" href="/files/www/res/style.css">
|
|
</head>
|
|
<body>
|
|
<div id="preview" class="terminal">
|
|
{}
|
|
</div>
|
|
</body>
|
|
</html>""".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])) 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; encoding=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()
|