use actix_web::dev::{ServiceRequest, ServiceResponse};
use actix_web::error::ErrorUnauthorized;
use actix_web::http::header::{HeaderValue, AUTHORIZATION};
use actix_web::{test, Error};
use serde::Deserialize;
use std::collections::HashSet;
use std::hash::Hash;
use std::str::FromStr;

pub const ROLE_ADMIN: &str = "ROLE_ADMIN";
pub const ROLE_MANAGER: &str = "ROLE_MANAGER";

#[derive(parse_display::Display, parse_display::FromStr, PartialEq, Eq, Hash)]
#[display(style = "SNAKE_CASE")]
#[allow(clippy::upper_case_acronyms)]
pub enum Role {
    ADMIN,
    MANAGER,
}

#[derive(parse_display::Display, parse_display::FromStr, PartialEq, Eq, Hash)]
#[display(style = "SNAKE_CASE")]
#[allow(clippy::upper_case_acronyms)]
pub enum Permission {
    READ,
    WRITE,
}

pub async fn extract(req: &ServiceRequest) -> Result<HashSet<String>, Error> {
    let auth_header: Option<&str> = req
        .headers()
        .get(AUTHORIZATION)
        .map(HeaderValue::to_str)
        .filter(Result::is_ok)
        .map(Result::unwrap);

    auth_header
        .map(|header| header.split(',').map(str::to_string).collect())
        .ok_or_else(|| ErrorUnauthorized("Authorization header incorrect!"))
}

pub async fn enum_extract<T: FromStr + Eq + Hash>(
    req: &ServiceRequest,
) -> Result<HashSet<T>, Error> {
    let auth_header: Option<&str> = req
        .headers()
        .get(AUTHORIZATION)
        .map(HeaderValue::to_str)
        .filter(Result::is_ok)
        .map(Result::unwrap);

    auth_header
        .map(|header| {
            header
                .split(',')
                .filter_map(|name| T::from_str(name).ok())
                .collect()
        })
        .ok_or_else(|| ErrorUnauthorized("Authorization header incorrect!"))
}

pub async fn test_body<B: actix_web::body::MessageBody>(
    resp: ServiceResponse<B>,
    expected_body: &str,
) {
    let body = test::read_body(resp).await;

    assert_eq!(String::from_utf8(body.to_vec()).unwrap(), expected_body);
}

#[derive(Deserialize)]
pub struct NamePayload {
    pub name: Option<String>,
}