custom rooms

apps/storage/src/storage.app.ts

@@ -0,0 +1,115 @@
+import { Hono } from 'hono'
+import { useWorkersLogger } from 'workers-tagged-logger'
+
+import { withNotFound, withOnError } from '@repo/hono-helpers'
+
+import { validateAndGetAccountId } from './jwt'
+
+import type { Context } from 'hono'
+import type { App } from './context'
+
+/**
+ * Storage worker. Handles client file uploads (`POST /upload`) into the shared
+ * CDN R2 bucket, foldered by the posted `FileType` so the `cdn` worker can serve
+ * them back.
+ */
+
+/**
+ * The client's `UploadFileType` enum → the R2 subfolder uploads of that type are
+ * stored under. `Unknown` (0) is intentionally absent: like the reference
+ * server's `makeUploadName`, an unrecognized type has no destination and is
+ * rejected rather than stored.
+ */
+const UPLOAD_SUBFOLDER: Record<number, string> = {
+	1: 'roomsave',
+	2: 'holotar',
+	3: 'image',
+	4: 'video',
+	5: 'invention',
+	6: 'roommetadata',
+}
+
+/** Resolve the storage subfolder for a posted FileType, or `undefined` when unknown. */
+function subfolderForFileType(fileType: string): string | undefined {
+	return UPLOAD_SUBFOLDER[Number.parseInt(fileType, 10)]
+}
+
+/** Read a text form field by any of its accepted names, matched case-insensitively. */
+function textField(body: Record<string, unknown>, ...names: string[]): string | undefined {
+	for (const [key, value] of Object.entries(body)) {
+		if (typeof value === 'string' && names.includes(key.toLowerCase())) return value
+	}
+	return undefined
+}
+
+/**
+ * Resolve the account id from a Bearer token. Returns `null` when the header is
+ * missing, the token is invalid, or the `sub` claim isn't an integer.
+ */
+async function authedId(c: Context<App>): Promise<number | null> {
+	const authHeader = c.req.header('Authorization') ?? ''
+	if (!authHeader.toLowerCase().startsWith('bearer ')) return null
+
+	const token = authHeader.slice('Bearer '.length)
+	const accountId = await validateAndGetAccountId(token)
+	if (!accountId) return null
+
+	const id = Number.parseInt(accountId, 10)
+	return Number.isNaN(id) ? null : id
+}
+
+const app = new Hono<App>()
+	.use(
+		'*',
+		// middleware
+		(c, next) =>
+			useWorkersLogger(c.env.NAME, {
+				environment: c.env.ENVIRONMENT,
+				release: c.env.SENTRY_RELEASE,
+			})(c, next)
+	)
+
+	.onError(withOnError())
+	.notFound(withNotFound())
+
+	.get('/', async (c) => {
+		return c.text('hello, world!')
+	})
+
+	// File upload. Auth-gated — any valid account token is allowed (no role check).
+	// Multipart form with `FileType` (the client's UploadFileType enum) and a binary
+	// part. Stores the file in the shared CDN R2 bucket under
+	// `<type-subfolder>/<random-name>` and returns the generated filename the client
+	// references it by. Also accepts a name-only post (no binary) that just echoes
+	// back an explicit `name`/`filename`/`imagename`. Mirrors the reference `Upload`.
+	.post('/upload', async (c) => {
+		const id = await authedId(c)
+		if (id === null) return c.body(null, 401)
+
+		const body = await c.req.parseBody().catch(() => ({}) as Record<string, unknown>)
+
+		// The binary part is identified by being a file (filename/content-type),
+		// not by its field name — matching the reference's part detection.
+		const file = Object.values(body).find((v): v is File => v instanceof File)
+
+		if (file) {
+			const subfolder = subfolderForFileType(textField(body, 'filetype') ?? '0')
+			if (subfolder === undefined) {
+				// makeUploadName == "" → no destination for an unknown/missing type.
+				return c.json({ error: 'missing or unknown FileType' }, 400)
+			}
+			const filename = crypto.randomUUID().replace(/-/g, '')
+			await c.env.CDN_ASSETS.put(`${subfolder}/${filename}`, await file.arrayBuffer(), {
+				httpMetadata: { contentType: file.type || 'application/octet-stream' },
+			})
+			return c.json({ filename })
+		}
+
+		// No binary — accept an explicit name and echo it straight back.
+		const explicitName = textField(body, 'imagename', 'filename', 'name')
+		if (explicitName) return c.json({ filename: explicitName })
+
+		return c.json({ error: 'missing filename or valid upload data' }, 400)
+	})
+
+export default app