Restructure

packages/stitcher/src/adapters/kv/cloudflare-kv.ts

@@ -0,0 +1,14 @@
+import type { KVNamespace } from "@cloudflare/workers-types";
+
+// Make sure wrangler.toml has a binding named "kv".
+const kv = process.env["kv"] as unknown as KVNamespace;
+
+export async function set(key: string, value: string, ttl: number) {
+  await kv.put(key, value, {
+    expirationTtl: ttl,
+  });
+}
+
+export async function get(key: string) {
+  return await kv.get(key);
+}

packages/stitcher/src/adapters/kv/index.ts

@@ -0,0 +1,14 @@
+import { env } from "../../env";
+
+interface KeyValue {
+  set(key: string, value: string, ttl: number): Promise<void>;
+  get(key: string): Promise<string | null>;
+}
+
+export let kv: KeyValue;
+
+if (env.SERVERLESS) {
+  kv = await import("./cloudflare-kv");
+} else {
+  kv = await import("./redis");
+}

packages/stitcher/src/adapters/kv/redis.ts

@@ -0,0 +1,23 @@
+import { createClient } from "redis";
+import { env } from "../../env";
+
+const REDIS_PREFIX = "stitcher";
+
+const client = createClient({
+  socket: {
+    host: env.REDIS_HOST,
+    port: env.REDIS_PORT,
+  },
+});
+
+await client.connect();
+
+export async function set(key: string, value: string, ttl: number) {
+  await client.set(`${REDIS_PREFIX}:${key}`, value, {
+    EX: ttl,
+  });
+}
+
+export async function get(key: string) {
+  return await client.get(`${REDIS_PREFIX}:${key}`);
+}

packages/stitcher/src/session.ts

@@ -1,6 +1,6 @@
 import { randomUUID } from "crypto";
 import { DateTime } from "luxon";
-import { kv } from "./kv";
+import { kv } from "./adapters/kv";
 import { JSON } from "./lib/json";
 import { resolveUri } from "./lib/url";
 import { fetchVmap } from "./vmap";