Add Durable Objects plugin stubs

Author: mrbbot

packages/tre/src/plugins/do/gateway.ts

@@ -0,0 +1,24 @@
+import { Clock, Storage } from "@miniflare/shared";
+
+export class DurableObjectsStorageGateway {
+  constructor(
+    private readonly storage: Storage,
+    private readonly clock: Clock
+  ) {}
+
+  async get(key: string) {
+    throw new Error("Not yet implemented!");
+  }
+
+  async put(key: string, value: Uint8Array) {
+    throw new Error("Not yet implemented!");
+  }
+
+  async delete(key: string) {
+    throw new Error("Not yet implemented!");
+  }
+
+  async list() {
+    throw new Error("Not yet implemented!");
+  }
+}

packages/tre/src/plugins/do/index.ts

@@ -0,0 +1,41 @@
+import { z } from "zod";
+import { PersistenceSchema, Plugin } from "../shared";
+import { DurableObjectsStorageGateway } from "./gateway";
+import { DurableObjectsStorageRouter } from "./router";
+
+export const DurableObjectsOptionsSchema = z.object({
+  durableObjects: z
+    .record(
+      z.union([
+        z.string(),
+        z.object({
+          className: z.string(),
+          scriptName: z.string().optional(),
+        }),
+      ])
+    )
+    .optional(),
+});
+export const DurableObjectsSharedOptionsSchema = z.object({
+  durableObjectsPersist: PersistenceSchema,
+});
+
+export const DURABLE_OBJECTS_PLUGIN_NAME = "do";
+export const DURABLE_OBJECTS_PLUGIN: Plugin<
+  typeof DurableObjectsOptionsSchema,
+  typeof DurableObjectsSharedOptionsSchema,
+  DurableObjectsStorageGateway
+> = {
+  gateway: DurableObjectsStorageGateway,
+  router: DurableObjectsStorageRouter,
+  options: DurableObjectsOptionsSchema,
+  sharedOptions: DurableObjectsSharedOptionsSchema,
+  getBindings(options) {
+    return undefined;
+  },
+  getServices(options) {
+    return undefined;
+  },
+};
+
+export * from "./gateway";

packages/tre/src/plugins/do/router.ts

@@ -0,0 +1,51 @@
+import { Response } from "undici";
+import {
+  DELETE,
+  GET,
+  PUT,
+  RouteHandler,
+  Router,
+  decodePersist,
+} from "../shared";
+import { DurableObjectsStorageGateway } from "./gateway";
+
+export interface DurableObjectStorageParams {
+  namespace: string;
+  key: string;
+}
+export class DurableObjectsStorageRouter extends Router<DurableObjectsStorageGateway> {
+  @GET("/:bucket/:key")
+  get: RouteHandler<DurableObjectStorageParams> = async (req, params) => {
+    const persist = decodePersist(req.headers);
+    const gateway = this.gatewayFactory.get(params.namespace, persist);
+    await gateway.get(params.key);
+    return new Response();
+  };
+
+  @PUT("/:bucket/:key")
+  put: RouteHandler<DurableObjectStorageParams> = async (req, params) => {
+    const persist = decodePersist(req.headers);
+    const gateway = this.gatewayFactory.get(params.namespace, persist);
+    await gateway.put(params.key, new Uint8Array(await req.arrayBuffer()));
+    return new Response();
+  };
+
+  @DELETE("/:bucket/:key")
+  delete: RouteHandler<DurableObjectStorageParams> = async (req, params) => {
+    const persist = decodePersist(req.headers);
+    const gateway = this.gatewayFactory.get(params.namespace, persist);
+    await gateway.delete(params.key);
+    return new Response();
+  };
+
+  @GET("/:bucket/")
+  list: RouteHandler<Omit<DurableObjectStorageParams, "key">> = async (
+    req,
+    params
+  ) => {
+    const persist = decodePersist(req.headers);
+    const gateway = this.gatewayFactory.get(params.namespace, persist);
+    await gateway.list();
+    return new Response();
+  };
+}