init

workers/site/api-router.ts

@@ -0,0 +1,106 @@
+import { Router, withParams } from "itty-router";
+import { createServerContext } from "./context";
+
+export function createRouter() {
+  return (
+    Router()
+      .get("/assets/*", (r, e, c) => {
+        const { assetService } = createServerContext(e, c);
+        return assetService.handleStaticAssets(r, e, c);
+      })
+
+      .post("/api/contact", (r, e, c) => {
+        const { contactService } = createServerContext(e, c);
+        return contactService.handleContact(r);
+      })
+
+      .post("/api/chat", (r, e, c) => {
+        const { chatService } = createServerContext(e, c);
+        return chatService.handleChatRequest(r);
+      })
+
+      .get(
+        "/api/streams/:streamId",
+        withParams,
+        async ({ streamId }, env, ctx) => {
+          const { chatService } = createServerContext(env, ctx);
+          return chatService.handleSseStream(streamId); // Handles SSE for streamId
+        },
+      )
+
+      .get(
+        "/api/streams/webhook/:streamId",
+        withParams,
+        async ({ streamId }, env, ctx) => {
+          const { chatService } = createServerContext(env, ctx);
+          return chatService.proxyWebhookStream(streamId); // Handles SSE for streamId
+        },
+      )
+
+      .post("/api/feedback", async (r, e, c) => {
+        const { feedbackService } = createServerContext(e, c);
+        return feedbackService.handleFeedback(r);
+      })
+
+      .post("/api/tx", async (r, e, c) => {
+        const { transactionService } = createServerContext(e, c);
+        return transactionService.handleTransact(r);
+      })
+
+      // used for file handling, can be enabled but is not fully implemented in this fork.
+      // .post('/api/documents', async (r, e, c) => {
+      //     const {documentService} = createServerContext(e, c);
+      //     return documentService.handlePutDocument(r)
+      // })
+      //
+      // .get('/api/documents', async (r, e, c) => {
+      //     const {documentService} = createServerContext(e, c);
+      //     return documentService.handleGetDocument(r)
+      // })
+
+      .all("/api/metrics/*", async (r, e, c) => {
+        const { metricsService } = createServerContext(e, c);
+        return metricsService.handleMetricsRequest(r);
+      })
+
+      .get("*", async (r, e, c) => {
+        const { assetService } = createServerContext(e, c);
+
+        console.log("Request received:", { url: r.url, headers: r.headers });
+
+        // First attempt to serve pre-rendered HTML
+        const preRenderedHtml = await assetService.handleStaticAssets(r, e, c);
+
+        if (
+          preRenderedHtml !== null &&
+          typeof preRenderedHtml === "object" &&
+          Object.keys(preRenderedHtml).length > 0
+        ) {
+          console.log("Serving pre-rendered HTML for:", r.url);
+          console.log({ preRenderedHtml });
+          return preRenderedHtml;
+        }
+
+        // If no pre-rendered HTML, attempt SSR
+        console.log("No pre-rendered HTML found, attempting SSR for:", r.url);
+        const ssrResponse = await assetService.handleSsr(r.url, r.headers, e);
+        if (
+          ssrResponse !== null &&
+          typeof ssrResponse === "object" &&
+          Object.keys(ssrResponse).length > 0
+        ) {
+          console.log("SSR successful for:", r.url);
+          return ssrResponse;
+        }
+
+        // If no 404.html exists, fall back to static assets
+        console.log("Serving not found:", r.url);
+
+        const url = new URL(r.url);
+
+        url.pathname = "/404.html";
+        // Finally, try to serve 404.html for not found pages
+        return assetService.handleStaticAssets(new Request(url, r), e, c);
+      })
+  );
+}