tRPC middleware infer type from another protectedProcedure

Question

Hello, protectedProcedure check and adds a not nullable user to the ctx. However the studyMiddleware does not know that user is not nullable, is there a way for the middleware to be aware of it?
In my current code I had to validate the user is not nullable inside the middleware and had to manually pass the user to the ctx user: opts.ctx.user, to make user not nullable inside the mutation.

Here is a sample of my code

const studyMiddleware = middleware(async (opts) = {
  if (!opts.ctx.user) {
    throw new TRPCError({
      code: "UNAUTHORIZED",
      message: "Unauthorized",
    });
  }

const parsedInput = z
    .object({
      studyId: z.string().cuid(),
    })
    .parse(opts.input);

const study = await studyRepository.findOneByIdAndClientId({
    id: parsedInput.studyId,
    clientId: opts.ctx.user.client.id,
  });

if (!study) {
    throw new TRPCError({
      code: "NOT_FOUND",
      message: "Study not found",
    });
  }

return opts.next({
    ctx: {
      ...opts.ctx,
      user: opts.ctx.user,
      study,
    },
  });
});

export const budgetRouter = createTRPCRouter({
  create: protectedProcedure
    .use(studyMiddleware)
    .input(createBudgetInputSchema)
    .mutation(({ input, ctx: { user, study } }) =
      budgetService.create(input, user, study),
    ),
});

`

tRPC middleware infer type from another protectedProcedure

Looking for more? Join the community!