feat(metrics): expose the res FastifyReply object in AuroraPortalContext

.changeset/chilly-guests-design.md

@@ -0,0 +1,5 @@
+---
+"@cobaltcore-dev/aurora": patch
+---
+
+Expose the `res` (FastifyReply) object in `AuroraPortalContext` to allow consumer tRPC procedures to set response headers and cookies.

packages/aurora/src/server/Compute/routers/flavorRouter.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect, vi, beforeEach } from "vitest"
-import type { FastifyRequest } from "fastify"
+import type { FastifyRequest, FastifyReply } from "fastify"
 import { flavorRouter } from "./flavorRouter"
 import { Flavor } from "../types/flavor"
 import { TRPCError } from "@trpc/server"
@@ -21,6 +21,7 @@ vi.mock("../helpers/flavorHelpers", () => ({
 
 const createMockContext = (shouldFailAuth = false, shouldFailRescope = false, shouldFailCompute = false) => ({
   req: { headers: {} } as FastifyRequest,
+  res: { setCookie: vi.fn() } as unknown as FastifyReply,
   signal: new AbortController().signal,
   validateSession: vi.fn().mockReturnValue(!shouldFailAuth),
   identityEndpoint: "http://identity.example.com/",

packages/aurora/src/server/context.ts

@@ -1,5 +1,5 @@
 import type { CreateFastifyContextOptions } from "@trpc/server/adapters/fastify"
-import type { FastifyRequest } from "fastify"
+import type { FastifyRequest, FastifyReply } from "fastify"
 import { SignalOpenstackSession, SignalOpenstackSessionType } from "@cobaltcore-dev/signal-openstack"
 import { SessionCookie } from "./sessionCookie"
 import { AuthConfig } from "./Authentication/types/models"
@@ -57,6 +57,8 @@ export interface FormFieldData {
 
 export interface AuroraPortalContext extends AuroraContext {
   req: FastifyRequest
+  /** Fastify reply object for setting response headers/cookies */
+  res: FastifyReply
   /** Normalised identity endpoint (always ends with /) */
   identityEndpoint: string
   /** Ceph/S3 region from consumer config */
@@ -417,6 +419,7 @@ export async function createContext(
 
   return {
     req: opts.req,
+    res: opts.res,
     signal: abortController.signal,
     identityEndpoint: normalizedEndpoint,
     cephRegion: config.cephRegion,