feat: Allow to override ctx type when middleware is used

Describe the feature you’d like to request

I have this middleware which refines types on my ctx:

const isCompanyMemberMiddleware = t.middleware(({ ctx, next }) => {
  if (ctx.user?.system_role !== SystemRole.COMPANY_MEMBER) {
    throw new TRPCError({ code: 'UNAUTHORIZED' })
  }

  if (!ctx.user.company_id) {
    throw new TRPCError({ code: 'UNAUTHORIZED' })
  }
  return next({ ctx })
})

const companyProc = t.procedure.use(isCompanyMemberMiddleware)

but in my jobs query which uses this middleware typescript does not know that:

Describe the solution you’d like to see

It shure would be nice if we could override the ctx type with something like:

const companyProc = t.procedure.use(isCompanyMemberMiddleware).context<{user: {system_role: 'COMPANY_MEMBER', company_id: string}}>()

Desribe alternate solutions

for the moment, I am just writing this in every query/mutation:

companyProc.input(z.object({})).query(async ({ input, ctx }) => {
    const user = ctx.user as ISpottedSessionCompanyMember
    return ctx.prisma.job.findMany({
      where: {
        company_id: user.company_id
      }
    })
  })

which works, but it’s kind of annoying having to do it in every query/mutation

Additional information

No response

👨‍👧‍👦 Contributing

• 🙋‍♂️ Yes, I’d be down to file a PR implementing this feature!

[KATT]

Try

return next({
    ctx: { user: { ...ctx.user } }
  })

[KATT]

Hmmmmmmmmmmm, if ctx.user.system_role is a disctiminated union i feel like it should work

Maybe try

return next({
    ctx: { prisma: ctx.prisma, user: { ...ctx.user } }
  })