v-common 0.25.0

use v_individual_model::onto::individual::Individual;
use v_individual_model::onto::parser::parse_raw;
use v_storage::{Storage, StorageId, StorageResult};
use v_storage::lmdb_storage::LMDBStorage;
use crate::v_api::common_type::ResultCode;
use crate::v_authorization::common::{Access, Trace};
use futures::lock::Mutex;
use std::io;
use std::io::{Error, ErrorKind};
use v_storage::tt_wrapper::{Client, IteratorType};

pub use super::authorization_provider::AuthorizationProvider;

pub const INDIVIDUALS_SPACE_ID: i32 = 512;
pub const TICKETS_SPACE_ID: i32 = 513;

pub struct AStorage {
    pub tt: Option<Client>,
    pub lmdb: Option<Mutex<LMDBStorage>>,
}

pub async fn check_indv_access_read(
    mut indv: Individual,
    uri: &str,
    user_uri: &str,
    az: Option<&AuthorizationProvider>,
) -> io::Result<(Individual, ResultCode)> {
    if indv.get_id().is_empty() {
        return Ok((indv, ResultCode::NotFound));
    }

    if let Some(a) = az {
        if a.authorize(uri, user_uri, Access::CanRead as u8, false).await.unwrap_or(0) != Access::CanRead as u8 {
            return Ok((indv, ResultCode::NotAuthorized));
        }
    }

    indv.parse_all();
    Ok((indv, ResultCode::Ok))
}

pub async fn check_user_in_group(user_id: &str, group_id: &str, az: Option<&AuthorizationProvider>) -> io::Result<bool> {
    if let Some(a) = az {
        let mut tr = Trace {
            acl: &mut "".to_string(),
            is_acl: false,
            group: &mut String::new(),
            is_group: true,
            info: &mut "".to_string(),
            is_info: false,
            str_num: 0,
        };
        if a.authorize_and_trace(user_id, user_id, 0xF, false, &mut tr).await.is_ok() {
            for gr in tr.group.split('\n') {
                if gr == group_id {
                    return Ok(true);
                }
            }
        } else {
            return Err(Error::new(ErrorKind::Other, "fail authorize_and_trace"));
        }
    }

    Ok(false)
}

pub async fn get_individual_from_db(uri: &str, user_uri: &str, db: &AStorage, az: Option<&AuthorizationProvider>) -> io::Result<(Individual, ResultCode)> {
    get_individual_use_storage_id(StorageId::Individuals, uri, user_uri, db, az).await
}

pub async fn get_individual_use_storage_id(
    storage_id: StorageId,
    uri: &str,
    user_uri: &str,
    db: &AStorage,
    az: Option<&AuthorizationProvider>,
) -> io::Result<(Individual, ResultCode)> {
    if let Some(tt) = &db.tt {
        let space_id = match storage_id {
            StorageId::Tickets => TICKETS_SPACE_ID,
            StorageId::Individuals => INDIVIDUALS_SPACE_ID,
            StorageId::Az => 514,
        };

        let response = tt.select(space_id, 0, &(uri,), 0, 100, IteratorType::EQ).await?;

        let mut iraw = Individual::default();
        iraw.set_raw(&response.data[5..]);
        if parse_raw(&mut iraw).is_ok() {
            return check_indv_access_read(iraw, uri, user_uri, az).await;
        }
        return Ok((iraw, ResultCode::UnprocessableEntity));
    }
    if let Some(lmdb) = &db.lmdb {
        let mut iraw = Individual::default();
        let res = lmdb.lock().await.get_individual(storage_id, uri, &mut iraw);
        match res {
            StorageResult::Ok(()) => {
                return check_indv_access_read(iraw, uri, user_uri, az).await;
            }
            StorageResult::NotFound => {
                return Ok((Individual::default(), ResultCode::NotFound));
            }
            _ => {
                return Ok((Individual::default(), ResultCode::UnprocessableEntity));
            }
        }
    }

    Ok((Individual::default(), ResultCode::UnprocessableEntity))
}