mod context;
mod document;
pub(crate) mod error;
mod fields;
mod template_checker;

use std::collections::BTreeMap;
use std::convert::TryFrom;
use std::ops::Deref;

use error::{NewPromptError, RenderPromptError};

use self::context::Context;
use self::document::Document;
use crate::update::del_add::DelAdd;
use crate::{FieldId, FieldsIdsMap};

pub struct Prompt {
    template: liquid::Template,
    template_text: String,
}

#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
pub struct PromptData {
    pub template: String,
}

impl From<Prompt> for PromptData {
    fn from(value: Prompt) -> Self {
        Self { template: value.template_text }
    }
}

impl TryFrom<PromptData> for Prompt {
    type Error = NewPromptError;

    fn try_from(value: PromptData) -> Result<Self, Self::Error> {
        Prompt::new(value.template)
    }
}

impl Clone for Prompt {
    fn clone(&self) -> Self {
        let template_text = self.template_text.clone();
        Self { template: new_template(&template_text).unwrap(), template_text }
    }
}

fn new_template(text: &str) -> Result<liquid::Template, liquid::Error> {
    liquid::ParserBuilder::with_stdlib().build().unwrap().parse(text)
}

fn default_template() -> liquid::Template {
    new_template(default_template_text()).unwrap()
}

fn default_template_text() -> &'static str {
    "{% for field in fields %}\
    {% if field.is_searchable and field.value != nil %}\
    {{ field.name }}: {{ field.value }}\n\
    {% endif %}\
    {% endfor %}"
}

impl Default for Prompt {
    fn default() -> Self {
        Self { template: default_template(), template_text: default_template_text().into() }
    }
}

impl Default for PromptData {
    fn default() -> Self {
        Self { template: default_template_text().into() }
    }
}

impl Prompt {
    pub fn new(template: String) -> Result<Self, NewPromptError> {
        let this = Self {
            template: liquid::ParserBuilder::with_stdlib()
                .build()
                .unwrap()
                .parse(&template)
                .map_err(NewPromptError::cannot_parse_template)?,
            template_text: template,
        };

        // render template with special object that's OK with `doc.*` and `fields.*`
        this.template
            .render(&template_checker::TemplateChecker)
            .map_err(NewPromptError::invalid_fields_in_template)?;

        Ok(this)
    }

    pub fn render(
        &self,
        document: obkv::KvReaderU16<'_>,
        side: DelAdd,
        field_id_map: &FieldsIdsMapWithMetadata,
    ) -> Result<String, RenderPromptError> {
        let document = Document::new(document, side, field_id_map);
        let context = Context::new(&document, field_id_map);

        self.template.render(&context).map_err(RenderPromptError::missing_context)
    }
}

pub struct FieldsIdsMapWithMetadata<'a> {
    fields_ids_map: &'a FieldsIdsMap,
    metadata: BTreeMap<FieldId, FieldMetadata>,
}

impl<'a> FieldsIdsMapWithMetadata<'a> {
    pub fn new(fields_ids_map: &'a FieldsIdsMap, searchable_fields_ids: &'_ [FieldId]) -> Self {
        let mut metadata: BTreeMap<FieldId, FieldMetadata> =
            fields_ids_map.ids().map(|id| (id, Default::default())).collect();
        for searchable_field_id in searchable_fields_ids {
            let Some(metadata) = metadata.get_mut(searchable_field_id) else { continue };
            metadata.searchable = true;
        }
        Self { fields_ids_map, metadata }
    }

    pub fn metadata(&self, field_id: FieldId) -> Option<FieldMetadata> {
        self.metadata.get(&field_id).copied()
    }
}

impl<'a> Deref for FieldsIdsMapWithMetadata<'a> {
    type Target = FieldsIdsMap;

    fn deref(&self) -> &Self::Target {
        self.fields_ids_map
    }
}

#[derive(Debug, Default, Clone, Copy)]
pub struct FieldMetadata {
    pub searchable: bool,
}

#[cfg(test)]
mod test {
    use super::Prompt;
    use crate::error::FaultSource;
    use crate::prompt::error::{NewPromptError, NewPromptErrorKind};

    #[test]
    fn default_template() {
        // does not panic
        Prompt::default();
    }

    #[test]
    fn empty_template() {
        Prompt::new("".into()).unwrap();
    }

    #[test]
    fn template_ok() {
        Prompt::new("{{doc.title}}: {{doc.overview}}".into()).unwrap();
    }

    #[test]
    fn template_syntax() {
        assert!(matches!(
            Prompt::new("{{doc.title: {{doc.overview}}".into()),
            Err(NewPromptError {
                kind: NewPromptErrorKind::CannotParseTemplate(_),
                fault: FaultSource::User
            })
        ));
    }

    #[test]
    fn template_missing_doc() {
        assert!(matches!(
            Prompt::new("{{title}}: {{overview}}".into()),
            Err(NewPromptError {
                kind: NewPromptErrorKind::InvalidFieldsInTemplate(_),
                fault: FaultSource::User
            })
        ));
    }

    #[test]
    fn template_nested_doc() {
        Prompt::new("{{doc.actor.firstName}}: {{doc.actor.lastName}}".into()).unwrap();
    }

    #[test]
    fn template_fields() {
        Prompt::new("{% for field in fields %}{{field}}{% endfor %}".into()).unwrap();
    }

    #[test]
    fn template_fields_ok() {
        Prompt::new("{% for field in fields %}{{field.name}}: {{field.value}}{% endfor %}".into())
            .unwrap();
    }

    #[test]
    fn template_fields_invalid() {
        assert!(matches!(
            // intentionally garbled field
            Prompt::new("{% for field in fields %}{{field.vaelu}} {% endfor %}".into()),
            Err(NewPromptError {
                kind: NewPromptErrorKind::InvalidFieldsInTemplate(_),
                fault: FaultSource::User
            })
        ));
    }
}