From 1b609f5f18373af18f991ff2e6766a484edd8e85 Mon Sep 17 00:00:00 2001
From: Rafa Moreno <rmoreno@tinybird.co>
Date: Tue, 9 Jan 2024 22:14:00 +0100
Subject: [PATCH] try edge functions

---
 api/cors.js  | 121 +++++++++++++++++++++++++++++++++++++++++++++++++++
 api/index.js |  50 ++++++++++++---------
 2 files changed, 150 insertions(+), 21 deletions(-)
 create mode 100644 api/cors.js

diff --git a/api/cors.js b/api/cors.js
new file mode 100644
index 0000000..190285e
--- /dev/null
+++ b/api/cors.js
@@ -0,0 +1,121 @@
+const defaultOptions = {
+    origin: "*",
+    methods: "GET,HEAD,PUT,PATCH,POST,DELETE",
+    preflightContinue: false,
+    optionsSuccessStatus: 204,
+};
+
+function isOriginAllowed(origin, allowed) {
+    return Array.isArray(allowed)
+        ? allowed.some((o) => isOriginAllowed(origin, o))
+        : typeof allowed === "string"
+        ? origin === allowed
+        : allowed instanceof RegExp
+        ? allowed.test(origin)
+        : !!allowed;
+}
+
+function getOriginHeaders(reqOrigin, origin) {
+    const headers = new Headers();
+
+    if (origin === "*") {
+        // Allow any origin
+        headers.set("Access-Control-Allow-Origin", "*");
+    } else if (typeof origin === "string") {
+        // Fixed origin
+        headers.set("Access-Control-Allow-Origin", origin);
+        headers.append("Vary", "Origin");
+    } else {
+        const allowed = isOriginAllowed(reqOrigin ?? "", origin);
+
+        if (allowed && reqOrigin) {
+            headers.set("Access-Control-Allow-Origin", reqOrigin);
+        }
+        headers.append("Vary", "Origin");
+    }
+
+    return headers;
+}
+
+// originHeadersFromReq
+
+async function originHeadersFromReq(req, origin) {
+    const reqOrigin = req.headers.get("Origin") || undefined;
+    const value =
+        typeof origin === "function" ? await origin(reqOrigin, req) : origin;
+
+    if (!value) return;
+    return getOriginHeaders(reqOrigin, value);
+}
+
+function getAllowedHeaders(req, allowed) {
+    const headers = new Headers();
+
+    if (!allowed) {
+        allowed = req.headers.get("Access-Control-Request-Headers");
+        headers.append("Vary", "Access-Control-Request-Headers");
+    } else if (Array.isArray(allowed)) {
+        // If the allowed headers is an array, turn it into a string
+        allowed = allowed.join(",");
+    }
+    if (allowed) {
+        headers.set("Access-Control-Allow-Headers", allowed);
+    }
+
+    return headers;
+}
+
+export default async function cors(req, res, options) {
+    const opts = { ...defaultOptions, ...options };
+    const { headers } = res;
+    const originHeaders = await originHeadersFromReq(req, opts.origin ?? false);
+    const mergeHeaders = (v, k) => {
+        if (k === "Vary") headers.append(k, v);
+        else headers.set(k, v);
+    };
+
+    // If there's no origin we won't touch the response
+    if (!originHeaders) return res;
+
+    originHeaders.forEach(mergeHeaders);
+
+    if (opts.credentials) {
+        headers.set("Access-Control-Allow-Credentials", "true");
+    }
+
+    const exposed = Array.isArray(opts.exposedHeaders)
+        ? opts.exposedHeaders.join(",")
+        : opts.exposedHeaders;
+
+    if (exposed) {
+        headers.set("Access-Control-Expose-Headers", exposed);
+    }
+
+    // Handle the preflight request
+    if (req.method === "OPTIONS") {
+        if (opts.methods) {
+            const methods = Array.isArray(opts.methods)
+                ? opts.methods.join(",")
+                : opts.methods;
+
+            headers.set("Access-Control-Allow-Methods", methods);
+        }
+
+        getAllowedHeaders(req, opts.allowedHeaders).forEach(mergeHeaders);
+
+        if (typeof opts.maxAge === "number") {
+            headers.set("Access-Control-Max-Age", String(opts.maxAge));
+        }
+
+        if (opts.preflightContinue) return res;
+
+        headers.set("Content-Length", "0");
+        return new Response(null, {
+            status: opts.optionsSuccessStatus,
+            headers,
+        });
+    }
+
+    // If we got here, it's a normal request
+    return res;
+}
diff --git a/api/index.js b/api/index.js
index a391227..6b2757a 100644
--- a/api/index.js
+++ b/api/index.js
@@ -1,44 +1,52 @@
-import express from 'express';
-import { Kafka } from 'kafkajs';
-import cors from 'cors';
+import { Kafka } from "kafkajs";
+import cors from "./cors";
+
+export const config = {
+    runtime: "edge",
+};
 
 const kafka = new Kafka({
     clientId: process.env.KAFKA_CLIENT_ID,
     brokers: [process.env.KAFKA_BROKER_URL],
     ssl: true,
     sasl: {
-        mechanism: 'plain',
+        mechanism: "plain",
         username: process.env.KAFKA_API_KEY,
         password: process.env.KAFKA_API_SECRET,
     },
 });
 
 const producer = kafka.producer();
-const app = express();
-
-app.use(express.json());
-app.use(cors());    
 
-app.post('/api/send-kafka', async (req, res) => {
+export default async function handler(req) {
     const payload = req.body;
-    const topic = 'demo_flappy';
+    const topic = "demo_flappy";
 
     try {
         await producer.connect();
         await producer.send({
             topic,
-            messages: [
-                { key: 'vercel_edge', value: JSON.stringify(payload) },
-            ],
+            messages: [{ key: "vercel_edge", value: JSON.stringify(payload) }],
         });
-
-        res.json({ status: 'Message sent to Kafka' });
+        await producer.disconnect();
+        return cors(
+            req,
+            new Response(JSON.stringify({ status: "Message sent to Kafka" }), {
+                status: 200,
+                headers: { "Content-Type": "application/json" },
+            })
+        );
     } catch (error) {
-        console.error('Error sending message to Kafka:', error);
-        res.json({ error: error.message });
-    } finally {
         await producer.disconnect();
+        return cors(
+            req,
+            new Response(
+                JSON.stringify({ status: "Error sending message to Kafka" }),
+                {
+                    status: 500,
+                    headers: { "Content-Type": "application/json" },
+                }
+            )
+        );
     }
-});
-
-export default app;
\ No newline at end of file
+}