How to set cookies in trpc response?

import { api } from "~/trpc/react";
[...]
  const login = api.auth.login.useMutation({ ... });

import { cookies } from "next/headers";

  publicProcedure
    .input(...)
    .mutation(async ({ ctx, input }) => {
      [...]

      // somehow set a cookie
      cookies().set("session", session.data.token, {
        httpOnly: true,
      });

      [...]
    }),