Augment user type in JWT callback

callbacks: {
    jwt: async ({ token, user }) => {
      if (user) {
        token.id = user.id;
        // token.role = user.
      }
      return token;
    },
    session: ({ session, token }) => {
      if (session.user) {
        session.id = token.id;
        session.user.role = token.role;
      }
      return session;
    },
  },

import { DefaultSession } from "next-auth";

declare module "next-auth" {
  /**
   * Returned by `useSession`, `getSession` and received as a prop on the `SessionProvider` React Context
   */
  interface Session {
    id: string;
    user?: {
      id: string;
      sponsorId?: string;
      role: string;
    } & DefaultSession["user"];
  }
}

declare module "next-auth/jwt" {
  interface JWT {
    id: string;
    role: string;
    sponsorId?: string;
  }
}