|
1 | | -import { Kv, ResponseBuilder } from '@fermyon/spin-sdk'; |
| 1 | +// https://itty.dev/itty-router/routers/autorouter |
| 2 | +import { AutoRouter } from 'itty-router'; |
| 3 | +import { Kv } from '@fermyon/spin-sdk'; |
2 | 4 |
|
3 | 5 | const decoder = new TextDecoder(); |
4 | 6 |
|
5 | | -export async function handler(req: Request, res: ResponseBuilder) { |
6 | | - let store = Kv.openDefault(); |
7 | | - let status = 200; |
8 | | - let body; |
| 7 | +let router = AutoRouter(); |
9 | 8 |
|
10 | | - switch (req.method) { |
11 | | - case 'POST': |
12 | | - store.set(req.url, (await req.text()) || new Uint8Array().buffer); |
13 | | - break; |
14 | | - case 'GET': |
15 | | - let val; |
16 | | - val = store.get(req.url); |
17 | | - if (!val) { |
18 | | - status = 404; |
19 | | - } else { |
20 | | - body = decoder.decode(val); |
21 | | - } |
22 | | - break; |
23 | | - case 'DELETE': |
24 | | - store.delete(req.url); |
25 | | - break; |
26 | | - case 'HEAD': |
27 | | - if (!store.exists(req.url)) { |
28 | | - status = 404; |
29 | | - } |
30 | | - break; |
31 | | - default: |
32 | | - } |
| 9 | +// Route ordering matters, the first route that matches will be used |
| 10 | +// Any route that does not return will be treated as a middleware |
| 11 | +// Any unmatched route will return a 404 |
| 12 | +router |
| 13 | + .get("*", async (req: Request) => { |
| 14 | + let store = Kv.openDefault(); |
| 15 | + let status = 200; |
| 16 | + let body; |
33 | 17 |
|
34 | | - res.status(status); |
35 | | - res.send(body); |
36 | | -} |
| 18 | + switch (req.method) { |
| 19 | + case 'POST': |
| 20 | + store.set(req.url, (await req.text()) || new Uint8Array().buffer); |
| 21 | + break; |
| 22 | + case 'GET': |
| 23 | + let val; |
| 24 | + val = store.get(req.url); |
| 25 | + if (!val) { |
| 26 | + status = 404; |
| 27 | + } else { |
| 28 | + body = decoder.decode(val); |
| 29 | + } |
| 30 | + break; |
| 31 | + case 'DELETE': |
| 32 | + store.delete(req.url); |
| 33 | + break; |
| 34 | + case 'HEAD': |
| 35 | + if (!store.exists(req.url)) { |
| 36 | + status = 404; |
| 37 | + } |
| 38 | + break; |
| 39 | + default: |
| 40 | + } |
| 41 | + |
| 42 | + return new Response(body, { status }); |
| 43 | + }) |
| 44 | + |
| 45 | +//@ts-ignore |
| 46 | +addEventListener('fetch', async (event: FetchEvent) => { |
| 47 | + event.respondWith(router.fetch(event.request)); |
| 48 | +}); |
0 commit comments