File size: 2,880 Bytes
a5d718a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | // src/validator/validator.ts
import { getCookie } from "../helper/cookie/index.js";
import { HTTPException } from "../http-exception.js";
import { bufferToFormData } from "../utils/buffer.js";
var jsonRegex = /^application\/([a-z-\.]+\+)?json(;\s*[a-zA-Z0-9\-]+\=([^;]+))*$/;
var multipartRegex = /^multipart\/form-data(;\s?boundary=[a-zA-Z0-9'"()+_,\-./:=?]+)?$/;
var urlencodedRegex = /^application\/x-www-form-urlencoded(;\s*[a-zA-Z0-9\-]+\=([^;]+))*$/;
var validator = (target, validationFunc) => {
return async (c, next) => {
let value = {};
const contentType = c.req.header("Content-Type");
switch (target) {
case "json":
if (!contentType || !jsonRegex.test(contentType)) {
break;
}
try {
value = await c.req.json();
} catch {
const message = "Malformed JSON in request body";
throw new HTTPException(400, { message });
}
break;
case "form": {
if (!contentType || !(multipartRegex.test(contentType) || urlencodedRegex.test(contentType))) {
break;
}
let formData;
if (c.req.bodyCache.formData) {
formData = await c.req.bodyCache.formData;
} else {
try {
const arrayBuffer = await c.req.arrayBuffer();
formData = await bufferToFormData(arrayBuffer, contentType);
c.req.bodyCache.formData = formData;
} catch (e) {
let message = "Malformed FormData request.";
message += e instanceof Error ? ` ${e.message}` : ` ${String(e)}`;
throw new HTTPException(400, { message });
}
}
const form = /* @__PURE__ */ Object.create(null);
formData.forEach((value2, key) => {
if (key.endsWith("[]")) {
;
(form[key] ??= []).push(value2);
} else if (Array.isArray(form[key])) {
;
form[key].push(value2);
} else if (Object.hasOwn(form, key)) {
form[key] = [form[key], value2];
} else {
form[key] = value2;
}
});
value = form;
break;
}
case "query":
value = Object.fromEntries(
Object.entries(c.req.queries()).map(([k, v]) => {
return v.length === 1 ? [k, v[0]] : [k, v];
})
);
break;
case "param":
value = c.req.param();
break;
case "header":
value = c.req.header();
break;
case "cookie":
value = getCookie(c);
break;
}
const res = await validationFunc(value, c);
if (res instanceof Response) {
return res;
}
c.req.addValidatedData(target, res);
return await next();
};
};
export {
validator
};
|