From 1c1ec2f6bccba95a3acc75d867e30783c2c2470e Mon Sep 17 00:00:00 2001
From: Joshua <lio@werewolf.design>
Date: Sun, 18 Sep 2022 15:06:31 +0200
Subject: [PATCH] Add Domain 'Middleware'

---
 src/index.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/index.ts b/src/index.ts
index 69b857b..cfdec87 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,5 +1,6 @@
 import { pogo, serve } from "../deps.ts";
 import routes from "./routes.generated.ts"
+import domain from "../utils/middleware/domain.ts";
 
 const env = {
     port: Number(Deno.env.get("POGO_PORT")) || 3000,
@@ -16,7 +17,12 @@ server.router.get('/static/{file*}', (r, h) => {
 
 
 routes.forEach(route => {
-    server.route({ method: route.method, handler: route.handler, path: route.route })
+    server.route({
+        method: route.method, handler: (r, h) => {
+            if (!route.domains) route.domains = ["*"]
+            return domain({ r, h }, route.domains, route.handler)
+        }, path: route.route
+    })
 })