index.js 3.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. exports.PRAS = void 0;
  4. class PRAS {
  5. getPRASResultObject(statusCode = 200, status = "OK", message = "", data = null, code = "UNKNOWN_CODE", extra = null) {
  6. return {
  7. statusCode,
  8. status,
  9. message,
  10. data,
  11. code,
  12. extra,
  13. };
  14. }
  15. responseMiddleware() {
  16. // eslint-disable-next-line @typescript-eslint/ban-types
  17. return (req, res, next) => {
  18. res.pras = {
  19. ok: (message, data, code, extra) => {
  20. res.json(this.getPRASResultObject(200, "OK", message, data, code, extra));
  21. },
  22. created: (message, data, code, extra) => {
  23. res.json(this.getPRASResultObject(201, "OK", message, data, code, extra));
  24. },
  25. badRequest: (message, data, code, extra) => {
  26. res.json(this.getPRASResultObject(400, "FAIL", message, data, code, extra));
  27. },
  28. unauthorized: (message, data, code, extra) => {
  29. res.json(this.getPRASResultObject(401, "FAIL", message, data, code, extra));
  30. },
  31. forbidden: (message, data, code, extra) => {
  32. res.json(this.getPRASResultObject(403, "FAIL", message, data, code, extra));
  33. },
  34. notFound: (message, data, code, extra) => {
  35. res.json(this.getPRASResultObject(404, "FAIL", message, data, code, extra));
  36. },
  37. serverError: (message, data, code, extra) => {
  38. res.json(this.getPRASResultObject(500, "ERROR", message, data, code, extra));
  39. },
  40. customMessage: (statusCode, message, data, code, extra) => {
  41. let status;
  42. if (statusCode >= 500) {
  43. status = "ERROR";
  44. }
  45. else if (statusCode >= 400) {
  46. status = "FAIL";
  47. }
  48. else {
  49. status = "OK";
  50. }
  51. res.json(this.getPRASResultObject(statusCode, status, message, data, code, extra));
  52. },
  53. };
  54. if (req.rawHeaders.hasOwnProperty("Accept") &&
  55. /application\/json/.test(req.rawHeaders["Accept"])) {
  56. const send = res.send;
  57. res.send = (body) => {
  58. try {
  59. if (typeof body === "string") {
  60. body = JSON.parse(body);
  61. }
  62. if (typeof body === "string") {
  63. body = { data: body };
  64. }
  65. else if (!body.hasOwnProperty("data")) {
  66. body = { data: body };
  67. }
  68. const prasBody = this.getPRASResultObject(body.statusCode, body.status, body.message, body.data, body.code, body.extra);
  69. res.status(prasBody.statusCode || 200);
  70. delete prasBody.statusCode;
  71. return send.call(res, JSON.stringify(prasBody));
  72. }
  73. catch (err) {
  74. return send.call(res, body);
  75. }
  76. };
  77. }
  78. next();
  79. };
  80. }
  81. }
  82. exports.PRAS = PRAS;
  83. exports.default = new PRAS();