UNPKG

hono

Version:

Web framework built on Web Standards

86 lines (85 loc) 2.67 kB
// 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(; boundary=[a-zA-Z0-9'"()+_,\-./:=?]+)?$/; var urlencodedRegex = /^application\/x-www-form-urlencoded$/; 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 = {}; formData.forEach((value2, key) => { if (key.endsWith("[]")) { if (form[key] === void 0) { form[key] = [value2]; } else if (Array.isArray(form[key])) { ; form[key].push(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); await next(); }; }; export { validator };