kellnr-http 7.0.0

use std::borrow::Cow;
use std::net::SocketAddr;
use std::sync::Arc;

use axum::Router;
use axum_extra::extract::cookie::Key;
use cookie::{Cookie, CookieJar};
use kellnr_common::account_name::UserName;
use kellnr_common::background_tasks::BackgroundTasks;
use kellnr_cratesio_sync::CratesioSync;
use kellnr_db::download_counter::DownloadCounter;
use kellnr_db::{Database, DbProvider, UserFlags};
use kellnr_settings::Proxy;
use kellnr_storage::cratesio_crate_storage::CratesIoCrateStorage;
use kellnr_storage::docs_storage::DocsStorage;
use kellnr_storage::kellnr_crate_storage::KellnrCrateStorage;
use kellnr_storage::object_storage::MemoryStorage;
use kellnr_test_db::TestDb;
use tokio::net::TcpListener;
use tokio::task::JoinHandle;

use crate::identity::{Identity, Source};
use crate::login_throttle::LoginThrottle;
use crate::state::{AppStateData, default_settings_prov};
use crate::token::generate_token;
use crate::token_cache::TokenCacheManager;

/// App state with a `MockDb`, [`kellnr_settings::test_settings`], in-memory
/// storage and [`TEST_KEY`] as cookie signing key.
pub fn test_state() -> AppStateData {
    state_with_db(Arc::new(kellnr_db::mock::MockDb::new()))
}

/// App state like [`test_state`], but on a fresh, migrated SQLite database.
/// Seed and inspect the database through the returned [`TestDb`]. It holds
/// the user `admin` with password `123` and is removed when dropped.
pub async fn sqlite_test_state() -> (AppStateData, TestDb) {
    let db = TestDb::sqlite().await;
    (state_with_db(db.provider()), db)
}

/// Logs in the user `name`, adding it with `flags` and password `pwd` if it
/// does not exist yet. Returns a `Cookie` header value with the new session.
pub async fn session_cookie(db: &Database, name: &str, flags: UserFlags) -> String {
    let name = UserName::new(name).unwrap();
    if db.get_user(&name).await.is_err() {
        db.add_user(&name, "pwd", flags).await.unwrap();
    }
    let token = generate_token();
    db.add_session_token(&name, &token).await.unwrap();
    encode_cookies([(
        kellnr_settings::constants::COOKIE_SESSION_ID,
        token.into_secret(),
    )])
}

fn state_with_db(db: Arc<dyn DbProvider>) -> AppStateData {
    let settings = Arc::new(kellnr_settings::test_settings());
    let settings_prov = default_settings_prov();
    let crate_storage = Arc::new(KellnrCrateStorage::new(
        &settings,
        Box::new(MemoryStorage::new()),
    ));
    let cratesio_storage = Arc::new(CratesIoCrateStorage::new(
        &settings,
        Box::new(MemoryStorage::new()),
    ));
    let docs_storage = Arc::new(DocsStorage::new(Box::new(MemoryStorage::new())));
    let (cratesio_prefetch_sender, _) = flume::unbounded();
    let token_cache = Arc::new(TokenCacheManager::new(
        false,
        std::time::Duration::from_secs(60),
        1000,
    ));
    let login_throttle = Arc::new(LoginThrottle::new(
        settings.registry.auth_max_failed_attempts,
        settings.registry.auth_failed_attempts_window(),
    ));
    let download_counter = Arc::new(DownloadCounter::new(db.clone(), 30));
    let proxy_client = kellnr_cratesio_sync::cratesio_downloader::CLIENT.clone();
    let cratesio_sync = CratesioSync::new(db.clone(), proxy_client.clone(), &settings.proxy);
    AppStateData {
        db,
        signing_key: Key::from(TEST_KEY),
        settings,
        settings_prov,
        crate_storage,
        cratesio_storage,
        docs_storage,
        cratesio_prefetch_sender,
        token_cache,
        login_throttle,
        toolchain_storage: None, // Toolchain storage disabled in tests by default
        download_counter,
        proxy_client,
        cratesio_sync,
        download_filters: Arc::default(),
        background_tasks: BackgroundTasks::new(),
    }
}

/// Fixed cookie signing key of [`test_state`], so requests can carry cookies
/// created by [`encode_cookies`].
pub const TEST_KEY: &[u8] = &[1; 64];

/// Encodes cookies as a `Cookie` header value, encrypted with [`TEST_KEY`] the
/// same way `PrivateCookieJar` does.
pub fn encode_cookies<const N: usize, K: Into<Cow<'static, str>>, V: Into<Cow<'static, str>>>(
    cookies: [(K, V); N],
) -> String {
    let mut clear = CookieJar::new();
    let mut jar = clear.private_mut(&Key::from(TEST_KEY));
    for (k, v) in cookies {
        jar.add(Cookie::new(k, v));
    }
    clear
        .iter()
        .map(|c| c.encoded().to_string())
        .collect::<Vec<_>>()
        .join("; ")
}

/// A user logged in with a session cookie.
pub fn session_identity(name: &str, is_admin: bool) -> Identity {
    Identity {
        name: UserName::new(name).unwrap(),
        is_admin,
        is_read_only: false,
        source: Source::Session,
    }
}

/// An HTTP server on a free port of 127.0.0.1 for tests of code that calls
/// other servers, like crates.io or webhook targets. It stops when dropped.
pub struct TestServer {
    addr: SocketAddr,
    task: JoinHandle<()>,
}

impl TestServer {
    pub async fn start(router: Router) -> Self {
        let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
        let addr = listener.local_addr().unwrap();
        let task = tokio::spawn(async move { axum::serve(listener, router).await.unwrap() });
        Self { addr, task }
    }

    pub fn addr(&self) -> SocketAddr {
        self.addr
    }

    /// URL of `path` on this server, e.g. `url("/api/")`.
    pub fn url(&self, path: &str) -> String {
        format!("http://{}{path}", self.addr)
    }

    /// Enabled crates.io proxy settings for this server as a fake crates.io
    /// with the sparse index at `/index/`, the API at `/api/` and downloads at
    /// `/dl/`.
    pub fn cratesio_proxy(&self) -> Proxy {
        Proxy {
            enabled: true,
            index: self.url("/index/").parse().unwrap(),
            api: self.url("/api/").parse().unwrap(),
            url: self.url("/dl/").parse().unwrap(),
            ..Proxy::default()
        }
    }
}

impl Drop for TestServer {
    fn drop(&mut self) {
        self.task.abort();
    }
}

/// URL of a free port of 127.0.0.1 where nothing listens.
pub async fn closed_port_url() -> String {
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    format!("http://{}/", listener.local_addr().unwrap())
}