src/blastradius/web/create_html_report.py

#!/usr/bin/env python3
# Copyright (c) 2026, the R8 project authors. Please see the AUTHORS file
# for details. All rights reserved. Use of this source code is governed by a
# BSD-style license that can be found in the LICENSE file.

import argparse
import base64
import os

# Paths relative to this script
web_dir = os.path.dirname(os.path.abspath(__file__))
index_path = os.path.join(web_dir, 'index.html')
style_path = os.path.join(web_dir, 'style.css')
main_js_path = os.path.join(web_dir, 'main.js')
proto_path = os.path.join(web_dir, 'blastradius.proto')


def ParseOptions():
    parser = argparse.ArgumentParser(
        description='Create a standalone Blast Radius report.')
    parser.add_argument('input', help='Input .pb file')
    parser.add_argument('--output',
                        help='Output .html file',
                        default='blastradius.html')
    return parser.parse_args()


def readTextFile(path):
    with open(path, 'r') as f:
        return f.read()


def readBinaryFile(path):
    with open(path, 'rb') as f:
        return base64.b64encode(f.read()).decode('utf-8')


def main():
    options = ParseOptions()

    # Check if files exist
    for path in [
            index_path, style_path, main_js_path, proto_path
    ]:
        if not os.path.exists(path):
            print(f"Error: Required file not found: {path}")
            return 1

    # Read HTML.
    html = readTextFile(index_path)

    # Inline CSS
    html = html.replace('<link rel="stylesheet" href="style.css">',
                        f'<style>\n{readTextFile(style_path)}\n</style>')

    # Embed proto schema, proto data, and main.js.
    embedded_data = f"""
<script id="blastradius-proto" type="text/plain">
{readTextFile(proto_path)}
</script>
<script id="blastradius-data" type="application/octet-stream">
{readBinaryFile(options.input)}
</script>
<script>
{readTextFile(main_js_path)}
</script>
"""
    html = html.replace(
        '<script src="main.js"></script>',
        embedded_data)

    # Write output.
    with open(options.output, 'w') as f:
        f.write(html)
    return 0


if __name__ == '__main__':
    exit(main())