fix: network context chainId header (#900)

apps/api/gql/resolver-context.ts

@@ -1,9 +1,4 @@
 import { Request } from 'express';
-import { networkContext } from '../../../modules/network/network-context.service';
-import {
-    initRequestScopedContext,
-    setRequestScopedContextValue,
-} from '../../../modules/context/request-scoped-context';
 
 function getHeader(req: Request, key: string): string | undefined {
     const value = req.headers[key.toLowerCase()];
@@ -21,11 +16,6 @@ export async function resolverContext(req: Request) {
     const adminApiKey = getHeader(req, 'AdminApiKey');
     const chainId = getHeader(req, 'ChainId');
 
-    if (chainId && networkContext.isValidChainId(chainId)) {
-        initRequestScopedContext();
-        setRequestScopedContextValue('chainId', chainId);
-    }
-
     // Initialize context if it doesn't exist
     return {
         accountAddress: accountAddress ? accountAddress.toLowerCase() : null,

apps/api/middleware/index.ts

@@ -1,2 +1,3 @@
 export * from './corsMiddleware';
 export * from './lowerCaseMiddleware';
+export * from './sessionMiddleware';

apps/api/middleware/sessionMiddleware.ts

@@ -0,0 +1,24 @@
+import { NextFunction, Request, Response } from 'express';
+import { networkContext } from '../../../modules/network/network-context.service';
+import {
+    initRequestScopedContext,
+    setRequestScopedContextValue,
+} from '../../../modules/context/request-scoped-context';
+
+function getHeader(req: Request, key: string): string | undefined {
+    const value = req.headers[key.toLowerCase()];
+    return Array.isArray(value) ? value[0] : value;
+}
+
+export async function sessionMiddleware(req: Request, res: Response, next: NextFunction) {
+    const chainId = getHeader(req, 'ChainId');
+
+    if (chainId && networkContext.isValidChainId(chainId)) {
+        initRequestScopedContext();
+        setRequestScopedContextValue('chainId', chainId);
+
+        next();
+    } else {
+        next();
+    }
+}

apps/api/server.ts

@@ -6,7 +6,7 @@ import helmet from 'helmet';
 import * as http from 'http';
 import { env } from '../env';
 import { loadRestRoutes } from './rest-routes';
-import { corsMiddleware, lowerCaseMiddleware } from './middleware';
+import { corsMiddleware, lowerCaseMiddleware, sessionMiddleware } from './middleware';
 import {
     ApolloServerPluginDrainHttpServer,
     ApolloServerPluginLandingPageGraphQLPlayground,
@@ -34,6 +34,7 @@ const configureHelmet = (app: express.Express) => {
 
 const configureMiddlewares = (app: express.Express) => {
     app.use(corsMiddleware);
+    app.use(sessionMiddleware);
     app.use(lowerCaseMiddleware);
 };