import contextlib
import http.server
import os
import pathlib
import ssl
import threading

# Generated with:
# $ openssl req -new -x509 -days 3650 -nodes -out cert.pem -keyout cert.pem
CERT_FILE = str(pathlib.Path(__file__).parent / "certs" / "cert.pem")


class HttpServerThread(threading.Thread):
    def __init__(self, handler, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.server = http.server.HTTPServer(("localhost", 7777), handler)

    def run(self):
        self.server.serve_forever(poll_interval=0.01)

    def terminate(self):
        self.server.shutdown()
        self.server.server_close()
        self.join()


class HttpsServerThread(HttpServerThread):
    def __init__(self, handler, *args, **kwargs):
        super().__init__(handler, *args, **kwargs)
        self.server.socket = ssl.wrap_socket(
            self.server.socket,
            certfile=CERT_FILE,
            server_side=True,
        )


def create_server(thread_class):
    def server(handler):
        server_thread = thread_class(handler, daemon=True)
        server_thread.start()
        try:
            yield server_thread
        finally:
            server_thread.terminate()
    return contextlib.contextmanager(server)


http_server = create_server(HttpServerThread)
https_server = create_server(HttpsServerThread)


@contextlib.contextmanager
def modify_env(**env):
    original_env = os.environ.copy()
    for k, v in env.items():
        os.environ[k] = v
    yield
    for k in env:
        try:
            os.environ[k] = original_env[k]
        except KeyError:
            os.unsetenv(k)