feat: no need for another middleware (#4630)

Signed-off-by: Mariusz Jasuwienas <[email protected]>

Author: jasuwienas

packages/server/src/compliance.ts

@@ -56,7 +56,7 @@ const makeSureBodyExistsAndCanBeChecked = (ctx: IResponseContext) => {
  *
  * @param {IResponseContext & ParameterizedContext} ctx - Koa context containing status and body.
  */
-export const jsonRpcComplianceLayer = async (ctx: IResponseContext & ParameterizedContext) => {
+export const jsonRpcComplianceLayer = (ctx: IResponseContext & ParameterizedContext) => {
   if (!makeSureBodyExistsAndCanBeChecked(ctx)) return;
   if (ctx.status === 400) {
     if (!ctx.body.error || !ctx.body.error.code) ctx.body.error = FALLBACK_RESPONSE_BODY.error;

packages/server/src/server.ts

@@ -309,13 +309,11 @@ export async function initializeServer() {
 
   const rpcApp = koaJsonRpc.rpcApp();
 
-  app.use(async (ctx, next) => {
+  app.use(async (ctx) => {
     await rpcApp(ctx);
-    await next();
+    jsonRpcComplianceLayer(ctx);
   });
 
-  app.use(jsonRpcComplianceLayer);
-
   process.on('unhandledRejection', (reason, p) => {
     logger.error(`Unhandled Rejection at: Promise: ${JSON.stringify(p)}, reason: ${reason}`);
   });