How to make _middleware wait for token refresh before validating?

[DonHuskini]

I have a custom backend with accessToken and refreshToken. When the accessToken expires, I try to refresh it. I added the logic to refresh the access token on the jwt callback, however, the _middleware doesn't wait for the session or token to be updated so when I access the token param on the _middleware, it contains the old data and then user is redirected to /login.

Do you know how can I fix it?

[...nextauth].js

callbacks: {
  async jwt() {
    if (user) {
        return { ... }
    }

    if (Date.now() < token.accessTokenExpires) {
        return token
    }

    // Access token has expired, try to update it
    const newToken = await refreshAccessToken(token)

    token.accessToken = newToken.accessToken.token
    token.accessTokenExpires = newToken.accessToken.expireTime
    token.refreshToken = newToken.refreshToken.token
    token.refreshTokenExpires = newToken.refreshToken.expireTime
    return token
  },

  async session({ session, token }) {
    session.accessToken = token.accessToken
    session.accessTokenExpires = token.accessTokenExpires
    session.refreshToken = token.refreshToken
    session.refreshTokenExpires = token.refreshTokenExpires
  },
},

/pages/me/_middleware.ts

withAuth({
  callbacks: {
    authorized({ req, token }) {    
      // Tried with `await getToken({ req })` and didn't work as well
      return token && Date.now() < token.accessTokenExpires ? true : false
    },
  },
})

[bymoe]

Have u managed to fix this? Posted the same issue #5406