use std::num::NonZeroU32;

use scram_rs::ScramAuthClient;
use scram_rs::ScramCbHelper;
use scram_rs::ScramResult;
use scram_rs::ScramResultClient;
use scram_rs::ScramSha256RustNative;
use scram_rs::ScramNonce;
use scram_rs::ScramPassword;
use scram_rs::ScramAuthServer;
use scram_rs::scram_sync::SyncScramClient;
use scram_rs::scram_sync::{ScramServerDyn, SyncScramServer};
use scram_rs::ScramCommon;
use scram_rs::SCRAM_TYPES;

#[derive(Debug)]
struct AuthDB
{
}

impl AuthDB
{
    pub fn new() -> Self
    {
        return AuthDB{};
    }
}

impl ScramCbHelper for &AuthDB
{
    
}

impl ScramAuthServer<ScramSha256RustNative> for &AuthDB
{
    fn get_password_for_user(&self, username: &str) -> ScramResult<ScramPassword>
    {
        return 
            if username == "user"
            {
                Ok(ScramPassword::found_secret_base64_password(
                    "xeR41ZKIyEGqUw22hFxMjZYok6ABzk4RpJY4c6qYE0o=".to_string(),
                        "c2FsdA==".to_string(), 
                        unsafe { NonZeroU32::new_unchecked(4096) },
                    None
                )?)
            }
            else 
            {
                ScramPassword::not_found::<ScramSha256RustNative>()
            };      
    }
}

#[derive(Debug)]
struct AuthClient
{
    username: String,
    password: String,
    key: scram_rs::ScramKey,
}

impl ScramAuthClient for &AuthClient
{
    fn get_username(&self) -> &str
    {
        return &self.username;
    }

    fn get_password(&self) -> &str
    {
        return &self.password;
    }

    fn get_scram_keys(&self) -> &scram_rs::ScramKey 
    {
        return &self.key;
    }
}

impl ScramCbHelper for &AuthClient
{
    
}

impl AuthClient
{
    pub 
    fn new(u: &'static str, p: &'static str) -> Self
    {
        return AuthClient{ username: u.to_string(), password: p.to_string(), key: scram_rs::ScramKey::new() };
    }
}

/// This example will not run, because it requires server, see tests in scram_sync.rs
pub fn main()
{
    let client = AuthClient::new("user", "password");
    
    let (clie_send, serv_recv) = std::sync::mpsc::channel::<String>();
    let (serv_send, clie_recv) = std::sync::mpsc::channel::<String>();

    let hndl = 
        std::thread::spawn(move || 
            {
                let authdb = AuthDB::new();
                let scramtype = SCRAM_TYPES.get_scramtype("SCRAM-SHA-256").unwrap();
            
                let mut server = 
                    SyncScramServer::<ScramSha256RustNative, &AuthDB, &AuthDB>::new(&authdb, &authdb, ScramNonce::none(), scramtype).unwrap();
            
                loop
                {
                    let client_data = serv_recv.recv().unwrap();

                    let serv_data = server.parse_response_base64(client_data.as_bytes());
                    serv_send.send(serv_data.encode_base64()).unwrap();

                    match serv_data
                    {
                        scram_rs::ScramResultServer::Error(e) => 
                        {
                            println!("server: error: {}", e);
                            break;
                        },
                        scram_rs::ScramResultServer::Final(_) =>
                        {
                            println!("server: final!");
                            break;
                        },
                        _ => {}
                    }
                }
            }
        );

    let mut client =
        SyncScramClient
            ::<ScramSha256RustNative, &AuthClient, &AuthClient>
            ::new(&client, ScramNonce::None, scram_rs::ChannelBindType::None, &client)
                .unwrap();

    // client sends initial message: cli -> serv
    let ci = client.init_client().encode_output_base64().unwrap();
    clie_send.send(ci).unwrap();

    
    loop
    {
        let serv_data = clie_recv.recv().unwrap();

        match client.parse_response_base64(serv_data)
        {
            Ok(ScramResultClient::Completed) => 
            {
                println!("client: completed auth successfully");
                break;
            },
            Ok(out) => clie_send.send(out.encode_output_base64().unwrap()).unwrap(),
            Err(e) =>
            {
                println!("client: error: {}", e);
                break;
            }
        }

    }

    hndl.join().unwrap();

    return;
}