index.js 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. // src/middleware/csrf/index.ts
  2. import { HTTPException } from "../../http-exception.js";
  3. var secFetchSiteValues = ["same-origin", "same-site", "none", "cross-site"];
  4. var isSecFetchSite = (value) => secFetchSiteValues.includes(value);
  5. var isSafeMethodRe = /^(GET|HEAD)$/;
  6. var isRequestedByFormElementRe = /^\b(application\/x-www-form-urlencoded|multipart\/form-data|text\/plain)\b/i;
  7. var csrf = (options) => {
  8. const originHandler = ((optsOrigin) => {
  9. if (!optsOrigin) {
  10. return (origin, c) => origin === new URL(c.req.url).origin;
  11. } else if (typeof optsOrigin === "string") {
  12. return (origin) => origin === optsOrigin;
  13. } else if (typeof optsOrigin === "function") {
  14. return optsOrigin;
  15. } else {
  16. return (origin) => optsOrigin.includes(origin);
  17. }
  18. })(options?.origin);
  19. const isAllowedOrigin = async (origin, c) => {
  20. if (origin === void 0) {
  21. return false;
  22. }
  23. return await originHandler(origin, c);
  24. };
  25. const secFetchSiteHandler = ((optsSecFetchSite) => {
  26. if (!optsSecFetchSite) {
  27. return (secFetchSite) => secFetchSite === "same-origin";
  28. } else if (typeof optsSecFetchSite === "string") {
  29. return (secFetchSite) => secFetchSite === optsSecFetchSite;
  30. } else if (typeof optsSecFetchSite === "function") {
  31. return optsSecFetchSite;
  32. } else {
  33. return (secFetchSite) => optsSecFetchSite.includes(secFetchSite);
  34. }
  35. })(options?.secFetchSite);
  36. const isAllowedSecFetchSite = async (secFetchSite, c) => {
  37. if (secFetchSite === void 0) {
  38. return false;
  39. }
  40. if (!isSecFetchSite(secFetchSite)) {
  41. return false;
  42. }
  43. return await secFetchSiteHandler(secFetchSite, c);
  44. };
  45. return async function csrf2(c, next) {
  46. if (!isSafeMethodRe.test(c.req.method) && isRequestedByFormElementRe.test(c.req.header("content-type") || "text/plain") && !await isAllowedSecFetchSite(c.req.header("sec-fetch-site"), c) && !await isAllowedOrigin(c.req.header("origin"), c)) {
  47. const res = new Response("Forbidden", { status: 403 });
  48. throw new HTTPException(403, { res });
  49. }
  50. await next();
  51. };
  52. };
  53. export {
  54. csrf
  55. };