use failure::*;

use crate::tools;
use crate::api_schema::*;
use crate::api_schema::router::*;
use crate::tools::ticket::*;
use crate::auth_helpers::*;

use hyper::StatusCode;

use serde_json::{json, Value};

fn authenticate_user(username: &str, password: &str) -> Result<(), Error> {

    let ticket_lifetime = tools::ticket::TICKET_LIFETIME;

    if password.starts_with("PBS:") {
        if let Ok((_age, Some(ticket_username))) = tools::ticket::verify_rsa_ticket(public_auth_key(), "PBS", password, None, -300, ticket_lifetime) {
            if ticket_username == username {
                return Ok(());
            } else {
                bail!("ticket login failed - wrong username");
            }
        }
    }

    if username == "root@pam" {
        let mut auth = pam::Authenticator::with_password("proxmox-backup-auth").unwrap();
        auth.get_handler().set_credentials("root", password);
        auth.authenticate()?;
        return Ok(());
    }

    bail!("inavlid credentials");
}

fn create_ticket(
    param: Value,
    _info: &ApiMethod,
    _rpcenv: &mut RpcEnvironment,
) -> Result<Value, Error> {

    let username = tools::required_string_param(&param, "username")?;
    let password = tools::required_string_param(&param, "password")?;

    match authenticate_user(username, password) {
        Ok(_) => {

            let ticket = assemble_rsa_ticket( private_auth_key(), "PBS", Some(username), None)?;

            let token = assemble_csrf_prevention_token(csrf_secret(), username);

            log::info!("successful auth for user '{}'", username);

            return Ok(json!({
                "username": username,
                "ticket": ticket,
                "CSRFPreventionToken": token,
            }));
        }
        Err(err) => {
	    let client_ip = "unknown"; // $rpcenv->get_client_ip() || '';
            log::error!("authentication failure; rhost={} user={} msg={}", client_ip, username, err.to_string());
            return Err(http_err!(UNAUTHORIZED, "permission check failed.".into()));
        }
    }
}

pub fn router() -> Router {

    let route = Router::new()
        .get(ApiMethod::new(
            || Ok(json!([
                {"subdir": "ticket"}
            ])),
            ObjectSchema::new("Directory index.")))
        .subdir(
            "ticket",
            Router::new()
                .post(
                    ApiMethod::new(
                        create_ticket,
                        ObjectSchema::new("Create or verify authentication ticket.")
                            .required(
                                "username",
                                StringSchema::new("User name.")
                                    .max_length(64)
                            )
                            .required(
                                "password",
                                StringSchema::new("The secret password. This can also be a valid ticket.")
                            )
                    ).returns(
                        ObjectSchema::new("Returns authentication ticket with additional infos.")
                            .required("username", StringSchema::new("User name."))
                            .required("ticket", StringSchema::new("Auth ticket."))
                            .required("CSRFPreventionToken", StringSchema::new("Cross Site Request Forgery Prevention Token."))
                    ).protected(true)
                )
        );

    route
}