import os
import sys
from http import server


def main(argv):
    build_workspace_directory = os.environ["BUILD_WORKSPACE_DIRECTORY"]
    docs_directory = argv[1]
    serve_directory = os.path.join(build_workspace_directory, docs_directory)

    class DirectoryHandler(server.SimpleHTTPRequestHandler):
        def __init__(self, *args, **kwargs):
            super().__init__(directory=serve_directory, *args, **kwargs)

    address = ("0.0.0.0", 8000)
    with server.ThreadingHTTPServer(address, DirectoryHandler) as httpd:
        print(f"Serving...")
        print(f"  Address: http://{address[0]}:{address[1]}")
        print(f"  Serving directory: {serve_directory}")
        print(f"  CWD: {os.getcwd()}")
        print()
        print("*** You do not need to restart this server to see changes ***")
        print()
        try:
            httpd.serve_forever()
        except KeyboardInterrupt:
            pass
    return 0


if __name__ == "__main__":
    sys.exit(main(sys.argv))