From e35d332916a46052a6ab164961d829f01532a197 Mon Sep 17 00:00:00 2001
From: Luke Kingland <luke@lukekingland.com>
Date: Tue, 24 Sep 2024 01:01:33 +0900
Subject: [PATCH] initial ASGI implementation (#1)

---
 http/serve.py | 143 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 143 insertions(+)
 create mode 100644 http/serve.py

diff --git a/http/serve.py b/http/serve.py
new file mode 100644
index 00000000..f79898ee
--- /dev/null
+++ b/http/serve.py
@@ -0,0 +1,143 @@
+# ASGI main
+import asyncio
+import logging
+import os
+import hypercorn.config
+import hypercorn.asyncio
+
+DEFAULT_LOG_LEVEL = logging.INFO
+DEFAULT_LISTEN_ADDRESS = "127.0.0.1:8080"
+
+logging.basicConfig(level=DEFAULT_LOG_LEVEL)
+
+
+def serve(f):
+    """serve a function f by wrapping it in an ASGI web application
+    and starting.  The function can be either a constructor for a functon
+    instance (named "new") or a simple ASGI handler function (named "handle").
+    """
+    logging.debug("func runtime creating function instance")
+
+    if f.__name__ == 'new':
+        return ASGIApplication(f()).serve()
+    elif f.__name__ == 'handle':
+        return ASGIApplication(DefaultFunction(f)).serve()
+    else:
+        raise ValueError("function must be either be a constructor 'new' or a "
+                         "handler 'handle'.")
+
+
+class DefaultFunction:
+    """DefaultFunction is used when the provided functon is not a constructor
+    for a funciton instance, but rather a simple handler function"""
+
+    def __init__(self, handler):
+        self.handle = handler
+
+
+class ASGIApplication():
+    def __init__(self, f):
+        self.f = f
+        # Inform the user via logs that defaults will be used for health
+        # endpoints if no matchin methods were provided.
+        if hasattr(self.f, "alive") is not True:
+            logging.info(
+                "function does not implement 'alive'. Using default "
+                "implementation for liveness checks."
+            )
+
+        if hasattr(self.f, "ready") is not True:
+            logging.info(
+                "function does not implement 'ready'. Using default "
+                "implementation for readiness checks."
+            )
+
+    def serve(self):
+        """serve serving this ASGIhandler, delegating implementation of
+           methods as necessary to the wrapped Function instance"""
+        cfg = hypercorn.config.Config()
+        cfg.bind = [os.getenv('LISTEN_ADDRESS', DEFAULT_LISTEN_ADDRESS)]
+
+        logging.debug(f"function starting on {cfg.bind}")
+        return asyncio.run(hypercorn.asyncio.serve(self, cfg))
+
+    async def on_start(self):
+        """on_start handles the ASGI server start event, delegating control
+           to the internal Function instance if it has a "start" method."""
+        if hasattr(self.f, "start"):
+            self.f.start(os.environ.copy())
+        else:
+            logging.info("function does not implement 'start'. Skipping.")
+
+    async def on_stop(self):
+        if hasattr(self.f, "stop"):
+            self.f.stop()
+        else:
+            logging.info("function does not implement 'stop'. Skipping.")
+
+    # Register ASGIFunctoin as a callable ASGI Function
+    async def __call__(self, scope, receive, send):
+        if scope['type'] == 'lifespan':
+            while True:
+                message = await receive()
+                if message['type'] == 'lifespan.startup':
+                    await self.on_start()
+                    await send({'type': 'lifespan.startup.complete'})
+                elif message['type'] == 'lifespan.shutdown':
+                    await self.on_stop()
+                    await send({'type': 'lifespan.shutdown.complete'})
+                    return
+                else:
+                    break
+
+        # Assert request is HTTP
+        if scope["type"] != "http":
+            await send_exception(send, 400,
+                                 "Functions currenly only support ASGI/HTTP "
+                                 f"connections. Got {scope['type']}"
+                                 )
+            return
+
+        # Route request
+        try:
+            if scope['path'] == '/health/liveness':
+                await self.handle_liveness(scope, receive, send)
+            elif scope['path'] == '/health/readiness':
+                await self.handle_readiness(scope, receive, send)
+            else:
+                if hasattr(self.f, "handle"):
+                    await self.f.handle(scope, receive, send)
+                else:
+                    raise Exception("function does not implement handle")
+        except Exception as e:
+            await send_exception(send, 500, f"Error: {e}")
+
+    async def handle_liveness(self, scope, receive, send):
+        if hasattr(self.f, "alive"):
+            self.f.alive()
+        else:
+            await send({'type': 'http.response.start', 'status': 200,
+                        'headers': [[b'content-type', b'text/plain']]})
+            await send({'type': 'http.response.body',
+                        'body': b'OK',
+                        })
+
+    async def handle_readiness(self, scope, receive, send):
+        if hasattr(self.f, "ready"):
+            self.f.ready()
+        else:
+            await send({'type': 'http.response.start', 'status': 200,
+                        'headers': [[b'content-type', b'text/plain']]})
+            await send({'type': 'http.response.body',
+                        'body': b'OK',
+                        })
+
+
+async def send_exception(send, code, message):
+    await send({
+        'type': 'http.response.start', 'status': code,
+        'headers': [[b'content-type', b'text/plain']],
+    })
+    await send({
+        'type': 'http.response.body', 'body': message,
+    })