fix: test do

Author: yjl9903

packages/vite-plugin-cloudflare-functions/src/vite/generate.ts

@@ -22,6 +22,9 @@ export async function generate(functionsRoot: string, dtsPath: string) {
     if (!file.startsWith('/')) {
       file = '/' + file;
     }
+    if (file.endsWith('/index')) {
+      file = file.replace(/index$/, '');
+    }
     file = file.replace(/\[\[([\w]+)\]\]$/g, '**:$1');
     file = file.replace(/\[([\w]+)\]/g, ':$1');
     return file;

playground/app/cloudflare.d.ts

@@ -11,6 +11,9 @@ declare module 'vite-plugin-cloudflare-functions/client' {
     '/api/**': {
       ALL: CloudflareResponseBody<typeof import('../functions/api/_middleware')['onRequest']>;
     };
+    '/api/counter/': {
+      GET: CloudflareResponseBody<typeof import('../functions/api/counter/index')['onRequestGet']>;
+    };
     '/api/state/:key': {
       GET: CloudflareResponseBody<typeof import('../functions/api/state/[key]')['onRequestGet']>;
       POST: CloudflareResponseBody<typeof import('../functions/api/state/[key]')['onRequestPost']>;

playground/app/vite.config.ts

@@ -16,7 +16,13 @@ export default defineConfig({
     CloudflarePagesFunctions({
       root: '../functions',
       outDir: '../../',
-      wrangler: { log: false, kv: 'STORE', binding: { USER: 'yjl9903' } }
+      wrangler: {
+        log: true,
+        compatibilityDate: '2023-10-26',
+        kv: 'STORE',
+        binding: { USER: 'yjl9903' },
+        do: { COUNTER: { class: 'Counter', script: 'do' } }
+      }
     })
   ]
 });

playground/functions/api/counter/index.ts

@@ -0,0 +1,13 @@
+import { makePagesFunction } from 'vite-plugin-cloudflare-functions/worker';
+
+export const onRequestGet = makePagesFunction(async ({ request, env }) => {
+  const id = env.COUNTER.idFromName('root');
+  const counter = env.COUNTER.get(id);
+
+  console.log(id, counter);
+
+  const url = new URL(request.url);
+  const resp = await counter.fetch(url.origin);
+
+  return { value: await resp.text() };
+});

playground/functions/cloudflare.d.ts

@@ -10,6 +10,8 @@ declare module 'vite-plugin-cloudflare-functions/worker' {
     STORE: KVNamespace;
 
     USER: string;
+
+    COUNTER: DurableObjectNamespace;
   }
 
   interface PagesFunctionData {}

playground/functions/do.ts

@@ -0,0 +1,39 @@
+export class Counter implements DurableObject {
+  private state: DurableObjectState;
+
+  constructor(state: DurableObjectState) {
+    this.state = state;
+  }
+
+  // Handle HTTP requests from clients.
+  async fetch(request: Request) {
+    // Apply requested action.
+    const url = new URL(request.url);
+
+    // Durable Object storage is automatically cached in-memory, so reading the
+    // same key every request is fast.
+    // You could also store the value in a class member if you prefer.
+    let value = +(await this.state.storage.get('value')) || 0;
+
+    switch (url.pathname) {
+      case '/increment':
+        ++value;
+        break;
+      case '/decrement':
+        --value;
+        break;
+      case '/':
+        // Serves the current value.
+        break;
+      default:
+        return new Response('Not found', { status: 404 });
+    }
+
+    // You do not have to worry about a concurrent request having modified the value in storage.
+    // "input gates" will automatically protect against unwanted concurrency.
+    // Read-modify-write is safe.
+    await this.state.storage.put('value', value);
+
+    return new Response('' + value);
+  }
+}