Initial websockets getToken() returns null: next-auth + websockets :)

import { getToken } from "next-auth/jwt";

type Incoming = IncomingMessage & {
  cookies: Partial<{ [key: string]: string }>;
};

export const createTRPCContext = async (
  opts:
    | CreateNextContextOptions
    | NodeHTTPCreateContextFnOptions<IncomingMessage, ws>
) => {
  const { req } = opts;
  const token = await getToken({ req: req as Incoming });

  if (token) {
    return { session: { user: token } };
  }

  return { session: null };
};

api.messages.whoIsTyping.useSubscription(
    {
      slug,
    },
    {
      onData(data) {
        setCurrentlyTyping(data);
      },
      onError(err) {
        console.error("Subscription error:", err);
      },
    }
  );