app/lib/media/upload.server.ts

Functions

Imports

It imports app/db/index.ts, app/lib/error-message.mjs, app/lib/media/classify.mjs, app/lib/media/core.server.ts, app/lib/media/upload-contract.mjs, app/lib/seo.ts. Imported by app/lib/carrel/media-adapter.server.ts, app/lib/operator/upload-media.server.ts, app/routes/admin.media.upload.ts.

Source

1import { upsertMediaRecord } from "~/db";
2import { classify, contentKey, roleOf } from "./classify.mjs";
3import { measureDimensions } from "./core.server";
4import { ALLOWED, validateUpload } from "./upload-contract.mjs";
5import { errorMessage } from "~/lib/error-message.mjs";
6import { IMMUTABLE_CACHE_CONTROL } from "~/lib/seo";
7
8// `type` is the caller's claim: checked against `ALLOWED`, never sniffed.
9type UploadInput = {
10  bytes: ArrayBuffer;
11  type: string;
12  name: string;
13};
14
15type StoreUploadResult =
16  | {
17      ok: true;
18      key: string;
19      bytes: number;
20      width: number | null;
21      height: number | null;
22      // False when the D1 row failed; the object is stored regardless.
23      recorded: boolean;
24    }
25  | { ok: false; code: string; message: string; status: number };
26
27// Refuses before measuring, so an unsupported type never reaches Images and an oversized blob is never hashed.
28export async function storeUpload(env: Env, file: UploadInput): Promise<StoreUploadResult> {
29  const refusal = validateUpload({ type: file.type, bytes: file.bytes });
30  if (refusal) return { ok: false, ...refusal };
31
32  // A throw, not a default extension: `"bin"` would store an object `classify()` throws on for every reader.
33  const extension = ALLOWED.get(file.type);
34  if (!extension) {
35    throw new Error(
36      `storeUpload: "${file.type}" passed validateUpload but is not in ALLOWED. ` +
37        `The two have drifted; both are in upload-contract.mjs.`,
38    );
39  }
40
41  // Measured before the key exists, because the key carries `-<w>x<h>`. Null (an SVG) is a real answer;
42  // a failed measurement refuses, since the key would be permanent and sizeless and never render.
43  let dimensions: { width: number; height: number } | null;
44  try {
45    dimensions = await measureDimensions(env, file.bytes);
46  } catch (error) {
47    console.error(JSON.stringify({ alert: "upload-unmeasurable", detail: errorMessage(error) }));
48    return {
49      ok: false,
50      code: "images-unavailable",
51      message: `${errorMessage(error)}. Nothing was stored; try again.`,
52      status: 503,
53    };
54  }
55  // The key's name segment is a lossy slug, so the original name is also kept in customMetadata.
56  const key = contentKey(
57    await crypto.subtle.digest("SHA-256", file.bytes),
58    extension,
59    dimensions,
60    file.name,
61  );
62
63  // Idempotent: the key is a function of the bytes and the name, so there is no exists check.
64  await env.MEDIA.put(key, file.bytes, {
65    httpMetadata: {
66      contentType: file.type,
67      cacheControl: IMMUTABLE_CACHE_CONTROL,
68    },
69    // On the object, not only the row: R2 is the truth and D1 must be re-derivable from it.
70    customMetadata: { originalName: file.name },
71  });
72
73  // Non-fatal: the object is already stored. `recorded: false` tells an HTTP caller the row did not land.
74  let recorded = true;
75  try {
76    const { kind, mime } = classify(key);
77    await upsertMediaRecord(env, {
78      key,
79      alt: "",
80      storage: "r2",
81      kind,
82      role: roleOf(key),
83      mime,
84      bytes: file.bytes.byteLength,
85      originalName: file.name,
86      uploadedAt: new Date().toISOString(),
87      width: dimensions?.width ?? null,
88      height: dimensions?.height ?? null,
89    });
90  } catch (error) {
91    recorded = false;
92    console.error(JSON.stringify({ alert: "media-record-write-failed", detail: errorMessage(error) }));
93  }
94
95  return {
96    ok: true,
97    key,
98    bytes: file.bytes.byteLength,
99    width: dimensions?.width ?? null,
100    height: dimensions?.height ?? null,
101    recorded,
102  };
103}