#!/usr/bin/env python3.6 # Copyright (C) 2018 Andrew Hamilton. All rights reserved. # Licensed under the Artistic License 2.0. import gzip import http.server import os import sys import pickle import vigil.fill3 as fill3 import vigil.tools as tools USAGE = """Usage: vigil-webserver Example: vigil-webserver my_project """ def make_page(body_html, title): return (f"{title}{body_html}" ).encode("utf-8") class Webserver(http.server.BaseHTTPRequestHandler): def _set_headers(self): self.send_response(200) self.send_header("Content-type", "text/html") self.end_headers() def do_GET(self): self._set_headers() if self.path == "/": page = summary_page elif "/" in self.path[1:]: path, tool = os.path.split(self.path[1:]) result = index[(path, tool)] body = fill3.appearance_as_html( fill3.Border(result).appearance_min()) page = make_page(body, f"{tool} of {path}") else: return self.wfile.write(page) def do_HEAD(self): self._set_headers() def do_POST(self): self._set_headers() self.wfile.write("posted".encode("utf-8")) def make_summary_page(project_name, summary): summary_html, summary_styles = summary.as_html() body_html = ("\n".join(style.as_html() for style in summary_styles) + "\n" + summary_html) return make_page(body_html, "Summary of " + project_name) def run(server_class=http.server.HTTPServer, handler_class=Webserver, port=80): server_address = ("", port) httpd = server_class(server_address, handler_class) print("Starting httpd...") httpd.serve_forever() def main(): global summary_page, index if len(sys.argv) == 1: print(USAGE) sys.exit(1) project_path = os.path.abspath(sys.argv[1]) os.chdir(project_path) project_name = os.path.basename(project_path) pickle_path = os.path.join(project_path, tools.CACHE_PATH, "summary.pickle") with gzip.open(pickle_path, "rb") as file_: screen = pickle.load(file_) summary_page = make_summary_page(project_name, screen._summary) index = {} for row in screen._summary._column: for result in row: index[(result.path[2:], result.tool.__name__)] = result.result run() if __name__ == "__main__": main()