Can't use `self.pool` in impl of UserStore<CustomRole> for PostgresStore<CustomUser> about axum-login HOT 6 CLOSED

  let user_store = PostgresStore::<User, UserRole>::new(pool.clone());

thread 'tokio-runtime-worker' panicked at 'Auth extension missing. Is the auth layer installed?: MissingExtension(MissingExtension(Error { inner: "Extension of type `axum_login::extractors::AuthContext<be::entities::user::User, axum_login::sqlx_store::SqlxStore<sqlx_core::pool::Pool<sqlx_core::postgres::database::Postgres>, be::entities::user::User>, be::entities::user::UserRole>` was not found. Perhaps you forgot to add it? See `axum::Extension`." }))

use axum::{
    extract::State,
    routing::{get, post},
    Json, Router,
};
use axum_login::{
    axum_sessions::{async_session::MemoryStore, SessionLayer},
    AuthLayer, PostgresStore,
};
use rand::Rng;
use sqlx::postgres::PgPoolOptions;
use sqlx::PgPool;
use std::net::SocketAddr;

mod entities;

use entities::user::{User, UserRole};

#[tokio::main]
async fn main() {
    let secret = rand::thread_rng().gen::<[u8; 64]>();
    let session_store = MemoryStore::new();
    let session_layer = SessionLayer::new(session_store, &secret).with_secure(false);

    dotenvy::dotenv().ok();
    let database_connection_uri = dotenvy::var("DATABASE_URL").unwrap();

    let pool = PgPoolOptions::new()
        .max_connections(5)
        .connect(&database_connection_uri)
        .await
        .unwrap();

    let user_store = PostgresStore::<User, UserRole>::new(pool.clone());
    let auth_layer = AuthLayer::new(user_store, &secret);

    let app = Router::new()
        .route("/login", post(login))
        .layer(auth_layer)
        .layer(session_layer)
        .with_state(pool);

    let addr = SocketAddr::from(([127, 0, 0, 1], 5000));
    axum::Server::bind(&addr)
        .serve(app.into_make_service())
        .await
        .unwrap();
}

type AuthContext = axum_login::extractors::AuthContext<User, PostgresStore<User>, UserRole>;

#[derive(serde::Deserialize)]
struct Login {
    username: String,
    password: String,
}

#[axum::debug_handler]
async fn login(
    mut auth: AuthContext,
    State(pool): State<PgPool>,
    Json(payload): Json<Login>,
) -> impl axum::response::IntoResponse {
    let password_hash = hash_password(payload.password);
    let user_result =
        sqlx::query_as::<_, User>("SELECT * FROM users WHERE username = $1 AND password_hash = $2")
            .bind(payload.username)
            .bind(password_hash)
            .fetch_one(&pool)
            .await;

    let user: User = match user_result {
        Ok(value) => value,
        Err(_) => return Err((axum::http::StatusCode::UNAUTHORIZED, ())),
    };

    auth.login(&user).await.unwrap();
    Ok(())
}