Add R2 plugin stubs

Author: mrbbot

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

@@ -0,0 +1,24 @@
+import { Clock, Storage } from "@miniflare/shared";
+
+export class R2Gateway {
+  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/r2/index.ts

@@ -0,0 +1,31 @@
+import { z } from "zod";
+import { PersistenceSchema, Plugin } from "../shared";
+import { R2Gateway } from "./gateway";
+import { R2Router } from "./router";
+
+export const R2OptionsSchema = z.object({
+  r2Buckets: z.record(z.string()).optional(),
+});
+export const R2SharedOptionsSchema = z.object({
+  r2Persist: PersistenceSchema,
+});
+
+export const R2_PLUGIN_NAME = "r2";
+export const R2_PLUGIN: Plugin<
+  typeof R2OptionsSchema,
+  typeof R2SharedOptionsSchema,
+  R2Gateway
+> = {
+  gateway: R2Gateway,
+  router: R2Router,
+  options: R2OptionsSchema,
+  sharedOptions: R2SharedOptionsSchema,
+  getBindings(options) {
+    return undefined;
+  },
+  getServices(options) {
+    return undefined;
+  },
+};
+
+export * from "./gateway";

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

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