Custom middleware for authentication with an optional state

[Alepheia]

Summary

I'm working on a handler for Google's OAuth and the idea is to validade the JWT token extracted from a PrivateCookieJar, refresh the token if its expired, and return a struct Claims with all the relevant fields. I initally attempted the method shown in this example using FromRequestParts:
https://github.com/tokio-rs/axum/blob/main/examples/oauth/src/main.rs

This works great, I can even use user: Option<User> (https://github.com/tokio-rs/axum/blob/main/examples/oauth/src/main.rs#L135C16-L135C34) in the handler for cases when the claims is optional. However I couldn't find a way pass the updated cookie jar the browser so I looked into another way with a custom middleware:
https://docs.rs/axum/latest/axum/middleware/index.html#passing-state-from-middleware-to-handlers

I'm able to do everything including update the browser with the updated cookie jar Ok((updated_jar, next.run(req).await)) but there's one small issue: I can't find a way to make the claims set by the middleware optional in the same way I can do user: Option<User> above. Is there a way around this?

axum version

0.7.5

[Alepheia]

Got it working by returning Option<Claims> in another custom middleware alongside the one I've already implemented returning just Claims. I'm not sure if this is the best way to do it but it works for now.