devoyage-subgraph 0.0.15

use log::{debug, error, trace};
use serde::{Deserialize, Serialize};

use service_entity_field::ServiceEntityFieldConfig;

use crate::resolver_type::ResolverType;

use super::{cors::MethodOption, guard::Guard};

pub mod service_entity_field;

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ServiceEntityResolversConfig {
    pub find_one: Option<ServiceEntityResolverConfig>,
    pub find_many: Option<ServiceEntityResolverConfig>,
    pub create_one: Option<ServiceEntityResolverConfig>,
    pub update_one: Option<ServiceEntityResolverConfig>,
    pub update_many: Option<ServiceEntityResolverConfig>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct QueryPair(pub String, pub String);

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ServiceEntityDataSourceConfig {
    pub from: Option<String>,
    pub collection: Option<String>,
    pub table: Option<String>,
    pub path: Option<String>,
    pub search_query: Option<Vec<QueryPair>>,
    pub resolvers: Option<ServiceEntityResolversConfig>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ServiceEntityResolverConfig {
    pub fields: Option<Vec<ServiceEntityFieldConfig>>,
    pub path: Option<String>,
    pub search_query: Option<Vec<QueryPair>>,
    pub http_method: Option<MethodOption>,
    pub guards: Option<Vec<Guard>>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ServiceEntityConfig {
    pub name: String,
    pub fields: Vec<ServiceEntityFieldConfig>,
    pub data_source: Option<ServiceEntityDataSourceConfig>,
    pub guards: Option<Vec<Guard>>,
    pub required: Option<bool>,
    pub exclude_from_output: Option<bool>,
}

impl ServiceEntityConfig {
    pub fn get_resolvers(
        service_entity: ServiceEntityConfig,
    ) -> Option<ServiceEntityResolversConfig> {
        debug!("Get Resolvers From Service Entity: {:?}", service_entity);
        let data_source = service_entity.data_source;
        if data_source.is_some() {
            let resolvers = data_source.unwrap().resolvers;
            if resolvers.is_some() {
                let resolvers = resolvers.unwrap();
                debug!("Service Entity Resolvers: {:?}", resolvers);
                return Some(resolvers);
            }
        }
        debug!("No Resolvers Found");
        None
    }

    pub fn get_resolver(
        service_entity: &ServiceEntityConfig,
        resolver_type: ResolverType,
    ) -> Option<ServiceEntityResolverConfig> {
        debug!("Get Resolver: {:?}", resolver_type);
        let resolvers = ServiceEntityConfig::get_resolvers(service_entity.clone());
        if resolvers.is_none() {
            return None;
        }
        let resolvers = resolvers.unwrap();
        match resolver_type {
            ResolverType::FindOne => {
                if resolvers.find_one.is_some() {
                    debug!("Found Find One Resolver, {:?}", resolvers.find_one);
                    return resolvers.find_one;
                }
            }
            ResolverType::FindMany => {
                if resolvers.find_many.is_some() {
                    debug!("Found Find Many Resolver, {:?}", resolvers.find_many);
                    return resolvers.find_many;
                }
            }
            ResolverType::CreateOne => {
                if resolvers.create_one.is_some() {
                    debug!("Found Create One Resolver, {:?}", resolvers.create_one);
                    return resolvers.create_one;
                }
            }
            ResolverType::UpdateOne => {
                if resolvers.update_one.is_some() {
                    debug!("Found Update One Resolver, {:?}", resolvers.update_one);
                    return resolvers.update_one;
                }
            }
            ResolverType::UpdateMany => {
                if resolvers.update_many.is_some() {
                    debug!("Found Update Many Resolver, {:?}", resolvers.update_many);
                    return resolvers.update_many;
                }
            }
            _ => {}
        }
        None
    }

    pub fn get_entity_data_source(
        service_entity: &ServiceEntityConfig,
    ) -> Option<ServiceEntityDataSourceConfig> {
        debug!("Get Data Source From Service Entity: {:?}", service_entity);
        let data_source = &service_entity.data_source;
        if data_source.is_some() {
            let data_source = data_source.clone().unwrap();
            debug!("Data Source: {:?}", data_source);
            return Some(data_source.clone());
        }
        None
    }

    pub fn get_mongo_collection_name(entity: &ServiceEntityConfig) -> String {
        debug!("Found Entity Data Source: {:?}", entity.data_source);
        let data_source = ServiceEntityConfig::get_entity_data_source(entity);
        if data_source.is_none() {
            return entity.name.clone();
        }
        let collection = data_source.unwrap().collection;
        if collection.is_none() {
            return entity.name.clone();
        }
        collection.unwrap()
    }

    /// Returns vector of fields for a given entity.
    /// If field is nested, it returns all fields leading to the final field.
    pub fn get_fields_recursive(
        entity: &ServiceEntityConfig,
        field_name: &str,
    ) -> Result<Vec<ServiceEntityFieldConfig>, async_graphql::Error> {
        debug!("Get Field: {:?}", field_name);
        let entity_fields = &entity.fields;
        if field_name.contains(".") {
            debug!("Field is Nested");
            let mut fields = vec![];
            let mut field_names = ServiceEntityFieldConfig::split_field_names(field_name)?;
            let first_field =
                ServiceEntityConfig::get_field(entity.clone(), field_names[0].to_string().clone())?;
            fields.push(first_field.clone());
            let nested_fields = first_field.fields;
            if nested_fields.is_none() {
                return Ok(fields);
            }
            field_names.remove(0);
            let rest_fields = ServiceEntityFieldConfig::get_fields_recursive(
                nested_fields.unwrap(),
                field_names.join("."),
            )?;
            fields.extend(rest_fields);
            return Ok(fields);
        } else {
            for field in entity_fields {
                if field.name == field_name {
                    return Ok(vec![field.clone()]);
                }
            }
            Err(async_graphql::Error::new(format!(
                "Field {} not found in entity {}",
                field_name, entity.name
            )))
        }
    }

    /// Gets a field from a given entity.
    /// If field is nested, only returns nested field.
    pub fn get_field(
        entity: ServiceEntityConfig,
        field_name: String,
    ) -> Result<ServiceEntityFieldConfig, async_graphql::Error> {
        debug!("Get Field {:?} From {:?}", field_name, entity);
        let entity_fields = &entity.fields;
        if field_name.contains(".") {
            trace!("Field is Nested");
            let mut field_names = ServiceEntityFieldConfig::split_field_names(&field_name)?;
            let first_field_name = field_names[0];
            let first_field = ServiceEntityConfig::get_field(entity, first_field_name.to_string())?;
            let nested_fields = first_field.fields;
            if nested_fields.is_none() {
                error!("Field {} is not a nested field", field_name);
                return Err(async_graphql::Error::new(format!(
                    "Field {} is not a nested field",
                    field_name
                )));
            }
            field_names.remove(0);
            let field =
                ServiceEntityFieldConfig::get_field(nested_fields.unwrap(), field_names.join("."))?;
            trace!("Found Field: {:?}", field);
            return Ok(field);
        } else {
            for field in entity_fields {
                if field.name == field_name {
                    return Ok(field.clone());
                }
            }
            error!("Field {} not found when executing 'get_field'.", field_name);
            Err(async_graphql::Error::new(format!(
                "Field {} not found.",
                field_name
            )))
        }
    }

    pub fn get_primary_key_field(
        entity: &ServiceEntityConfig,
    ) -> Result<ServiceEntityFieldConfig, async_graphql::Error> {
        debug!("Get Primary Key Field");
        let entity_fields = &entity.fields;
        // Make sure the entity does not have multiple primary keys
        if entity_fields
            .iter()
            .filter(|field| field.primary_key.unwrap_or(false))
            .count()
            > 1
        {
            error!("Entity {} has multiple primary keys", entity.name);
            return Err(async_graphql::Error::from(format!(
                "Entity {} has multiple primary keys.",
                entity.name
            )));
        }

        for field in entity_fields {
            if field.primary_key.unwrap_or(false) {
                return Ok(field.clone());
            }
        }
        // Try to find the id or the _id field
        for field in entity_fields {
            if field.name == "id" || field.name == "_id" {
                return Ok(field.clone());
            }
        }
        Err(async_graphql::Error::from(format!(
            "No primary key field found for entity {}",
            entity.name
        )))
    }
}