#[allow(async_fn_in_trait)]
pub mod auth_handler;

use std::{env, net::SocketAddr, sync::Arc};
use tokio::net::TcpStream;

use nex_account::{grpc::Pid, grpc_client};
use rnex_reggie_protos::reggie::{EdgeNodeHolderConnectOption::DontRegister, RemoteEdgeNodeHolder};
use rnex_rmc::{
    OnlyRemote, new_rmc_gateway_connection,
    serialization::RmcSerialize,
    util::{SplittableBufferConnection, account::Account},
};
use rnex_server::{RnexManager, RnexModule, WeakPassthroughInitModule};
use tracing::info;

use crate::auth_handler::AuthHandler;

#[derive(Debug)]
pub struct AuthManager {
    pub destination_server_acct: Account,
    pub build_name: &'static str,
    pub control_server: Arc<OnlyRemote<RemoteEdgeNodeHolder>>,
}
#[derive(Debug)]
pub struct AuthModule;

impl RnexManager for AuthManager {
    type User = AuthHandler;
    type InitData = SocketAddr;
    async fn init_new_user(
        this: rnex_server::PassthroughInitModule<Self>,
        _: &rnex_server::ModuleHolder,
        _: &rnex_rmc::RmcConnection,
        init_data: &Self::InitData,
        _: WeakPassthroughInitModule<Self::User>,
    ) -> Self::User {
        info!(target: "proxy_connections", address = ?init_data, "user connected");
        Self::User { am: this }
    }
}

/*
pub static FORWARD_EDGE_NODE_HOLDER: Lazy<SocketAddrV4> = Lazy::new(|| {
    env::var("FORWARD_EDGE_NODE_HOLDER")
        .ok()
        .and_then(|s| Some(s.parse().unwrap()))
        .expect("FORWARD_EDGE_NODE_HOLDER not set")
});*/

impl RnexModule for AuthModule {
    type Manager = AuthManager;
    type InitError = anyhow::Error;
    async fn create_manager(
        _: &rnex_server::ModuleHolder,
    ) -> Result<Self::Manager, Self::InitError> {
        let conn = TcpStream::connect(env::var("FORWARD_EDGE_NODE_HOLDER")?)
            .await
            .unwrap();

        let conn: SplittableBufferConnection = conn.into();

        conn.send(DontRegister.to_data().unwrap()).await;

        let conn = new_rmc_gateway_connection(conn, async |r| {
            Arc::new(OnlyRemote::<RemoteEdgeNodeHolder>::new(r))
        })
        .await;
        Ok(AuthManager {
            build_name: option_env!("AUTH_REPORT_VERSION").unwrap_or("no version specified"),
            control_server: conn,
            // todo: update nex-account to allow pulling the entire account info for rnex
            destination_server_acct: Account::new_raw_key(
                2,
                "Quazal Rendez-Vous",
                grpc_client()
                    .await?
                    .get_nex_key_by_pid(Pid { pid: 2 })
                    .await?
                    .into_inner()
                    .key
                    .try_into()
                    .map_err(|_| anyhow::Error::msg("invalid key size"))?,
            ),
        })
    }
}